opentelemetry-instrumentation-llamaindex 0.24.0__py3-none-any.whl → 0.25.1__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
@@ -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/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/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/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/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/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/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.24.0"
+__version__ = "0.25.1"

{opentelemetry_instrumentation_llamaindex-0.24.0.dist-info → opentelemetry_instrumentation_llamaindex-0.25.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opentelemetry-instrumentation-llamaindex
-Version: 0.24.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
@@ -18,7 +18,7 @@ Requires-Dist: inflection (>=0.5.1,<0.6.0)
 Requires-Dist: opentelemetry-api (>=1.25.0,<2.0.0)
 Requires-Dist: opentelemetry-instrumentation (>=0.46b0,<0.47)
 Requires-Dist: opentelemetry-semantic-conventions (>=0.46b0,<0.47)
-Requires-Dist: opentelemetry-semantic-conventions-ai (==0.3.3)
+Requires-Dist: opentelemetry-semantic-conventions-ai (==0.3.4)
 Project-URL: Repository, https://github.com/traceloop/openllmetry/tree/main/packages/opentelemetry-instrumentation-llamaindex
 Description-Content-Type: text/markdown
 

{opentelemetry_instrumentation_llamaindex-0.24.0.dist-info → opentelemetry_instrumentation_llamaindex-0.25.1.dist-info}/RECORD

@@ -1,16 +1,16 @@
 opentelemetry/instrumentation/llamaindex/__init__.py,sha256=xVgDrVrDSVLC93OAk9hIvzBybBoKCtquGSV2-2Y_P5E,2318
-opentelemetry/instrumentation/llamaindex/base_agent_instrumentor.py,sha256=5orAAgeGzNh5uIQzMZML8WMHSAJpFo0C5EINoif4pTc,2246
+opentelemetry/instrumentation/llamaindex/base_agent_instrumentor.py,sha256=rJPlI4w-2LgKlKEEwPiU4Zvr7uot-BOzFYhH5mkU-mE,2485
 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/base_retriever_instrumentor.py,sha256=FBm4lvzbwBU0S9RpjP-yqTgnfLCGau8YDcCt2KtNE0g,2194
+opentelemetry/instrumentation/llamaindex/base_synthesizer_instrumentor.py,sha256=voT7sIlKN_ftdKS2wmKY4pej6YWjQEDzhbfa61LRtxs,2213
+opentelemetry/instrumentation/llamaindex/base_tool_instrumentor.py,sha256=Krvy1h8W-Cs7gPBh6Dg6HfRuY_HbhXxs6KuvsvC4q0w,2598
 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=DxtMZD542lg_xb6icrE2d5JOY8oUi-v34i2Ar63ddvs,23
-opentelemetry_instrumentation_llamaindex-0.24.0.dist-info/METADATA,sha256=yepK5Vr6EhpGUZAfR4_YZUBEWbkptwxQw6eRkjr-Gow,2285
-opentelemetry_instrumentation_llamaindex-0.24.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-opentelemetry_instrumentation_llamaindex-0.24.0.dist-info/entry_points.txt,sha256=gtV40W4oFCp6VNvgowTKa0zQjfIrvfdlYflgGdSsA5A,106
-opentelemetry_instrumentation_llamaindex-0.24.0.dist-info/RECORD,,
+opentelemetry/instrumentation/llamaindex/query_pipeline_instrumentor.py,sha256=6cbZSOqyn1uRC7TMuIYxfmgTvQhuFu_1IeY_ClX_Fcg,2336
+opentelemetry/instrumentation/llamaindex/retriever_query_engine_instrumentor.py,sha256=ZSD0PwcZeh50uUJ_b_C4qFRsc0yjNFDDlsEOUbEIQpc,2467
+opentelemetry/instrumentation/llamaindex/utils.py,sha256=u5F80iSqg4sZst6KYwcxvJeC73nr2IN3hFACKf-mtBc,2349
+opentelemetry/instrumentation/llamaindex/version.py,sha256=ACu2Z3Q3TFgYpAno_eu9ssJ1QULjNXvjGvyqDSHrQ_o,23
+opentelemetry_instrumentation_llamaindex-0.25.1.dist-info/METADATA,sha256=sKRSOSya6mC5-KLYFHzdczJnzJF5WVWNI-G7LZhHE-M,2285
+opentelemetry_instrumentation_llamaindex-0.25.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+opentelemetry_instrumentation_llamaindex-0.25.1.dist-info/entry_points.txt,sha256=gtV40W4oFCp6VNvgowTKa0zQjfIrvfdlYflgGdSsA5A,106
+opentelemetry_instrumentation_llamaindex-0.25.1.dist-info/RECORD,,