opentelemetry-instrumentation-openai 0.31.2__tar.gz → 0.31.3__tar.gz

{opentelemetry_instrumentation_openai-0.31.2 → opentelemetry_instrumentation_openai-0.31.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opentelemetry-instrumentation-openai
-Version: 0.31.2
+Version: 0.31.3
 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.31.2 → opentelemetry_instrumentation_openai-0.31.3}/opentelemetry/instrumentation/openai/__init__.py

@@ -1,4 +1,5 @@
-from typing import Callable, Collection
+from typing import Callable, Collection, Optional
+from typing_extensions import Coroutine
 
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 
@@ -19,7 +20,9 @@ class OpenAIInstrumentor(BaseInstrumentor):
         enrich_token_usage: bool = False,
         exception_logger=None,
         get_common_metrics_attributes: Callable[[], dict] = lambda: {},
-        upload_base64_image: Callable[[str, str, str, str], str] = lambda *args: "",
+        upload_base64_image: Optional[
+            Callable[[str, str, str, str], Coroutine[None, None, str]]
+        ] = lambda *args: "",
     ):
         super().__init__()
         Config.enrich_assistant = enrich_assistant

{opentelemetry_instrumentation_openai-0.31.2 → opentelemetry_instrumentation_openai-0.31.3}/opentelemetry/instrumentation/openai/shared/chat_wrappers.py

@@ -1,3 +1,4 @@
+import asyncio
 import copy
 import json
 import logging
@@ -69,15 +70,15 @@ def chat_wrapper(
         SUPPRESS_LANGUAGE_MODEL_INSTRUMENTATION_KEY
     ):
         return wrapped(*args, **kwargs)
-
     # span needs to be opened and closed manually because the response is a generator
+
     span = tracer.start_span(
         SPAN_NAME,
         kind=SpanKind.CLIENT,
         attributes={SpanAttributes.LLM_REQUEST_TYPE: LLM_REQUEST_TYPE.value},
     )
 
-    _handle_request(span, kwargs, instance)
+    asyncio.run(_handle_request(span, kwargs, instance))
 
     try:
         start_time = time.time()
@@ -167,7 +168,7 @@ async def achat_wrapper(
         kind=SpanKind.CLIENT,
         attributes={SpanAttributes.LLM_REQUEST_TYPE: LLM_REQUEST_TYPE.value},
     )
-    _handle_request(span, kwargs, instance)
+    await _handle_request(span, kwargs, instance)
 
     try:
         start_time = time.time()
@@ -236,11 +237,11 @@ async def achat_wrapper(
 
 
 @dont_throw
-def _handle_request(span, kwargs, instance):
+async def _handle_request(span, kwargs, instance):
     _set_request_attributes(span, kwargs)
     _set_client_attributes(span, instance)
     if should_send_prompts():
-        _set_prompts(span, kwargs.get("messages"))
+        await _set_prompts(span, kwargs.get("messages"))
         if kwargs.get("functions"):
             _set_functions_attributes(span, kwargs.get("functions"))
         elif kwargs.get("tools"):
@@ -331,33 +332,28 @@ def _is_base64_image(item):
     if not isinstance(item, dict):
         return False
 
-    if not isinstance(item.get('image_url'), dict):
+    if not isinstance(item.get("image_url"), dict):
         return False
 
-    if 'data:image/' not in item.get('image_url', {}).get('url', ''):
+    if "data:image/" not in item.get("image_url", {}).get("url", ""):
         return False
 
     return True
 
 
-def _process_image_item(item, trace_id, span_id, message_index, content_index):
+async def _process_image_item(item, trace_id, span_id, message_index, content_index):
     if not Config.upload_base64_image:
         return item
 
     image_format = item["image_url"]["url"].split(";")[0].split("/")[1]
     image_name = f"message_{message_index}_content_{content_index}.{image_format}"
     base64_string = item["image_url"]["url"].split(",")[1]
-    url = Config.upload_base64_image(trace_id, span_id, image_name, base64_string)
+    url = await Config.upload_base64_image(trace_id, span_id, image_name, base64_string)
 
-    return {
-        'type': 'image_url',
-        'image_url': {
-            'url': url
-        }
-    }
+    return {"type": "image_url", "image_url": {"url": url}}
 
 
-def _set_prompts(span, messages):
+async def _set_prompts(span, messages):
     if not span.is_recording() or messages is None:
         return
 
@@ -369,9 +365,13 @@ def _set_prompts(span, messages):
             content = copy.deepcopy(msg.get("content"))
             if isinstance(content, list):
                 content = [
-                    _process_image_item(item, span.context.trace_id, span.context.span_id, i, j)
-                    if _is_base64_image(item)
-                    else item
+                    (
+                        await _process_image_item(
+                            item, span.context.trace_id, span.context.span_id, i, j
+                        )
+                        if _is_base64_image(item)
+                        else item
+                    )
                     for j, item in enumerate(content)
                 ]
 

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

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

{opentelemetry_instrumentation_openai-0.31.2 → opentelemetry_instrumentation_openai-0.31.3}/pyproject.toml

@@ -8,7 +8,7 @@ show_missing = true
 
 [tool.poetry]
 name = "opentelemetry-instrumentation-openai"
-version = "0.31.2"
+version = "0.31.3"
 description = "OpenTelemetry OpenAI instrumentation"
 authors = [
   "Gal Kleinman <gal@traceloop.com>",

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

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