OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac: model preset handling behaviour

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2024-11-15 22:04:33 -08:00
parent 19c98b74fa
commit d5f84d6234
7 changed files with 305 additions and 206 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/open_webui/apps/webui/models/models.py
View File

@@ -238,7 +238,7 @@ class ModelsTable:
result = (
db.query(Model)
.filter_by(id=id)
.update(model.model_dump(exclude={"id"}, exclude_none=True))
.update(model.model_dump(exclude={"id"}))
)
db.commit()

10
backend/open_webui/main.py
View File

@@ -948,7 +948,7 @@ async def get_all_models():
models = await get_all_base_models()
# If there are no models, return an empty list
if len([model for model in models if model["owned_by"] != "arena"]) == 0:
if len([model for model in models if not model.get("arena", False)]) == 0:
return []
global_action_ids = [
@@ -975,7 +975,7 @@ async def get_all_models():
action_ids.extend(model["info"]["meta"].get("actionIds", []))
model["action_ids"] = action_ids
else:
elif custom_model.id not in [model["id"] for model in models]:
owned_by = "openai"
pipe = None
action_ids = []
@@ -997,7 +997,7 @@ async def get_all_models():
models.append(
{
"id": f"open-webui.{custom_model.id}",
"id": f"{custom_model.id}",
"name": custom_model.name,
"object": "model",
"created": custom_model.created_at,
@@ -1164,10 +1164,6 @@ async def generate_chat_completions(
"selected_model_id": selected_model_id,
}
if model_id.startswith("open-webui."):
model_id = model_id[len("open-webui.") :]
form_data["model"] = model_id
if model.get("pipe"):
# Below does not require bypass_filter because this is the only route the uses this function and it is already bypassing the filter
return await generate_function_chat_completion(form_data, user=user)