opentelemetry-instrumentation-llamaindex 0.25.1__py3-none-any.whl → 0.25.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of opentelemetry-instrumentation-llamaindex might be problematic. Click here for more details.

Files changed (12) hide show
  1. opentelemetry/instrumentation/llamaindex/base_agent_instrumentor.py +6 -4
  2. opentelemetry/instrumentation/llamaindex/base_embedding_instrumentor.py +2 -0
  3. opentelemetry/instrumentation/llamaindex/base_retriever_instrumentor.py +2 -0
  4. opentelemetry/instrumentation/llamaindex/base_synthesizer_instrumentor.py +2 -0
  5. opentelemetry/instrumentation/llamaindex/base_tool_instrumentor.py +6 -4
  6. opentelemetry/instrumentation/llamaindex/query_pipeline_instrumentor.py +2 -0
  7. opentelemetry/instrumentation/llamaindex/retriever_query_engine_instrumentor.py +2 -0
  8. opentelemetry/instrumentation/llamaindex/version.py +1 -1
  9. {opentelemetry_instrumentation_llamaindex-0.25.1.dist-info → opentelemetry_instrumentation_llamaindex-0.25.3.dist-info}/METADATA +1 -1
  10. {opentelemetry_instrumentation_llamaindex-0.25.1.dist-info → opentelemetry_instrumentation_llamaindex-0.25.3.dist-info}/RECORD +12 -12
  11. {opentelemetry_instrumentation_llamaindex-0.25.1.dist-info → opentelemetry_instrumentation_llamaindex-0.25.3.dist-info}/WHEEL +0 -0
  12. {opentelemetry_instrumentation_llamaindex-0.25.1.dist-info → opentelemetry_instrumentation_llamaindex-0.25.3.dist-info}/entry_points.txt +0 -0
opentelemetry/instrumentation/llamaindex/base_agent_instrumentor.py CHANGED
@@ -50,11 +50,13 @@ class BaseAgentInstrumentor:
50
50
 
51
51
  @_with_tracer_wrapper
52
52
  def query_wrapper(tracer, wrapped, instance, args, kwargs):
53
- with tracer.start_as_current_span(f"{instance.__class__.__name__}.agent") as span:
53
+ name = instance.__class__.__name__
54
+ with tracer.start_as_current_span(f"{name}.agent") as span:
54
55
  span.set_attribute(
55
56
  SpanAttributes.TRACELOOP_SPAN_KIND,
56
57
  TraceloopSpanKindValues.AGENT.value,
57
58
  )
59
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, name)
58
60
 
59
61
  process_request(span, args, kwargs)
60
62
  res = wrapped(*args, **kwargs)
@@ -64,13 +66,13 @@ def query_wrapper(tracer, wrapped, instance, args, kwargs):
64
66
 
65
67
  @_with_tracer_wrapper
66
68
  async def aquery_wrapper(tracer, wrapped, instance, args, kwargs):
67
- async with start_as_current_span_async(
68
- tracer=tracer, name=f"{instance.__class__.__name__}.agent"
69
- ) as span:
69
+ name = instance.__class__.__name__
70
+ async with start_as_current_span_async(tracer=tracer, name=f"{name}.agent") as span:
70
71
  span.set_attribute(
71
72
  SpanAttributes.TRACELOOP_SPAN_KIND,
72
73
  TraceloopSpanKindValues.AGENT.value,
73
74
  )
75
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, name)
74
76
 
75
77
  process_request(span, args, kwargs)
76
78
  res = await wrapped(*args, **kwargs)
opentelemetry/instrumentation/llamaindex/base_embedding_instrumentor.py CHANGED
@@ -49,6 +49,7 @@ def get_query_embedding_wrapper(tracer, wrapped, instance, args, kwargs):
49
49
  SpanAttributes.TRACELOOP_SPAN_KIND,
50
50
  TraceloopSpanKindValues.TASK.value,
51
51
  )
52
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
52
53
 
53
54
  return wrapped(*args, **kwargs)
54
55
 
