From 0c6b4f98b329a629fe6655e0161117cbcc777b4c Mon Sep 17 00:00:00 2001 From: AI Christianson Date: Thu, 19 Dec 2024 15:08:11 -0500 Subject: [PATCH] Improve task completed panel. --- ra_aid/__main__.py | 2 +- ra_aid/tools/memory.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/ra_aid/__main__.py b/ra_aid/__main__.py index 0787b71..1f37ca0 100644 --- a/ra_aid/__main__.py +++ b/ra_aid/__main__.py @@ -149,7 +149,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( - f"[green]{completion_msg}[/green]", + completion_msg, title="✅ Task Completed", style="green" )) diff --git a/ra_aid/tools/memory.py b/ra_aid/tools/memory.py index 81fea2d..e4ad11a 100644 --- a/ra_aid/tools/memory.py +++ b/ra_aid/tools/memory.py @@ -317,6 +317,7 @@ 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]: