opentelemetry-instrumentation-llamaindex 0.25.0__tar.gz → 0.25.1__tar.gz

{opentelemetry_instrumentation_llamaindex-0.25.0 → opentelemetry_instrumentation_llamaindex-0.25.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opentelemetry-instrumentation-llamaindex
-Version: 0.25.0
+Version: 0.25.1
 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.25.0 → opentelemetry_instrumentation_llamaindex-0.25.1}/opentelemetry/instrumentation/llamaindex/base_agent_instrumentor.py

@@ -4,6 +4,8 @@ from wrapt import wrap_function_wrapper
 
 from opentelemetry.instrumentation.llamaindex.utils import (
     _with_tracer_wrapper,
+    process_request,
+    process_response,
     start_as_current_span_async,
 )
 from opentelemetry.semconv.ai import SpanAttributes, TraceloopSpanKindValues
@@ -54,7 +56,10 @@ def query_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.AGENT.value,
         )
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
@@ -67,4 +72,7 @@ async def aquery_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.AGENT.value,
         )
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

{opentelemetry_instrumentation_llamaindex-0.25.0 → opentelemetry_instrumentation_llamaindex-0.25.1}/opentelemetry/instrumentation/llamaindex/base_retriever_instrumentor.py

@@ -4,6 +4,8 @@ from wrapt import wrap_function_wrapper
 
 from opentelemetry.instrumentation.llamaindex.utils import (
     _with_tracer_wrapper,
+    process_request,
+    process_response,
     start_as_current_span_async,
 )
 from opentelemetry.semconv.ai import SpanAttributes, TraceloopSpanKindValues
@@ -47,7 +49,10 @@ def retrieve_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.TASK.value,
         )
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
@@ -60,4 +65,7 @@ async def aretrieve_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.TASK.value,
         )
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

{opentelemetry_instrumentation_llamaindex-0.25.0 → opentelemetry_instrumentation_llamaindex-0.25.1}/opentelemetry/instrumentation/llamaindex/base_synthesizer_instrumentor.py

@@ -4,6 +4,8 @@ from wrapt import wrap_function_wrapper
 
 from opentelemetry.instrumentation.llamaindex.utils import (
     _with_tracer_wrapper,
+    process_request,
+    process_response,
     start_as_current_span_async,
 )
 from opentelemetry.semconv.ai import SpanAttributes, TraceloopSpanKindValues
@@ -46,7 +48,10 @@ def synthesize_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.TASK.value,
         )
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
@@ -59,4 +64,7 @@ async def asynthesize_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.TASK.value,
         )
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

{opentelemetry_instrumentation_llamaindex-0.25.0 → opentelemetry_instrumentation_llamaindex-0.25.1}/opentelemetry/instrumentation/llamaindex/base_tool_instrumentor.py

@@ -4,6 +4,8 @@ from wrapt import wrap_function_wrapper
 
 from opentelemetry.instrumentation.llamaindex.utils import (
     _with_tracer_wrapper,
+    process_request,
+    process_response,
     start_as_current_span_async,
 )
 from opentelemetry.semconv.ai import SpanAttributes, TraceloopSpanKindValues
@@ -56,7 +58,10 @@ def query_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.TOOL.value,
         )
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
@@ -69,4 +74,7 @@ async def aquery_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.TOOL.value,
         )
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

{opentelemetry_instrumentation_llamaindex-0.25.0 → opentelemetry_instrumentation_llamaindex-0.25.1}/opentelemetry/instrumentation/llamaindex/query_pipeline_instrumentor.py

@@ -5,6 +5,8 @@ from opentelemetry.context import attach, set_value
 
 from opentelemetry.instrumentation.llamaindex.utils import (
     _with_tracer_wrapper,
+    process_request,
+    process_response,
     start_as_current_span_async,
 )
 from opentelemetry.semconv.ai import SpanAttributes, TraceloopSpanKindValues
