Rework AgentOps implementation for new agent executor

src/crewai/llm.py

@@ -1,5 +1,4 @@
 from typing import Any, Dict, List
-from litellm import completion
 import litellm
 
 
@@ -10,7 +9,7 @@ def __init__(self, model: str, stop: List[str] = [], callbacks: List[Any] = []):
         litellm.callbacks = callbacks
 
     def call(self, messages: List[Dict[str, str]]) -> Dict[str, Any]:
-        response = completion(
+        response = litellm.completion(
             stop=self.stop, model=self.model, messages=messages, num_retries=5
         )
         return response["choices"][0]["message"]["content"]

src/crewai/tools/tool_usage.py

@@ -11,11 +11,10 @@
 from crewai.utilities import I18N, Converter, ConverterError, Printer
 
 agentops = None
-if os.environ.get("AGENTOPS_API_KEY"):
-    try:
-        import agentops  # type: ignore
-    except ImportError:
-        pass
+try:
+    import agentops  # type: ignore
+except ImportError:
+    pass
 
 OPENAI_BIGGER_MODELS = ["gpt-4", "gpt-4o"]
 

src/crewai/utilities/internal_instructor.py

@@ -1,7 +1,7 @@
 from typing import Any, Optional, Type
 
 import instructor
-from litellm import completion
+import litellm
 
 
 class InternalInstructor:
@@ -29,7 +29,7 @@ def set_instructor(self):
             self.llm = self.agent.function_calling_llm or self.agent.llm
 
         self._client = instructor.from_litellm(
-            completion,
+            litellm.completion,
             mode=instructor.Mode.TOOLS,
         )