From 84052a21792bfad263e3d72e9bc7038c175a191e Mon Sep 17 00:00:00 2001
From: snaily <snaily@linux.do>
Date: Fri, 28 Mar 2025 23:44:40 +0800
Subject: [PATCH] =?UTF-8?q?feat(auth):=20=E5=A2=9E=E5=BC=BAGemini=20API?=
 =?UTF-8?q?=E7=9A=84=E8=AE=A4=E8=AF=81=E6=9C=BA=E5=88=B6=E6=94=AF=E6=8C=81?=
 =?UTF-8?q?URL=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 将generate_content和stream_generate_content端点的认证依赖从verify_goog_api_key更改为verify_key_or_goog_api_key
- 使Gemini API同时支持URL参数中的key和请求头中的x-goog-api-key进行认证
- 提高API的灵活性，便于不同客户端集成
---
 app/router/gemini_routes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/app/router/gemini_routes.py b/app/router/gemini_routes.py
index 50a9efc..a8163c0 100644
--- a/app/router/gemini_routes.py
+++ b/app/router/gemini_routes.py
@@ -86,7 +86,7 @@ async def list_models(
 async def generate_content(
     model_name: str,
     request: GeminiRequest,
-    _=Depends(security_service.verify_goog_api_key),
+    _=Depends(security_service.verify_key_or_goog_api_key),
     api_key: str = Depends(get_next_working_key),
     key_manager: KeyManager = Depends(get_key_manager)
 ):
@@ -118,7 +118,7 @@ async def generate_content(
 async def stream_generate_content(
     model_name: str,
     request: GeminiRequest,
-    _=Depends(security_service.verify_goog_api_key),
+    _=Depends(security_service.verify_key_or_goog_api_key),
     api_key: str = Depends(get_next_working_key),
     key_manager: KeyManager = Depends(get_key_manager)
 ):