"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "5d6182eb7fc2edd8ed5d6b28aa2c2ca62c0757af"
Commit 5f2d37dc authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

fix: valves

parent 2eb15ea1
...@@ -270,8 +270,9 @@ async def get_function_call_response( ...@@ -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): ...@@ -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 ...@@ -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)): ...@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment