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

key snippets

This commit is contained in:
AI Christianson 2025-03-04 17:49:17 -05:00
parent 47cb6b65ef
commit d907a0ea9c
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ra_aid/agent_utils.py
View File

@ -396,7 +396,7 @@ def run_research_agent(
except RuntimeError as e: except RuntimeError as e:
logger.error(f"Failed to access key fact repository: {str(e)}") logger.error(f"Failed to access key fact repository: {str(e)}")
key_facts = "" key_facts = ""
code_snippets = _global_memory.get("code_snippets", "") key_snippets = format_key_snippets_dict(get_key_snippet_repository().get_snippets_dict())
related_files = get_related_files() related_files = get_related_files()
current_date = datetime.now().strftime("%Y-%m-%d %H:%M:%S") current_date = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@ -435,7 +435,7 @@ def run_research_agent(
web_research_section=web_research_section, web_research_section=web_research_section,
key_facts=key_facts, key_facts=key_facts,
work_log=get_memory_value("work_log"), work_log=get_memory_value("work_log"),
code_snippets=code_snippets, key_snippets=key_snippets,
related_files=related_files, related_files=related_files,
project_info=formatted_project_info, project_info=formatted_project_info,
new_project_hints=NEW_PROJECT_HINTS if project_info.is_new else "", new_project_hints=NEW_PROJECT_HINTS if project_info.is_new else "",
@ -551,7 +551,11 @@ def run_web_research_agent(
except RuntimeError as e: except RuntimeError as e:
logger.error(f"Failed to access key fact repository: {str(e)}") logger.error(f"Failed to access key fact repository: {str(e)}")
key_facts = "" key_facts = ""
code_snippets = _global_memory.get("code_snippets", "") try:
key_snippets = format_key_snippets_dict(get_key_snippet_repository().get_snippets_dict())
except RuntimeError as e:
logger.error(f"Failed to access key snippet repository: {str(e)}")
key_snippets = ""
related_files = get_related_files() related_files = get_related_files()
current_date = datetime.now().strftime("%Y-%m-%d %H:%M:%S") current_date = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@ -565,7 +569,7 @@ def run_web_research_agent(
human_section=human_section, human_section=human_section,
key_facts=key_facts, key_facts=key_facts,
work_log=get_memory_value("work_log"), work_log=get_memory_value("work_log"),
code_snippets=code_snippets, key_snippets=key_snippets,
related_files=related_files, related_files=related_files,
) )

2
ra_aid/prompts/research_prompts.py
View File

@ -27,7 +27,7 @@ Context from Previous Research (if available):
</key facts> </key facts>
<relevant code snippets> <relevant code snippets>
{code_snippets} {key_snippets}
</relevant code snippets> </relevant code snippets>
<related files> <related files>