GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-09-08 17:36:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 为v1beta API添加路由

Browse Source
This commit is contained in:
yinpeng
2024-12-27 20:19:51 +08:00
parent 870b1ecc17
commit 9f9197f8c8
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/api/gemini_routes.py
+5 -2
View File
@@ -21,6 +21,7 @@ model_service = ModelService(settings.MODEL_SEARCH)
@router.get("/models") @router.get("/models")
@router_v1beta.get("/models")
async def list_models( async def list_models(
key: str = None, key: str = None,
token: str = Depends(security_service.verify_key), token: str = Depends(security_service.verify_key),
@@ -35,11 +36,12 @@ async def list_models(
return models_json return models_json
@router.post("/models/{model_name}:generateContent") @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") @RetryHandler(max_retries=3, key_manager=key_manager, key_arg="api_key")
async def generate_content( async def generate_content(
model_name: str, model_name: str,
request: GeminiRequest, 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), api_key: str = Depends(key_manager.get_next_working_key),
): ):
chat_service = GeminiChatService(settings.BASE_URL, key_manager) chat_service = GeminiChatService(settings.BASE_URL, key_manager)
@@ -63,11 +65,12 @@ async def generate_content(
@router.post("/models/{model_name}:streamGenerateContent") @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") @RetryHandler(max_retries=3, key_manager=key_manager, key_arg="api_key")
async def stream_generate_content( async def stream_generate_content(
model_name: str, model_name: str,
request: GeminiRequest, 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), api_key: str = Depends(key_manager.get_next_working_key),
): ):
chat_service = GeminiChatService(settings.BASE_URL, key_manager) chat_service = GeminiChatService(settings.BASE_URL, key_manager)