From 031c2d165a952eb99ca9ac17c4277dfcece878b3 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Thu, 10 Apr 2025 15:59:37 +0000
Subject: [PATCH] fix(tests): Update task evaluator test assertion for agent
 arg

Co-Authored-By: Joe Moura <joao@crewai.com>
---
 tests/utilities/evaluators/test_task_evaluator.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/utilities/evaluators/test_task_evaluator.py b/tests/utilities/evaluators/test_task_evaluator.py
index e4de1db62..cfcc44fcc 100644
--- a/tests/utilities/evaluators/test_task_evaluator.py
+++ b/tests/utilities/evaluators/test_task_evaluator.py
@@ -46,6 +46,7 @@ def test_evaluate_training_data(converter_mock):
     converter_mock.assert_has_calls(
         [
             mock.call(
+                agent=original_agent, # Add agent argument
                 llm=original_agent.llm,
                 text="Assess the quality of the training data based on the llm output, human feedback , and llm "
                 "output improved result.\n\nIteration: data1\nInitial Output:\nInitial output 1\n\nHuman Feedback:\nHuman feedback "