UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 05:24:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: valves

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-23 19:37:35 -07:00
parent 2eb15ea1fc
commit 5f2d37dce5
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
backend/main.py
View File

@ -270,8 +270,9 @@ async def get_function_call_response(
if hasattr(toolkit_module, "valves") and hasattr( if hasattr(toolkit_module, "valves") and hasattr(
toolkit_module, "Valves" toolkit_module, "Valves"
): ):
valves = Tools.get_tool_valves_by_id(tool_id)
toolkit_module.valves = toolkit_module.Valves( toolkit_module.valves = toolkit_module.Valves(
**Tools.get_tool_valves_by_id(tool_id) **(valves if valves else {})
) )
function = getattr(toolkit_module, result["name"]) function = getattr(toolkit_module, result["name"])
@ -417,8 +418,9 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
if hasattr(function_module, "valves") and hasattr( if hasattr(function_module, "valves") and hasattr(
function_module, "Valves" function_module, "Valves"
): ):
valves = Functions.get_function_valves_by_id(filter_id)
function_module.valves = function_module.Valves( function_module.valves = function_module.Valves(
**Functions.get_function_valves_by_id(filter_id) **(valves if valves else {})
) )
try: try:
@ -906,8 +908,10 @@ async def generate_chat_completions(form_data: dict, user=Depends(get_verified_u
if hasattr(function_module, "valves") and hasattr( if hasattr(function_module, "valves") and hasattr(
function_module, "Valves" function_module, "Valves"
): ):
valves = Functions.get_function_valves_by_id(pipe_id)
function_module.valves = function_module.Valves( function_module.valves = function_module.Valves(
**Functions.get_function_valves_by_id(pipe_id) **(valves if valves else {})
) )
pipe = function_module.pipe pipe = function_module.pipe
@ -1134,8 +1138,9 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)):
if hasattr(function_module, "valves") and hasattr( if hasattr(function_module, "valves") and hasattr(
function_module, "Valves" function_module, "Valves"
): ):
valves = Functions.get_function_valves_by_id(filter_id)
function_module.valves = function_module.Valves( function_module.valves = function_module.Valves(
**Functions.get_function_valves_by_id(filter_id) **(valves if valves else {})
) )
try: try: