Browse Source

refac: error handling

Timothy Jaeryang Baek 2 tháng trước cách đây
mục cha
commit
d3547f0f54
2 tập tin đã thay đổi với 4 bổ sung2 xóa
  1. 1 1
      backend/open_webui/main.py
  2. 3 1
      backend/open_webui/routers/openai.py

+ 1 - 1
backend/open_webui/main.py

@@ -1296,7 +1296,7 @@ async def get_models(
         models = get_filtered_models(models, user)
 
     log.debug(
-        f"/api/models returned filtered models accessible to the user: {json.dumps([model['id'] for model in models])}"
+        f"/api/models returned filtered models accessible to the user: {json.dumps([model.get('id') for model in models])}"
     )
     return {"data": models}
 

+ 3 - 1
backend/open_webui/routers/openai.py

@@ -362,7 +362,9 @@ async def get_all_models_responses(request: Request, user: UserModel) -> list:
                 response if isinstance(response, list) else response.get("data", [])
             ):
                 if prefix_id:
-                    model["id"] = f"{prefix_id}.{model['id']}"
+                    model["id"] = (
+                        f"{prefix_id}.{model.get('id', model.get('name', ''))}"
+                    )
 
                 if tags:
                     model["tags"] = tags