@@ -62,5 +63,6 @@ async def aget_query_embedding_wrapper(tracer, wrapped, instance, args, kwargs):
62
63
  SpanAttributes.TRACELOOP_SPAN_KIND,
63
64
  TraceloopSpanKindValues.TASK.value,
64
65
  )
66
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
65
67
 
66
68
  return await wrapped(*args, **kwargs)
opentelemetry/instrumentation/llamaindex/base_retriever_instrumentor.py CHANGED
@@ -48,6 +48,7 @@ def retrieve_wrapper(tracer, wrapped, instance, args, kwargs):
48
48
  SpanAttributes.TRACELOOP_SPAN_KIND,
49
49
  TraceloopSpanKindValues.TASK.value,
50
50
  )
51
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
51
52
 
52
53
  process_request(span, args, kwargs)
53
54
  res = wrapped(*args, **kwargs)
@@ -64,6 +65,7 @@ async def aretrieve_wrapper(tracer, wrapped, instance, args, kwargs):
64
65
  SpanAttributes.TRACELOOP_SPAN_KIND,
65
66
  TraceloopSpanKindValues.TASK.value,
66
67
  )
68
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
67
69
 
68
70
  process_request(span, args, kwargs)
69
71
  res = await wrapped(*args, **kwargs)
opentelemetry/instrumentation/llamaindex/base_synthesizer_instrumentor.py CHANGED
@@ -47,6 +47,7 @@ def synthesize_wrapper(tracer, wrapped, instance, args, kwargs):
47
47
  SpanAttributes.TRACELOOP_SPAN_KIND,
48
48
  TraceloopSpanKindValues.TASK.value,
49
49
  )
50
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
50
51
 
51
52
  process_request(span, args, kwargs)
52
53
  res = wrapped(*args, **kwargs)
@@ -63,6 +64,7 @@ async def asynthesize_wrapper(tracer, wrapped, instance, args, kwargs):
63
64
  SpanAttributes.TRACELOOP_SPAN_KIND,
64
65
  TraceloopSpanKindValues.TASK.value,
65
66
  )
67
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
66
68
 
67
69
  process_request(span, args, kwargs)
68
70
  res = await wrapped(*args, **kwargs)
opentelemetry/instrumentation/llamaindex/base_tool_instrumentor.py CHANGED
@@ -52,11 +52,13 @@ class BaseToolInstrumentor:
52
52
 
53
53
  @_with_tracer_wrapper
54
54
  def query_wrapper(tracer, wrapped, instance, args, kwargs):
55
- with tracer.start_as_current_span(f"{instance.__class__.__name__}.tool") as span:
55
+ name = instance.__class__.__name__
56
+ with tracer.start_as_current_span(f"{name}.tool") as span:
56
57
  span.set_attribute(
57
58
  SpanAttributes.TRACELOOP_SPAN_KIND,
58
59
  TraceloopSpanKindValues.TOOL.value,
59
60
  )
61
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, name)
60
62
 
61
63
  process_request(span, args, kwargs)
62
64
  res = wrapped(*args, **kwargs)
@@ -66,13 +68,13 @@ def query_wrapper(tracer, wrapped, instance, args, kwargs):
66
68
 
67
69
  @_with_tracer_wrapper
68
70
  async def aquery_wrapper(tracer, wrapped, instance, args, kwargs):
69
- async with start_as_current_span_async(
70
- tracer=tracer, name=f"{instance.__class__.__name__}.tool"
71
- ) as span:
71
+ name = instance.__class__.__name__
72
+ async with start_as_current_span_async(tracer=tracer, name=f"{name}.tool") as span:
72
73
  span.set_attribute(
73
74
  SpanAttributes.TRACELOOP_SPAN_KIND,
74
75
  TraceloopSpanKindValues.TOOL.value,
75
76
  )
77
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, name)
76
78
 
77
79
  process_request(span, args, kwargs)
78
80
  res = await wrapped(*args, **kwargs)
opentelemetry/instrumentation/llamaindex/query_pipeline_instrumentor.py CHANGED
@@ -53,6 +53,7 @@ def run_wrapper(tracer, wrapped, instance, args, kwargs):
53
53
  SpanAttributes.TRACELOOP_SPAN_KIND,
