ioa-observe-sdk 1.0.25__py3-none-any.whl → 1.0.27__py3-none-any.whl

ioa_observe/sdk/connectors/slim.py

@@ -3,7 +3,6 @@
 
 import inspect
 import string
-import time
 from functools import wraps
 from json import JSONEncoder
 from typing import Optional
@@ -82,23 +81,23 @@ def process_slim_msg(name: Optional[str] = None):
                     span, ctx, ctx_token = _setup_span(entity_name)
                     _handle_span_input(span, args, kwargs, cls=JSONEncoder)
 
-                    start_time = time.time()
+                    # start_time = time.time()
 
                     try:
                         async for item in _ahandle_generator(
                             span, ctx_token, fn(*args, **kwargs)
                         ):
                             # Measure throughput and processing time per item
-                            item_process_time = time.time() - start_time
-                            TracerWrapper().processing_time.record(
-                                item_process_time, {"agent": entity_name}
-                            )
+                            # item_process_time = time.time() - start_time
+                            # TracerWrapper().processing_time.record(
+                            #     item_process_time, {"agent": entity_name}
+                            # )
                             TracerWrapper().throughput_counter.add(
                                 1, {"agent": entity_name}
                             )
 
                             # Reset timer for next item
-                            start_time = time.time()
+                            # start_time = time.time()
                             # Count each yielded item as a published message
                             TracerWrapper().messages_received_counter.add(
                                 1, {"agent": entity_name}
@@ -122,16 +121,16 @@ def process_slim_msg(name: Optional[str] = None):
 
                     # span, ctx, ctx_token = _setup_span(entity_name)
                     # _handle_span_input(span, args, kwargs, cls=JSONEncoder)
-                    start_time = time.time()
+                    # start_time = time.time()
 
                     try:
                         res = await fn(*args, **kwargs)
 
                         # Measure processing time
-                        process_time = time.time() - start_time
-                        TracerWrapper().processing_time.record(
-                            process_time, {"agent": entity_name}
-                        )
+                        # process_time = time.time() - start_time
+                        # TracerWrapper().processing_time.record(
+                        #     process_time, {"agent": entity_name}
+                        # )
                         TracerWrapper().throughput_counter.add(
                             1, {"agent": entity_name}
                         )
@@ -172,16 +171,16 @@ def process_slim_msg(name: Optional[str] = None):
                 # span, ctx, ctx_token = _setup_span(entity_name)
                 # _handle_span_input(span, args, kwargs, cls=JSONEncoder)
 
-                start_time = time.time()
+                # start_time = time.time()
 
                 try:
                     res = fn(*args, **kwargs)
 
                     # Measure processing time
-                    process_time = time.time() - start_time
-                    TracerWrapper().processing_time.record(
-                        process_time, {"agent": entity_name}
-                    )
+                    # process_time = time.time() - start_time
+                    # TracerWrapper().processing_time.record(
+                    #     process_time, {"agent": entity_name}
+                    # )
                     TracerWrapper().throughput_counter.add(1, {"agent": entity_name})
                     # span will be ended in the generator
                     # if isinstance(res, types.GeneratorType):

ioa_observe/sdk/decorators/base.py

@@ -312,18 +312,18 @@ def _cleanup_span(span, ctx_token):
     """End the span process and detach the context token"""
 
     # Calculate agent chain completion time before ending span
-    span_kind = span.attributes.get(OBSERVE_SPAN_KIND)
-    if span_kind == ObserveSpanKindValues.AGENT.value:
-        start_time = span.attributes.get("agent_chain_start_time")
-        if start_time is not None:
-            import time
-
-            completion_time = time.time() - start_time
-
-            # Emit the metric
-            TracerWrapper().agent_chain_completion_time_histogram.record(
-                completion_time, attributes=span.attributes
-            )
+    # span_kind = span.attributes.get(OBSERVE_SPAN_KIND)
+    # if span_kind == ObserveSpanKindValues.AGENT.value:
+    #     start_time = span.attributes.get("agent_chain_start_time")
+    #     if start_time is not None:
+    #         import time
+    #
+    #         # completion_time = time.time() - start_time
+    #
+    #         # Emit the metric
+    #         # TracerWrapper().agent_chain_completion_time_histogram.record(
+    #         #     completion_time, attributes=span.attributes
+    #         # )
     span.end()
     context_api.detach(ctx_token)
 

ioa_observe/sdk/instrumentations/a2a.py

@@ -5,6 +5,7 @@ from typing import Collection
 import functools
 import threading
 
+from opentelemetry.context import get_value
 from opentelemetry import baggage
 from opentelemetry.baggage.propagation import W3CBaggagePropagator
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
@@ -47,8 +48,10 @@ class A2AInstrumentor(BaseInstrumentor):
                 session_id = None
                 if traceparent:
                     session_id = kv_store.get(f"execution.{traceparent}")
-                    if session_id:
-                        kv_store.set(f"execution.{traceparent}", session_id)
+                    if not session_id:
+                        session_id = get_value("session.id")
+                        if session_id:
+                            kv_store.set(f"execution.{traceparent}", session_id)
 
                 # Ensure metadata dict exists
                 try:
@@ -101,8 +104,10 @@ class A2AInstrumentor(BaseInstrumentor):
                     session_id = None
                     if traceparent:
                         session_id = kv_store.get(f"execution.{traceparent}")
-                        if session_id:
-                            kv_store.set(f"execution.{traceparent}", session_id)
+                        if not session_id:
+                            session_id = get_value("session.id")
+                            if session_id:
+                                kv_store.set(f"execution.{traceparent}", session_id)
 
                     # Ensure metadata dict exists
                     try:
@@ -199,6 +204,122 @@ class A2AInstrumentor(BaseInstrumentor):
 
         # original_server_on_message_send = DefaultRequestHandler.on_message_send
 
+        # Instrumentation for slima2a
+        if importlib.util.find_spec("slima2a"):
+            from slima2a.client_transport import SRPCTransport
+
+            # send_message
+            original_srpc_transport_send_message = SRPCTransport.send_message
+
+            @functools.wraps(original_srpc_transport_send_message)
+            async def instrumented_srpc_transport_send_message(
+                self, request, *args, **kwargs
+            ):
+                # Put context into A2A message metadata instead of HTTP headers
+                with _global_tracer.start_as_current_span("slima2a.send_message"):
+                    traceparent = get_current_traceparent()
+                    session_id = None
+                    if traceparent:
+                        session_id = kv_store.get(f"execution.{traceparent}")
+                        if not session_id:
+                            session_id = get_value("session.id")
+                            if session_id:
+                                kv_store.set(f"execution.{traceparent}", session_id)
+
+                    # Ensure metadata dict exists
+                    try:
+                        md = getattr(request.params, "metadata", None)
+                    except AttributeError:
+                        md = None
+                    metadata = md if isinstance(md, dict) else {}
+
+                    observe_meta = dict(metadata.get("observe", {}))
+
+                    # Inject W3C trace context + baggage into observe_meta
+                    TraceContextTextMapPropagator().inject(carrier=observe_meta)
+                    W3CBaggagePropagator().inject(carrier=observe_meta)
+
+                    if traceparent:
+                        observe_meta["traceparent"] = traceparent
+                    if session_id:
+                        observe_meta["session_id"] = session_id
+                        baggage.set_baggage(f"execution.{traceparent}", session_id)
+
+                    metadata["observe"] = observe_meta
+
+                    # Write back metadata (pydantic models are mutable by default in v2)
+                    try:
+                        request.metadata = metadata
+                    except Exception:
+                        # Fallback
+                        request = request.model_copy(update={"metadata": metadata})
+
+                # Call through without transport-specific kwargs
+                return await original_srpc_transport_send_message(
+                    self, request, *args, **kwargs
+                )
+
+            SRPCTransport.send_message = instrumented_srpc_transport_send_message
+
+            # send_message_streaming
+            original_srpc_transport_send_message_streaming = (
+                SRPCTransport.send_message_streaming
+            )
+
+            @functools.wraps(original_srpc_transport_send_message_streaming)
+            async def instrumented_srpc_transport_send_message_streaming(
+                self, request, *args, **kwargs
+            ):
+                # Put context into A2A message metadata instead of HTTP headers
+                with _global_tracer.start_as_current_span(
+                    "slima2a.send_message_streaming"
+                ):
+                    traceparent = get_current_traceparent()
+                    session_id = None
+                    if traceparent:
+                        session_id = kv_store.get(f"execution.{traceparent}")
+                        if not session_id:
+                            session_id = get_value("session.id")
+                            if session_id:
+                                kv_store.set(f"execution.{traceparent}", session_id)
+
+                    # Ensure metadata dict exists
+                    try:
+                        md = getattr(request.params, "metadata", None)
+                    except AttributeError:
+                        md = None
+                    metadata = md if isinstance(md, dict) else {}
+
+                    observe_meta = dict(metadata.get("observe", {}))
+
+                    # Inject W3C trace context + baggage into observe_meta
+                    TraceContextTextMapPropagator().inject(carrier=observe_meta)
+                    W3CBaggagePropagator().inject(carrier=observe_meta)
+
+                    if traceparent:
+                        observe_meta["traceparent"] = traceparent
+                    if session_id:
+                        observe_meta["session_id"] = session_id
+                        baggage.set_baggage(f"execution.{traceparent}", session_id)
+
+                    metadata["observe"] = observe_meta
+
+                    # Write back metadata (pydantic models are mutable by default in v2)
+                    try:
+                        request.metadata = metadata
+                    except Exception:
+                        # Fallback
+                        request = request.model_copy(update={"metadata": metadata})
+
+                # Call through without transport-specific kwargs
+                return await original_srpc_transport_send_message_streaming(
+                    self, request, *args, **kwargs
+                )
+
+            SRPCTransport.send_message_streaming = (
+                instrumented_srpc_transport_send_message_streaming
+            )
+
     def _uninstrument(self, **kwargs):
         import importlib
 
@@ -222,3 +343,20 @@ class A2AInstrumentor(BaseInstrumentor):
         DefaultRequestHandler.on_message_send = (
             DefaultRequestHandler.on_message_send.__wrapped__
         )
+
+        # handle slima2a
+        if importlib.util.find_spec("slima2a"):
+            from slima2a.client_transport import SRPCTransport
+
+            # Uninstrument `send_message`
+            if hasattr(SRPCTransport, "send_message") and hasattr(
+                SRPCTransport.send_message, "__wrapped__"
+            ):
+                SRPCTransport.send_message = SRPCTransport.send_message.__wrapped__
+
+            if hasattr(SRPCTransport, "send_message_streaming") and hasattr(
+                SRPCTransport.send_message_streaming, "__wrapped__"
+            ):
+                SRPCTransport.send_message_streaming = (
+                    SRPCTransport.send_message.__wrapped__
+                )

ioa_observe/sdk/instrumentations/mcp.py

@@ -10,6 +10,7 @@ import traceback
 import re
 from http import HTTPStatus
 
+from opentelemetry.context import get_value
 from opentelemetry import context, propagate
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.instrumentation.utils import unwrap
@@ -212,8 +213,10 @@ class McpInstrumentor(BaseInstrumentor):
                 session_id = None
                 if traceparent:
                     session_id = kv_store.get(f"execution.{traceparent}")
-                    if session_id:
-                        kv_store.set(f"execution.{traceparent}", session_id)
+                    if not session_id:
+                        session_id = get_value("session.id")
+                        if session_id:
+                            kv_store.set(f"execution.{traceparent}", session_id)
 
                         meta = meta or {}
                         if isinstance(meta, dict):

ioa_observe/sdk/instrumentations/nats.py

@@ -7,6 +7,7 @@ import json
 import base64
 import threading
 
+from opentelemetry.context import get_value
 from opentelemetry import baggage, context
 from opentelemetry.baggage.propagation import W3CBaggagePropagator
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
@@ -53,8 +54,10 @@ class NATSInstrumentor(BaseInstrumentor):
             if traceparent:
                 with _kv_lock:
                     session_id = kv_store.get(f"execution.{traceparent}")
-                    if session_id:
-                        kv_store.set(f"execution.{traceparent}", session_id)
+                    if not session_id:
+                        session_id = get_value("session.id")
+                        if session_id:
+                            kv_store.set(f"execution.{traceparent}", session_id)
 
             headers = {
                 "session_id": session_id if session_id else None,
@@ -104,8 +107,10 @@ class NATSInstrumentor(BaseInstrumentor):
             if traceparent:
                 with _kv_lock:
                     session_id = kv_store.get(f"execution.{traceparent}")
-                    if session_id:
-                        kv_store.set(f"execution.{traceparent}", session_id)
+                    if not session_id:
+                        session_id = get_value("session.id")
+                        if session_id:
+                            kv_store.set(f"execution.{traceparent}", session_id)
 
             headers = {
                 "session_id": session_id if session_id else None,

ioa_observe/sdk/instrumentations/slim.py

@@ -7,6 +7,7 @@ import json
 import base64
 import threading
 
+from opentelemetry.context import get_value
 from opentelemetry import baggage, context
 from opentelemetry.baggage.propagation import W3CBaggagePropagator
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
@@ -15,6 +16,7 @@ from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapProp
 from ioa_observe.sdk import TracerWrapper
 from ioa_observe.sdk.client import kv_store
 from ioa_observe.sdk.tracing import set_session_id, get_current_traceparent
+from ioa_observe.sdk.tracing.context_manager import get_tracer
 
 _instruments = ("slim-bindings >= 0.4.0",)
 _global_tracer = None
@@ -79,8 +81,10 @@ class SLIMInstrumentor(BaseInstrumentor):
                 if traceparent:
                     with _kv_lock:
                         session_id = kv_store.get(f"execution.{traceparent}")
-                        if session_id:
-                            kv_store.set(f"execution.{traceparent}", session_id)
+                        if not session_id:
+                            session_id = get_value("session.id")
+                            if session_id:
+                                kv_store.set(f"execution.{traceparent}", session_id)
 
                 headers = {
                     "session_id": session_id if session_id else None,
@@ -138,8 +142,10 @@ class SLIMInstrumentor(BaseInstrumentor):
                 if traceparent:
                     with _kv_lock:
                         session_id = kv_store.get(f"execution.{traceparent}")
-                        if session_id:
-                            kv_store.set(f"execution.{traceparent}", session_id)
+                        if not session_id:
+                            session_id = get_value("session.id")
+                            if session_id:
+                                kv_store.set(f"execution.{traceparent}", session_id)
 
                 headers = {
                     "session_id": session_id if session_id else None,
@@ -201,8 +207,10 @@ class SLIMInstrumentor(BaseInstrumentor):
                 if traceparent:
                     with _kv_lock:
                         session_id = kv_store.get(f"execution.{traceparent}")
-                        if session_id:
-                            kv_store.set(f"execution.{traceparent}", session_id)
+                        if not session_id:
+                            session_id = get_value("session.id")
+                            if session_id:
+                                kv_store.set(f"execution.{traceparent}", session_id)
 
                 headers = {
                     "session_id": session_id if session_id else None,
@@ -523,7 +531,15 @@ class SLIMInstrumentor(BaseInstrumentor):
             if timeout is not None:
                 kwargs["timeout"] = timeout
 
-            result = await original_get_message(self, **kwargs)
+            if _global_tracer:
+                with _global_tracer.start_as_current_span(
+                    "session.get_message"
+                ) as span:
+                    if hasattr(self, "id"):
+                        span.set_attribute("slim.session.id", str(self.id))
+                    result = await original_get_message(self, **kwargs)
+            else:
+                result = await original_get_message(self, **kwargs)
 
             # Handle different return types from get_message
             if result is None:
@@ -670,6 +686,8 @@ class SLIMInstrumentor(BaseInstrumentor):
                         if traceparent:
                             with _kv_lock:
                                 session_id = kv_store.get(f"execution.{traceparent}")
+                                if not session_id:
+                                    session_id = get_value("session.id")
                                 if session_id:
                                     kv_store.set(f"execution.{traceparent}", session_id)
 
@@ -722,6 +740,10 @@ class SLIMInstrumentor(BaseInstrumentor):
                     if traceparent:
                         with _kv_lock:
                             session_id = kv_store.get(f"execution.{traceparent}")
+                            if not session_id:
+                                session_id = get_value("session.id")
+                                if session_id:
+                                    kv_store.set(f"execution.{traceparent}", session_id)
 
                     if traceparent or session_id:
                         headers = {
@@ -778,7 +800,8 @@ class SLIMInstrumentor(BaseInstrumentor):
         @functools.wraps(original_method)
         async def instrumented_session_method(self, *args, **kwargs):
             if _global_tracer:
-                with _global_tracer.start_as_current_span(f"session.{method_name}"):
+                tracer = get_tracer()
+                with tracer.start_as_current_span(f"session.{method_name}"):
                     traceparent = get_current_traceparent()
 
                     # Handle message wrapping for publish methods
@@ -834,6 +857,10 @@ class SLIMInstrumentor(BaseInstrumentor):
                     if traceparent:
                         with _kv_lock:
                             session_id = kv_store.get(f"execution.{traceparent}")
+                            if not session_id:
+                                session_id = get_value("session.id")
+                                if session_id:
+                                    kv_store.set(f"execution.{traceparent}", session_id)
 
                     if traceparent or session_id:
                         headers = {

ioa_observe/sdk/metrics/metrics.py

@@ -194,26 +194,4 @@ def metric_views() -> Sequence[View]:
                 ]
             ),
         ),
-        # response latency in ms
-        View(
-            instrument_name="gen_ai.ioa.llm.response_latency",
-            aggregation=ExplicitBucketHistogramAggregation(
-                [
-                    0.01,
-                    0.02,
-                    0.04,
-                    0.08,
-                    0.16,
-                    0.32,
-                    0.64,
-                    1.28,
-                    2.56,
-                    5.12,
-                    10.24,
-                    20.48,
-                    40.96,
-                    81.92,
-                ]
-            ),
-        ),
     ]

ioa_observe/sdk/tracing/tracing.py

@@ -7,6 +7,7 @@ import json
 import logging
 import os
 import re
+import threading
 import time
 import uuid
 
@@ -64,6 +65,9 @@ from typing import Callable, Dict, Optional, Set
 TRACER_NAME = "ioa.observe.tracer"
 APP_NAME = ""
 
+SESSION_IDLE_TIMEOUT_SECONDS = 300  # e.g. 5 minutes
+SESSION_WATCHER_INTERVAL_SECONDS = 30
+
 
 def determine_reliability_score(span):
     if "observe.entity.output" in span.attributes:
@@ -132,6 +136,9 @@ class TracerWrapper(object):
             if not TracerWrapper.endpoint:
                 return obj
 
+            # session activity tracking
+            obj._session_last_activity: dict[str, float] = {}
+            obj._session_lock = threading.Lock()
             obj.__image_uploader = image_uploader
             # {(agent_name): [success_count, total_count]}
             obj._agent_execution_counts = {}
@@ -205,11 +212,11 @@ class TracerWrapper(object):
                 description="Counts agent failures by agent and reason",
                 unit="1",
             )
-            obj.response_latency_histogram = meter.create_histogram(
-                name="response_latency",
-                description="Records the latency of responses",
-                unit="ms",
-            )
+            # obj.response_latency_histogram = meter.create_histogram(
+            #     name="response_latency",
+            #     description="Records the latency of responses",
+            #     unit="ms",
+            # )
             obj.messages_received_counter = meter.create_counter(
                 "slim.messages.received",
                 description="Number of SLIM messages received per agent",
@@ -226,10 +233,10 @@ class TracerWrapper(object):
                 "slim.messages.processed",
                 description="Number of SLIM messages processed",
             )
-            obj.processing_time = meter.create_histogram(
-                "slim.message.processing_time",
-                description="Time taken to process SLIM messages",
-            )
+            # obj.processing_time = meter.create_histogram(
+            #     "slim.message.processing_time",
+            #     description="Time taken to process SLIM messages",
+            # )
             obj.throughput_counter = meter.create_counter(
                 "slim.message.throughput",
                 description="Message throughput for SLIM operations",
@@ -237,11 +244,11 @@ class TracerWrapper(object):
             obj.error_counter = meter.create_counter(
                 "slim.errors", description="Number of SLIM message errors or drops"
             )
-            obj.agent_chain_completion_time_histogram = meter.create_histogram(
-                name="gen_ai.client.ioa.agent.end_to_end_chain_completion_time",
-                description="Records the end-to-end chain completion time for a single agent",
-                unit="s",
-            )
+            # obj.agent_chain_completion_time_histogram = meter.create_histogram(
+            #     name="gen_ai.client.ioa.agent.end_to_end_chain_completion_time",
+            #     description="Records the end-to-end chain completion time for a single agent",
+            #     unit="s",
+            # )
             obj.agent_execution_success_rate = meter.create_observable_gauge(
                 name="gen_ai.client.ioa.agent.execution_success_rate",
                 description="Success rate of agent executions",
@@ -270,12 +277,70 @@ class TracerWrapper(object):
 
             obj.__content_allow_list = ContentAllowList()
 
+            # start background watcher for session end
+            obj._start_session_watcher()
+
             # Force flushes for debug environments (e.g. local development)
             atexit.register(obj.exit_handler)
 
         return cls.instance
 
+    def _start_session_watcher(self) -> None:
+        t = threading.Thread(
+            target=self._session_watcher_loop,
+            name="ioa-session-watcher",
+            daemon=True,
+        )
+        t.start()
+
+    def _session_watcher_loop(self) -> None:
+        while True:
+            time.sleep(SESSION_WATCHER_INTERVAL_SECONDS)
+            now = time.time()
+            expired: dict[str, float] = {}
+
+            # Find idle sessions and remove them from _session_last_activity
+            with self._session_lock:
+                for session_id, last_ts in list(self._session_last_activity.items()):
+                    if now - last_ts > SESSION_IDLE_TIMEOUT_SECONDS:
+                        expired[session_id] = last_ts
+                        del self._session_last_activity[session_id]
+
+            if not expired:
+                continue
+
+            tracer = self.get_tracer()
+
+            # Iterate over a snapshot and do *not* modify `expired` in the loop
+            for session_id, _last_ts in list(expired.items()):
+                print("ending session", session_id)
+                with tracer.start_as_current_span("session.end") as span:
+                    span.set_attribute("session.id", session_id)
+                    workflow_name = get_value("workflow_name")
+                    if workflow_name:
+                        span.set_attribute(OBSERVE_WORKFLOW_NAME, workflow_name)
+                    span.set_attribute("session.ended_at", _last_ts)
+
+            # ensure end spans are exported reasonably fast
+            self.flush()
+
     def exit_handler(self):
+        # emit end spans for any sessions that never went idle
+        now = time.time()
+        tracer = self.get_tracer()
+
+        with self._session_lock:
+            remaining_ids = list(self._session_last_activity.keys())
+            self._session_last_activity.clear()
+
+        for session_id in remaining_ids:
+            with tracer.start_as_current_span("session.end") as span:
+                span.set_attribute("session.id", session_id)
+                workflow_name = get_value("workflow_name")
+                if workflow_name:
+                    span.set_attribute(OBSERVE_WORKFLOW_NAME, workflow_name)
+                span.set_attribute("session.ended_at", now)
+
         self.flush()
 
     def _span_processor_on_start(self, span, parent_context):
@@ -362,8 +427,14 @@ class TracerWrapper(object):
         # Mark this span as processed
         self._processed_spans.add(span_id)
 
+        # update last activity per session
+        session_id = span.attributes.get("session.id")
+        if session_id:
+            with self._session_lock:
+                self._session_last_activity[session_id] = time.time()
+
         determine_reliability_score(span)
-        start_time = span.attributes.get("ioa_start_time")
+        # start_time = span.attributes.get("ioa_start_time")
 
         # Apply transformations if enabled
         apply_transform = get_value("apply_transform")
@@ -382,10 +453,6 @@ class TracerWrapper(object):
                 except Exception as e:
                     logging.error(f"Error applying span transformation: {e}")
 
-        if start_time is not None:
-            latency = (time.time() - start_time) * 1000
-            self.response_latency_histogram.record(latency, attributes=span.attributes)
-
         # Call original on_end method if it exists
         if (
             hasattr(self, "_TracerWrapper__spans_processor_original_on_end")
@@ -684,28 +751,24 @@ def set_session_id(session_id: str, traceparent: str = None) -> None:
 
     # Check if we have an active span first
     current_span = trace.get_current_span()
+    extracted_traceparent = None
 
     if current_span.is_recording():
         # We have an active span, use its context
         carrier = {}
         TraceContextTextMapPropagator().inject(carrier)
         extracted_traceparent = carrier.get("traceparent")
-    else:
-        # Only create new span if absolutely necessary (no existing context)
-        tracer = trace.get_tracer(__name__)
-        with tracer.start_as_current_span("set_session_id"):
-            carrier = {}
-            TraceContextTextMapPropagator().inject(carrier)
-            extracted_traceparent = carrier.get("traceparent")
-
-    # Store execution ID with traceparent as key
+        # Store execution ID with traceparent as key
+        if extracted_traceparent:
+            kv_key = f"execution.{extracted_traceparent}"
+            if kv_store.get(kv_key) is None:
+                kv_store.set(kv_key, session_id)
+    # If there is no active span, we do nothing now and when we need the traceparent
+    # e.g. when propagating context, we update the kv_store with the session_id then
+
+    # Finally, attach to context session.id and current_traceparent if present
+    attach(set_value("session.id", session_id))
     if extracted_traceparent:
-        kv_key = f"execution.{extracted_traceparent}"
-        if kv_store.get(kv_key) is None:
-            kv_store.set(kv_key, session_id)
-
-        # Also store in OpenTelemetry context
-        attach(set_value("session.id", session_id))
         attach(set_value("current_traceparent", extracted_traceparent))
 
 

{ioa_observe_sdk-1.0.25.dist-info → ioa_observe_sdk-1.0.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.25
+Version: 1.0.27
 Summary: IOA Observability SDK
 License-Expression: Apache-2.0
 Requires-Python: >=3.10

{ioa_observe_sdk-1.0.25.dist-info → ioa_observe_sdk-1.0.27.dist-info}/RECORD

@@ -8,21 +8,21 @@ ioa_observe/sdk/client/client.py,sha256=6TVOo_E1ulE3WO_CYG7oPgeucs-qegOA09uTO3yQ
 ioa_observe/sdk/client/http.py,sha256=LdLYSQPFIhKN5BTB-N78jLO7ITl7jGjA0-qpewEIvO4,1724
 ioa_observe/sdk/config/__init__.py,sha256=8aVNaw0yRNLFPxlf97iOZLlJVcV81ivSDnudH2m1OIo,572
 ioa_observe/sdk/connectors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ioa_observe/sdk/connectors/slim.py,sha256=NwbKEV7d5NIOqmG8zKqtgGigSJl7kf3QJ65z2gxpsY8,8498
+ioa_observe/sdk/connectors/slim.py,sha256=A_ojiJ_zm-GWuVkZPlwhY7nCt3w2kAohKgnkGO_S634,8518
 ioa_observe/sdk/decorators/__init__.py,sha256=qCpJAv98eLKs3I5EMXJVTV0s49Nc6QDSOHNh5rW5vLg,4268
-ioa_observe/sdk/decorators/base.py,sha256=pnoj73UrpvaZmOPbcpGRRYCwg7LBAgO8PiBDh5ntn0c,32694
+ioa_observe/sdk/decorators/base.py,sha256=gezoLMLOo-CWsxzsXbQMsj5HxmchUEmlhTspbqzOrPc,32732
 ioa_observe/sdk/decorators/helpers.py,sha256=I9HXMBivkZpGDtPe9Ad_UU35p_m_wEPate4r_fU0oOA,2705
 ioa_observe/sdk/decorators/util.py,sha256=IebvH9gwZN1en3LblYJUh4bAV2STl6xmp8WpZzBDH2g,30068
 ioa_observe/sdk/instrumentations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ioa_observe/sdk/instrumentations/a2a.py,sha256=6BOMxrGeMiii-5rptouqkUd2RDpGL01ZyDotgo7pFyM,9088
-ioa_observe/sdk/instrumentations/mcp.py,sha256=vRM3ofnn7AMmry2RrfyZnZVPEutLWiDMghx2TSnm0Wk,18569
-ioa_observe/sdk/instrumentations/nats.py,sha256=UOp2AJlm1JkYkwF3xzU_izzohQVQkByjL-AX4n_JRfo,14476
-ioa_observe/sdk/instrumentations/slim.py,sha256=HEl4IUKqoGN55FHKiWgP78KZWA1uCcfdDcKgtRhcfU8,44226
+ioa_observe/sdk/instrumentations/a2a.py,sha256=LHvvQUluPtmvbRNuoqf-EkWlZJa_NicerTpM3t1UZws,15299
+ioa_observe/sdk/instrumentations/mcp.py,sha256=UG82fWitFT-K-ecnkTKZa-uFTDAs-j2HymvhUwCBRLY,18721
+ioa_observe/sdk/instrumentations/nats.py,sha256=MEYCFqHyRK5HtkjrbGVV5qIWplZ1ZjJjFCB8vtQl6qE,14736
+ioa_observe/sdk/instrumentations/slim.py,sha256=c_Q1cb6pdjZLOw1MbclUf1eRFrc_sUEe9pQJvAyoD1Y,45712
 ioa_observe/sdk/logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ioa_observe/sdk/logging/logging.py,sha256=HZxW9s8Due7jgiNkdI38cIjv5rC9D-Flta3RQMOnpow,2891
 ioa_observe/sdk/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ioa_observe/sdk/metrics/agent.py,sha256=I4ygssTqHm_BPI4IB5rsId5EmRNvW9Te1gRbBp9aSJ0,731
-ioa_observe/sdk/metrics/metrics.py,sha256=CZI1ng6xDHlsct8Vf5dj5J88I8W5itu1d_vXw-2VcEs,6542
+ioa_observe/sdk/metrics/metrics.py,sha256=J1aRMZ9B6bbTE7Yv5ehs4bczZOvXm1KByk0lpRk56ik,5942
 ioa_observe/sdk/metrics/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ioa_observe/sdk/metrics/agents/agent_connections.py,sha256=r-g49czOQLnyIy3bMyJW15NsG3Pgeve9sxcgbpKbSXo,4314
 ioa_observe/sdk/metrics/agents/availability.py,sha256=761o2NiA9SomUWIZrzKjttsz5Fthf72nsgDNfDzUk8s,5028
@@ -35,15 +35,15 @@ ioa_observe/sdk/tracing/content_allow_list.py,sha256=1fAkpIwUQ7vDwCTkIVrqeltWQtr
 ioa_observe/sdk/tracing/context_manager.py,sha256=O0JEXYa9h8anhW78R8KKBuqS0j4by1E1KXxNIMPnLr8,400
 ioa_observe/sdk/tracing/context_utils.py,sha256=-sYS9vPLI87davV9ubneq5xqbV583CC_c0SmOQS1TAs,2933
 ioa_observe/sdk/tracing/manual.py,sha256=KS6WN-zw9vAACzXYmnMoJm9d1fenYMfvzeK1GrGDPDE,1937
-ioa_observe/sdk/tracing/tracing.py,sha256=MgKclJe_IRBh_EKTKwVJkrgrsBX1NUU05eRNcfYnTzE,47380
+ioa_observe/sdk/tracing/tracing.py,sha256=7eJAEqaAEBZ2gjIXrdb7SyUidA8P10GnpBLzr8InLKI,50007
 ioa_observe/sdk/tracing/transform_span.py,sha256=XTApi_gJxum7ynvhtcoCfDyK8VVOj91Q1DT6hAeLHA8,8419
 ioa_observe/sdk/utils/__init__.py,sha256=UPn182U-UblF_XwXaFpx8F-TmQTbm1LYf9y89uSp5Hw,704
 ioa_observe/sdk/utils/const.py,sha256=d67dUTAH9UpWvUV9GLBUqn1Sc2knJ55dy-e6YoLrvSo,1318
 ioa_observe/sdk/utils/in_memory_span_exporter.py,sha256=H_4TRaThMO1H6vUQ0OpQvzJk_fZH0OOsRAM1iZQXsR8,2112
 ioa_observe/sdk/utils/json_encoder.py,sha256=g4NQ0tTqgWssY6I1D7r4zo0G6PiUo61jhofTAw5-jno,639
 ioa_observe/sdk/utils/package_check.py,sha256=1d1MjxhwoEZIx9dumirT2pRsEWgn-m-SI4npDeEalew,576
-ioa_observe_sdk-1.0.25.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
-ioa_observe_sdk-1.0.25.dist-info/METADATA,sha256=QmZACsAli5POXMbvcTB3xuCdRl28w0FD8d2Nmt_920Y,7997
-ioa_observe_sdk-1.0.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ioa_observe_sdk-1.0.25.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
-ioa_observe_sdk-1.0.25.dist-info/RECORD,,
+ioa_observe_sdk-1.0.27.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
+ioa_observe_sdk-1.0.27.dist-info/METADATA,sha256=OrhfkqzR0Qj2qwSGXtSmL9rYjS1OMk-IYmTU5wHMHdE,7997
+ioa_observe_sdk-1.0.27.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ioa_observe_sdk-1.0.27.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
+ioa_observe_sdk-1.0.27.dist-info/RECORD,,

{ioa_observe_sdk-1.0.25.dist-info → ioa_observe_sdk-1.0.27.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any