diff --git a/aider/repomap.py b/aider/repomap.py index 50c95b50b..0b438114c 100644 --- a/aider/repomap.py +++ b/aider/repomap.py @@ -333,8 +333,8 @@ def call_map(): for ident in idents: definers = defines[ident] num_defs = len(definers) - if num_defs > 3: - continue + # if num_defs > 3: + # continue for referencer, num_refs in Counter(references[ident]).items(): for definer in definers: @@ -353,6 +353,10 @@ def call_map(): dangling=personalization, ) + for node in G.nodes: + # todo: sum up the weight of every inbound edge to node + pass + N = 20 top_10_nodes = sorted(ranked, key=ranked.get, reverse=True)[:N] nodes_to_remove = [node for node in G.nodes if node not in top_10_nodes] @@ -390,7 +394,7 @@ def call_map(): dump(refs, defs, data) color = get_random_color() - weight = weight * 2 + weight = weight dot.edge(refs, defs, penwidth=str(weight), color=color, fontcolor=color, label=label) top_rank = sorted([(rank, node) for (node, rank) in ranked.items()], reverse=True)