ioa-observe-sdk 1.0.19__py3-none-any.whl → 1.0.20__py3-none-any.whl

ioa_observe/sdk/decorators/base.py

@@ -16,10 +16,10 @@ from ioa_observe.sdk.decorators.helpers import (
     _is_async_generator,
 )
 
-
 from langgraph.graph.state import CompiledStateGraph
 from opentelemetry import trace
 from opentelemetry import context as context_api
+from opentelemetry.context import get_value, attach, set_value
 from pydantic_core import PydanticSerializationError
 from typing_extensions import ParamSpec
 
@@ -113,7 +113,49 @@ def _setup_span(
     with get_tracer() as tracer:
         span = tracer.start_span(span_name)
         ctx = trace.set_span_in_context(span)
+
+        # Preserve existing context values before attaching new context
+        session_id = get_value("session.id")
+        current_traceparent = get_value("current_traceparent")
+        agent_id = get_value("agent_id")
+        application_id_ctx = get_value("application_id")
+        association_properties = get_value("association_properties")
+        managed_prompt = get_value("managed_prompt")
+        prompt_key = get_value("prompt_key")
+        prompt_version = get_value("prompt_version")
+        prompt_version_name = get_value("prompt_version_name")
+        prompt_version_hash = get_value("prompt_version_hash")
+        prompt_template = get_value("prompt_template")
+        prompt_template_variables = get_value("prompt_template_variables")
+
         ctx_token = context_api.attach(ctx)
+
+        # Re-attach preserved context values to the new context
+        if session_id is not None:
+            attach(set_value("session.id", session_id))
+        if current_traceparent is not None:
+            attach(set_value("current_traceparent", current_traceparent))
+        if agent_id is not None:
+            attach(set_value("agent_id", agent_id))
+        if application_id_ctx is not None:
+            attach(set_value("application_id", application_id_ctx))
+        if association_properties is not None:
+            attach(set_value("association_properties", association_properties))
+        if managed_prompt is not None:
+            attach(set_value("managed_prompt", managed_prompt))
+        if prompt_key is not None:
+            attach(set_value("prompt_key", prompt_key))
+        if prompt_version is not None:
+            attach(set_value("prompt_version", prompt_version))
+        if prompt_version_name is not None:
+            attach(set_value("prompt_version_name", prompt_version_name))
+        if prompt_version_hash is not None:
+            attach(set_value("prompt_version_hash", prompt_version_hash))
+        if prompt_template is not None:
+            attach(set_value("prompt_template", prompt_template))
+        if prompt_template_variables is not None:
+            attach(set_value("prompt_template_variables", prompt_template_variables))
+
         if tlp_span_kind == ObserveSpanKindValues.AGENT:
             with trace.get_tracer(__name__).start_span(
                 "agent_start_event", context=trace.set_span_in_context(span)
@@ -127,9 +169,6 @@ def _setup_span(
                     },
                 )
             # start_span.end()  # end the span immediately
-        # session_id = get_value("session.id")
-        # if session_id is not None:
-        #     span.set_attribute("session.id", session_id)
         if tlp_span_kind in [
             ObserveSpanKindValues.TASK,
             ObserveSpanKindValues.TOOL,

ioa_observe/sdk/instrumentations/slim.py

@@ -349,8 +349,17 @@ class SLIMInstrumentor(BaseInstrumentor):
                             # Store in kv_store with thread safety
                             with _kv_lock:
                                 kv_store.set(f"execution.{traceparent}", session_id)
-                    finally:
+
+                        # DON'T detach the context yet - we need it to persist for the callback
+                        # The context will be cleaned up later or by the garbage collector
+
+                    except Exception as e:
+                        # Only detach on error
                         context.detach(token)
+                        raise e
+                elif traceparent and session_id and session_id != "None":
+                    # Even without carrier context, set session ID if we have the data
+                    set_session_id(session_id, traceparent=traceparent)
 
                 # Fallback: check stored execution ID if not found in headers
                 if traceparent and (not session_id or session_id == "None"):

{ioa_observe_sdk-1.0.19.dist-info → ioa_observe_sdk-1.0.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.19
+Version: 1.0.20
 Summary: IOA Observability SDK
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{ioa_observe_sdk-1.0.19.dist-info → ioa_observe_sdk-1.0.20.dist-info}/RECORD

@@ -10,13 +10,13 @@ ioa_observe/sdk/config/__init__.py,sha256=8aVNaw0yRNLFPxlf97iOZLlJVcV81ivSDnudH2
 ioa_observe/sdk/connectors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ioa_observe/sdk/connectors/slim.py,sha256=NwbKEV7d5NIOqmG8zKqtgGigSJl7kf3QJ65z2gxpsY8,8498
 ioa_observe/sdk/decorators/__init__.py,sha256=qCpJAv98eLKs3I5EMXJVTV0s49Nc6QDSOHNh5rW5vLg,4268
-ioa_observe/sdk/decorators/base.py,sha256=eYep1xHXscdxIGisXgXwSGbkCrdpqqT_2s5BMP07ldg,30646
+ioa_observe/sdk/decorators/base.py,sha256=pnoj73UrpvaZmOPbcpGRRYCwg7LBAgO8PiBDh5ntn0c,32694
 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=ZpqvPl4u-yheQzSdBfxnZhWFZ8ntbKni_uaW3IDyjqw,6309
 ioa_observe/sdk/instrumentations/mcp.py,sha256=vRM3ofnn7AMmry2RrfyZnZVPEutLWiDMghx2TSnm0Wk,18569
-ioa_observe/sdk/instrumentations/slim.py,sha256=UM_XCuzetOzeDFQ-bFY0Q93OLffH7OsvbgNaS58E0RA,22632
+ioa_observe/sdk/instrumentations/slim.py,sha256=WIe0E2y6eg5heQ1EA6l9BUDniYpYIjWQ24Pcv_8ZfPk,23147
 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
@@ -41,8 +41,8 @@ ioa_observe/sdk/utils/const.py,sha256=d67dUTAH9UpWvUV9GLBUqn1Sc2knJ55dy-e6YoLrvS
 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.19.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
-ioa_observe_sdk-1.0.19.dist-info/METADATA,sha256=ynlTiy3cP00dX-e_PjBgTP3wB8FoT20ztUbOLeSKQEg,7027
-ioa_observe_sdk-1.0.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ioa_observe_sdk-1.0.19.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
-ioa_observe_sdk-1.0.19.dist-info/RECORD,,
+ioa_observe_sdk-1.0.20.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
+ioa_observe_sdk-1.0.20.dist-info/METADATA,sha256=rq00vlT-d5xj5wwvTp0j8LeERhmSX1vC90mDkOi9tKI,7027
+ioa_observe_sdk-1.0.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ioa_observe_sdk-1.0.20.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
+ioa_observe_sdk-1.0.20.dist-info/RECORD,,