Update mock_server.py

Author: jackalcooper

tests/mock_server.py

@@ -641,8 +641,20 @@ def _prepare_proxy_and_headers(
     request_id = request.headers.get("x-request-id", str(uuid.uuid4()))
     api_key = DUMMY_KEY
 
+    # get x-api-key
+    x_api_key = request.headers.get("x-api-key")
+
     if SERVER_STATE.is_mock_mode:
         api_key = _MOCK_ENV_API_KEY or "mock-key"
+
+    # [Modification]: If x-api-key exists, override authorization header
+    elif x_api_key:
+        api_key = x_api_key
+        logger.debug(
+            f"Request {request_id}: Using x-api-key for authorization override."
+        )
+
+    # Only check Authorization Header if x-api-key is not present
     elif authorization:
         if not authorization.startswith("Bearer "):
             logger.warning(
@@ -660,12 +672,14 @@ def _prepare_proxy_and_headers(
         "host",
         "content-length",
         "content-type",
-        "authorization",
+        "authorization",  # Authorization has been regenerated by AsyncOpenAI client through api_key parameter, so it is filtered out here
         "connection",
         "upgrade",
         "accept-encoding",
         "transfer-encoding",
     }
+
+    # x-api-key 不在 unsafe_headers 中，因此会被保留在 forward_headers 中发送给上游
     forward_headers = {
         k: v for k, v in request.headers.items() if k.lower() not in unsafe_headers
     }