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(