UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-03 12:31:32 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

formatting

Browse Source
This commit is contained in:
Chris Kanich 2025-03-11 15:51:13 -05:00
parent c6e32715e3
commit 3bc97cf3fb
3 changed files with 25 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/open_webui/utils/chat.py
View File

@ -397,8 +397,12 @@ async def chat_action(request: Request, action_id: str, form_data: dict, user: A
if hasattr(function_module, "valves") and hasattr(function_module, "Valves"):
valves = Functions.get_function_valves_by_id(action_id)
model_valves = model.get("info", {}).get("meta", {}).get("valves", {}).get(action_id, {})
function_module.valves = function_module.Valves(**(valves if valves else {}), **model_valves)
model_valves = (
model.get("info", {}).get("meta", {}).get("valves", {}).get(action_id, {})
)
function_module.valves = function_module.Valves(
**(valves if valves else {}), **model_valves
)
if hasattr(function_module, "action"):
try:

9
backend/open_webui/utils/filter.py
View File

@ -62,7 +62,14 @@ async def process_filter_functions(
if hasattr(function_module, "valves") and hasattr(function_module, "Valves"):
valves = Functions.get_function_valves_by_id(filter_id)
# overwrite global valves with model valves
model_valves = extra_params.get("__model__", {}).get("info", {}).get("meta", {}).get("valves", {}).get("functions", {}).get(filter_id, {})
model_valves = (
extra_params.get("__model__", {})
.get("info", {})
.get("meta", {})
.get("valves", {})
.get("functions", {})
.get(filter_id, {})
)
valves = {**valves, **model_valves}
function_module.valves = function_module.Valves(

12
backend/open_webui/utils/tools.py
View File

@ -54,7 +54,17 @@ def get_tools(
if hasattr(module, "valves") and hasattr(module, "Valves"):
valves = Tools.get_tool_valves_by_id(tool_id) or {}
# overwrite global valves with model valves
valves = { **valves, **(extra_params.get("__model__", {}).get("info", {}).get("meta", {}).get("valves", {}).get("tools", {}).get(tool_id, {}))}
valves = {
**valves,
**(
extra_params.get("__model__", {})
.get("info", {})
.get("meta", {})
.get("valves", {})
.get("tools", {})
.get(tool_id, {})
),
}
module.valves = module.Valves(**valves)
if hasattr(module, "UserValves"):