File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 8484_thread_local_data = threading .local ()
8585
8686MAX_WORKERS = 100
87- AGENT_MAX_WORKERS = 10
87+ AGENT_MAX_WORKERS = 20
8888CONTENT = _evals_constant .CONTENT
8989PARTS = _evals_constant .PARTS
9090USER_AUTHOR = _evals_constant .USER_AUTHOR
@@ -2036,7 +2036,7 @@ async def _execute_local_agent_run_with_retry_async(
20362036 )
20372037 if attempt == max_retries - 1 :
20382038 return {"error" : f"Resource exhausted after retries: { e } " }
2039- time .sleep (2 ** attempt )
2039+ await asyncio .sleep (2 ** attempt )
20402040 except Exception as e : # pylint: disable=broad-exception-caught
20412041 logger .error (
20422042 "Unexpected error during generate_content on attempt %d/%d: %s" ,
@@ -2047,7 +2047,7 @@ async def _execute_local_agent_run_with_retry_async(
20472047
20482048 if attempt == max_retries - 1 :
20492049 return {"error" : f"Failed after retries: { e } " }
2050- time .sleep (1 )
2050+ await asyncio .sleep (1 )
20512051 return {"error" : f"Failed to get agent run results after { max_retries } retries" }
20522052
20532053
You can’t perform that action at this time.
0 commit comments