format + add python version to manifest

Author: sohankshirsagar

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

@@ -88,6 +88,7 @@ def redis_keys(pattern):
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+
 @app.route("/test/mget-mset", methods=["GET"])
 def test_mget_mset():
     """Test MGET/MSET - multiple key operations."""
@@ -102,6 +103,7 @@ def test_mget_mset():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+
 @app.route("/test/pipeline-basic", methods=["GET"])
 def test_pipeline_basic():
     """Test basic pipeline operations."""
@@ -134,6 +136,7 @@ def test_pipeline_no_transaction():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+
 @app.route("/test/async-pipeline", methods=["GET"])
 def test_async_pipeline():
     """Test async pipeline operations using asyncio."""
@@ -146,7 +149,7 @@ async def run_async_pipeline():
             host=os.getenv("REDIS_HOST", "redis"),
             port=int(os.getenv("REDIS_PORT", "6379")),
             db=0,
-            decode_responses=True
+            decode_responses=True,
         )
 
         try:
@@ -167,13 +170,11 @@ async def run_async_pipeline():
 
     try:
         results = asyncio.run(run_async_pipeline())
-        return jsonify({
-            "success": True,
-            "results": results
-        })
+        return jsonify({"success": True, "results": results})
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+
 @app.route("/test/binary-data", methods=["GET"])
 def test_binary_data():
     """Test binary data that cannot be decoded as UTF-8."""
@@ -183,7 +184,7 @@ def test_binary_data():
             host=os.getenv("REDIS_HOST", "redis"),
             port=int(os.getenv("REDIS_PORT", "6379")),
             db=0,
-            decode_responses=False
+            decode_responses=False,
         )
 
         # Binary data that cannot be decoded as UTF-8
@@ -198,15 +199,18 @@ def test_binary_data():
         # Clean up
         binary_client.delete("test:binary:key")
 
-        return jsonify({
-            "success": True,
-            "original_hex": binary_value.hex(),
-            "retrieved_hex": retrieved.hex() if retrieved else None,
-            "match": binary_value == retrieved
-        })
+        return jsonify(
+            {
+                "success": True,
+                "original_hex": binary_value.hex(),
+                "retrieved_hex": retrieved.hex() if retrieved else None,
+                "match": binary_value == retrieved,
+            }
+        )
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+
 @app.route("/test/transaction-watch", methods=["GET"])
 def test_transaction_watch():
     """Test transaction with WATCH pattern.
@@ -235,12 +239,7 @@ def test_transaction_watch():
         # Clean up
         redis_client.delete("test:watch:counter")
 
-        return jsonify({
-            "success": True,
-            "initial_value": 10,
-            "expected_final": 15,
-            "results": results
-        })
+        return jsonify({"success": True, "initial_value": 10, "expected_final": 15, "results": results})
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 

drift/instrumentation/redis/e2e-tests/src/test_requests.py

@@ -33,8 +33,7 @@ def make_request(method, endpoint, **kwargs):
     # Get operations
     make_request("GET", "/redis/get/test_key")
     make_request("GET", "/redis/get/test_key_expiry")
-    # TODO: figure out why this test fails during replay
-    # make_request("GET", "/redis/get/nonexistent_key")
+    make_request("GET", "/redis/get/nonexistent_key")
 
     # Increment operations
     make_request("POST", "/redis/incr/counter")

drift/instrumentation/redis/instrumentation.py

@@ -732,8 +732,7 @@ def _build_command_input_value(self, command_str: str, args: tuple) -> dict[str,
     def _build_pipeline_input_value(self, command_str: str, command_stack: list) -> dict[str, Any]:
         """Build input_value for pipeline operations (used by both record and replay)."""
         serialized_commands = [
-            self._serialize_args(cmd.args if hasattr(cmd, "args") else cmd[0])
-            for cmd in command_stack
+            self._serialize_args(cmd.args if hasattr(cmd, "args") else cmd[0]) for cmd in command_stack
         ]
         return {
             "command": command_str,

scripts/generate_manifest.py

@@ -176,6 +176,7 @@ def main() -> int:
     manifest = {
         "sdkVersion": sdk_version,
         "language": "python",
+        "pythonVersion": ">=3.12",
         "generatedAt": datetime.now(UTC).isoformat().replace("+00:00", "Z"),
         "instrumentations": instrumentations,
     }