54
54
  TraceloopSpanKindValues.WORKFLOW.value,
55
55
  )
56
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, WORKFLOW_NAME)
56
57
 
57
58
  process_request(span, args, kwargs)
58
59
  res = wrapped(*args, **kwargs)
@@ -71,6 +72,7 @@ async def arun_wrapper(tracer, wrapped, instance, args, kwargs):
71
72
  SpanAttributes.TRACELOOP_SPAN_KIND,
72
73
  TraceloopSpanKindValues.WORKFLOW.value,
73
74
  )
75
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, WORKFLOW_NAME)
74
76
 
75
77
  process_request(span, args, kwargs)
76
78
  res = await wrapped(*args, **kwargs)
opentelemetry/instrumentation/llamaindex/retriever_query_engine_instrumentor.py CHANGED
@@ -54,6 +54,7 @@ def query_wrapper(tracer, wrapped, instance, args, kwargs):
54
54
  SpanAttributes.TRACELOOP_SPAN_KIND,
55
55
  TraceloopSpanKindValues.WORKFLOW.value,
56
56
  )
57
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, WORKFLOW_NAME)
57
58
 
58
59
  process_request(span, args, kwargs)
59
60
  res = wrapped(*args, **kwargs)
@@ -72,6 +73,7 @@ async def aquery_wrapper(tracer, wrapped, instance, args, kwargs):
72
73
  SpanAttributes.TRACELOOP_SPAN_KIND,
73
74
  TraceloopSpanKindValues.WORKFLOW.value,
74
75
  )
76
+ span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, WORKFLOW_NAME)
75
77
 
76
78
  process_request(span, args, kwargs)
77
79
  res = await wrapped(*args, **kwargs)
opentelemetry/instrumentation/llamaindex/version.py CHANGED
@@ -1 +1 @@
1
- __version__ = "0.25.1"
1
+ __version__ = "0.25.3"
{opentelemetry_instrumentation_llamaindex-0.25.1.dist-info → opentelemetry_instrumentation_llamaindex-0.25.3.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: opentelemetry-instrumentation-llamaindex
3
- Version: 0.25.1
3
+ Version: 0.25.3
4
4
  Summary: OpenTelemetry LlamaIndex instrumentation
5
5
  Home-page: https://github.com/traceloop/openllmetry/tree/main/packages/opentelemetry-instrumentation-llamaindex
6
6
  License: Apache-2.0
{opentelemetry_instrumentation_llamaindex-0.25.1.dist-info → opentelemetry_instrumentation_llamaindex-0.25.3.dist-info}/RECORD RENAMED
@@ -1,16 +1,16 @@
1
1
  opentelemetry/instrumentation/llamaindex/__init__.py,sha256=xVgDrVrDSVLC93OAk9hIvzBybBoKCtquGSV2-2Y_P5E,2318
2
- opentelemetry/instrumentation/llamaindex/base_agent_instrumentor.py,sha256=rJPlI4w-2LgKlKEEwPiU4Zvr7uot-BOzFYhH5mkU-mE,2485
3
- opentelemetry/instrumentation/llamaindex/base_embedding_instrumentor.py,sha256=xbw3gD4OT_CObMOIV3vvLTV1V4obr3PgoiON5u2zd64,2055
4
- opentelemetry/instrumentation/llamaindex/base_retriever_instrumentor.py,sha256=FBm4lvzbwBU0S9RpjP-yqTgnfLCGau8YDcCt2KtNE0g,2194
5
- opentelemetry/instrumentation/llamaindex/base_synthesizer_instrumentor.py,sha256=voT7sIlKN_ftdKS2wmKY4pej6YWjQEDzhbfa61LRtxs,2213
6
- opentelemetry/instrumentation/llamaindex/base_tool_instrumentor.py,sha256=Krvy1h8W-Cs7gPBh6Dg6HfRuY_HbhXxs6KuvsvC4q0w,2598
2
+ opentelemetry/instrumentation/llamaindex/base_agent_instrumentor.py,sha256=FgchRWt7Tb4KpsTtZbe3HIeuMglH7Slek1sv8FH3xvU,2645
3
+ opentelemetry/instrumentation/llamaindex/base_embedding_instrumentor.py,sha256=6m3qnBfvU2dzAFuJCu_A576gyEmReCVKxOTeiaIoTX4,2207
4
+ opentelemetry/instrumentation/llamaindex/base_retriever_instrumentor.py,sha256=LPdBeoV3817P2mOCQSL8Y46zvxkHATa6oL89f8ZmVY8,2346
5
+ opentelemetry/instrumentation/llamaindex/base_synthesizer_instrumentor.py,sha256=HW-pO4ihrXE324tDg-KBh7U6e3zCRNuP8GZFl6vxGio,2365
6
+ opentelemetry/instrumentation/llamaindex/base_tool_instrumentor.py,sha256=z1Gj1JvBgMyeGjtp8a2GiKjPPBBSLR9-QBP1tHvdSJE,2758
7
7
  opentelemetry/instrumentation/llamaindex/config.py,sha256=CtypZov_ytI9nSrfN9lWnjcufbAR9sfkXRA0OstDEUw,42
8
8
  opentelemetry/instrumentation/llamaindex/custom_llm_instrumentor.py,sha256=9b9I3WDE0icnYtDpkiFChpl7fwadQnUCryHc3pPi6nM,5936
9
- opentelemetry/instrumentation/llamaindex/query_pipeline_instrumentor.py,sha256=6cbZSOqyn1uRC7TMuIYxfmgTvQhuFu_1IeY_ClX_Fcg,2336
10
- opentelemetry/instrumentation/llamaindex/retriever_query_engine_instrumentor.py,sha256=ZSD0PwcZeh50uUJ_b_C4qFRsc0yjNFDDlsEOUbEIQpc,2467
9
+ opentelemetry/instrumentation/llamaindex/query_pipeline_instrumentor.py,sha256=_XejlSUWusotWdkffKVfLaLsoj6G3Rf_RALs8XWJ0f4,2496
10
+ opentelemetry/instrumentation/llamaindex/retriever_query_engine_instrumentor.py,sha256=JkwdDVR6wrWUKgljnl23AusSw6zeb4JEE8ceM9lBilU,2627
11
11
  opentelemetry/instrumentation/llamaindex/utils.py,sha256=u5F80iSqg4sZst6KYwcxvJeC73nr2IN3hFACKf-mtBc,2349
12
- opentelemetry/instrumentation/llamaindex/version.py,sha256=ACu2Z3Q3TFgYpAno_eu9ssJ1QULjNXvjGvyqDSHrQ_o,23
13
- opentelemetry_instrumentation_llamaindex-0.25.1.dist-info/METADATA,sha256=sKRSOSya6mC5-KLYFHzdczJnzJF5WVWNI-G7LZhHE-M,2285
14
- opentelemetry_instrumentation_llamaindex-0.25.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
15
- opentelemetry_instrumentation_llamaindex-0.25.1.dist-info/entry_points.txt,sha256=gtV40W4oFCp6VNvgowTKa0zQjfIrvfdlYflgGdSsA5A,106
16
- opentelemetry_instrumentation_llamaindex-0.25.1.dist-info/RECORD,,
12
+ opentelemetry/instrumentation/llamaindex/version.py,sha256=mXrC0wG0d9x4Qj0b-1a15tGJ6JHnWNOm82J2nX_CL2k,23
13
+ opentelemetry_instrumentation_llamaindex-0.25.3.dist-info/METADATA,sha256=FpC3QoJ4rfSlBNwL8xIzSiVhMawXBaP2fZrQpKjXWkQ,2285
14
+ opentelemetry_instrumentation_llamaindex-0.25.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
15
+ opentelemetry_instrumentation_llamaindex-0.25.3.dist-info/entry_points.txt,sha256=gtV40W4oFCp6VNvgowTKa0zQjfIrvfdlYflgGdSsA5A,106
16
+ opentelemetry_instrumentation_llamaindex-0.25.3.dist-info/RECORD,,