From fb4b8d27e36769140b1ca9bfab0d45d37922309e Mon Sep 17 00:00:00 2001 From: lvliang-intel Date: Mon, 9 Sep 2024 07:34:44 +0800 Subject: [PATCH] Fix LVM streaming issue (#637) * Fix LVM streaming issue Signed-off-by: lvliang-intel * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Signed-off-by: lvliang-intel Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> --- comps/cores/mega/orchestrator.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/comps/cores/mega/orchestrator.py b/comps/cores/mega/orchestrator.py index 2410140b8..94ff5c330 100644 --- a/comps/cores/mega/orchestrator.py +++ b/comps/cores/mega/orchestrator.py @@ -136,7 +136,10 @@ async def execute( # send the cur_node request/reply endpoint = self.services[cur_node].endpoint_path llm_parameters_dict = llm_parameters.dict() - if self.services[cur_node].service_type == ServiceType.LLM: + if ( + self.services[cur_node].service_type == ServiceType.LLM + or self.services[cur_node].service_type == ServiceType.LVM + ): for field, value in llm_parameters_dict.items(): if inputs.get(field) != value: inputs[field] = value