From 7f5635fb9ef9aaafc4283fdf1a892d8b5c425220 Mon Sep 17 00:00:00 2001 From: Braelyn Boynton Date: Wed, 17 Apr 2024 17:19:38 -0700 Subject: [PATCH] cleanup --- 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 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"]