ioa-observe-sdk 1.0.5__tar.gz → 1.0.6__tar.gz

{ioa_observe_sdk-1.0.5 → ioa_observe_sdk-1.0.6}/PKG-INFO

@@ -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 → ioa_observe_sdk-1.0.6}/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-1.0.5 → ioa_observe_sdk-1.0.6}/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 → ioa_observe_sdk-1.0.6}/ioa_observe_sdk.egg-info/PKG-INFO

@@ -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 → ioa_observe_sdk-1.0.6}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ioa-observe-sdk"
-version = "1.0.5"
+version = "1.0.6"
 description = "IOA Observability SDK"
 readme = "README.md"
 requires-python = ">=3.10"