diff --git a/tests/cli/test_restart_command.py b/tests/cli/test_restart_command.py index 8b079d4e7..697d5fc17 100644 --- a/tests/cli/test_restart_command.py +++ b/tests/cli/test_restart_command.py @@ -148,7 +148,7 @@ class TestRestartCommand: assert response is not None assert "Model: test-model" in response.content assert "Tokens: 0 in / 0 out" in response.content - assert "Context: 20k/64k (31%)" in response.content + assert "Context: 20k/65k (31%)" in response.content assert "Session: 3 messages" in response.content assert "Uptime: 2m 5s" in response.content assert response.metadata == {"render_as": "text"} @@ -186,7 +186,7 @@ class TestRestartCommand: assert response is not None assert "Tokens: 1200 in / 34 out" in response.content - assert "Context: 1k/64k (1%)" in response.content + assert "Context: 1k/65k (1%)" in response.content @pytest.mark.asyncio async def test_process_direct_preserves_render_metadata(self):