opentelemetry-instrumentation-crewai 0.43.1__py3-none-any.whl → 0.44.1__py3-none-any.whl

opentelemetry/instrumentation/crewai/crewai_span_attributes.py

@@ -1,4 +1,5 @@
 from opentelemetry.trace import Span
+from opentelemetry.semconv_ai import SpanAttributes
 import json
 
 
@@ -13,7 +14,7 @@ class CrewAISpanAttributes:
     def __init__(self, span: Span, instance) -> None:
         self.span = span
         self.instance = instance
-        self.crew = {"tasks": [], "agents": [], "llms": []}
+        self.crew = {"tasks": [], "agents": []}
         self.process_instance()
 
     def process_instance(self):
@@ -44,9 +45,37 @@ class CrewAISpanAttributes:
             self._set_attribute(f"crewai.task.{key}", value)
 
     def _process_llm(self):
-        llm_data = self._populate_llm_attributes()
-        for key, value in llm_data.items():
-            self._set_attribute(f"crewai.llm.{key}", value)
+        fields = [
+            "model",
+            "temperature",
+            "top_p",
+            "n",
+            "stop",
+            "max_completion_tokens",
+            "max_tokens",
+            "presence_penalty",
+            "frequency_penalty",
+            "seed"
+        ]
+
+        for field in fields:
+            value = getattr(self.instance, field, None)
+            if value is None:
+                continue
+            if field == "model":
+                self._set_attribute(SpanAttributes.LLM_REQUEST_MODEL, value)
+            elif field == "temperature":
+                self._set_attribute(SpanAttributes.LLM_REQUEST_TEMPERATURE, value)
+            elif field == "top_p":
+                self._set_attribute(SpanAttributes.LLM_REQUEST_TOP_P, value)
+            elif field == "max_tokens":
+                self._set_attribute(SpanAttributes.LLM_REQUEST_MAX_TOKENS, value)
+            elif field == "presence_penalty":
+                self._set_attribute(SpanAttributes.LLM_PRESENCE_PENALTY, value)
+            elif field == "frequency_penalty":
+                self._set_attribute(SpanAttributes.LLM_FREQUENCY_PENALTY, value)
+            else:
+                self._set_attribute(f"llm.{field}", value)
 
     def _populate_crew_attributes(self):
         for key, value in self.instance.__dict__.items():
@@ -56,8 +85,6 @@ class CrewAISpanAttributes:
                 self._parse_tasks(value)
             elif key == "agents":
                 self._parse_agents(value)
-            elif key == "llms":
-                self._parse_llms(value)
             else:
                 self.crew[key] = str(value)
 
@@ -92,20 +119,6 @@ class CrewAISpanAttributes:
             for task in tasks
         ]
 
-    def _parse_llms(self, llms):
-        self.crew["tasks"] = [
-            {
-                "temperature": llm.temperature,
-                "max_tokens": llm.max_tokens,
-                "max_completion_tokens": llm.max_completion_tokens,
-                "top_p": llm.top_p,
-                "n": llm.n,
-                "seed": llm.seed,
-                "base_url": llm.base_url,
-                "api_version": llm.api_version, }
-            for llm in llms
-        ]
-
     def _extract_agent_data(self, agent):
         model = (
             getattr(agent.llm, "model", None)
@@ -146,5 +159,5 @@ class CrewAISpanAttributes:
         )
 
     def _set_attribute(self, key, value):
-        if value:
+        if value is not None:
             set_span_attribute(self.span, key, str(value) if isinstance(value, list) else value)

{opentelemetry_instrumentation_crewai-0.43.1.dist-info → opentelemetry_instrumentation_crewai-0.44.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: opentelemetry-instrumentation-crewai
-Version: 0.43.1
+Version: 0.44.1
 Summary: OpenTelemetry crewAI instrumentation
 License: Apache-2.0
 Author: Gal Kleinman

{opentelemetry_instrumentation_crewai-0.43.1.dist-info → opentelemetry_instrumentation_crewai-0.44.1.dist-info}/RECORD

@@ -1,8 +1,8 @@
 opentelemetry/instrumentation/crewai/__init__.py,sha256=zdNyWV62d5IaueXR9clUD_c9F32MF39AIchZf66XI2c,245
-opentelemetry/instrumentation/crewai/crewai_span_attributes.py,sha256=9ax6fn0IT4dqH2iaRG8VIZ0SWTkxQrMTiF7DWUdwfy0,4945
+opentelemetry/instrumentation/crewai/crewai_span_attributes.py,sha256=9pLK-fFbbq4PQiHcj58wlh1h0PQ0zruOxccewEaPKno,5523
 opentelemetry/instrumentation/crewai/instrumentation.py,sha256=AhQpPgpgtx9DKdnRDcrCKXn4AIuErdfznxfLlfs9OTc,7913
 opentelemetry/instrumentation/crewai/version.py,sha256=ITH8IVCEQOekRajZpPo6sKNlMBNDRQRnzipIN0dqXoc,23
-opentelemetry_instrumentation_crewai-0.43.1.dist-info/METADATA,sha256=8Trm2dYWDWA4QKekavc_NscD5pJq_QvxUSckVKQIDeA,2095
-opentelemetry_instrumentation_crewai-0.43.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-opentelemetry_instrumentation_crewai-0.43.1.dist-info/entry_points.txt,sha256=4Zs98MjB2HIaX8fk8evs41x1J1xBNfztvPN428bf_b8,93
-opentelemetry_instrumentation_crewai-0.43.1.dist-info/RECORD,,
+opentelemetry_instrumentation_crewai-0.44.1.dist-info/METADATA,sha256=DZ-I9BUBC5aduf13yvageQ88Z3iq36M7zeK-lBSOJag,2095
+opentelemetry_instrumentation_crewai-0.44.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+opentelemetry_instrumentation_crewai-0.44.1.dist-info/entry_points.txt,sha256=4Zs98MjB2HIaX8fk8evs41x1J1xBNfztvPN428bf_b8,93
+opentelemetry_instrumentation_crewai-0.44.1.dist-info/RECORD,,