opentelemetry-instrumentation-replicate 0.16.5__tar.gz → 0.16.6__tar.gz

{opentelemetry_instrumentation_replicate-0.16.5 → opentelemetry_instrumentation_replicate-0.16.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opentelemetry-instrumentation-replicate
-Version: 0.16.5
+Version: 0.16.6
 Summary: OpenTelemetry Replicate instrumentation
 Home-page: https://github.com/traceloop/openllmetry/tree/main/packages/opentelemetry-instrumentation-replicate
 License: Apache-2.0

{opentelemetry_instrumentation_replicate-0.16.5 → opentelemetry_instrumentation_replicate-0.16.6}/opentelemetry/instrumentation/replicate/__init__.py

@@ -1,8 +1,11 @@
 """OpenTelemetry Replicate instrumentation"""
+
 import logging
 import os
 import types
 from typing import Collection
+from opentelemetry.instrumentation.replicate.config import Config
+from opentelemetry.instrumentation.replicate.utils import dont_throw
 from wrapt import wrap_function_wrapper
 
 from opentelemetry import context as context_api
@@ -88,19 +91,7 @@ def _set_input_attributes(span, args, kwargs):
     return
 
 
-def _set_span_completions(span, response):
-    if response is None:
-        return
-    if isinstance(response, list):
-        for index, item in enumerate(response):
-            prefix = f"{SpanAttributes.LLM_COMPLETIONS}.{index}"
-            _set_span_attribute(span, f"{prefix}.content", item)
-    elif isinstance(response, str):
-        _set_span_attribute(
-            span, f"{SpanAttributes.LLM_COMPLETIONS}.0.content", response
-        )
-
-
+@dont_throw
 def _set_response_attributes(span, response):
     if should_send_prompts():
         if isinstance(response, list):
@@ -128,28 +119,17 @@ def _build_from_streaming_response(span, response):
     span.end()
 
 
+@dont_throw
 def _handle_request(span, args, kwargs):
-    try:
-        if span.is_recording():
-            _set_input_attributes(span, args, kwargs)
-
-    except Exception as ex:  # pylint: disable=broad-except
-        logger.warning(
-            "Failed to set input attributes for replicate span, error: %s", str(ex)
-        )
+    if span.is_recording():
+        _set_input_attributes(span, args, kwargs)
 
 
+@dont_throw
 def _handle_response(span, response):
-    try:
-        if span.is_recording():
-            _set_response_attributes(span, response)
-
-    except Exception as ex:  # pylint: disable=broad-except
-        logger.warning(
-            "Failed to set response attributes for replicate span, error: %s",
-            str(ex),
-        )
     if span.is_recording():
+        _set_response_attributes(span, response)
+
         span.set_status(Status(StatusCode.OK))
 
 
@@ -198,6 +178,10 @@ def _wrap(tracer, to_wrap, wrapped, instance, args, kwargs):
 class ReplicateInstrumentor(BaseInstrumentor):
     """An instrumentor for Replicate's client library."""
 
+    def __init__(self, exception_logger=None):
+        super().__init__()
+        Config.exception_logger = exception_logger
+
     def instrumentation_dependencies(self) -> Collection[str]:
         return _instruments
 

opentelemetry_instrumentation_replicate-0.16.6/opentelemetry/instrumentation/replicate/config.py

@@ -0,0 +1,2 @@
+class Config:
+    exception_logger = None

opentelemetry_instrumentation_replicate-0.16.6/opentelemetry/instrumentation/replicate/utils.py

@@ -0,0 +1,23 @@
+import logging
+from opentelemetry.instrumentation.replicate.config import Config
+
+
+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
+    """
+    # Obtain a logger specific to the function's module
+    logger = logging.getLogger(func.__module__)
+
+    def wrapper(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except Exception as e:
+            logger.warning("Failed to execute %s, error: %s", func.__name__, str(e))
+            if Config.exception_logger:
+                Config.exception_logger(e)
+
+    return wrapper

opentelemetry_instrumentation_replicate-0.16.6/opentelemetry/instrumentation/replicate/version.py

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

{opentelemetry_instrumentation_replicate-0.16.5 → opentelemetry_instrumentation_replicate-0.16.6}/pyproject.toml

@@ -8,7 +8,7 @@ show_missing = true
 
 [tool.poetry]
 name = "opentelemetry-instrumentation-replicate"
-version = "0.16.5"
+version = "0.16.6"
 description = "OpenTelemetry Replicate instrumentation"
 authors = [ "Kartik Prajapati <kartik@ktklab.org>" ]
 repository = "https://github.com/traceloop/openllmetry/tree/main/packages/opentelemetry-instrumentation-replicate"

opentelemetry_instrumentation_replicate-0.16.5/opentelemetry/instrumentation/replicate/version.py

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