update

.clinerules

@@ -1,6 +1,9 @@
 # 自定义指令
 CUSTOM_INSTRUCTIONS = """
-
+# 自动激活conda环境
+# 在每次会话开始时，自动激活名为 'airs' 的conda环境。
+# 这确保了所有后续的Python相关命令都在正确的环境中执行。
+conda activate airs
 """
 
 

app.py

@@ -1,7 +1,8 @@
 # uvicorn app:app --host 0.0.0.0 --port 7860 --reload 
 
 from fastapi import FastAPI, Request, HTTPException
-import requests
+from fastapi.responses import Response # 导入Response
+import httpx # 使用httpx替代requests，因为requests是同步的，而FastAPI是异步的
 import os
 from dotenv import load_dotenv
 
@@ -13,33 +14,45 @@ app = FastAPI(title="Gemini Proxy API")
 def greet_json():
     return {"Hello": "World!"}
 
+# 3. 健康检查接口（可选，用于验证服务是否正常）
+@app.get("/health")
+async def health_check():
+    return {"status": "ok", "message": "Proxy service is running."}
+
 # 1. 配置Gemini真实信息
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")  # 从环境变量获取密钥
-GEMINI_BASE_URL = "https://generativelanguage.googleapis.com/v1beta/models"  # Gemini真实Base URL
+GEMINI_BASE_URL = os.getenv("GEMINI_BASE_URL")  # 从环境变量获取Gemini真实Base URL
+print(f"Using Gemini Base URL: {GEMINI_BASE_URL}")
+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):
-    # 拼接Gemini真实请求URL（如客户端访问 /models/gemini-pro:generateContent → 转发到Gemini的对应路径）
-    gemini_url = f"{GEMINI_BASE_URL}/{path}?key={GEMINI_API_KEY}"
+    # 拼接Gemini真实请求URL
+    gemini_url = f"{GEMINI_BASE_URL}/{path}"
     
     # 提取客户端请求的headers和body，透传给Gemini
     client_headers = dict(request.headers)
     # 移除FastAPI自带的Host头，避免Gemini校验报错
     client_headers.pop("host", None)
     
+    # 添加Authorization头，用于Gemini的OpenAI兼容API
+    if GEMINI_API_KEY:
+        client_headers["Authorization"] = f"Bearer {GEMINI_API_KEY}"
+    
     try:
         # 读取客户端请求体
         client_body = await request.body()
         
-        # 向Gemini发起请求（转发）
-        response = requests.request(
-            method=request.method,
-            url=gemini_url,
-            headers=client_headers,
-            data=client_body,
-            timeout=30  # 超时时间，可调整
-        )
+        # 使用httpx异步客户端发起请求
+        async with httpx.AsyncClient() as client:
+            response = await client.request(
+                method=request.method,
+                url=gemini_url,
+                headers=client_headers,
+                content=client_body, # httpx使用content参数传递请求体
+                timeout=30  # 超时时间，可调整
+            )
         
         # 将Gemini的响应透传给客户端
         return Response(
@@ -48,10 +61,9 @@ async def proxy_gemini(request: Request, path: str):
             headers=dict(response.headers)
         )
     
+    except httpx.RequestError as e:
+        raise HTTPException(status_code=500, detail=f"转发请求失败：{str(e)}")
+    except httpx.HTTPStatusError as e:
+        raise HTTPException(status_code=e.response.status_code, detail=f"目标API返回错误：{e.response.text}")
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"转发失败：{str(e)}")
-
-# 3. 健康检查接口（可选，用于验证服务是否正常）
-@app.get("/health")
-async def health_check():
-    return {"status": "ok", "proxy_target": "Google Gemini"}

requirements.txt

@@ -1,2 +1,4 @@
 fastapi
 uvicorn[standard]
+httpx
+python-dotenv