Skip to content

Commit

Permalink
Revert "cleanup"
Browse files Browse the repository at this point in the history
This reverts commit 7f5635f.
  • Loading branch information
bboynton97 committed Apr 20, 2024
1 parent 4d1b460 commit cea33d9
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 2 deletions.
8 changes: 7 additions & 1 deletion src/crewai/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,13 @@
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()
Expand Down
5 changes: 4 additions & 1 deletion src/crewai/tools/tool_usage.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,10 @@
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"]

Expand Down

0 comments on commit cea33d9

Please sign in to comment.