opentelemetry-instrumentation-openai 0.33.0__tar.gz → 0.33.1__tar.gz

{opentelemetry_instrumentation_openai-0.33.0 → opentelemetry_instrumentation_openai-0.33.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opentelemetry-instrumentation-openai
-Version: 0.33.0
+Version: 0.33.1
 Summary: OpenTelemetry OpenAI instrumentation
 Home-page: https://github.com/traceloop/openllmetry/tree/main/packages/opentelemetry-instrumentation-openai
 License: Apache-2.0

{opentelemetry_instrumentation_openai-0.33.0 → opentelemetry_instrumentation_openai-0.33.1}/opentelemetry/instrumentation/openai/shared/chat_wrappers.py

@@ -432,7 +432,11 @@ def _set_completions(span, choices):
             return
 
         _set_span_attribute(span, f"{prefix}.role", message.get("role"))
-        _set_span_attribute(span, f"{prefix}.content", message.get("content"))
+
+        if message.get("refusal"):
+            _set_span_attribute(span, f"{prefix}.refusal", message.get("refusal"))
+        else:
+            _set_span_attribute(span, f"{prefix}.content", message.get("content"))
 
         function_call = message.get("function_call")
         if function_call:

{opentelemetry_instrumentation_openai-0.33.0 → opentelemetry_instrumentation_openai-0.33.1}/opentelemetry/instrumentation/openai/v1/__init__.py

@@ -197,6 +197,32 @@ class OpenAIV1Instrumentor(BaseInstrumentor):
                 "Assistants.create",
                 assistants_create_wrapper(tracer),
             )
+            wrap_function_wrapper(
+                "openai.resources.beta.chat.completions",
+                "Completions.parse",
+                chat_wrapper(
+                    tracer,
+                    tokens_histogram,
+                    chat_choice_counter,
+                    duration_histogram,
+                    chat_exception_counter,
+                    streaming_time_to_first_token,
+                    streaming_time_to_generate,
+                ),
+            )
+            wrap_function_wrapper(
+                "openai.resources.beta.chat.completions",
+                "AsyncCompletions.parse",
+                achat_wrapper(
+                    tracer,
+                    tokens_histogram,
+                    chat_choice_counter,
+                    duration_histogram,
+                    chat_exception_counter,
+                    streaming_time_to_first_token,
+                    streaming_time_to_generate,
+                ),
+            )
             wrap_function_wrapper(
                 "openai.resources.beta.threads.runs",
                 "Runs.create",
@@ -217,7 +243,7 @@ class OpenAIV1Instrumentor(BaseInstrumentor):
                 "Messages.list",
                 messages_list_wrapper(tracer),
             )
-        except AttributeError:
+        except (AttributeError, ModuleNotFoundError):
             pass
 
     def _uninstrument(self, **kwargs):

opentelemetry_instrumentation_openai-0.33.1/opentelemetry/instrumentation/openai/version.py

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

{opentelemetry_instrumentation_openai-0.33.0 → opentelemetry_instrumentation_openai-0.33.1}/pyproject.toml

@@ -8,7 +8,7 @@ show_missing = true
 
 [tool.poetry]
 name = "opentelemetry-instrumentation-openai"
-version = "0.33.0"
+version = "0.33.1"
 description = "OpenTelemetry OpenAI instrumentation"
 authors = [
   "Gal Kleinman <gal@traceloop.com>",
@@ -39,7 +39,7 @@ pytest = "^8.2.2"
 pytest-sugar = "1.0.0"
 vcrpy = "^6.0.1"
 pytest-recording = "^0.13.1"
-openai = {extras = ["datalib"], version = "^1.31.1"}
+openai = {extras = ["datalib"], version = ">=1.50.0"}
 opentelemetry-sdk = "^1.27.0"
 pytest-asyncio = "^0.23.7"
 

opentelemetry_instrumentation_openai-0.33.0/opentelemetry/instrumentation/openai/version.py

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