소스 검색

fix: prefix issue

Timothy Jaeryang Baek 5 달 전
부모
커밋
d5c65e36c9
1개의 변경된 파일3개의 추가작업 그리고 1개의 파일을 삭제
  1. 3 1
      backend/open_webui/apps/openai/main.py

+ 3 - 1
backend/open_webui/apps/openai/main.py

@@ -313,7 +313,9 @@ async def get_all_models_responses() -> list:
             prefix_id = api_config.get("prefix_id", None)
 
             if prefix_id:
-                for model in response["data"]:
+                for model in (
+                    response if isinstance(response, list) else response.get("data", [])
+                ):
                     model["id"] = f"{prefix_id}.{model['id']}"
 
     log.debug(f"get_all_models:responses() {responses}")