warpzone-sdk 15.0.0.dev5__tar.gz → 15.0.0.dev6__tar.gz

{warpzone_sdk-15.0.0.dev5 → warpzone_sdk-15.0.0.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: warpzone-sdk
-Version: 15.0.0.dev5
+Version: 15.0.0.dev6
 Summary: The main objective of this package is to centralize logic used to interact with Azure Functions, Azure Service Bus and Azure Table Storage
 Author: Team Enigma
 Author-email: enigma@energinet.dk

{warpzone_sdk-15.0.0.dev5 → warpzone_sdk-15.0.0.dev6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "warpzone-sdk"
-version = "15.0.0.dev5"
+version = "15.0.0.dev6"
 description = "The main objective of this package is to centralize logic used to interact with Azure Functions, Azure Service Bus and Azure Table Storage"
 authors = [{ name = "Team Enigma", email = "enigma@energinet.dk" }]
 requires-python = ">=3.10"

{warpzone_sdk-15.0.0.dev5 → warpzone_sdk-15.0.0.dev6}/warpzone/monitor/traces.py

@@ -78,12 +78,16 @@ def set_trace_context(trace_parent: str, span_name: str = "function_execution"):
     carrier = {"traceparent": trace_parent}
     ctx = TraceContextTextMapPropagator().extract(carrier=carrier)
 
-    # Start a recording span as a child of the propagated context.
-    # This ensures logs are properly correlated with the trace.
-    tracer = trace.get_tracer(__name__)
-    with trace.use_span(trace.get_current_span(ctx), end_on_exit=False):
+    # Explicitly attach the extracted context first
+    token = context.attach(ctx)
+    try:
+        # Now start a recording span as a child of the propagated context
+        tracer = trace.get_tracer(__name__)
         with tracer.start_as_current_span(span_name):
             yield
+    finally:
+        # Explicitly detach to restore previous context
+        context.detach(token)
 
 
 def get_tracer(name: str):