diff --git a/nova/utils/formatting.py b/nova/utils/formatting.py index 0e96500..95ed1bf 100644 --- a/nova/utils/formatting.py +++ b/nova/utils/formatting.py @@ -13,7 +13,7 @@ def print_message(role: str, content: str, timestamp: str = None): if role.lower() == "user": color = "blue" icon = "👤" - display_role = role.title() + display_role = "You" elif role.lower() == "assistant": color = "green" icon = "🤖" diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py index 0f29374..937382d 100644 --- a/tests/unit/test_utils.py +++ b/tests/unit/test_utils.py @@ -159,7 +159,7 @@ def test_print_message_user(self, mock_console): mock_console.print.assert_called_once() # Verify the panel was created with correct styling panel_arg = mock_console.print.call_args[0][0] - assert "👤 User (12:34:56)" in str(panel_arg.title) + assert "👤 You (12:34:56)" in str(panel_arg.title) assert panel_arg.border_style == "blue" @patch("nova.utils.formatting.console") @@ -189,7 +189,7 @@ def test_print_message_no_timestamp(self, mock_console): mock_console.print.assert_called_once() panel_arg = mock_console.print.call_args[0][0] - assert "👤 User" in str(panel_arg.title) + assert "👤 You" in str(panel_arg.title) assert "(" not in str(panel_arg.title) # No timestamp @patch("nova.utils.formatting.console")