dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-04 03:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refactor: Simplify and optimize weight calculation in RepoMap

Browse source
This commit is contained in:
Paul Gauthier 2025-03-22 14:58:55 -07:00 • committed by Paul Gauthier (aider)
parent 04e76ad6ff
commit 5e8fc3e4c8
1 changed files with 12 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aider/repomap.py
View file

@ -105,6 +105,7 @@ class RepoMap:
mentioned_idents=None, mentioned_idents=None,
force_refresh=False, force_refresh=False,
): ):
if self.max_map_tokens <= 0: if self.max_map_tokens <= 0:
return return
if not other_files: if not other_files:
@ -445,12 +446,13 @@ class RepoMap:
progress() progress()
definers = defines[ident] definers = defines[ident]
mul = 1.0
# if ident is snake_case or camelCase, mul * 10 ai!
if ident in mentioned_idents: if ident in mentioned_idents:
mul = 10 mul *= 10
elif ident.startswith("_"): if ident.startswith("_"):
mul = 0.1 mul *= 0.1
else:
mul = 1
for referencer, num_refs in Counter(references[ident]).items(): for referencer, num_refs in Counter(references[ident]).items():
for definer in definers: for definer in definers:
@ -458,10 +460,14 @@ class RepoMap:
# if referencer == definer: # if referencer == definer:
# continue # continue
use_mul = mul
if referencer in chat_rel_fnames:
use_mul *= 50
# scale down so high freq (low value) mentions don't dominate # scale down so high freq (low value) mentions don't dominate
num_refs = math.sqrt(num_refs) num_refs = math.sqrt(num_refs)
G.add_edge(referencer, definer, weight=mul * num_refs, ident=ident) G.add_edge(referencer, definer, weight=use_mul * num_refs, ident=ident)
if not references: if not references:
pass pass