UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-12-28 06:42:47 +00:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-12-15 22:44:47 -08:00
parent ff8e94e41f
commit ddfed87b15
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/open_webui/functions.py
View File

@ -65,12 +65,12 @@ def get_function_module_by_id(request: Request, pipe_id: str):
return function_module return function_module
async def get_function_models(): async def get_function_models(request):
pipes = Functions.get_functions_by_type("pipe", active_only=True) pipes = Functions.get_functions_by_type("pipe", active_only=True)
pipe_models = [] pipe_models = []
for pipe in pipes: for pipe in pipes:
function_module = get_function_module_by_id(pipe.id) function_module = get_function_module_by_id(request, pipe.id)
# Check if function is a manifold # Check if function is a manifold
if hasattr(function_module, "pipes"): if hasattr(function_module, "pipes"):
@ -253,7 +253,7 @@ async def generate_function_chat_completion(
form_data = apply_model_system_prompt_to_body(params, form_data, user) form_data = apply_model_system_prompt_to_body(params, form_data, user)
pipe_id = get_pipe_id(form_data) pipe_id = get_pipe_id(form_data)
function_module = get_function_module_by_id(pipe_id) function_module = get_function_module_by_id(request, pipe_id)
pipe = function_module.pipe pipe = function_module.pipe
params = get_function_params(function_module, form_data, user, extra_params) params = get_function_params(function_module, form_data, user, extra_params)

2
backend/open_webui/utils/models.py
View File

@ -52,7 +52,7 @@ async def get_all_base_models(request: Request):
for model in ollama_models["models"] for model in ollama_models["models"]
] ]
function_models = await get_function_models() function_models = await get_function_models(request)
models = function_models + openai_models + ollama_models models = function_models + openai_models + ollama_models
return models return models