Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 03:47:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek
2024-06-20 00:54:58 -07:00
parent 27f8afebab
commit 43e08c6afa
6 changed files with 308 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/apps/webui/utils.py
View File

@@ -33,9 +33,9 @@ def load_function_module_by_id(function_id):
spec.loader.exec_module(module)
print(f"Loaded module: {module.__name__}")
if hasattr(module, "Pipe"):
return module.Pipe()
return module.Pipe(), "pipe"
elif hasattr(module, "Filter"):
return module.Filter()
return module.Filter(), "filter"
else:
raise Exception("No Function class found")
except Exception as e: