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

Notice if the user mentions the basename of a filename; fixes to repomap context

Browse source
This commit is contained in:
Paul Gauthier 2024-06-24 15:12:58 -07:00
parent 7fad06e74c
commit 965c35870c
2 changed files with 29 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

13
aider/repomap.py
View file

@ -1,4 +1,5 @@
import colorsys
import math
import os
import random
import sys
@ -242,7 +243,7 @@ class RepoMap:
# Default personalization for unspecified files is 1/num_nodes
# https://networkx.org/documentation/stable/_modules/networkx/algorithms/link_analysis/pagerank_alg.html#pagerank
personalize = 10 / len(fnames)
personalize = 100 / len(fnames)
if self.cache_missing:
fnames = tqdm(fnames)
@ -268,7 +269,7 @@ class RepoMap:
personalization[rel_fname] = personalize
chat_rel_fnames.add(rel_fname)
if fname in mentioned_fnames:
if rel_fname in mentioned_fnames:
personalization[rel_fname] = personalize
tags = list(self.get_tags(fname, rel_fname))
@ -300,12 +301,20 @@ class RepoMap:
definers = defines[ident]
if ident in mentioned_idents:
mul = 10
elif ident.startswith("_"):
mul = 0.1
else:
mul = 1
for referencer, num_refs in Counter(references[ident]).items():
for definer in definers:
# dump(referencer, definer, num_refs, mul)
# if referencer == definer:
# continue
# scale down so high freq (low value) mentions don't dominate
num_refs = math.sqrt(num_refs)
G.add_edge(referencer, definer, weight=mul * num_refs, ident=ident)
if not references: