diff --git a/src/crewai/agent.py b/src/crewai/agent.py index 80bd022db..a64dc489c 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -19,14 +19,17 @@ from crewai.utilities.constants import TRAINED_AGENTS_DATA_FILE, TRAINING_DATA_F 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 index 254bfc573..1ebcdb598 100644 --- a/src/crewai/utilities/evaluators/task_evaluator.py +++ b/src/crewai/utilities/evaluators/task_evaluator.py @@ -7,14 +7,17 @@ from pydantic import BaseModel, Field 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"):