fixing mock_agent_ops_provider

crewAIInc · Aug 10, 2024 · 94b52cf · 94b52cf
1 parent 82b1db1
commit 94b52cf
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/crewai/agent.py b/src/crewai/agent.py
@@ -19,14 +19,17 @@
 from crewai.utilities.token_counter_callback import TokenCalcHandler
 from crewai.utilities.training_handler import CrewTrainingHandler
 
-def mock_agent_ops_provider():
+
+def mock_agent_ops_provider(*args, **kwargs):
     def track_agent():
         def noop(f):
             return f
 
         return noop
+
     return track_agent
 
+
 agentops = None
 
 if os.environ.get("AGENTOPS_API_KEY"):

diff --git a/src/crewai/utilities/evaluators/task_evaluator.py b/src/crewai/utilities/evaluators/task_evaluator.py
@@ -7,14 +7,17 @@
 from crewai.utilities import Converter
 from crewai.utilities.pydantic_schema_parser import PydanticSchemaParser
 
-def mock_agent_ops_provider():
+
+def mock_agent_ops_provider(*args, **kwargs):
     def track_agent():
         def noop(f):
             return f
 
         return noop
+
     return track_agent
 
+
 agentops = None
 
 if os.environ.get("AGENTOPS_API_KEY"):