From d7f8a459169f912db9ecb37571cc92c411370aec Mon Sep 17 00:00:00 2001
From: AI Christianson <ai.christianson@christianson.ai>
Date: Thu, 19 Dec 2024 15:17:54 -0500
Subject: [PATCH] Fix duplicate task completed panel.

---
 ra_aid/__main__.py     | 15 ++++-----------
 ra_aid/tools/memory.py |  1 -
 2 files changed, 4 insertions(+), 12 deletions(-)

diff --git a/ra_aid/__main__.py b/ra_aid/__main__.py
index 1f37ca0..d7b911a 100644
--- a/ra_aid/__main__.py
+++ b/ra_aid/__main__.py
@@ -2,6 +2,7 @@ import argparse
 import sys
 from typing import Optional
 from rich.panel import Panel
+from rich.markdown import Markdown
 from rich.console import Console
 from langchain_core.messages import HumanMessage
 from langgraph.checkpoint.memory import MemorySaver
@@ -149,7 +150,7 @@ def run_agent_with_retry(agent, prompt: str, config: dict) -> Optional[str]:
                 if _global_memory.get('task_completed'):
                     completion_msg = _global_memory.get('completion_message', 'Task was completed successfully.')
                     console.print(Panel(
-                        completion_msg,
+                        Markdown(completion_msg),
                         title="✅ Task Completed",
                         style="green"
                     ))
@@ -306,16 +307,8 @@ def main():
         research_only_note='' if args.research_only else ' Only request implementation if the user explicitly asked for changes to be made.'
     )
 
-    # Run research agent and check for one-shot completion
-    output = run_agent_with_retry(research_agent, research_prompt, config)
-    if isinstance(output, str) and "one_shot_completed" in str(output):
-        print_stage_header("Task Completed")
-        console.print(Panel(
-            "[green]Task was completed successfully as a one-shot operation.[/green]",
-            title="Task Completed",
-            style="green"
-        ))
-        sys.exit(0)
+    # Run research agent
+    run_agent_with_retry(research_agent, research_prompt, config)
     
     # Run any research subtasks
     run_research_subtasks(base_task, config, model, expert_enabled=expert_enabled)
diff --git a/ra_aid/tools/memory.py b/ra_aid/tools/memory.py
index e4ad11a..81fea2d 100644
--- a/ra_aid/tools/memory.py
+++ b/ra_aid/tools/memory.py
@@ -317,7 +317,6 @@ def one_shot_completed(message: str) -> str:
         
     _global_memory['task_completed'] = True
     _global_memory['completion_message'] = message
-    console.print(Panel(Markdown(message), title="✨ Task Completed", border_style="green"))
     return message
 
 def get_related_files() -> Set[str]: