fix formatting

Author: sohankshirsagar

drift/instrumentation/requests/e2e-tests/src/app.py

@@ -261,6 +261,7 @@ def text_response():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+
 @app.route("/test/session-send-direct", methods=["GET"])
 def session_send_direct():
     try:
@@ -273,37 +274,33 @@ def session_send_direct():
         # Call send() directly - bypasses request() method
         response = session.send(prepared, timeout=10)
 
-        return jsonify({
-            "status_code": response.status_code,
-            "data": response.json(),
-            "test": "session-send-direct"
-        })
+        return jsonify({"status_code": response.status_code, "data": response.json(), "test": "session-send-direct"})
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+
 @app.route("/test/streaming-iter-lines", methods=["GET"])
 def test_streaming_iter_lines():
     """Test streaming response using iter_lines."""
     try:
         response = requests.get(
             "https://httpbin.org/stream/5",  # Returns 5 JSON lines
             stream=True,
-            timeout=10
+            timeout=10,
         )
         # Consume the stream using iter_lines
         lines = []
         for line in response.iter_lines():
             if line:
-                lines.append(line.decode('utf-8'))
+                lines.append(line.decode("utf-8"))
 
-        return jsonify({
-            "test": "streaming-iter-lines",
-            "status_code": response.status_code,
-            "lines_received": len(lines)
-        })
+        return jsonify(
+            {"test": "streaming-iter-lines", "status_code": response.status_code, "lines_received": len(lines)}
+        )
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+
 @app.route("/test/response-hooks", methods=["GET"])
 def test_response_hooks():
     """Test response hooks that modify response."""
@@ -316,18 +313,16 @@ def response_hook(response, *args, **kwargs):
             hook_status["value"] = response.status_code
             return response
 
-        response = requests.get(
-            "https://httpbin.org/get",
-            hooks={"response": response_hook},
-            timeout=10
-        )
+        response = requests.get("https://httpbin.org/get", hooks={"response": response_hook}, timeout=10)
 
-        return jsonify({
-            "test": "response-hooks",
-            "status_code": response.status_code,
-            "hook_called": hook_called["value"],
-            "hook_status": hook_status["value"]
-        })
+        return jsonify(
+            {
+                "test": "response-hooks",
+                "status_code": response.status_code,
+                "hook_called": hook_called["value"],
+                "hook_status": hook_status["value"],
+            }
+        )
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 

drift/instrumentation/requests/instrumentation.py

@@ -131,9 +131,7 @@ def original_call():
             # REPLAY mode: Use handle_replay_mode for proper background request handling
             if sdk.mode == TuskDriftMode.REPLAY:
                 return handle_replay_mode(
-                    replay_mode_handler=lambda: instrumentation_self._handle_replay_send(
-                        sdk, request, **kwargs
-                    ),
+                    replay_mode_handler=lambda: instrumentation_self._handle_replay_send(sdk, request, **kwargs),
                     no_op_request_handler=lambda: instrumentation_self._get_default_response(url),
                     is_server_request=False,
                 )
@@ -182,7 +180,6 @@ def _handle_record_send(
         Similar to _handle_record but works with PreparedRequest objects.
 
         Args:
-            sdk: TuskDrift instance
             session_self: Session instance
             prepared_request: PreparedRequest object
             is_pre_app_start: Whether this is before app start
@@ -319,6 +316,7 @@ def _handle_replay_send(
                     # Dispatch response hooks (matches Session.send() behavior)
                     # This ensures hooks registered via hooks={"response": callback} are called
                     from requests.hooks import dispatch_hook
+
                     mock_response = dispatch_hook("response", prepared_request.hooks, mock_response, **kwargs)
                     return mock_response