ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

添加 transform_user_model_if_needed 逻辑，以适配 /api/chat/completed 接口对私有模型的处理

Browse source
This commit is contained in:
Gaofeng 2025-12-08 15:45:22 +08:00
parent 945c9f7c4d
commit b660a8fe7f
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
backend/open_webui/utils/models.py
View file

@ -374,6 +374,22 @@ def get_filtered_models(models, user):
async def transform_user_model_if_needed(form_data: dict, user: UserModel):
# If model_item is missing, try to reconstruct it from chat history
if "model_item" not in form_data or not form_data.get("model_item"):
chat_id = form_data.get("chat_id")
if chat_id:
from open_webui.models.chats import Chats
messages_map = Chats.get_messages_map_by_chat_id(chat_id)
if messages_map:
for message in messages_map.values():
if "models" in message and isinstance(message["models"], list) and message["models"]:
for model_ref in message["models"]:
if isinstance(model_ref, str) and model_ref.startswith("user:"):
form_data["model_item"] = { "credential_id": model_ref }
break
if "model_item" in form_data:
break
model_id = form_data.get("model")
model_item = form_data.get("model_item", {})