From c0005e112e30b85f7581c6bc02f3085202230875 Mon Sep 17 00:00:00 2001
From: Vikram Guhan Subbiah <45928972+tiovikram@users.noreply.github.com>
Date: Sat, 10 Aug 2024 05:47:13 -0700
Subject: [PATCH] AgentOps ENG-525: Decouple CrewAI and AgentOps (#1033)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* Make AgentOps import optional upon AGENTOPS_API_KEY
    being set

Co-authored-by: João Moura <joaomdmoura@gmail.com>
---
 src/crewai/agent.py                           | 19 ++++++++++++------
 src/crewai/crew.py                            | 12 +++++++----
 src/crewai/tools/tool_usage.py                | 11 ++++++----
 .../utilities/evaluators/task_evaluator.py    | 20 +++++++++++++------
 4 files changed, 42 insertions(+), 20 deletions(-)

diff --git a/src/crewai/agent.py b/src/crewai/agent.py
index 08ff2ad70..80bd022db 100644
--- a/src/crewai/agent.py
+++ b/src/crewai/agent.py
@@ -19,17 +19,24 @@ from crewai.utilities.constants import TRAINED_AGENTS_DATA_FILE, TRAINING_DATA_F
 from crewai.utilities.token_counter_callback import TokenCalcHandler
 from crewai.utilities.training_handler import CrewTrainingHandler
 
-agentops = None
-try:
-    import agentops  # type: ignore # Name "agentops" already defined on line 21
-    from agentops import track_agent
-except ImportError:
-
+def mock_agent_ops_provider():
     def track_agent():
         def noop(f):
             return f
 
         return noop
+    return track_agent
+
+agentops = None
+
+if os.environ.get("AGENTOPS_API_KEY"):
+    try:
+        import agentops  # type: ignore # Name "agentops" already defined on line 21
+        from agentops import track_agent
+    except ImportError:
+        track_agent = mock_agent_ops_provider()
+else:
+    track_agent = mock_agent_ops_provider()
 
 
 @track_agent()
diff --git a/src/crewai/crew.py b/src/crewai/crew.py
index 109cbd28f..98f7b0ae0 100644
--- a/src/crewai/crew.py
+++ b/src/crewai/crew.py
@@ -3,6 +3,7 @@ import json
 import uuid
 from concurrent.futures import Future
 from hashlib import md5
+import os
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from langchain_core.callbacks import BaseCallbackHandler
@@ -47,10 +48,13 @@ from crewai.utilities.planning_handler import CrewPlanner
 from crewai.utilities.task_output_storage_handler import TaskOutputStorageHandler
 from crewai.utilities.training_handler import CrewTrainingHandler
 
-try:
-    import agentops
-except ImportError:
-    agentops = None
+
+agentops = None
+if os.environ.get("AGENTOPS_API_KEY"):
+    try:
+        import agentops
+    except ImportError:
+        pass
 
 if TYPE_CHECKING:
     from crewai.pipeline.pipeline import Pipeline
diff --git a/src/crewai/tools/tool_usage.py b/src/crewai/tools/tool_usage.py
index f205dd098..3beb99461 100644
--- a/src/crewai/tools/tool_usage.py
+++ b/src/crewai/tools/tool_usage.py
@@ -1,5 +1,6 @@
 import ast
 from difflib import SequenceMatcher
+import os
 from textwrap import dedent
 from typing import Any, List, Union
 
@@ -11,10 +12,12 @@ 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
+agentops = None
+if os.environ.get("AGENTOPS_API_KEY"):
+    try:
+        import agentops
+    except ImportError:
+        pass
 
 OPENAI_BIGGER_MODELS = ["gpt-4o"]
 
diff --git a/src/crewai/utilities/evaluators/task_evaluator.py b/src/crewai/utilities/evaluators/task_evaluator.py
index 46cbef2d8..254bfc573 100644
--- a/src/crewai/utilities/evaluators/task_evaluator.py
+++ b/src/crewai/utilities/evaluators/task_evaluator.py
@@ -1,3 +1,4 @@
+import os
 from typing import List
 
 from langchain_openai import ChatOpenAI
@@ -6,16 +7,23 @@ from pydantic import BaseModel, Field
 from crewai.utilities import Converter
 from crewai.utilities.pydantic_schema_parser import PydanticSchemaParser
 
-agentops = None
-try:
-    from agentops import track_agent
-except ImportError:
-
-    def track_agent(name):
+def mock_agent_ops_provider():
+    def track_agent():
         def noop(f):
             return f
 
         return noop
+    return track_agent
+
+agentops = None
+
+if os.environ.get("AGENTOPS_API_KEY"):
+    try:
+        from agentops import track_agent
+    except ImportError:
+        track_agent = mock_agent_ops_provider()
+else:
+    track_agent = mock_agent_ops_provider()
 
 
 class Entity(BaseModel):