Skip to content

Commit 0b06e56

Browse files
committed
refactor: Simplify file mention handling logic in ContextCoder
1 parent fb5a32f commit 0b06e56

1 file changed

Lines changed: 16 additions & 7 deletions

File tree

aider/coders/context_coder.py

Lines changed: 16 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,20 +15,29 @@ def reply_completed(self):
1515
if not content or not content.strip():
1616
return True
1717

18-
dump(repr(content))
18+
#dump(repr(content))
1919
current_rel_fnames = set(self.get_inchat_relative_files())
2020
mentioned_rel_fnames = set(self.get_file_mentions(content, ignore_current=True))
2121

2222
dump(current_rel_fnames)
2323
dump(mentioned_rel_fnames)
2424
dump(current_rel_fnames == mentioned_rel_fnames)
2525

26-
if mentioned_rel_fnames != current_rel_fnames:
27-
self.abs_fnames = set()
28-
for fname in mentioned_rel_fnames:
29-
self.add_rel_fname(fname)
30-
dump(self.get_inchat_relative_files())
31-
self.reflected_message = self.gpt_prompts.try_again
26+
if mentioned_rel_fnames == current_rel_fnames:
27+
return True
28+
29+
if self.num_reflections >= self.max_reflections-1:
30+
return True
31+
32+
self.abs_fnames = set()
33+
for fname in mentioned_rel_fnames:
34+
self.add_rel_fname(fname)
35+
dump(self.get_inchat_relative_files())
36+
37+
self.reflected_message = self.gpt_prompts.try_again
38+
39+
#mentioned_idents = self.get_ident_mentions(cur_msg_text)
40+
#if mentioned_idents:
3241

3342
return True
3443

0 commit comments

Comments
 (0)