Merge pull request #252 from Scale3-Labs/development

Skip missing module instrumentation for crewai + Add llm.system attribute

Author: alizenhom

src/langtrace_python_sdk/instrumentation/crewai/instrumentation.py

@@ -33,21 +33,24 @@ def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")
         tracer = get_tracer(__name__, "", tracer_provider)
         version = v("crewai")
-        _W(
-            "crewai.crew",
-            "Crew.kickoff",
-            patch_crew("Crew.kickoff", version, tracer),
-        )
-        _W(
-            "crewai.agent",
-            "Agent.execute_task",
-            patch_crew("Agent.execute_task", version, tracer),
-        )
-        _W(
-            "crewai.task",
-            "Task.execute",
-            patch_crew("Task.execute", version, tracer),
-        )
+        try:
+            _W(
+                "crewai.crew",
+                "Crew.kickoff",
+                patch_crew("Crew.kickoff", version, tracer),
+            )
+            _W(
+                "crewai.agent",
+                "Agent.execute_task",
+                patch_crew("Agent.execute_task", version, tracer),
+            )
+            _W(
+                "crewai.task",
+                "Task.execute",
+                patch_crew("Task.execute", version, tracer),
+            )
+        except Exception as e:
+            pass
 
     def _uninstrument(self, **kwargs):
         pass

src/langtrace_python_sdk/utils/llm.py

@@ -88,6 +88,7 @@ def get_langtrace_attributes(version, service_provider, vendor_type="llm"):
         SpanAttributes.LANGTRACE_SERVICE_VERSION: version,
         SpanAttributes.LANGTRACE_SERVICE_NAME: service_provider,
         SpanAttributes.LANGTRACE_SERVICE_TYPE: vendor_type,
+        SpanAttributes.LLM_SYSTEM: service_provider,
     }
 
 

src/langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "2.2.5"
+__version__ = "2.2.6"