update

app.py

@@ -6,6 +6,7 @@ from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials # 导入HT
 import httpx # 使用httpx替代requests，因为requests是同步的，而FastAPI是异步的
 import os
 from dotenv import load_dotenv
+from enum import Enum
 
 # 加载环境变量
 load_dotenv()
@@ -16,6 +17,11 @@ PROXY_API_KEY = os.getenv("PROXY_API_KEY")
 # 定义HTTPBearer安全方案
 security = HTTPBearer()
 
+# 枚举校验协议类型
+class ProtocolType(str, Enum):
+    HTTP = "http"
+    HTTPS = "https"
+
 # 依赖函数：验证代理的API密钥
 def verify_proxy_api_key(credentials: HTTPAuthorizationCredentials = Depends(security)):
     if not PROXY_API_KEY or credentials.credentials != PROXY_API_KEY:
@@ -44,10 +50,20 @@ GEMINI_BASE_URL = os.getenv("GEMINI_BASE_URL")  # 从环境变量获取Gemini真
 # print(f"Using Gemini API Key: {GEMINI_API_KEY}")
 
 # 2. 通用转发接口（匹配Gemini的/predict等端点，路径动态匹配）
-@app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
-async def proxy_gemini(request: Request, path: str, proxy_api_key: str = Depends(verify_proxy_api_key)): # 添加代理认证依赖
+@app.api_route("/v1/{protocol}/{host}/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
+async def proxy_gemini(request: Request, protocol: ProtocolType, host:str, path: str, proxy_api_key: str = Depends(verify_proxy_api_key)): # 添加代理认证依赖
     # 拼接Gemini真实请求URL
     gemini_url = f"{GEMINI_BASE_URL}/{path}"
+
+
+    print('gemini_url', gemini_url)
+
+    print('protocol.value', protocol.value)
+    print('host', host)
+    print('path', path)
+
+    real_url = f"{protocol.value}://{host}/{path}"
+    print('real_url', real_url)
     
     # 提取客户端请求的headers和body，透传给Gemini
     client_headers = dict(request.headers)
@@ -70,7 +86,8 @@ async def proxy_gemini(request: Request, path: str, proxy_api_key: str = Depends
         async with httpx.AsyncClient() as client:
             response = await client.request(
                 method=request.method,
-                url=gemini_url,
+                # url=gemini_url,
+                url=real_url,
                 headers=client_headers,
                 content=client_body, # httpx使用content参数传递请求体
                 timeout=30  # 超时时间，可调整