diff --git a/ra_aid/agents/implementation_agent.py b/ra_aid/agents/implementation_agent.py index 352e241..447f565 100644 --- a/ra_aid/agents/implementation_agent.py +++ b/ra_aid/agents/implementation_agent.py @@ -202,6 +202,7 @@ def run_task_implementation_agent( related_files="\n".join(related_files), env_inv=env_inv, tool_metadata=formatted_tool_metadata, + project_info=formatted_project_info, ) # Show the reasoning assist query in a panel diff --git a/ra_aid/agents/planning_agent.py b/ra_aid/agents/planning_agent.py index e6dd8e8..42355b6 100644 --- a/ra_aid/agents/planning_agent.py +++ b/ra_aid/agents/planning_agent.py @@ -197,6 +197,7 @@ def run_planning_agent( related_files=related_files, env_inv=env_inv, tool_metadata=formatted_tool_metadata, + project_info=formatted_project_info, ) # Show the reasoning assist query in a panel diff --git a/ra_aid/agents/research_agent.py b/ra_aid/agents/research_agent.py index 381cb9e..4acde08 100644 --- a/ra_aid/agents/research_agent.py +++ b/ra_aid/agents/research_agent.py @@ -216,6 +216,7 @@ def run_research_agent( related_files=related_files, env_inv=get_env_inv(), tool_metadata=formatted_tool_metadata, + project_info=formatted_project_info, ) # Show the reasoning assist query in a panel @@ -511,8 +512,8 @@ def run_web_research_agent( console.print(Panel(Markdown(console_message), title="🔬 Researching...")) logger.debug("Web research agent completed successfully") - none_or_fallback_handler = init_fallback_handler(agent, tools) - _result = run_agent_with_retry(agent, prompt, none_or_fallback_handler) + none_or_fallback_handler = agent_utils.init_fallback_handler(agent, tools) + _result = agent_utils.run_agent_with_retry(agent, prompt, none_or_fallback_handler) if _result: # Log web research completion log_work_event(f"Completed web research phase for: {query}") diff --git a/ra_aid/prompts/reasoning_assist_prompt.py b/ra_aid/prompts/reasoning_assist_prompt.py index dbde1e3..ab50738 100644 --- a/ra_aid/prompts/reasoning_assist_prompt.py +++ b/ra_aid/prompts/reasoning_assist_prompt.py @@ -23,6 +23,10 @@ Working Directory: {working_directory} {related_files} + +{project_info} + + {env_inv} @@ -85,6 +89,10 @@ Working Directory: {working_directory} {related_files} + +{project_info} + + {env_inv} @@ -147,6 +155,10 @@ Working Directory: {working_directory} {related_files} + +{project_info} + + {env_inv} @@ -166,6 +178,8 @@ Given the available information, tools, and base task or query, write a couple p The agent is so dumb it needs you to explicitly say how to use the parameters to the tools as well. +ONLY FOR NEW PROJECTS: If this is a new project, most of the focus needs to be on asking the expert, reading/research available library files, emitting key snippets/facts, and most importantly research notes to lay out that we have a new project and what we are building. DO NOT INSTRUCT THE AGENT TO LIST PROJECT DIRECTORIES/READ FILES IF WE ALREADY KNOW THERE ARE NO PROJECT FILES. + Answer quickly and confidently with five sentences at most. DO NOT WRITE CODE