From 829c1b02ec855e5cefc499af5472457d0e7f55e2 Mon Sep 17 00:00:00 2001 From: yinpeng <2291314224@qq.com> Date: Sun, 15 Dec 2024 21:09:11 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9Elist=5Fmodels=E6=8E=A5?= =?UTF-8?q?=E5=8F=A3=E7=9A=84=E6=8E=88=E6=9D=83=E9=AA=8C=E8=AF=81=E9=80=BB?= =?UTF-8?q?=E8=BE=91=EF=BC=8C=E5=AE=8C=E5=96=84=E6=A8=A1=E5=9E=8B=E6=A0=BC?= =?UTF-8?q?=E5=BC=8F=E8=BD=AC=E6=8D=A2=E7=BB=93=E6=9E=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/api/routes.py | 5 ++++- app/services/model_service.py | 4 ++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/app/api/routes.py b/app/api/routes.py index b05fdb3..e87c8da 100644 --- a/app/api/routes.py +++ b/app/api/routes.py @@ -24,7 +24,10 @@ embedding_service = EmbeddingService(settings.BASE_URL) @router.get("/v1/models") @router.get("/hf/v1/models") -async def list_models(): +async def list_models( + authorization: str = Header(None), + token: str = Depends(security_service.verify_authorization), +): logger.info("Handling models list request") api_key = await key_manager.get_next_working_key() logger.info(f"Using API key: {api_key}") diff --git a/app/services/model_service.py b/app/services/model_service.py index 49fcdd5..056f924 100644 --- a/app/services/model_service.py +++ b/app/services/model_service.py @@ -31,7 +31,7 @@ class ModelService: def convert_to_openai_models_format( self, gemini_models: Dict[str, Any] ) -> Dict[str, Any]: - openai_format = {"object": "list", "data": []} + openai_format = {"object": "list", "data": [],"success": True} for model in gemini_models.get("models", []): model_id = model["name"].split("/")[-1] @@ -43,7 +43,7 @@ class ModelService: "permission": [], "root": model["name"], "parent": None, - "success": True, + } openai_format["data"].append(openai_model)