From bee8526425a7a6ad43ff11e721afcce22ff4fe72 Mon Sep 17 00:00:00 2001 From: Damien Berezenko Date: Sun, 1 Mar 2026 00:37:38 +0000 Subject: [PATCH] fix(llm): add retries to RAG orchestrator to handle output validation issues --- codebase_rag/services/llm.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codebase_rag/services/llm.py b/codebase_rag/services/llm.py index 204b40b2b..74bbebf92 100644 --- a/codebase_rag/services/llm.py +++ b/codebase_rag/services/llm.py @@ -111,6 +111,7 @@ def create_rag_orchestrator(tools: list[Tool]) -> Agent: model=llm, system_prompt=RAG_ORCHESTRATOR_SYSTEM_PROMPT, tools=tools, + retries=3, # Increase retries to handle output validation issues ) except Exception as e: raise LLMGenerationError(f"Failed to initialize RAG Orchestrator: {e}") from e