zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

store trajectory for tasks

This commit is contained in:
AI Christianson 2025-03-11 08:41:59 -04:00
parent 7038a61693
commit d0985b6b84
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ra_aid/console/formatting.py
View File

@ -61,6 +61,20 @@ def print_task_header(task: str) -> None:
""" """
console.print(Panel(Markdown(task), title="🔧 Task", border_style="yellow bold")) 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: def print_error(message: str) -> None:
"""Print an error message in a red-bordered panel with warning emoji. """Print an error message in a red-bordered panel with warning emoji.