From 5205021e94e7378562e0135c6850e18fc5203f49 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:35:08 +0000 Subject: [PATCH] test: update test assertions to use llm parameter Co-Authored-By: Joe Moura --- tests/crew_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/crew_test.py b/tests/crew_test.py index 39b56bdf2..d19106951 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["openai_model_name"] == "gpt-4" + assert kwargs["llm"] == "gpt-4" def test_manager_llm_requirement_for_hierarchical_process(): task = Task( @@ -2846,7 +2846,7 @@ def test_crew_testing_function(kickoff_mock, copy_mock, crew_evaluator): crew_evaluator.assert_has_calls( [ - mock.call(crew, "gpt-4o-mini"), + mock.call(crew, llm="gpt-4o-mini"), mock.call().set_iteration(1), mock.call().set_iteration(2), mock.call().print_crew_evaluation_result(),