From dfc3d03082ed5806640c3260a025542828124716 Mon Sep 17 00:00:00 2001
From: PeteFedora <tristan.peter.schneider@gmail.com>
Date: Mon, 8 Jun 2026 19:58:39 +0200
Subject: [PATCH] fix: add timeout to ChatOpen response to prevent infinite
 hangs

---
 treesearch/llm/query.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/treesearch/llm/query.py b/treesearch/llm/query.py
index 035a9e2..771ae0f 100644
--- a/treesearch/llm/query.py
+++ b/treesearch/llm/query.py
@@ -93,8 +93,14 @@ async def run(
         input = prompt_to_md(input)
         tools = await self._get_all_tools()
 
+        from config import get_config
+        _cfg = get_config()
         model = ChatOpenAI(
-            model=self._model, temperature=self._temperature, use_responses_api=True
+            model=self._model,
+            temperature=self._temperature,
+            use_responses_api=True,
+            timeout=_cfg.agent.code.request_timeout,
+            max_retries=_cfg.agent.code.max_retries,
         )
 
         agent = Agent(