Skip to content

Commit

Permalink
Enlarge megaservice request timeout for supporting high concurrency (#…
Browse files Browse the repository at this point in the history
…279)

Signed-off-by: lvliang-intel <[email protected]>
  • Loading branch information
lvliang-intel authored Jul 17, 2024
1 parent 2458e2f commit 876ca50
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion comps/cores/mega/orchestrator.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,8 @@ def flow_to(self, from_service, to_service):
async def schedule(self, initial_inputs: Dict, llm_parameters: LLMParams = LLMParams()):
result_dict = {}

async with aiohttp.ClientSession(trust_env=True) as session:
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)) for node in self.ind_nodes()}

while pending:
Expand Down

0 comments on commit 876ca50

Please sign in to comment.