This commit is contained in:
Paul Gauthier 2023-05-18 18:17:02 -07:00
parent 7fc02ccd66
commit 78af36ffda
4 changed files with 6 additions and 4 deletions

View file

@ -156,18 +156,18 @@ class Commands:
def cmd_add(self, args): def cmd_add(self, args):
"Add matching files to the chat session" "Add matching files to the chat session"
from aider.dump import dump
added_fnames = [] added_fnames = []
files = self.coder.get_all_relative_files() files = self.coder.get_all_relative_files()
for word in args.split(): for word in args.split():
matched_files = [file for file in files if word in file] matched_files = [file for file in files if word in file]
dump(word, matched_files)
if not matched_files: if not matched_files:
if self.coder.repo is not None: if self.coder.repo is not None:
create_file = Confirm.ask( create_file = Confirm.ask(
f"No files matched '{word}'. Do you want to create the file and add it to git?", (
f"No files matched '{word}'. Do you want to create the file and add it"
" to git?"
),
) )
else: else:
create_file = Confirm.ask( create_file = Confirm.ask(

View file

View file

View file

@ -10,6 +10,8 @@ from aider.coder import Coder
class TestCommands(unittest.TestCase): class TestCommands(unittest.TestCase):
def test_cmd_add(self): def test_cmd_add(self):
with tempfile.TemporaryDirectory() as tmpdir: with tempfile.TemporaryDirectory() as tmpdir:
os.chdir(tmpdir)
io = IO(pretty=False, yes=True) io = IO(pretty=False, yes=True)
coder = Coder( coder = Coder(
main_model="gpt-3.5-turbo", main_model="gpt-3.5-turbo",