feat(run): add retry mechanism with error handling to agent entry point

Implement retry logic in the run() function with max_retries parameter (default 3).
On failure, append error context to help the agent retry with a simpler approach.
Add recursion_limit config of 30 to prevent infinite loops. Also update
answer extractor output to wrap answer in quotes for better visibility.

agent/agent.py

@@ -29,17 +29,50 @@ def supervisor_agent():
     )
 
 
-def run(query: str) -> str:
+def run(query: str, max_retries: int = 3) -> str:
     """Entry point: let the supervisor agent finish the work."""
-    print(f"{Fore.CYAN}[Supervisor] Processing query...{Style.RESET_ALL}")
-    agent = supervisor_agent()
-    result = agent.invoke({"messages": [HumanMessage(content=query)]})
-    content = result["messages"][-1].content
-    if isinstance(content, list):
-        content = content[0].get("text", "")
-    else:
-        content = str(content)
-    return extract_answer(content, query)
+    last_error: str | None = None
+
+    for attempt in range(1, max_retries + 1):
+        print(
+            f"{Fore.CYAN}[Supervisor] Processing query (attempt {attempt}/{max_retries})...{Style.RESET_ALL}"
+        )
+        agent = supervisor_agent()
+
+        messages = [HumanMessage(content=query)]
+        if last_error:
+            messages.append(
+                HumanMessage(
+                    content=(
+                        f"Previous attempt failed with error: {last_error}\n"
+                        f"Try thinking about the problem more simply. "
+                        f"Use fewer steps and a more straightforward approach."
+                    )
+                )
+            )
+
+        try:
+            result = agent.invoke(
+                {"messages": messages},
+                # Max agent action is 30 turn.
+                config={"recursion_limit": 30},
+            )
+            content = result["messages"][-1].content
+            if isinstance(content, list):
+                content = content[0].get("text", "")
+            else:
+                content = str(content)
+            return extract_answer(content, query)
+        except Exception as e:
+            last_error = str(e)
+            print(
+                f"{Fore.RED}[Supervisor] Attempt {attempt} failed: {last_error}{Style.RESET_ALL}"
+            )
+
+    print(f"{Fore.RED}[Supervisor] All {max_retries} attempts failed.{Style.RESET_ALL}")
+    return extract_answer(
+        f"Agent failed after {max_retries} attempts. Last error: {last_error}", query
+    )
 
 
 if __name__ == "__main__":

agent/agents/answer_extractor.py

@@ -80,5 +80,5 @@ Output:
     print(
         f"{Fore.CYAN}[AnswerExtractor] Process: {extracted.extract_process_description}{Style.RESET_ALL}"
     )
-    print(f"{Fore.CYAN}[AnswerExtractor] Answer: {extracted.answer}{Style.RESET_ALL}")
+    print(f"{Fore.CYAN}[AnswerExtractor] Answer: '{extracted.answer}'{Style.RESET_ALL}")
     return extracted.answer