@@ -52,7 +54,10 @@ def run_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.WORKFLOW.value,
         )
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
@@ -67,4 +72,7 @@ async def arun_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.WORKFLOW.value,
         )
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

{opentelemetry_instrumentation_llamaindex-0.25.0 → opentelemetry_instrumentation_llamaindex-0.25.1}/opentelemetry/instrumentation/llamaindex/retriever_query_engine_instrumentor.py

@@ -5,6 +5,8 @@ from opentelemetry.context import attach, set_value
 
 from opentelemetry.instrumentation.llamaindex.utils import (
     _with_tracer_wrapper,
+    process_request,
+    process_response,
     start_as_current_span_async,
 )
 from opentelemetry.semconv.ai import SpanAttributes, TraceloopSpanKindValues
@@ -53,7 +55,10 @@ def query_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.WORKFLOW.value,
         )
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
@@ -68,4 +73,7 @@ async def aquery_wrapper(tracer, wrapped, instance, args, kwargs):
             TraceloopSpanKindValues.WORKFLOW.value,
         )
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

{opentelemetry_instrumentation_llamaindex-0.25.0 → opentelemetry_instrumentation_llamaindex-0.25.1}/opentelemetry/instrumentation/llamaindex/utils.py

@@ -1,3 +1,5 @@
+import dataclasses
+import json
 import os
 import logging
 import traceback
@@ -5,6 +7,7 @@ from contextlib import asynccontextmanager
 
 from opentelemetry import context as context_api
 from opentelemetry.instrumentation.llamaindex.config import Config
+from opentelemetry.semconv.ai import SpanAttributes
 
 
 def _with_tracer_wrapper(func):
@@ -52,3 +55,32 @@ def dont_throw(func):
                 Config.exception_logger(e)
 
     return wrapper
+
+
+class JSONEncoder(json.JSONEncoder):
+    def default(self, o):
+        if dataclasses.is_dataclass(o):
+            return dataclasses.asdict(o)
+        elif hasattr(o, "json"):
+            return o.json()
+        elif hasattr(o, "to_json"):
+            return o.to_json()
+        return super().default(o)
+
+
+@dont_throw
+def process_request(span, args, kwargs):
+    if should_send_prompts():
+        span.set_attribute(
+            SpanAttributes.TRACELOOP_ENTITY_INPUT,
+            json.dumps({"args": args, "kwargs": kwargs}, cls=JSONEncoder),
+        )
+
+
+@dont_throw
+def process_response(span, res):
+    if should_send_prompts():
+        span.set_attribute(
+            SpanAttributes.TRACELOOP_ENTITY_OUTPUT,
+            json.dumps(res, cls=JSONEncoder),
+        )

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

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

{opentelemetry_instrumentation_llamaindex-0.25.0 → opentelemetry_instrumentation_llamaindex-0.25.1}/pyproject.toml

@@ -8,7 +8,7 @@ show_missing = true
 
 [tool.poetry]
 name = "opentelemetry-instrumentation-llamaindex"
-version = "0.25.0"
+version = "0.25.1"
 description = "OpenTelemetry LlamaIndex instrumentation"
 authors = [
   "Gal Kleinman <gal@traceloop.com>",
@@ -43,9 +43,9 @@ openai = "^1.35.0"
 opentelemetry-sdk = "^1.23.0"
 llama-index = "^0.10.46"
 llama-index-postprocessor-cohere-rerank = "^0.1.7"
-opentelemetry-instrumentation-openai = "==0.25.0"
-opentelemetry-instrumentation-cohere = "==0.25.0"
-opentelemetry-instrumentation-chromadb = "==0.25.0"
+opentelemetry-instrumentation-openai = "==0.25.1"
+opentelemetry-instrumentation-cohere = "==0.25.1"
+opentelemetry-instrumentation-chromadb = "==0.25.1"
 sqlalchemy = "^2.0.31"
 llama-index-agent-openai = ">=0.2.7,<0.3.0"
 llama-index-vector-stores-chroma = "^0.1.9"

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

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