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

refac: pipe function support

Browse Source
This commit is contained in:
Timothy J. Baek
2024-06-20 04:47:40 -07:00
parent d6e4aef607
commit 59fa2f8f26
2 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/apps/webui/main.py
View File

@@ -120,16 +120,16 @@ async def get_pipe_models():
manifold_pipes = []
# Check if pipes is a function or a list
if callable(pipe.pipes):
manifold_pipes = pipe.pipes()
if callable(function_module.pipes):
manifold_pipes = function_module.pipes()
else:
manifold_pipes = pipe.pipes
manifold_pipes = function_module.pipes
for p in manifold_pipes:
manifold_pipe_id = f'{pipe.id}.{p["id"]}'
manifold_pipe_name = p["name"]
if hasattr(pipe, "name"):
if hasattr(function_module, "name"):
manifold_pipe_name = f"{pipe.name}{manifold_pipe_name}"
pipe_models.append(