opentelemetry-instrumentation-groq 0.38.12__tar.gz → 0.39.0__tar.gz

{opentelemetry_instrumentation_groq-0.38.12 → opentelemetry_instrumentation_groq-0.39.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: opentelemetry-instrumentation-groq
-Version: 0.38.12
+Version: 0.39.0
 Summary: OpenTelemetry Groq instrumentation
 License: Apache-2.0
 Author: Gal Kleinman

{opentelemetry_instrumentation_groq-0.38.12 → opentelemetry_instrumentation_groq-0.39.0}/opentelemetry/instrumentation/groq/__init__.py

@@ -196,24 +196,35 @@ def _set_response_attributes(span, response, token_histogram):
             span, SpanAttributes.LLM_USAGE_TOTAL_TOKENS, usage.get("total_tokens")
         )
         set_span_attribute(
-            span,
-            SpanAttributes.LLM_USAGE_COMPLETION_TOKENS, completion_tokens
-        )
-        set_span_attribute(
-            span, SpanAttributes.LLM_USAGE_PROMPT_TOKENS, prompt_tokens
+            span, SpanAttributes.LLM_USAGE_COMPLETION_TOKENS, completion_tokens
         )
+        set_span_attribute(span, SpanAttributes.LLM_USAGE_PROMPT_TOKENS, prompt_tokens)
 
-    if isinstance(prompt_tokens, int) and prompt_tokens >= 0 and token_histogram is not None:
-        token_histogram.record(prompt_tokens, attributes={
-            SpanAttributes.LLM_TOKEN_TYPE: "input",
-            SpanAttributes.LLM_RESPONSE_MODEL: response.get("model")
-        })
+    if (
+        isinstance(prompt_tokens, int)
+        and prompt_tokens >= 0
+        and token_histogram is not None
+    ):
+        token_histogram.record(
+            prompt_tokens,
+            attributes={
+                SpanAttributes.LLM_TOKEN_TYPE: "input",
+                SpanAttributes.LLM_RESPONSE_MODEL: response.get("model"),
+            },
+        )
 
-    if isinstance(completion_tokens, int) and completion_tokens >= 0 and token_histogram is not None:
-        token_histogram.record(completion_tokens, attributes={
-            SpanAttributes.LLM_TOKEN_TYPE: "output",
-            SpanAttributes.LLM_RESPONSE_MODEL: response.get("model")
-        })
+    if (
+        isinstance(completion_tokens, int)
+        and completion_tokens >= 0
+        and token_histogram is not None
+    ):
+        token_histogram.record(
+            completion_tokens,
+            attributes={
+                SpanAttributes.LLM_TOKEN_TYPE: "output",
+                SpanAttributes.LLM_RESPONSE_MODEL: response.get("model"),
+            },
+        )
 
     choices = response.get("choices")
     if should_send_prompts() and choices:
@@ -575,7 +586,7 @@ class GroqInstrumentor(BaseInstrumentor):
                 token_histogram,
                 choice_counter,
                 duration_histogram,
-            ) = (None, None, None, None)
+            ) = (None, None, None)
 
         for wrapped_method in WRAPPED_METHODS:
             wrap_package = wrapped_method.get("package")

opentelemetry_instrumentation_groq-0.39.0/opentelemetry/instrumentation/groq/version.py

@@ -0,0 +1 @@
+__version__ = "0.39.0"

{opentelemetry_instrumentation_groq-0.38.12 → opentelemetry_instrumentation_groq-0.39.0}/pyproject.toml

@@ -8,7 +8,7 @@ show_missing = true
 
 [tool.poetry]
 name = "opentelemetry-instrumentation-groq"
-version = "0.38.12"
+version = "0.39.0"
 description = "OpenTelemetry Groq instrumentation"
 authors = ["Gal Kleinman <gal@traceloop.com>", "Nir Gazit <nir@traceloop.com>"]
 repository = "https://github.com/traceloop/openllmetry/tree/main/packages/opentelemetry-instrumentation-groq"

opentelemetry_instrumentation_groq-0.38.12/opentelemetry/instrumentation/groq/version.py

@@ -1 +0,0 @@
-__version__ = "0.38.12"