From fd7de7f2ebef6f5c68a3a32855c1a0d53e543beb Mon Sep 17 00:00:00 2001 From: Braelyn Boynton Date: Fri, 19 Apr 2024 19:08:22 -0700 Subject: [PATCH] Revert "Revert "cleanup"" This reverts commit cea33d9a5d37289b9e257046c1354057b27fbe7f. --- src/crewai/agent.py | 8 +------- src/crewai/tools/tool_usage.py | 5 +---- 2 files changed, 2 insertions(+), 11 deletions(-) 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 ba9ea19dc..da3f1be50 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"]