From b28e5b30b4c873b6431b85e1e382e17d5148726d Mon Sep 17 00:00:00 2001 From: user Date: Sun, 22 Dec 2024 15:29:23 -0500 Subject: [PATCH] Do not show file ID in user-facing panel. --- ra_aid/tools/agent.py | 12 ++++++------ ra_aid/tools/memory.py | 2 +- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/ra_aid/tools/agent.py b/ra_aid/tools/agent.py index c20df2c..c41eca5 100644 --- a/ra_aid/tools/agent.py +++ b/ra_aid/tools/agent.py @@ -39,7 +39,7 @@ def request_research(query: str) -> ResearchResult: return { "completion_message": "Research stopped - maximum recursion depth reached", "key_facts": get_memory_value("key_facts"), - "related_files": list(get_related_files()), + "related_files": get_related_files(), "research_notes": get_memory_value("research_notes"), "key_snippets": get_memory_value("key_snippets"), "success": False, @@ -85,7 +85,7 @@ def request_research(query: str) -> ResearchResult: return { "completion_message": completion_message, "key_facts": get_memory_value("key_facts"), - "related_files": list(get_related_files()), + "related_files": get_related_files(), "research_notes": get_memory_value("research_notes"), "key_snippets": get_memory_value("key_snippets"), "success": success, @@ -133,11 +133,11 @@ def request_research_and_implementation(query: str) -> Dict[str, Any]: _global_memory['completion_message'] = '' _global_memory['task_completed'] = False _global_memory['plan_completed'] = False - + return { "completion_message": completion_message, "key_facts": get_memory_value("key_facts"), - "related_files": list(get_related_files()), + "related_files": get_related_files(), "research_notes": get_memory_value("research_notes"), "key_snippets": get_memory_value("key_snippets"), "success": success, @@ -194,7 +194,7 @@ def request_task_implementation(task_spec: str) -> Dict[str, Any]: return { "key_facts": get_memory_value("key_facts"), - "related_files": list(get_related_files()), + "related_files": get_related_files(), "key_snippets": get_memory_value("key_snippets"), "completion_message": completion_message, "success": success, @@ -245,7 +245,7 @@ def request_implementation(task_spec: str) -> Dict[str, Any]: return { "completion_message": completion_message, "key_facts": get_memory_value("key_facts"), - "related_files": list(get_related_files()), + "related_files": get_related_files(), "key_snippets": get_memory_value("key_snippets"), "success": success, "reason": reason diff --git a/ra_aid/tools/memory.py b/ra_aid/tools/memory.py index 8a6e7f0..e0f6572 100644 --- a/ra_aid/tools/memory.py +++ b/ra_aid/tools/memory.py @@ -375,7 +375,7 @@ def emit_related_files(files: List[str]) -> str: # Rich output - single consolidated panel if added_files: - files_added_md = '\n'.join(f"- ID#{id}: `{file}`" for id, file in added_files) + files_added_md = '\n'.join(f"- `{file}`" for id, file in added_files) md_content = f"**Files Noted:**\n{files_added_md}" console.print(Panel(Markdown(md_content), title="📁 Related Files Noted",