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

Use a thread to import slow modules in the background

Browse source
This commit is contained in:
Paul Gauthier 2024-07-03 13:55:29 -03:00
parent b3f7f0a250
commit b5cd5f0e23
2 changed files with 17 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

15
aider/repomap.py
View file

@ -58,15 +58,6 @@ class RepoMap:
self.token_count = main_model.token_count
self.repo_content_prefix = repo_content_prefix
def import_nx(self):
"""Import nx as needed, to avoid 250ms on launch"""
if self.nx:
return
import networkx as nx
self.nx = nx
def get_repo_map(self, chat_files, other_files, mentioned_fnames=None, mentioned_idents=None):
if self.max_map_tokens <= 0:
return
@ -239,7 +230,7 @@ class RepoMap:
)
def get_ranked_tags(self, chat_fnames, other_fnames, mentioned_fnames, mentioned_idents):
self.import_nx()
import networkx as nx
defines = defaultdict(set)
references = defaultdict(list)
@ -306,7 +297,7 @@ class RepoMap:
idents = set(defines.keys()).intersection(set(references.keys()))
G = self.nx.MultiDiGraph()
G = nx.MultiDiGraph()
for ident in idents:
definers = defines[ident]
@ -337,7 +328,7 @@ class RepoMap:
pers_args = dict()
try:
ranked = self.nx.pagerank(G, weight="weight", **pers_args)
ranked = nx.pagerank(G, weight="weight", **pers_args)
except ZeroDivisionError:
return []