diff --git a/benchmarks/commit0/run_infer.py b/benchmarks/commit0/run_infer.py index 156564fe..c1067b1c 100644 --- a/benchmarks/commit0/run_infer.py +++ b/benchmarks/commit0/run_infer.py @@ -291,7 +291,8 @@ def _log_event(ev): metadata=self.metadata, ) conversation.send_message(instruction) - conversation.run() + run_timeout = int(os.getenv("CONVERSATION_TIMEOUT", "3600")) + conversation.run(timeout=run_timeout) history = list(conversation.state.events)