refactor: Enhance file mention handling and context coder logic

This commit is contained in:
Paul Gauthier 2025-03-22 13:57:15 -07:00 committed by Paul Gauthier (aider)
parent f543c1ee1c
commit 5661d1428e
3 changed files with 44 additions and 7 deletions

View file

@ -1,3 +1,5 @@
from aider.dump import dump
from .base_coder import Coder
from .context_prompts import ContextPrompts
@ -7,3 +9,28 @@ class ContextCoder(Coder):
edit_format = "context"
gpt_prompts = ContextPrompts()
def reply_completed(self):
content = self.partial_response_content
if not content or not content.strip():
return True
dump(repr(content))
current_rel_fnames = set(self.get_inchat_relative_files())
mentioned_rel_fnames = set(self.get_file_mentions(content, ignore_current=True))
dump(current_rel_fnames)
dump(mentioned_rel_fnames)
dump(current_rel_fnames == mentioned_rel_fnames)
if mentioned_rel_fnames != current_rel_fnames:
self.abs_fnames = set()
for fname in mentioned_rel_fnames:
self.add_rel_fname(fname)
dump(self.get_inchat_relative_files())
self.reflected_message = self.gpt_prompts.try_again
return True
def check_for_file_mentions(self, content):
pass