dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 10:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix: Improve filename detection in find_filename function

Browse source
This commit is contained in:
Paul Gauthier 2024-08-26 13:52:35 -07:00 • committed by Paul Gauthier (aider)
parent a716fda897
commit 66e9c3834a
2 changed files with 10 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

10
aider/coders/editblock_coder.py
View file

@ -532,12 +532,9 @@ def find_filename(lines, fence, valid_fnames):
if not line.startswith(fence[0]):
break
if not len(filenames):
if not filenames:
return
if len(filenames) == 1:
return filenames[0]
# pick the *best* filename found
# Check for exact match first
@ -553,7 +550,7 @@ def find_filename(lines, fence, valid_fnames):
# Perform fuzzy matching with valid_fnames
for fname in filenames:
close_matches = difflib.get_close_matches(fname, valid_fnames, n=1, cutoff=0.6)
close_matches = difflib.get_close_matches(fname, valid_fnames, n=1, cutoff=0.8)
if len(close_matches) == 1:
return close_matches[0]
@ -562,7 +559,8 @@ def find_filename(lines, fence, valid_fnames):
if "." in fname:
return fname
return filenames[0] if filenames else None
if filenames:
return filenames[0]
def find_similar_lines(search_lines, content_lines, threshold=0.6):