adjust conftest

Author: alizenhom

src/tests/openai/conftest.py

@@ -26,59 +26,27 @@ def async_openai_client():
 
 @pytest.fixture(scope="module")
 def vcr_config():
-    def before_record_response(response):
-        # Mask rate limit information and sensitive headers
-        headers = response["headers"]
-        sensitive_headers = {
-            "x-request-id", "x-Request-id", "X-Request-ID",
-            "openai-organization", "openai-Organization", "openai-ORGANIZATION",
-            "set-cookie", "Set-Cookie",
-            "_cfuvid", "__cf_bm",
-            "openai-processing-ms"
-        }
-
-        for key in list(headers.keys()):
-            # Case insensitive header matching
-            if any(sensitive.lower() == key.lower() for sensitive in sensitive_headers):
-                headers[key] = ["[FILTERED]"]
-            elif key.lower().startswith(("x-ratelimit-", "openai-processing")):
-                headers[key] = ["0"]
-
-        # Keep response structure but mask actual values
-        if "body" in response and "string" in response["body"]:
-            try:
-                import json
-                import base64
-                decoded = json.loads(base64.b64decode(response["body"]["string"]))
-                if "data" in decoded and isinstance(decoded["data"], list):
-                    for item in decoded["data"]:
-                        if "embedding" in item:
-                            item["embedding"] = [0.1] * len(item["embedding"])
-                response["body"]["string"] = base64.b64encode(json.dumps(decoded).encode()).decode()
-            except:
-                pass
-        return response
-
     return {
         "filter_headers": [
-            "authorization",
-            "api-key",
-            "user-agent",
-            "x-stainless-arch",
-            "x-stainless-async",
-            "x-stainless-lang",
-            "x-stainless-os",
-            "x-stainless-package-version",
-            "x-stainless-runtime",
-            "x-stainless-runtime-version",
-            "host",
-            "cf-ray",
+            ("cookie", "test_cookie"),
+            ("authorization", "Bearer test_openai_api_key"),
+            ("openai-organization", "test_openai_org_id"),
+            ("openai-project", "test_openai_project_id"),
         ],
-        "filter_query_parameters": ["api-version"],
-        "before_record_response": before_record_response,
+        "decode_compressed_response": True,
+        "before_record_response": scrub_response_headers,
     }
 
 
 @pytest.fixture(scope="session", autouse=True)
 def instrument():
     OpenAIInstrumentation().instrument()
+
+
+def scrub_response_headers(response):
+    """
+    This scrubs sensitive response headers. Note they are case-sensitive!
+    """
+    response["headers"]["openai-organization"] = "test_openai_org_id"
+    response["headers"]["Set-Cookie"] = "test_set_cookie"
+    return response