UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-18 00:30:51 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #5155 from open-webui/dev

Browse Source 
fix: tools & functions import
This commit is contained in:
Timothy Jaeryang Baek 2024-09-04 19:00:52 +02:00 committed by GitHub
commit a988c53949
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 26 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
backend/open_webui/apps/webui/utils.py
View File

@ -54,9 +54,20 @@ def load_toolkit_module_by_id(toolkit_id):
if not os.path.exists(toolkit_path): if not os.path.exists(toolkit_path):
tool = Tools.get_tool_by_id(toolkit_id) tool = Tools.get_tool_by_id(toolkit_id)
if tool:
if tool and not os.path.exists(f"{toolkit_path}.error"):
with open(toolkit_path, "w") as file: with open(toolkit_path, "w") as file:
file.write(tool.content) content = tool.content
content = content.replace("from utils", "from open_webui.utils")
content = content.replace("from apps", "from open_webui.apps")
content = content.replace("from main", "from open_webui.main")
content = content.replace("from config", "from open_webui.config")
if tool.content != content:
print(f"Replaced imports for: {toolkit_id}")
Tools.update_tool_by_id(toolkit_id, {"content": content})
file.write(content)
else: else:
raise Exception(f"Toolkit not found: {toolkit_id}") raise Exception(f"Toolkit not found: {toolkit_id}")
@ -84,9 +95,20 @@ def load_function_module_by_id(function_id):
if not os.path.exists(function_path): if not os.path.exists(function_path):
function = Functions.get_function_by_id(function_id) function = Functions.get_function_by_id(function_id)
if function:
if function and not os.path.exists(f"{function_path}.error"):
with open(function_path, "w") as file: with open(function_path, "w") as file:
file.write(function.content) content = function.content
content = content.replace("from utils", "from open_webui.utils")
content = content.replace("from apps", "from open_webui.apps")
content = content.replace("from main", "from open_webui.main")
content = content.replace("from config", "from open_webui.config")
if function.content != content:
print(f"Replaced imports for: {function_id}")
Functions.update_function_by_id(function_id, {"content": content})
file.write(content)
else: else:
raise Exception(f"Function not found: {function_id}") raise Exception(f"Function not found: {function_id}")