OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-25 17:45:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-27 00:18:48 -07:00
parent 8c3f262ffd
commit 23ce9d396b
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main.py
View File

@ -156,10 +156,13 @@ async def generate_openai_chat_completion(form_data: OpenAIChatCompletionForm):
pipeline = app.state.PIPELINES[form_data.model]
pipeline_id = form_data.model
if pipeline.get("manifold", False):
pipeline_id = pipeline_id.split(".")[1]
print(pipeline_id)
get_response = pipeline["module"].get_response
if pipeline.get("manifold", False):
manifold_id, pipeline_id = pipeline_id.split(".", 1)
get_response = PIPELINE_MODULES[manifold_id].get_response
else:
get_response = PIPELINE_MODULES[pipeline_id].get_response
if form_data.stream: