From 9f9197f8c86e1d13e2a29167f06117d89a74fe8e Mon Sep 17 00:00:00 2001 From: yinpeng <2291314224@qq.com> Date: Fri, 27 Dec 2024 20:19:51 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E4=B8=BAv1beta=20API=E6=B7=BB=E5=8A=A0?= =?UTF-8?q?=E8=B7=AF=E7=94=B1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/api/gemini_routes.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/app/api/gemini_routes.py b/app/api/gemini_routes.py index e4b965c..a552744 100644 --- a/app/api/gemini_routes.py +++ b/app/api/gemini_routes.py @@ -21,6 +21,7 @@ model_service = ModelService(settings.MODEL_SEARCH) @router.get("/models") +@router_v1beta.get("/models") async def list_models( key: str = None, token: str = Depends(security_service.verify_key), @@ -35,11 +36,12 @@ async def list_models( return models_json @router.post("/models/{model_name}:generateContent") +@router_v1beta.post("/models/{model_name}:generateContent") @RetryHandler(max_retries=3, key_manager=key_manager, key_arg="api_key") async def generate_content( model_name: str, request: GeminiRequest, - # x_goog_api_key: str = Depends(security_service.verify_goog_api_key), + x_goog_api_key: str = Depends(security_service.verify_goog_api_key), api_key: str = Depends(key_manager.get_next_working_key), ): chat_service = GeminiChatService(settings.BASE_URL, key_manager) @@ -63,11 +65,12 @@ async def generate_content( @router.post("/models/{model_name}:streamGenerateContent") +@router_v1beta.post("/models/{model_name}:streamGenerateContent") @RetryHandler(max_retries=3, key_manager=key_manager, key_arg="api_key") async def stream_generate_content( model_name: str, request: GeminiRequest, - # x_goog_api_key: str = Depends(security_service.verify_goog_api_key), + x_goog_api_key: str = Depends(security_service.verify_goog_api_key), api_key: str = Depends(key_manager.get_next_working_key), ): chat_service = GeminiChatService(settings.BASE_URL, key_manager)