diff --git a/tests/llm_test.py b/tests/llm_test.py index 5db37eedb..f80637c60 100644 --- a/tests/llm_test.py +++ b/tests/llm_test.py @@ -376,6 +376,7 @@ def get_weather_tool_schema(): def test_context_window_exceeded_error_handling(): """Test that litellm.ContextWindowExceededError is converted to LLMContextLengthExceededException.""" from litellm.exceptions import ContextWindowExceededError + from crewai.utilities.exceptions.context_window_exceeding_exception import ( LLMContextLengthExceededException, )