Minor fixes

Author: karthikscale3

src/examples/crewai_example/simple_agent/agents.py

@@ -28,5 +28,5 @@ def create_poet_agent(self):
             goal="""Create a poem that captures the essence of a given theme or emotion""",
             allow_delegation=False,
             verbose=True,
-            llm=self.ollama,
+            llm=self.open_ai,
         )

src/examples/crewai_example/simple_agent/main.py

@@ -1,10 +1,12 @@
+import sys
+
+sys.path.insert(0, '/Users/karthikkalyanaraman/work/langtrace/langtrace-python-sdk/src')
+
 from crewai import Crew
-from textwrap import dedent
-from .agents import PoetryAgents
-from .tasks import PoetryTasks
+from agents import PoetryAgents
+from tasks import PoetryTasks
 from langtrace_python_sdk import langtrace
 from dotenv import load_dotenv
-import agentops
 
 load_dotenv()
 langtrace.init()

src/langtrace_python_sdk/instrumentation/crewai/patch.py

@@ -197,6 +197,12 @@ def set_task_attributes(self):
 
     def _parse_agents(self, agents):
         for agent in agents:
+            model = None
+            if agent.llm is not None:
+                if hasattr(agent.llm, "model"):
+                    model = agent.llm.model
+                elif hasattr(agent.llm, "model_name"):
+                    model = agent.llm.model_name
             self.crew["agents"].append(
                 {
                     "id": str(agent.id),
@@ -209,7 +215,7 @@ def _parse_agents(self, agents):
                     "allow_delegation": agent.allow_delegation,
                     "tools": agent.tools,
                     "max_iter": agent.max_iter,
-                    "llm": str(agent.llm.model),
+                    "llm": str(model if model is not None else ""),
                 }
             )