handling n to summarize prompt

2026-01-10 16:48:30 +00:00 · 2024-07-31 22:19:47 -07:00
parent 9f31fec084
commit 9393373ff0
4 changed files with 35 additions and 24 deletions
--- a/tests/agent_test.py
+++ b/tests/agent_test.py
@@ -1084,7 +1084,7 @@ def test_handle_context_length_exceeds_limit_cli_no():
        private_mock.assert_called_once()
        with patch("crewai.agents.executor.prompt") as mock_prompt:
            mock_prompt.return_value = "n"
-
+            pytest.raises(SystemExit)
            with patch.object(
                CrewAgentExecutor, "_handle_context_length"
            ) as mock_handle_context: