UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-20 12:00:22 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-08-22 14:53:14 +02:00
parent ed5761f18f
commit 1c89c91f07
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
backend/apps/webui/main.py
View File

@ -277,6 +277,7 @@ async def generate_function_chat_completion(form_data, user):
model_id = form_data.get("model") model_id = form_data.get("model")
model_info = Models.get_model_by_id(model_id) model_info = Models.get_model_by_id(model_id)
metadata = form_data.pop("metadata", {}) metadata = form_data.pop("metadata", {})
files = metadata.get("files", []) files = metadata.get("files", [])
tool_ids = metadata.get("tool_ids", []) tool_ids = metadata.get("tool_ids", [])
@ -299,15 +300,18 @@ async def generate_function_chat_completion(form_data, user):
"__event_call__": __event_call__, "__event_call__": __event_call__,
"__task__": __task__, "__task__": __task__,
} }
tools_params = {
extra_params["__tools__"] = get_tools(
app,
tool_ids,
user,
{
**extra_params, **extra_params,
"__model__": app.state.MODELS[form_data["model"]], "__model__": app.state.MODELS[form_data["model"]],
"__messages__": form_data["messages"], "__messages__": form_data["messages"],
"__files__": files, "__files__": files,
} },
)
tools = get_tools(app, tool_ids, user, tools_params)
extra_params["__tools__"] = tools
if model_info: if model_info:
if model_info.base_model_id: if model_info.base_model_id: