From 5f2d37dce5c657a05c5c1544657271e10afc1a33 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Sun, 23 Jun 2024 19:37:35 -0700 Subject: [PATCH] fix: valves --- backend/main.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/backend/main.py b/backend/main.py index 4a889f1b3..2f2e549bd 100644 --- a/backend/main.py +++ b/backend/main.py @@ -270,8 +270,9 @@ async def get_function_call_response( if hasattr(toolkit_module, "valves") and hasattr( toolkit_module, "Valves" ): + valves = Tools.get_tool_valves_by_id(tool_id) toolkit_module.valves = toolkit_module.Valves( - **Tools.get_tool_valves_by_id(tool_id) + **(valves if valves else {}) ) function = getattr(toolkit_module, result["name"]) @@ -417,8 +418,9 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware): if hasattr(function_module, "valves") and hasattr( function_module, "Valves" ): + valves = Functions.get_function_valves_by_id(filter_id) function_module.valves = function_module.Valves( - **Functions.get_function_valves_by_id(filter_id) + **(valves if valves else {}) ) 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( function_module, "Valves" ): + + valves = Functions.get_function_valves_by_id(pipe_id) function_module.valves = function_module.Valves( - **Functions.get_function_valves_by_id(pipe_id) + **(valves if valves else {}) ) 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( function_module, "Valves" ): + valves = Functions.get_function_valves_by_id(filter_id) function_module.valves = function_module.Valves( - **Functions.get_function_valves_by_id(filter_id) + **(valves if valves else {}) ) try: