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

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

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

@@ -50,11 +50,13 @@ 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)
 
         process_request(span, args, kwargs)
         res = wrapped(*args, **kwargs)
@@ -64,13 +66,13 @@ def query_wrapper(tracer, wrapped, instance, args, kwargs):
 
 @_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)
 
         process_request(span, args, kwargs)
         res = await wrapped(*args, **kwargs)

{opentelemetry_instrumentation_llamaindex-0.25.1 → opentelemetry_instrumentation_llamaindex-0.25.2}/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-0.25.1 → opentelemetry_instrumentation_llamaindex-0.25.2}/opentelemetry/instrumentation/llamaindex/base_retriever_instrumentor.py

@@ -48,6 +48,7 @@ def retrieve_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.TASK.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
 
         process_request(span, args, kwargs)
         res = wrapped(*args, **kwargs)
@@ -64,6 +65,7 @@ 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)
 
         process_request(span, args, kwargs)
         res = await wrapped(*args, **kwargs)

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

@@ -47,6 +47,7 @@ def synthesize_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.TASK.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, TASK_NAME)
 
         process_request(span, args, kwargs)
         res = wrapped(*args, **kwargs)
@@ -63,6 +64,7 @@ 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)
 
         process_request(span, args, kwargs)
         res = await wrapped(*args, **kwargs)

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

@@ -52,11 +52,13 @@ 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)
 
         process_request(span, args, kwargs)
         res = wrapped(*args, **kwargs)
@@ -66,13 +68,13 @@ def query_wrapper(tracer, wrapped, instance, args, kwargs):
 
 @_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)
 
         process_request(span, args, kwargs)
         res = await wrapped(*args, **kwargs)

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

@@ -53,6 +53,7 @@ def run_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.WORKFLOW.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, WORKFLOW_NAME)
 
         process_request(span, args, kwargs)
         res = wrapped(*args, **kwargs)
@@ -71,6 +72,7 @@ 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)
 
         process_request(span, args, kwargs)
         res = await wrapped(*args, **kwargs)

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

@@ -54,6 +54,7 @@ def query_wrapper(tracer, wrapped, instance, args, kwargs):
             SpanAttributes.TRACELOOP_SPAN_KIND,
             TraceloopSpanKindValues.WORKFLOW.value,
         )
+        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, WORKFLOW_NAME)
 
         process_request(span, args, kwargs)
         res = wrapped(*args, **kwargs)
@@ -72,6 +73,7 @@ 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)
 
         process_request(span, args, kwargs)
         res = await wrapped(*args, **kwargs)

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

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

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

@@ -8,7 +8,7 @@ show_missing = true
 
 [tool.poetry]
 name = "opentelemetry-instrumentation-llamaindex"
-version = "0.25.1"
+version = "0.25.2"
 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.1"
-opentelemetry-instrumentation-cohere = "==0.25.1"
-opentelemetry-instrumentation-chromadb = "==0.25.1"
+opentelemetry-instrumentation-openai = "==0.25.2"
+opentelemetry-instrumentation-cohere = "==0.25.2"
+opentelemetry-instrumentation-chromadb = "==0.25.2"
 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.1/opentelemetry/instrumentation/llamaindex/version.py

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