From cea33d9a5d37289b9e257046c1354057b27fbe7f Mon Sep 17 00:00:00 2001 From: Braelyn Boynton Date: Fri, 19 Apr 2024 19:02:20 -0700 Subject: [PATCH] Revert "cleanup" This reverts commit 7f5635fb9ef9aaafc4283fdf1a892d8b5c425220. --- src/crewai/agent.py | 8 +++++++- src/crewai/tools/tool_usage.py | 5 ++++- 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/src/crewai/agent.py b/src/crewai/agent.py index ac080f2d0..20b47dbe9 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -24,7 +24,13 @@ 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 -from agentops.agent import track_agent +try: + from agentops.agent import track_agent +except ImportError: + def track_agent(): + def noop(f): + return f + return noop @track_agent() diff --git a/src/crewai/tools/tool_usage.py b/src/crewai/tools/tool_usage.py index 83f0dd928..c5bbdbd33 100644 --- a/src/crewai/tools/tool_usage.py +++ b/src/crewai/tools/tool_usage.py @@ -9,7 +9,10 @@ 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 -import agentops +try: + import agentops +except ImportError: + agentops = None OPENAI_BIGGER_MODELS = ["gpt-4"]