Skip to content

Commit 876ca50

Browse files
Enlarge megaservice request timeout for supporting high concurrency (#279)
Signed-off-by: lvliang-intel <[email protected]>
1 parent 2458e2f commit 876ca50

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

comps/cores/mega/orchestrator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,8 @@ def flow_to(self, from_service, to_service):
4040
async def schedule(self, initial_inputs: Dict, llm_parameters: LLMParams = LLMParams()):
4141
result_dict = {}
4242

43-
async with aiohttp.ClientSession(trust_env=True) as session:
43+
timeout = aiohttp.ClientTimeout(total=1000)
44+
async with aiohttp.ClientSession(trust_env=True, timeout=timeout) as session:
4445
pending = {asyncio.create_task(self.execute(session, node, initial_inputs)) for node in self.ind_nodes()}
4546

4647
while pending:

0 commit comments

Comments
 (0)