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

Merge pull request #10566 from oloflarsson/feat/async-pipes

Browse Source 
feat: add optional support for async pipes
This commit is contained in:
Timothy Jaeryang Baek
2025-02-22 02:47:08 -08:00
committed by GitHub
parent 8f3fcdcf5c ab1f2ae914
commit 32fa8d5143
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/open_webui/functions.py
View File

@@ -2,6 +2,7 @@ import logging
import sys
import inspect
import json
import asyncio
from pydantic import BaseModel
from typing import AsyncGenerator, Generator, Iterator
@@ -76,11 +77,13 @@ async def get_function_models(request):
if hasattr(function_module, "pipes"):
sub_pipes = []
# Check if pipes is a function or a list
# Handle pipes being a list, sync function, or async function
try:
if callable(function_module.pipes):
sub_pipes = function_module.pipes()
if asyncio.iscoroutinefunction(function_module.pipes):
sub_pipes = await function_module.pipes()
else:
sub_pipes = function_module.pipes()
else:
sub_pipes = function_module.pipes
except Exception as e: