wip: admin models setting

backend/open_webui/apps/webui/models/models.py

@@ -188,6 +188,13 @@ class ModelsTable:
                 for model in db.query(Model).filter(Model.base_model_id != None).all()
             ]
 
+    def get_base_models(self) -> list[ModelModel]:
+        with get_db() as db:
+            return [
+                ModelModel.model_validate(model)
+                for model in db.query(Model).filter(Model.base_model_id == None).all()
+            ]
+
     def get_models_by_user_id(
         self, user_id: str, permission: str = "write"
     ) -> list[ModelModel]:

backend/open_webui/apps/webui/routers/models.py

@@ -28,6 +28,16 @@ async def get_models(id: Optional[str] = None, user=Depends(get_verified_user)):
         return Models.get_models_by_user_id(user.id)
 
 
+###########################
+# GetBaseModels
+###########################
+
+
+@router.get("/base", response_model=list[ModelResponse])
+async def get_base_models(user=Depends(get_admin_user)):
+    return Models.get_base_models()
+
+
 ############################
 # CreateNewModel
 ############################

backend/open_webui/main.py

@@ -993,7 +993,7 @@ async def get_all_models():
 
             models.append(
                 {
-                    "id": f"open-webui-{custom_model.id}",
+                    "id": f"open-webui.{custom_model.id}",
                     "name": custom_model.name,
                     "object": "model",
                     "created": custom_model.created_at,
@@ -1154,8 +1154,8 @@ async def generate_chat_completions(
                 "selected_model_id": selected_model_id,
             }
 
-    if model_id.startswith("open-webui-"):
-        model_id = model_id[len("open-webui-") :]
+    if model_id.startswith("open-webui."):
+        model_id = model_id[len("open-webui.") :]
         form_data["model"] = model_id
 
     if model.get("pipe"):