cleanup

AgentOps-AI · Apr 18, 2024 · 7f5635f · 7f5635f
1 parent 0ce8d14
commit 7f5635f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 11 deletions.
diff --git a/src/crewai/agent.py b/src/crewai/agent.py
@@ -24,13 +24,7 @@
 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
@@ -9,10 +9,7 @@
 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"]