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

trajectory

This commit is contained in:
AI Christianson 2025-03-11 08:56:12 -04:00
parent d0985b6b84
commit 5d899d3d13
4 changed files with 55 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
ra_aid/__main__.py
View File

@ -621,6 +621,18 @@ def main():
print_stage_header("Chat Mode")
# Record stage transition in trajectory
trajectory_repo = get_trajectory_repository()
human_input_id = get_human_input_repository().get_most_recent_id()
trajectory_repo.create(
step_data={
"stage": "chat_mode",
"display_title": "Chat Mode",
},
record_type="stage_transition",
human_input_id=human_input_id
)
# Get project info
try:
project_info = get_project_info(".", file_limit=2000)
@ -770,6 +782,18 @@ def main():
# Run research stage
print_stage_header("Research Stage")
# Record stage transition in trajectory
trajectory_repo = get_trajectory_repository()
human_input_id = get_human_input_repository().get_most_recent_id()
trajectory_repo.create(
step_data={
"stage": "research_stage",
"display_title": "Research Stage",
},
record_type="stage_transition",
human_input_id=human_input_id
)
# Initialize research model with potential overrides
research_provider = args.research_provider or args.provider
research_model_name = args.research_model or args.model

14
ra_aid/agents/planning_agent.py
View File

@ -24,6 +24,8 @@ from ra_aid.database.repositories.key_snippet_repository import get_key_snippet_
from ra_aid.database.repositories.research_note_repository import get_research_note_repository
from ra_aid.database.repositories.config_repository import get_config_repository
from ra_aid.database.repositories.work_log_repository import get_work_log_repository
from ra_aid.database.repositories.trajectory_repository import get_trajectory_repository
from ra_aid.database.repositories.human_input_repository import get_human_input_repository
from ra_aid.env_inv_context import get_env_inv
from ra_aid.exceptions import AgentInterrupt
from ra_aid.llm import initialize_expert_llm
@ -156,6 +158,18 @@ def run_planning_agent(
# Display the planning stage header before any reasoning assistance
print_stage_header("Planning Stage")
# Record stage transition in trajectory
trajectory_repo = get_trajectory_repository()
human_input_id = get_human_input_repository().get_most_recent_id()
trajectory_repo.create(
step_data={
"stage": "planning_stage",
"display_title": "Planning Stage",
},
record_type="stage_transition",
human_input_id=human_input_id
)
# Initialize expert guidance section
expert_guidance = ""

28
ra_aid/console/formatting.py
View File

@ -38,20 +38,6 @@ def print_stage_header(stage: str) -> None:
panel_content = f" {icon} {stage_title}"
console.print(Panel(panel_content, style="green bold", padding=0))
# Record trajectory event - focus on semantic meaning
trajectory_repo = get_trajectory_repository()
human_input_id = get_human_input_repository().get_most_recent_id()
trajectory_repo.create(
step_data={
"stage": stage_key,
"display_icon": icon,
"display_title": stage_title,
},
record_type="stage_transition",
human_input_id=human_input_id
)
def print_task_header(task: str) -> None:
"""Print a task header with yellow styling and wrench emoji. Content is rendered as Markdown.
@ -61,20 +47,6 @@ def print_task_header(task: str) -> None:
"""
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:
"""Print an error message in a red-bordered panel with warning emoji.

21
ra_aid/tools/agent.py
View File

@ -14,11 +14,12 @@ from ra_aid.agent_context import (
is_crashed,
reset_completion_flags,
)
from ra_aid.console.formatting import print_error
from ra_aid.database.repositories.human_input_repository import HumanInputRepository
from ra_aid.console.formatting import print_error, print_task_header
from ra_aid.database.repositories.human_input_repository import HumanInputRepository, get_human_input_repository
from ra_aid.database.repositories.key_fact_repository import get_key_fact_repository
from ra_aid.database.repositories.key_snippet_repository import get_key_snippet_repository
from ra_aid.database.repositories.config_repository import get_config_repository
from ra_aid.database.repositories.trajectory_repository import get_trajectory_repository
from ra_aid.database.repositories.related_files_repository import get_related_files_repository
from ra_aid.database.repositories.research_note_repository import get_research_note_repository
from ra_aid.exceptions import AgentInterrupt
@ -26,8 +27,7 @@ from ra_aid.model_formatters import format_key_facts_dict
from ra_aid.model_formatters.key_snippets_formatter import format_key_snippets_dict
from ra_aid.model_formatters.research_notes_formatter import format_research_notes_dict
from ..console import print_task_header
from ..llm import initialize_llm
from ra_aid.llm import initialize_llm
from .human import ask_human
from .memory import get_related_files, get_work_log
@ -346,6 +346,19 @@ def request_task_implementation(task_spec: str) -> str:
try:
print_task_header(task_spec)
# Record task display in trajectory
trajectory_repo = get_trajectory_repository()
human_input_id = get_human_input_repository().get_most_recent_id()
trajectory_repo.create(
step_data={
"task": task_spec,
"display_title": "Task",
},
record_type="task_display",
human_input_id=human_input_id
)
# Run implementation agent
from ..agents.implementation_agent import run_task_implementation_agent