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

overwrite action, tool, and filter valves

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

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

@ -397,7 +397,8 @@ 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)
function_module.valves = function_module.Valves(**(valves if valves else {}))
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:

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

@ -61,12 +61,10 @@ async def process_filter_functions(
# Apply valves to the function
if hasattr(function_module, "valves") and hasattr(function_module, "Valves"):
valves = Functions.get_function_valves_by_id(filter_id)
print(f"global valves for {filter_id}: {valves}")
# overwrite global valves with model valves
model_valves = extra_params.get("__model__", {}).get("info", {}).get("meta", {}).get("valves", {}).get("filters", {}).get(filter_id, {})
model_valves = extra_params.get("__model__", {}).get("info", {}).get("meta", {}).get("valves", {}).get("functions", {}).get(filter_id, {})
valves = {**valves, **model_valves}
print(f"applying model valves: {model_valves}")
function_module.valves = function_module.Valves(
**(valves if valves else {})
)

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

@ -53,6 +53,8 @@ def get_tools(
extra_params["__id__"] = tool_id
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, {}))}
module.valves = module.Valves(**valves)
if hasattr(module, "UserValves"):