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

Update list of files for programmer tool.

This commit is contained in:
user 2024-12-24 09:08:47 -05:00
parent 1bcf3b4544
commit eb5cc50291
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ra_aid/tools/programmer.py
View File

@ -1,5 +1,6 @@
import os import os
from typing import List, Optional, Dict, Union from typing import List, Optional, Dict, Union
from ra_aid.tools.memory import _global_memory
from langchain_core.tools import tool from langchain_core.tools import tool
from rich.console import Console from rich.console import Console
from rich.panel import Panel from rich.panel import Panel
@ -36,6 +37,9 @@ Args: instructions: Programming task instructions files: Optional; if not provid
Returns: { "output": stdout+stderr, "return_code": 0 if success, "success": True/False } Returns: { "output": stdout+stderr, "return_code": 0 if success, "success": True/False }
""" """
# Get related files if no specific files provided
file_paths = list(_global_memory['related_files'].values()) if 'related_files' in _global_memory else []
# Build command # Build command
command = [ command = [
"aider", "aider",
@ -58,8 +62,10 @@ Returns: { "output": stdout+stderr, "return_code": 0 if success, "success": True
command.append(input.instructions) command.append(input.instructions)
if input.files: # Add files to command
command.extend(input.files) files_to_use = file_paths + (input.files or [])
if files_to_use:
command.extend(files_to_use)
# Create a pretty display of what we're doing # Create a pretty display of what we're doing
task_display = [ task_display = [
@ -67,10 +73,10 @@ Returns: { "output": stdout+stderr, "return_code": 0 if success, "success": True
f"{input.instructions}\n" f"{input.instructions}\n"
] ]
if input.files: if files_to_use:
task_display.extend([ task_display.extend([
"\n## Files\n", "\n## Files\n",
*[f"- `{file}`\n" for file in input.files] *[f"- `{file}`\n" for file in files_to_use]
]) ])
markdown_content = "".join(task_display) markdown_content = "".join(task_display)