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

feat: add force_refresh arg to get_repo_map and pass to get_ranked_tags_map

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-17 06:30:48 -07:00
parent bcd6257c90
commit 551804c289
1 changed files with 16 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

30
aider/repomap.py
View file

@ -83,7 +83,7 @@ class RepoMap:
est_tokens = sample_tokens / len(sample_text) * len_text est_tokens = sample_tokens / len(sample_text) * len_text
return est_tokens return est_tokens
def get_repo_map(self, chat_files, other_files, mentioned_fnames=None, mentioned_idents=None): def get_repo_map(self, chat_files, other_files, mentioned_fnames=None, mentioned_idents=None, force_refresh=False):
if self.max_map_tokens <= 0: if self.max_map_tokens <= 0:
return return
if not other_files: if not other_files:
@ -109,7 +109,7 @@ class RepoMap:
try: try:
files_listing = self.get_ranked_tags_map( files_listing = self.get_ranked_tags_map(
chat_files, other_files, max_map_tokens, mentioned_fnames, mentioned_idents chat_files, other_files, max_map_tokens, mentioned_fnames, mentioned_idents, force_refresh
) )
except RecursionError: except RecursionError:
self.io.tool_error("Disabling repo map, git repo too large?") self.io.tool_error("Disabling repo map, git repo too large?")
@ -412,6 +412,7 @@ class RepoMap:
max_map_tokens=None, max_map_tokens=None,
mentioned_fnames=None, mentioned_fnames=None,
mentioned_idents=None, mentioned_idents=None,
force_refresh=False,
): ):
# Create a cache key # Create a cache key
cache_key = ( cache_key = (
@ -420,21 +421,22 @@ class RepoMap:
max_map_tokens, max_map_tokens,
) )
if self.refresh == "manual" and self.last_map: if not force_refresh:
return self.last_map if self.refresh == "manual" and self.last_map:
return self.last_map
if self.refresh == "always": if self.refresh == "always":
use_cache = False use_cache = False
elif self.refresh == "files": elif self.refresh == "files":
use_cache = True use_cache = True
elif self.refresh == "auto": elif self.refresh == "auto":
use_cache = (self.map_processing_time > 1.0) use_cache = (self.map_processing_time > 1.0)
# Check if the result is in the cache # Check if the result is in the cache
if use_cache and cache_key in self.map_cache: if use_cache and cache_key in self.map_cache:
return self.map_cache[cache_key] return self.map_cache[cache_key]
# If not in cache, generate the map # If not in cache or force_refresh is True, generate the map
start_time = time.time() start_time = time.time()
result = self.get_ranked_tags_map_uncached( result = self.get_ranked_tags_map_uncached(
chat_fnames, other_fnames, max_map_tokens, mentioned_fnames, mentioned_idents chat_fnames, other_fnames, max_map_tokens, mentioned_fnames, mentioned_idents