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

fix: valves

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-23 19:48:16 -07:00
parent 5f2d37dce5
commit 0250f69da0
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/apps/webui/models/functions.py
View File

@ -146,7 +146,7 @@ class FunctionsTable:
def get_function_valves_by_id(self, id: str) -> Optional[dict]: def get_function_valves_by_id(self, id: str) -> Optional[dict]:
try: try:
function = Function.get(Function.id == id) function = Function.get(Function.id == id)
return function.valves if "valves" in function and function.valves else {} return function.valves if function.valves else {}
except Exception as e: except Exception as e:
print(f"An error occurred: {e}") print(f"An error occurred: {e}")
return None return None

2
backend/apps/webui/models/tools.py
View File

@ -117,7 +117,7 @@ class ToolsTable:
def get_tool_valves_by_id(self, id: str) -> Optional[dict]: def get_tool_valves_by_id(self, id: str) -> Optional[dict]:
try: try:
tool = Tool.get(Tool.id == id) tool = Tool.get(Tool.id == id)
return tool.valves if "valves" in tool and tool.valves else {} return tool.valves if tool.valves else {}
except Exception as e: except Exception as e:
print(f"An error occurred: {e}") print(f"An error occurred: {e}")
return None return None