dreadnode 1.0.7__tar.gz → 1.0.8__tar.gz

{dreadnode-1.0.7 → dreadnode-1.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dreadnode
-Version: 1.0.7
+Version: 1.0.8
 Summary: Dreadnode SDK
 Author: Nick Landers
 Author-email: monoxgas@gmail.com

{dreadnode-1.0.7 → dreadnode-1.0.8}/dreadnode/api/models.py

@@ -122,7 +122,7 @@ class TraceSpan(BaseModel):
 class Metric(BaseModel):
     """Metric data for a span in a trace."""
 
-    value: float
+    value: float | None
     """Value of the metric."""
     step: int
     """Step or iteration number for the metric."""
@@ -268,7 +268,7 @@ class ArtifactDir(BaseModel):
 class RunSummary(BaseModel):
     """Summary of a run, containing metadata and basic information."""
 
-    id: ULID
+    id: ULID | str
     """Unique identifier for the run."""
     name: str
     """Name of the run."""

{dreadnode-1.0.7 → dreadnode-1.0.8}/dreadnode/tracing/span.py

@@ -23,6 +23,7 @@ from opentelemetry import propagate
 from opentelemetry import trace as trace_api
 from opentelemetry.sdk.trace import ReadableSpan
 from opentelemetry.trace import Tracer
+from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
 from opentelemetry.util import types as otel_types
 from ulid import ULID
 
@@ -358,8 +359,23 @@ class RunSpan(Span):
             raise RuntimeError("You cannot start a run span within another run")
 
         if self._remote_context is not None:
-            otel_context = propagate.extract(carrier=self._remote_context)
-            self._remote_token = context_api.attach(otel_context)
+            # If the global propagator is a NoExtract instance, we can't continue
+            # a trace, so we'll bypass it and use the W3C propagator directly.
+            global_propagator = propagate.get_global_textmap()
+            if "NoExtract" in type(global_propagator).__name__:
+                w3c_propagator = TraceContextTextMapPropagator()
+                otel_context = w3c_propagator.extract(carrier=self._remote_context)
+            else:
+                otel_context = propagate.extract(carrier=self._remote_context)
+
+            span_context = trace_api.get_current_span(otel_context).get_span_context()
+
+            # If we have a valid trace_id, we can attach the context and continue the trace.
+            if span_context.trace_id != 0:
+                self._remote_token = context_api.attach(otel_context)
+            else:
+                # Fall back to creating a new span if the context is invalid.
+                super().__enter__()
         else:
             super().__enter__()
 

{dreadnode-1.0.7 → dreadnode-1.0.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "dreadnode"
-version = "1.0.7"
+version = "1.0.8"
 description = "Dreadnode SDK"
 requires-python = ">=3.10,<3.14"