dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 09:44:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

fix: Add elapsed time logging to print statements

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-05 16:54:42 -03:00
parent 042a4f14d2
commit 276b30506c
1 changed files with 16 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

22
aider/repomap.py
View file

@ -3,6 +3,7 @@ import math
import os import os
import random import random
import sys import sys
import time
import warnings import warnings
from collections import Counter, defaultdict, namedtuple from collections import Counter, defaultdict, namedtuple
from importlib import resources from importlib import resources
@ -23,6 +24,15 @@ from aider.dump import dump # noqa: F402,E402
Tag = namedtuple("Tag", "rel_fname fname line name kind".split()) Tag = namedtuple("Tag", "rel_fname fname line name kind".split())
def print_elapsed(message):
current_time = time.time()
if hasattr(print_elapsed, 'last_time'):
elapsed = current_time - print_elapsed.last_time
print(f"{message}: {elapsed:.2f} seconds")
else:
print(f"{message}: (first measurement)")
print_elapsed.last_time = current_time
class RepoMap: class RepoMap:
CACHE_VERSION = 3 CACHE_VERSION = 3
@ -246,7 +256,7 @@ class RepoMap:
fnames = tqdm(fnames) fnames = tqdm(fnames)
self.cache_missing = False self.cache_missing = False
print('tags') print_elapsed('Starting tags processing')
for fname in fnames: for fname in fnames:
if not Path(fname).is_file(): if not Path(fname).is_file():
@ -284,7 +294,7 @@ class RepoMap:
elif tag.kind == "ref": elif tag.kind == "ref":
references[tag.name].append(rel_fname) references[tag.name].append(rel_fname)
print('graph') print_elapsed('Finished tags processing, starting graph creation')
## ##
# dump(defines) # dump(defines)
# dump(references) # dump(references)
@ -325,13 +335,13 @@ class RepoMap:
else: else:
pers_args = dict() pers_args = dict()
print('rank') print_elapsed('Starting pagerank calculation')
try: try:
ranked = nx.pagerank(G, weight="weight", **pers_args) ranked = nx.pagerank(G, weight="weight", **pers_args)
except ZeroDivisionError: except ZeroDivisionError:
return [] return []
print('dist rank') print_elapsed('Starting rank distribution')
# distribute the rank from each source node, across all of its out edges # distribute the rank from each source node, across all of its out edges
ranked_definitions = defaultdict(float) ranked_definitions = defaultdict(float)
@ -349,7 +359,7 @@ class RepoMap:
# dump(ranked_definitions) # dump(ranked_definitions)
print('ranked_tags') print_elapsed('Starting ranked tags processing')
for (fname, ident), rank in ranked_definitions: for (fname, ident), rank in ranked_definitions:
# print(f"{rank:.03f} {fname} {ident}") # print(f"{rank:.03f} {fname} {ident}")
if fname in chat_rel_fnames: if fname in chat_rel_fnames:
@ -370,7 +380,7 @@ class RepoMap:
for fname in rel_other_fnames_without_tags: for fname in rel_other_fnames_without_tags:
ranked_tags.append((fname,)) ranked_tags.append((fname,))
print('done') print_elapsed('Finished processing')
return ranked_tags return ranked_tags
def get_ranked_tags_map( def get_ranked_tags_map(