From e73486b7a84be2dab76d936fafac80d0eef8906c Mon Sep 17 00:00:00 2001
From: Yiorgis Gozadinos <ggozadinos@gmail.com>
Date: Thu, 15 Jan 2026 17:09:44 +0200
Subject: [PATCH] Set retries to 3 for agents that did not have it.

---
 haiku_rag_slim/haiku/rag/agents/chat/agent.py  | 1 +
 haiku_rag_slim/haiku/rag/agents/chat/search.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/haiku_rag_slim/haiku/rag/agents/chat/agent.py b/haiku_rag_slim/haiku/rag/agents/chat/agent.py
index 636eb08e..4417c5b6 100644
--- a/haiku_rag_slim/haiku/rag/agents/chat/agent.py
+++ b/haiku_rag_slim/haiku/rag/agents/chat/agent.py
@@ -29,6 +29,7 @@ def create_chat_agent(config: AppConfig) -> Agent[ChatDeps, str]:
         deps_type=ChatDeps,
         output_type=str,
         instructions=CHAT_SYSTEM_PROMPT,
+        retries=3,
     )
 
     @agent.tool
diff --git a/haiku_rag_slim/haiku/rag/agents/chat/search.py b/haiku_rag_slim/haiku/rag/agents/chat/search.py
index 5c71f31b..a6168539 100644
--- a/haiku_rag_slim/haiku/rag/agents/chat/search.py
+++ b/haiku_rag_slim/haiku/rag/agents/chat/search.py
@@ -21,6 +21,7 @@ class SearchAgent:
             deps_type=SearchDeps,
             output_type=str,
             instructions=SEARCH_SYSTEM_PROMPT,
+            retries=3,
         )
 
         @self._agent.tool