diff --git a/src/crewai/agent.py b/src/crewai/agent.py index 20b47dbe9..ac080f2d0 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -24,13 +24,7 @@ from crewai.agents import CacheHandler, CrewAgentExecutor, CrewAgentParser, Tool from crewai.memory.contextual.contextual_memory import ContextualMemory from crewai.utilities import I18N, Logger, Prompts, RPMController from crewai.utilities.token_counter_callback import TokenCalcHandler, TokenProcess -try: - from agentops.agent import track_agent -except ImportError: - def track_agent(): - def noop(f): - return f - return noop +from agentops.agent import track_agent @track_agent() diff --git a/src/crewai/tools/tool_usage.py b/src/crewai/tools/tool_usage.py index 2489cba66..b11f4ec2a 100644 --- a/src/crewai/tools/tool_usage.py +++ b/src/crewai/tools/tool_usage.py @@ -9,10 +9,7 @@ from crewai.agents.tools_handler import ToolsHandler from crewai.telemetry import Telemetry from crewai.tools.tool_calling import InstructorToolCalling, ToolCalling from crewai.utilities import I18N, Converter, ConverterError, Printer -try: - import agentops -except ImportError: - agentops = None +import agentops OPENAI_BIGGER_MODELS = ["gpt-4"]