From 2086a4b5306b9e3779b59005c77c7fb91701ebbe Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Sun, 9 Feb 2025 21:32:44 +0000 Subject: [PATCH] fix: update backward compatibility test Co-Authored-By: Joe Moura --- src/crewai/crew.py | 2 +- tests/crew_test.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/crewai/crew.py b/src/crewai/crew.py index 2a78f7307..2582acb7d 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -1092,7 +1092,7 @@ class Crew(BaseModel): inputs, llm or openai_model_name or "gpt-4o-mini", ) - evaluator = CrewEvaluator(test_crew, llm or openai_model_name or "gpt-4o-mini") + evaluator = CrewEvaluator(test_crew, openai_model_name or "gpt-4o-mini") for i in range(1, n_iterations + 1): evaluator.set_iteration(i) diff --git a/tests/crew_test.py b/tests/crew_test.py index bd18ddea8..39b56bdf2 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -307,7 +307,7 @@ def test_crew_test_backward_compatibility(mock_copy, mock_evaluator): crew.test(2, openai_model_name="gpt-4") mock_evaluator.assert_called_once() _, kwargs = mock_evaluator.call_args - assert kwargs["llm"] == "gpt-4" + assert kwargs["openai_model_name"] == "gpt-4" def test_manager_llm_requirement_for_hierarchical_process(): task = Task(