Fixed edge condition in orchestrator schedule method (#403)

Signed-off-by: Sanket Nayak <[email protected]> Co-authored-by: Sanket Nayak <[email protected]> Co-authored-by: lvliang-intel <[email protected]>
opea-project · Aug 7, 2024 · 76877c1 · 76877c1
1 parent 7185d6b
commit 76877c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()