fixing mock_agent_ops_provider

This commit is contained in:
João Moura
2024-08-10 17:21:21 -03:00
parent b80ea04fe1
commit ac48b7a669
2 changed files with 8 additions and 2 deletions

View File

@@ -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"):

View File

@@ -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"):