From 23ce9d396bdd21ecd8ee7e75c11cd27c73d56d1d Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Mon, 27 May 2024 00:18:48 -0700 Subject: [PATCH] fix --- main.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index b309c9a..7796aa2 100644 --- a/main.py +++ b/main.py @@ -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: