diff --git a/environments/SWE-INFINITE/agents/miniswe.py b/environments/SWE-INFINITE/agents/miniswe.py index 7a4972ff..d16135c3 100644 --- a/environments/SWE-INFINITE/agents/miniswe.py +++ b/environments/SWE-INFINITE/agents/miniswe.py @@ -162,6 +162,7 @@ def parse_action(self, response: dict) -> dict: model_name = f"openai/{model_name}" model_kwargs = {"temperature": self.config.temperature} + model_kwargs["timeout"] = min(self.config.timeout // 2, 300) if self.config.seed is not None: model_kwargs["seed"] = self.config.seed @@ -229,7 +230,10 @@ def parse_action(self, response: dict) -> dict: try: loop = asyncio.get_event_loop() - _, result = await loop.run_in_executor(None, self._agent.run, prompt) + _, result = await asyncio.wait_for( + loop.run_in_executor(None, self._agent.run, prompt), + timeout=self.config.timeout, + ) patch = result except Exception: import traceback