tusk-drift-python-sdk 0.1.20__tar.gz → 0.1.22__tar.gz

{tusk_drift_python_sdk-0.1.20 → tusk_drift_python_sdk-0.1.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tusk-drift-python-sdk
-Version: 0.1.20
+Version: 0.1.22
 Summary: Python SDK for Tusk Drift instrumentation and replay
 Author-email: Tusk <support@usetusk.ai>
 License: Apache-2.0

{tusk_drift_python_sdk-0.1.20 → tusk_drift_python_sdk-0.1.22}/drift/core/tracing/td_span_processor.py

@@ -171,6 +171,9 @@ class TdSpanProcessor(SpanProcessor):
                 replay_trace_id = replay_trace_id_context.get()
                 if replay_trace_id:
                     clean_span.trace_id = replay_trace_id
+                    # Rust path may have prebuilt bytes with the original OTel trace_id.
+                    # Force re-serialization so CLI receives the replay trace_id.
+                    clean_span.proto_span_bytes = None
                 else:
                     logger.error("No replay trace ID found, cannot send inbound span for replay")
                     return

{tusk_drift_python_sdk-0.1.20 → tusk_drift_python_sdk-0.1.22}/drift/instrumentation/django/middleware.py

@@ -144,15 +144,19 @@ class DriftMiddleware:
         try:
             with SpanUtils.with_span(span_info):
                 response = self.get_response(request)
-                # REPLAY mode: don't capture the span (it's already recorded)
-                # But do normalize the response so comparison succeeds
                 response = self._normalize_html_response(response)
+
+                # Capture response data on the span so the inbound replay span
+                # sent to the CLI includes the actual OUTPUT_VALUE for UI display
+                self._capture_replay_output(request, response, span_info)
+
                 return response
         finally:
-            # Reset context
+            # End span BEFORE resetting context so that TdSpanProcessor.on_end()
+            # can still read replay_trace_id_context to send the inbound span
+            span_info.span.end()
             span_kind_context.reset(span_kind_token)
             replay_trace_id_context.reset(replay_token)
-            span_info.span.end()
 
     def _record_request(self, request: HttpRequest, sdk, is_pre_app_start: bool) -> HttpResponse:
         """Handle request in RECORD mode.
@@ -262,6 +266,48 @@ class DriftMiddleware:
             if route:
                 request._drift_route_template = route  # type: ignore
 
+    def _capture_replay_output(self, request: HttpRequest, response: HttpResponse, span_info: SpanInfo) -> None:
+        """Capture response data on the span for REPLAY mode.
+
+        Sets OUTPUT_VALUE so the inbound replay span sent to the CLI includes
+        the actual response for UI comparison. Skips RECORD-mode concerns like
+        transforms, trace blocking, and schema merges.
+
+        Args:
+            request: Django HttpRequest object
+            response: Django HttpResponse object
+            span_info: SpanInfo containing trace/span IDs and span reference
+        """
+        if not span_info.span.is_recording():
+            return
+
+        status_code = response.status_code
+        status_message = response.reason_phrase if hasattr(response, "reason_phrase") else ""
+        response_headers = dict(response.items()) if hasattr(response, "items") else {}
+
+        response_body = None
+        if hasattr(response, "content"):
+            content = response.content
+            if isinstance(content, bytes) and len(content) > 0:
+                response_body = content
+
+        if response_body:
+            from .html_utils import normalize_html_body
+
+            content_type = response_headers.get("Content-Type", "")
+            content_encoding = response_headers.get("Content-Encoding", "")
+            response_body = normalize_html_body(response_body, content_type, content_encoding)
+
+        output_value = build_output_value(
+            status_code,
+            status_message,
+            response_headers,
+            response_body,
+            None,
+        )
+
+        span_info.span.set_attribute(TdSpanAttributes.OUTPUT_VALUE, json.dumps(output_value))
+
     def _normalize_html_response(self, response: HttpResponse) -> HttpResponse:
         """Normalize HTML response body for REPLAY mode comparison.
 

{tusk_drift_python_sdk-0.1.20 → tusk_drift_python_sdk-0.1.22}/drift/instrumentation/e2e_common/base_runner.py

@@ -285,7 +285,10 @@ class E2ETestRunnerBase:
         self.log("Phase 3: Running Tusk Tests", Colors.BLUE)
         self.log("=" * 50, Colors.BLUE)
 
-        env = {"TUSK_ANALYTICS_DISABLED": "1"}
+        env = {
+            "TUSK_ANALYTICS_DISABLED": "1",
+            "TUSK_REQUIRE_INBOUND_REPLAY_SPAN": "1",
+        }
 
         result = self.run_command(
             ["tusk", "run", "--print", "--output-format", "json", "--enable-service-logs"],

{tusk_drift_python_sdk-0.1.20 → tusk_drift_python_sdk-0.1.22}/drift/instrumentation/fastapi/instrumentation.py

@@ -226,10 +226,11 @@ async def _handle_replay_request(
                 transform_engine,
             )
     finally:
-        # Reset context
+        # End span BEFORE resetting context so that TdSpanProcessor.on_end()
+        # can still read replay_trace_id_context to send the inbound span
+        span_info.span.end()
         span_kind_context.reset(span_kind_token)
         replay_trace_id_context.reset(replay_token)
-        span_info.span.end()
 
 
 async def _record_request(

{tusk_drift_python_sdk-0.1.20 → tusk_drift_python_sdk-0.1.22}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tusk-drift-python-sdk"
-version = "0.1.20"
+version = "0.1.22"
 description = "Python SDK for Tusk Drift instrumentation and replay"
 readme = "README.md"
 requires-python = ">=3.9"

{tusk_drift_python_sdk-0.1.20 → tusk_drift_python_sdk-0.1.22}/tusk_drift_python_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tusk-drift-python-sdk
-Version: 0.1.20
+Version: 0.1.22
 Summary: Python SDK for Tusk Drift instrumentation and replay
 Author-email: Tusk <support@usetusk.ai>
 License: Apache-2.0