Improve task completed panel.

This commit is contained in:
AI Christianson 2024-12-19 15:08:11 -05:00
parent 5029106bdd
commit 0c6b4f98b3
2 changed files with 2 additions and 1 deletions

View File

@ -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"
))

View File

@ -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]: