From 7d33ceeae832897a608ff71daae60a614bdb66dc Mon Sep 17 00:00:00 2001 From: Sunny Nayak Date: Wed, 7 Aug 2024 00:38:13 -0700 Subject: [PATCH] Fixed edge condition in orchestrator schedule method (#403) Signed-off-by: Sanket Nayak Co-authored-by: Sanket Nayak Co-authored-by: lvliang-intel Signed-off-by: sharanshirodkar7 --- comps/cores/mega/orchestrator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comps/cores/mega/orchestrator.py b/comps/cores/mega/orchestrator.py index fb769edb04..d4f3ac9b74 100644 --- a/comps/cores/mega/orchestrator.py +++ b/comps/cores/mega/orchestrator.py @@ -47,7 +47,7 @@ async def schedule(self, initial_inputs: Dict, llm_parameters: LLMParams = LLMPa timeout = aiohttp.ClientTimeout(total=1000) async with aiohttp.ClientSession(trust_env=True, timeout=timeout) as session: pending = { - asyncio.create_task(self.execute(session, node, initial_inputs, runtime_graph)) + asyncio.create_task(self.execute(session, node, initial_inputs, runtime_graph, llm_parameters)) for node in self.ind_nodes() } ind_nodes = self.ind_nodes()