opentelemetry-instrumentation-llamaindex 0.25.0__py3-none-any.whl → 0.25.2__py3-none-any.whl

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
@@ -48,23 +50,31 @@ class BaseAgentInstrumentor:
 
 @_with_tracer_wrapper
 def query_wrapper(tracer, wrapped, instance, args, kwargs):
-    with tracer.start_as_current_span(f"{instance.__class__.__name__}.agent") as span:
+    name = instance.__class__.__name__
+    with tracer.start_as_current_span(f"{name}.agent") as span:
         span.set_attribute(
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.AGENT.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, name)
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
 async def aquery_wrapper(tracer, wrapped, instance, args, kwargs):
-    async with start_as_current_span_async(
-        tracer=tracer, name=f"{instance.__class__.__name__}.agent"
-    ) as span:
+    name = instance.__class__.__name__
+    async with start_as_current_span_async(tracer=tracer, name=f"{name}.agent") as span:
         span.set_attribute(
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.AGENT.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, name)
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

opentelemetry/instrumentation/llamaindex/base_embedding_instrumentor.py

@@ -49,6 +49,7 @@ def get_query_embedding_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.TASK.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
 
         return wrapped(*args, **kwargs)
 
@@ -62,5 +63,6 @@ async def aget_query_embedding_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.TASK.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
 
         return await wrapped(*args, **kwargs)

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
@@ -46,8 +48,12 @@ def retrieve_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.TASK.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
@@ -59,5 +65,9 @@ async def aretrieve_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.TASK.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

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
@@ -45,8 +47,12 @@ def synthesize_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.TASK.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
@@ -58,5 +64,9 @@ async def asynthesize_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.TASK.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

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
@@ -50,23 +52,31 @@ class BaseToolInstrumentor:
 
 @_with_tracer_wrapper
 def query_wrapper(tracer, wrapped, instance, args, kwargs):
-    with tracer.start_as_current_span(f"{instance.__class__.__name__}.tool") as span:
+    name = instance.__class__.__name__
+    with tracer.start_as_current_span(f"{name}.tool") as span:
         span.set_attribute(
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.TOOL.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, name)
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
 async def aquery_wrapper(tracer, wrapped, instance, args, kwargs):
-    async with start_as_current_span_async(
-        tracer=tracer, name=f"{instance.__class__.__name__}.tool"
-    ) as span:
+    name = instance.__class__.__name__
+    async with start_as_current_span_async(tracer=tracer, name=f"{name}.tool") as span:
         span.set_attribute(
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.TOOL.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, name)
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

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
@@ -51,8 +53,12 @@ def run_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.WORKFLOW.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, WORKFLOW_NAME)
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
@@ -66,5 +72,9 @@ async def arun_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.WORKFLOW.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, WORKFLOW_NAME)
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

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
@@ -52,8 +54,12 @@ def query_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.WORKFLOW.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, WORKFLOW_NAME)
 
-        return wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res
 
 
 @_with_tracer_wrapper
@@ -67,5 +73,9 @@ async def aquery_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.WORKFLOW.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, WORKFLOW_NAME)
 
-        return await wrapped(*args, **kwargs)
+        process_request(span, args, kwargs)
+        res = await wrapped(*args, **kwargs)
+        process_response(span, res)
+        return res

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/version.py

@@ -1 +1 @@
-__version__ = "0.25.0"
+__version__ = "0.25.2"

{opentelemetry_instrumentation_llamaindex-0.25.0.dist-info → opentelemetry_instrumentation_llamaindex-0.25.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opentelemetry-instrumentation-llamaindex
-Version: 0.25.0
+Version: 0.25.2
 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.2.dist-info/RECORD

@@ -0,0 +1,16 @@
+opentelemetry/instrumentation/llamaindex/__init__.py,sha256=xVgDrVrDSVLC93OAk9hIvzBybBoKCtquGSV2-2Y_P5E,2318
+opentelemetry/instrumentation/llamaindex/base_agent_instrumentor.py,sha256=FgchRWt7Tb4KpsTtZbe3HIeuMglH7Slek1sv8FH3xvU,2645
+opentelemetry/instrumentation/llamaindex/base_embedding_instrumentor.py,sha256=6m3qnBfvU2dzAFuJCu_A576gyEmReCVKxOTeiaIoTX4,2207
+opentelemetry/instrumentation/llamaindex/base_retriever_instrumentor.py,sha256=LPdBeoV3817P2mOCQSL8Y46zvxkHATa6oL89f8ZmVY8,2346
+opentelemetry/instrumentation/llamaindex/base_synthesizer_instrumentor.py,sha256=HW-pO4ihrXE324tDg-KBh7U6e3zCRNuP8GZFl6vxGio,2365
+opentelemetry/instrumentation/llamaindex/base_tool_instrumentor.py,sha256=z1Gj1JvBgMyeGjtp8a2GiKjPPBBSLR9-QBP1tHvdSJE,2758
+opentelemetry/instrumentation/llamaindex/config.py,sha256=CtypZov_ytI9nSrfN9lWnjcufbAR9sfkXRA0OstDEUw,42
+opentelemetry/instrumentation/llamaindex/custom_llm_instrumentor.py,sha256=9b9I3WDE0icnYtDpkiFChpl7fwadQnUCryHc3pPi6nM,5936
+opentelemetry/instrumentation/llamaindex/query_pipeline_instrumentor.py,sha256=_XejlSUWusotWdkffKVfLaLsoj6G3Rf_RALs8XWJ0f4,2496
+opentelemetry/instrumentation/llamaindex/retriever_query_engine_instrumentor.py,sha256=JkwdDVR6wrWUKgljnl23AusSw6zeb4JEE8ceM9lBilU,2627
+opentelemetry/instrumentation/llamaindex/utils.py,sha256=u5F80iSqg4sZst6KYwcxvJeC73nr2IN3hFACKf-mtBc,2349
+opentelemetry/instrumentation/llamaindex/version.py,sha256=NMzPG-AisbqV0qHNDcM-oIMtdXA6rLhjzeATPyNq6Nw,23
+opentelemetry_instrumentation_llamaindex-0.25.2.dist-info/METADATA,sha256=fH9AB9yqiDNgzk9VBXgF4u_m4DxvYsVlHZJ7ds3RmY4,2285
+opentelemetry_instrumentation_llamaindex-0.25.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+opentelemetry_instrumentation_llamaindex-0.25.2.dist-info/entry_points.txt,sha256=gtV40W4oFCp6VNvgowTKa0zQjfIrvfdlYflgGdSsA5A,106
+opentelemetry_instrumentation_llamaindex-0.25.2.dist-info/RECORD,,

opentelemetry_instrumentation_llamaindex-0.25.0.dist-info/RECORD

@@ -1,16 +0,0 @@
-opentelemetry/instrumentation/llamaindex/__init__.py,sha256=xVgDrVrDSVLC93OAk9hIvzBybBoKCtquGSV2-2Y_P5E,2318
-opentelemetry/instrumentation/llamaindex/base_agent_instrumentor.py,sha256=5orAAgeGzNh5uIQzMZML8WMHSAJpFo0C5EINoif4pTc,2246
-opentelemetry/instrumentation/llamaindex/base_embedding_instrumentor.py,sha256=xbw3gD4OT_CObMOIV3vvLTV1V4obr3PgoiON5u2zd64,2055
-opentelemetry/instrumentation/llamaindex/base_retriever_instrumentor.py,sha256=ipQozp1_4RE2pLPpWFuDgDOtv_GKPh5PocZRO3_LZWA,1955
-opentelemetry/instrumentation/llamaindex/base_synthesizer_instrumentor.py,sha256=NCgwwZthjq4fdZj-U0m14qZY-2BoEow4H7t5tVr4EAs,1974
-opentelemetry/instrumentation/llamaindex/base_tool_instrumentor.py,sha256=6a5NMPOI9B_oDy7zlAkA6-u5Kqt1fXcbnxx7Xw-E2M4,2359
-opentelemetry/instrumentation/llamaindex/config.py,sha256=CtypZov_ytI9nSrfN9lWnjcufbAR9sfkXRA0OstDEUw,42
-opentelemetry/instrumentation/llamaindex/custom_llm_instrumentor.py,sha256=9b9I3WDE0icnYtDpkiFChpl7fwadQnUCryHc3pPi6nM,5936
-opentelemetry/instrumentation/llamaindex/query_pipeline_instrumentor.py,sha256=Z84cNWnjjMlVRDUE2Jyb0LIYUVqMvf_7YsQ9vQjXrmw,2097
-opentelemetry/instrumentation/llamaindex/retriever_query_engine_instrumentor.py,sha256=mlpyxZ51nYDqdo_O8BSdzBdbmNpD9ejokbIYaejGFPs,2228
-opentelemetry/instrumentation/llamaindex/utils.py,sha256=39q5U64UkSoiMVTTzSFMX4xdFyzZFUhiXM5I_OH6kIU,1496
-opentelemetry/instrumentation/llamaindex/version.py,sha256=Mu4JbSLl5nr-J2figk5hmW2mrw4skf_oeIzxbnpcgwY,23
-opentelemetry_instrumentation_llamaindex-0.25.0.dist-info/METADATA,sha256=yScWbuL5GOC0R8tzRe82hC2-yJvWrcxaOxsaCuPIcTg,2285
-opentelemetry_instrumentation_llamaindex-0.25.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-opentelemetry_instrumentation_llamaindex-0.25.0.dist-info/entry_points.txt,sha256=gtV40W4oFCp6VNvgowTKa0zQjfIrvfdlYflgGdSsA5A,106
-opentelemetry_instrumentation_llamaindex-0.25.0.dist-info/RECORD,,