From 1c97b916d99ac1cc86e6af4f6426fbea3cf05b6e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Sat, 10 Aug 2024 17:26:45 -0300 Subject: [PATCH] fixing mock_agent_ops_provider --- src/crewai/agent.py | 4 ++-- src/crewai/utilities/evaluators/task_evaluator.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/crewai/agent.py b/src/crewai/agent.py index a64dc489ce..e137a1e412 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -20,8 +20,8 @@ from crewai.utilities.training_handler import CrewTrainingHandler -def mock_agent_ops_provider(*args, **kwargs): - def track_agent(): +def mock_agent_ops_provider(): + def track_agent(*args, **kwargs): def noop(f): return f diff --git a/src/crewai/utilities/evaluators/task_evaluator.py b/src/crewai/utilities/evaluators/task_evaluator.py index 1ebcdb5983..ed2df41ce9 100644 --- a/src/crewai/utilities/evaluators/task_evaluator.py +++ b/src/crewai/utilities/evaluators/task_evaluator.py @@ -8,8 +8,8 @@ from crewai.utilities.pydantic_schema_parser import PydanticSchemaParser -def mock_agent_ops_provider(*args, **kwargs): - def track_agent(): +def mock_agent_ops_provider(): + def track_agent(*args, **kwargs): def noop(f): return f