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

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-08-17 16:46:04 +02:00
parent 15f3ebba93
commit d5337917db
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/main.py
View File

@ -384,10 +384,6 @@ def get_tools(
webui_app.state.TOOLS[tool_id] = module webui_app.state.TOOLS[tool_id] = module
extra_params["__id__"] = tool_id extra_params["__id__"] = tool_id
has_citation = hasattr(module, "citation") and module.citation
handles_files = hasattr(module, "file_handler") and module.file_handler
if hasattr(module, "valves") and hasattr(module, "Valves"): if hasattr(module, "valves") and hasattr(module, "Valves"):
valves = Tools.get_tool_valves_by_id(tool_id) or {} valves = Tools.get_tool_valves_by_id(tool_id) or {}
module.valves = module.Valves(**valves) module.valves = module.Valves(**valves)
@ -411,11 +407,11 @@ def get_tools(
# TODO: This needs to be a pydantic model # TODO: This needs to be a pydantic model
tool_dict = { tool_dict = {
"spec": spec,
"citation": has_citation,
"file_handler": handles_files,
"toolkit_id": tool_id, "toolkit_id": tool_id,
"callable": callable, "callable": callable,
"spec": spec,
"file_handler": hasattr(module, "file_handler") and module.file_handler,
"citation": hasattr(module, "citation") and module.citation,
} }
# TODO: if collision, prepend toolkit name # TODO: if collision, prepend toolkit name
@ -507,7 +503,7 @@ async def chat_completion_tools_handler(
if tools[tool_function_name]["citation"]: if tools[tool_function_name]["citation"]:
citations.append( citations.append(
{ {
"source": {"name": f"TOOL:{tools[tool_function_name]["toolkit_id"]}/{tool_function_name}"}, "source": {"name": f"TOOL:{tools[tool_function_name]['toolkit_id']}/{tool_function_name}"},
"document": [tool_output], "document": [tool_output],
"metadata": [{"source": tool_function_name}], "metadata": [{"source": tool_function_name}],
} }