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