ioa-observe-sdk 1.0.5__py3-none-any.whl → 1.0.6__py3-none-any.whl

ioa_observe/sdk/tracing/context_utils.py

@@ -6,6 +6,7 @@ from ioa_observe.sdk import TracerWrapper
 from ioa_observe.sdk.client import kv_store
 from ioa_observe.sdk.tracing import set_execution_id, get_current_traceparent
 from opentelemetry import context as otel_context
+from opentelemetry.context import attach
 
 """
 Usage Example:
@@ -42,21 +43,21 @@ def set_context_from_headers(headers):
     """
     Restores the trace context, baggage, and execution_id from headers.
     """
-    # Extract trace context and baggage
-    _global_tracer = TracerWrapper().get_tracer()
-    with _global_tracer.start_as_current_span("set_context_from_headers"):
-        ctx = TraceContextTextMapPropagator().extract(carrier=headers)
-        ctx = W3CBaggagePropagator().extract(carrier=headers, context=ctx)
-        # Restore execution_id if present
-        if headers is not None:
-            if "traceparent" in headers:
-                traceparent = headers.get("traceparent")
-            if "executionID" in headers:
-                execution_id = headers.get("executionID")
-        if traceparent and execution_id and execution_id != "None":
-            set_execution_id(execution_id, traceparent=traceparent)
-            kv_store.set(f"execution.{traceparent}", execution_id)
-        return ctx
+    carrierHeaders = {}
+    if "traceparentID" in headers:
+        carrierHeaders["traceparent"] = headers["traceparentID"]
+    if "executionID" in headers:
+        carrierHeaders["execution_id"] = headers["executionID"]
+    ctx = TraceContextTextMapPropagator().extract(carrier=carrierHeaders)
+    ctx = W3CBaggagePropagator().extract(carrier=carrierHeaders, context=ctx)
+    attach(ctx)
+    # Restore execution_id if present
+    traceparent = headers.get("traceparentID")
+    execution_id = headers.get("executionID")
+    if traceparent and execution_id and execution_id != "None":
+        set_execution_id(execution_id, traceparent=traceparent)
+        kv_store.set(f"execution.{traceparent}", execution_id)
+    return ctx
 
 
 def set_baggage_item(key, value):

ioa_observe/sdk/tracing/tracing.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import atexit
+import contextlib
 import json
 import logging
 import os
@@ -412,7 +413,7 @@ def session_start():
     set_execution_id(execution_id)
     metadata = {
         "executionID": get_value("execution.id") or execution_id,
-        "traceparent": get_current_traceparent(),
+        "traceparentID": get_current_traceparent(),
     }
     import inspect
 
@@ -427,7 +428,7 @@ def session_start():
 
         return _cm()
     # Used as a normal function
-    return None
+    return contextlib.nullcontext(metadata)
 
 
 def set_execution_id(execution_id: str, traceparent: str = None) -> None:

{ioa_observe_sdk-1.0.5.dist-info → ioa_observe_sdk-1.0.6.dist-info}/METADATA

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

{ioa_observe_sdk-1.0.5.dist-info → ioa_observe_sdk-1.0.6.dist-info}/RECORD

@@ -29,16 +29,16 @@ ioa_observe/sdk/metrics/agents/tracker.py,sha256=KN3VFPXrgB1f5VF87ppqS4zG2vxn-9o
 ioa_observe/sdk/tracing/__init__.py,sha256=RBc_lmUEdCDy1ICW6TtRor_0Tew1SyzYFjT4Sn1pkdo,437
 ioa_observe/sdk/tracing/content_allow_list.py,sha256=1fAkpIwUQ7vDwCTkIVrqeltWQtrIbYvj8gz6_7P6NrE,945
 ioa_observe/sdk/tracing/context_manager.py,sha256=O0JEXYa9h8anhW78R8KKBuqS0j4by1E1KXxNIMPnLr8,400
-ioa_observe/sdk/tracing/context_utils.py,sha256=CzIXY402ner15E64Dssd1YMJYI3rsYXqFV47gY2O7Lg,3018
+ioa_observe/sdk/tracing/context_utils.py,sha256=vlLJEq0cqLCScqg1IGaKWQV19tKkHdHQzAt4-pOBbCg,2981
 ioa_observe/sdk/tracing/manual.py,sha256=KS6WN-zw9vAACzXYmnMoJm9d1fenYMfvzeK1GrGDPDE,1937
-ioa_observe/sdk/tracing/tracing.py,sha256=DuEd_VYdDkA7ZNtASDuDhO_O-LnQET8UwCgdqclGmcw,37879
+ioa_observe/sdk/tracing/tracing.py,sha256=ZvWAsKEG-iGuef5EeLiHtSgLoYUCrQPA5RdaQXF1DC0,37927
 ioa_observe/sdk/utils/__init__.py,sha256=UPn182U-UblF_XwXaFpx8F-TmQTbm1LYf9y89uSp5Hw,704
 ioa_observe/sdk/utils/const.py,sha256=GwbHakKPjBL4wLqAVkDrSoKB-8p18EUrbaqPuRuV_xg,1099
 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.5.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
-ioa_observe_sdk-1.0.5.dist-info/METADATA,sha256=kcLchJ5m1Hsvvj_LAsmOuYIjl394oWKTvQWZEE_Y8cQ,5428
-ioa_observe_sdk-1.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ioa_observe_sdk-1.0.5.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
-ioa_observe_sdk-1.0.5.dist-info/RECORD,,
+ioa_observe_sdk-1.0.6.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
+ioa_observe_sdk-1.0.6.dist-info/METADATA,sha256=uRkW5gpM6gtLbDCCvs5---NfvLYUYbC32IR4Dv6a33o,5428
+ioa_observe_sdk-1.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ioa_observe_sdk-1.0.6.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
+ioa_observe_sdk-1.0.6.dist-info/RECORD,,