opentelemetry-instrumentation-llamaindex 0.16.4__tar.gz → 0.16.6__tar.gz

{opentelemetry_instrumentation_llamaindex-0.16.4 → opentelemetry_instrumentation_llamaindex-0.16.6}/PKG-INFO

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

{opentelemetry_instrumentation_llamaindex-0.16.4 → opentelemetry_instrumentation_llamaindex-0.16.6}/opentelemetry/instrumentation/llamaindex/__init__.py

@@ -3,6 +3,7 @@
 import logging
 from typing import Collection
 
+from opentelemetry.instrumentation.llamaindex.config import Config
 from opentelemetry.trace import get_tracer
 
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
@@ -41,6 +42,10 @@ _instruments = ("llama-index >= 0.7.0",)
 class LlamaIndexInstrumentor(BaseInstrumentor):
     """An instrumentor for LlamaIndex SDK."""
 
+    def __init__(self, exception_logger=None):
+        super().__init__()
+        Config.exception_logger = exception_logger
+
     def instrumentation_dependencies(self) -> Collection[str]:
         return _instruments
 

opentelemetry_instrumentation_llamaindex-0.16.6/opentelemetry/instrumentation/llamaindex/config.py

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

{opentelemetry_instrumentation_llamaindex-0.16.4 → opentelemetry_instrumentation_llamaindex-0.16.6}/opentelemetry/instrumentation/llamaindex/custom_llm_instrumentor.py

@@ -10,6 +10,7 @@ from opentelemetry.instrumentation.utils import _SUPPRESS_INSTRUMENTATION_KEY
 from opentelemetry.semconv.ai import SpanAttributes, LLMRequestTypeValues
 from opentelemetry.instrumentation.llamaindex.utils import (
     _with_tracer_wrapper,
+    dont_throw,
     start_as_current_span_async,
     should_send_prompts,
 )
@@ -139,6 +140,7 @@ async def acomplete_wrapper(tracer, wrapped, instance: CustomLLM, args, kwargs):
         return response
 
 
+@dont_throw
 def _handle_request(span, llm_request_type, args, kwargs, instance: CustomLLM):
     _set_span_attribute(span, SpanAttributes.LLM_VENDOR, instance.__class__.__name__)
     _set_span_attribute(span, SpanAttributes.LLM_REQUEST_TYPE, llm_request_type.value)
@@ -164,6 +166,7 @@ def _handle_request(span, llm_request_type, args, kwargs, instance: CustomLLM):
     return
 
 
+@dont_throw
 def _handle_response(span, llm_request_type, instance, response):
     _set_span_attribute(
         span, SpanAttributes.LLM_RESPONSE_MODEL, instance.metadata.model_name

opentelemetry_instrumentation_llamaindex-0.16.6/opentelemetry/instrumentation/llamaindex/utils.py

@@ -0,0 +1,49 @@
+import os
+import logging
+from contextlib import asynccontextmanager
+
+from opentelemetry import context as context_api
+from opentelemetry.instrumentation.llamaindex.config import Config
+
+
+def _with_tracer_wrapper(func):
+    def _with_tracer(tracer):
+        def wrapper(wrapped, instance, args, kwargs):
+            return func(tracer, wrapped, instance, args, kwargs)
+
+        return wrapper
+
+    return _with_tracer
+
+
+@asynccontextmanager
+async def start_as_current_span_async(tracer, *args, **kwargs):
+    with tracer.start_as_current_span(*args, **kwargs) as span:
+        yield span
+
+
+def should_send_prompts():
+    return (
+        os.getenv("TRACELOOP_TRACE_CONTENT") or "true"
+    ).lower() == "true" or context_api.get_value("override_enable_content_tracing")
+
+
+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_llamaindex-0.16.6/opentelemetry/instrumentation/llamaindex/version.py

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

{opentelemetry_instrumentation_llamaindex-0.16.4 → opentelemetry_instrumentation_llamaindex-0.16.6}/pyproject.toml

@@ -8,7 +8,7 @@ show_missing = true
 
 [tool.poetry]
 name = "opentelemetry-instrumentation-llamaindex"
-version = "0.16.4"
+version = "0.16.6"
 description = "OpenTelemetry LlamaIndex instrumentation"
 authors = [
   "Gal Kleinman <gal@traceloop.com>",
@@ -44,9 +44,9 @@ openai = "=1.12.0"
 opentelemetry-sdk = "^1.23.0"
 llama-index = "^0.10.10"
 llama-index-postprocessor-cohere-rerank = "^0.1.2"
-opentelemetry-instrumentation-openai = "==0.16.4"
-opentelemetry-instrumentation-cohere = "==0.16.4"
-opentelemetry-instrumentation-chromadb = "==0.16.4"
+opentelemetry-instrumentation-openai = "==0.16.6"
+opentelemetry-instrumentation-cohere = "==0.16.6"
+opentelemetry-instrumentation-chromadb = "==0.16.6"
 sqlalchemy = "^2.0.27"
 llama-index-agent-openai = ">=0.1.5,<0.3.0"
 llama-index-vector-stores-chroma = "^0.1.5"

opentelemetry_instrumentation_llamaindex-0.16.4/opentelemetry/instrumentation/llamaindex/utils.py

@@ -1,27 +0,0 @@
-import os
-
-from contextlib import asynccontextmanager
-
-from opentelemetry import context as context_api
-
-
-def _with_tracer_wrapper(func):
-    def _with_tracer(tracer):
-        def wrapper(wrapped, instance, args, kwargs):
-            return func(tracer, wrapped, instance, args, kwargs)
-
-        return wrapper
-
-    return _with_tracer
-
-
-@asynccontextmanager
-async def start_as_current_span_async(tracer, *args, **kwargs):
-    with tracer.start_as_current_span(*args, **kwargs) as span:
-        yield span
-
-
-def should_send_prompts():
-    return (
-        os.getenv("TRACELOOP_TRACE_CONTENT") or "true"
-    ).lower() == "true" or context_api.get_value("override_enable_content_tracing")

opentelemetry_instrumentation_llamaindex-0.16.4/opentelemetry/instrumentation/llamaindex/version.py

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