opentelemetry-instrumentation-openai 0.33.8__tar.gz → 0.33.9__tar.gz

{opentelemetry_instrumentation_openai-0.33.8 → opentelemetry_instrumentation_openai-0.33.9}/PKG-INFO

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

@@ -231,6 +231,8 @@ def is_streaming_response(response):
 
 
 def model_as_dict(model):
+    if isinstance(model, dict):
+        return model
     if version("pydantic") < "2.0.0":
         return model.dict()
     if hasattr(model, "model_dump"):

{opentelemetry_instrumentation_openai-0.33.8 → opentelemetry_instrumentation_openai-0.33.9}/opentelemetry/instrumentation/openai/shared/chat_wrappers.py

@@ -353,6 +353,7 @@ async def _process_image_item(item, trace_id, span_id, message_index, content_in
     return {"type": "image_url", "image_url": {"url": url}}
 
 
+@dont_throw
 async def _set_prompts(span, messages):
     if not span.is_recording() or messages is None:
         return

{opentelemetry_instrumentation_openai-0.33.8 → opentelemetry_instrumentation_openai-0.33.9}/opentelemetry/instrumentation/openai/utils.py

@@ -115,26 +115,32 @@ async def start_as_current_span_async(tracer, *args, **kwargs):
 def dont_throw(func):
     """
     A decorator that wraps the passed in function and logs exceptions instead of throwing them.
-
-    @param func: The function to wrap
-    @return: The wrapper function
+    Works for both synchronous and asynchronous functions.
     """
-    # Obtain a logger specific to the function's module
     logger = logging.getLogger(func.__module__)
 
-    def wrapper(*args, **kwargs):
+    async def async_wrapper(*args, **kwargs):
         try:
-            return func(*args, **kwargs)
+            return await func(*args, **kwargs)
         except Exception as e:
-            logger.debug(
-                "OpenLLMetry failed to trace in %s, error: %s",
-                func.__name__,
-                traceback.format_exc(),
-            )
-            if Config.exception_logger:
-                Config.exception_logger(e)
+            _handle_exception(e, func, logger)
 
-    return wrapper
+    def sync_wrapper(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except Exception as e:
+            _handle_exception(e, func, logger)
+
+    def _handle_exception(e, func, logger):
+        logger.debug(
+            "OpenLLMetry failed to trace in %s, error: %s",
+            func.__name__,
+            traceback.format_exc(),
+        )
+        if Config.exception_logger:
+            Config.exception_logger(e)
+
+    return async_wrapper if asyncio.iscoroutinefunction(func) else sync_wrapper
 
 
 def run_async(method):

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

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

{opentelemetry_instrumentation_openai-0.33.8 → opentelemetry_instrumentation_openai-0.33.9}/pyproject.toml

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

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

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