Fixed edge condition in orchestrator schedule method (opea-project#403)

Signed-off-by: Sanket Nayak <[email protected]> Co-authored-by: Sanket Nayak <[email protected]> Co-authored-by: lvliang-intel <[email protected]> Signed-off-by: sharanshirodkar7 <[email protected]>
predictionguard · Aug 7, 2024 · 7d33cee · 7d33cee
1 parent ada09b6
commit 7d33cee
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()