From 47293c5ba9e1e4908799401194148fccf7e60fdb Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 2 Jun 2023 13:22:25 -0700 Subject: [PATCH] removed graph --- aider/repomap.py | 37 ------------------------------------- 1 file changed, 37 deletions(-) diff --git a/aider/repomap.py b/aider/repomap.py index 4f00ebeb8..68edb3c6d 100644 --- a/aider/repomap.py +++ b/aider/repomap.py @@ -8,7 +8,6 @@ import tempfile from collections import Counter, defaultdict # import shelve -import graphviz import networkx as nx import tiktoken from pygments.lexers import guess_lexer_for_filename @@ -347,48 +346,12 @@ class RepoMap: ident = data["ident"] ranked_definitions[(dst, ident)] += data["rank"] - draw_graph = False - - if draw_graph: - clusters = dict() - for fname in set(show_fnames): - clusters[fname] = graphviz.Digraph(f"cluster_{fname}") - clusters[fname].attr(label=fname, style="filled") - clusters[fname].node(f"invis_{fname}", style="invis", width="0", label="") - ranked_tags = [] ranked_definitions = sorted(ranked_definitions.items(), reverse=True, key=lambda x: x[1]) for (fname, ident), rank in ranked_definitions: print(f"{rank:.03f} {fname} {ident}") ranked_tags += list(definitions.get((fname, ident), [])) - if draw_graph: - sz = str(rank * 25) - font_sz = rank * 500 - font_sz = str(max(10, font_sz)) - clusters[fname].node( - str((fname, ident)), label=ident, width=sz, height=sz, fontsize=font_sz - ) - - if draw_graph: - dot = graphviz.Digraph(graph_attr={"ratio": ".5"}) - - for cluster in clusters.values(): - dot.subgraph(cluster) - - for src, dst, data in G.edges(data=True): - frm = f"invis_{src}" - ident = data["ident"] - to = str((dst, ident)) - - dot.edge( - frm, - to, - # penwidth=str(weight), color=color, fontcolor=color, label=label, - ) - - dot.render("tmp", format="pdf", view=True) - return ranked_tags def get_ranked_tags_map(self, chat_fnames, other_fnames=None):