From d0985b6b84f6426a026ce204a614dd9cf45df421 Mon Sep 17 00:00:00 2001 From: AI Christianson Date: Tue, 11 Mar 2025 08:41:59 -0400 Subject: [PATCH] store trajectory for tasks --- ra_aid/console/formatting.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/ra_aid/console/formatting.py b/ra_aid/console/formatting.py index 229204e..a10fffe 100644 --- a/ra_aid/console/formatting.py +++ b/ra_aid/console/formatting.py @@ -60,6 +60,20 @@ def print_task_header(task: str) -> None: task: The task text to print (supports Markdown formatting) """ console.print(Panel(Markdown(task), title="🔧 Task", border_style="yellow bold")) + + # Record trajectory event + trajectory_repo = get_trajectory_repository() + human_input_id = get_human_input_repository().get_most_recent_id() + + trajectory_repo.create( + step_data={ + "task": task, + "display_title": "Task", + "display_icon": "🔧", + }, + record_type="task_display", + human_input_id=human_input_id + ) def print_error(message: str) -> None: