Skip to content

Commit fd88871

Browse files
committed
feat: Increase timeout for HTTP client in run.py
1 parent bb1aa53 commit fd88871

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

scripts/run.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ async def single_turn_experiment(row, agent_url: str, workflow_name: str) -> dic
115115
span.set_attribute("workflow.name", workflow_name)
116116

117117
try:
118-
async with httpx.AsyncClient():
118+
async with httpx.AsyncClient(timeout=httpx.Timeout(300)):
119119
client = await initialize_client(agent_url)
120120

121121
# Get the input from the row
@@ -208,7 +208,7 @@ async def multi_turn_experiment(row, agent_url: str, workflow_name: str) -> dict
208208
# Validate input format
209209
user_input = validate_multi_turn_input(row.get("user_input"))
210210

211-
async with httpx.AsyncClient():
211+
async with httpx.AsyncClient(timeout=httpx.Timeout(300)):
212212
client = await initialize_client(agent_url)
213213

214214
# Extract only human messages (agent messages are from dataset, not sent)

0 commit comments

Comments
 (0)