OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-23 05:24:09 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: manifold

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-28 15:10:13 -07:00
parent 5cb9dfa972
commit 78e9bcf34c
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main.py
View File

@ -274,9 +274,10 @@ async def generate_openai_chat_completion(form_data: OpenAIChatCompletionForm):
user_message = get_last_user_message(form_data.messages)
messages = [message.model_dump() for message in form_data.messages]
if form_data.model not in app.state.PIPELINES or app.state.PIPELINES[
form_data.model
].get("filter", False):
if (
form_data.model not in app.state.PIPELINES
or app.state.PIPELINES[form_data.model]["type"] == "filter"
):
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail=f"Pipeline {form_data.model} not found",
@ -290,7 +291,7 @@ async def generate_openai_chat_completion(form_data: OpenAIChatCompletionForm):
print(pipeline_id)
if pipeline.get("manifold", False):
if pipeline["type"] == "manifold":
manifold_id, pipeline_id = pipeline_id.split(".", 1)
pipe = PIPELINE_MODULES[manifold_id].pipe
else: