From efa685ab49830afdfcc54381188247cea6978621 Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Thu, 23 Jan 2025 12:35:08 -0500 Subject: [PATCH] Fix tests --- tests/agent_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/agent_test.py b/tests/agent_test.py index 46b20004e..3ed51ebde 100644 --- a/tests/agent_test.py +++ b/tests/agent_test.py @@ -1638,13 +1638,13 @@ def test_litellm_auth_error_handling(): def test_crew_agent_executor_litellm_auth_error(): - """Test that CrewAgentExecutor properly identifies and handles LiteLLM authentication errors.""" + """Test that CrewAgentExecutor handles LiteLLM authentication errors by raising them.""" from litellm import AuthenticationError as LiteLLMAuthenticationError from crewai.agents.tools_handler import ToolsHandler from crewai.utilities import Printer - # Create an agent and executor with max_retry_limit=0 + # Create an agent and executor agent = Agent( role="test role", goal="test goal", @@ -1691,7 +1691,7 @@ def test_crew_agent_executor_litellm_auth_error(): # Verify error handling mock_printer.assert_any_call( - content="Authentication error with litellm occurred. Please check your API key and configuration.", + content="An unknown error occurred. Please check the details below.", color="red", ) mock_printer.assert_any_call(