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

refac

Browse source
This commit is contained in:
Paul Gauthier 2023-05-24 14:01:38 -07:00
parent 69b180cffc
commit 27199becd4
1 changed files with 22 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

40
aider/coder.py
View file

@ -171,24 +171,11 @@ class Coder:
files_content += self.get_files_content() files_content += self.get_files_content()
all_content += files_content all_content += files_content
res = self.choose_files_listing() other_files = set(self.get_all_abs_files()) - set(self.abs_fnames)
if res: repo_content = self.get_map(self.abs_fnames, other_files)
files_listing, ctags_msg = res if repo_content:
if self.abs_fnames:
other = "other "
else:
other = ""
repo_content = prompts.repo_content_prefix.format(
other=other,
ctags_msg=ctags_msg,
)
repo_content += files_listing
if all_content: if all_content:
all_content += "\n" all_content += "\n"
all_content += repo_content all_content += repo_content
files_messages = [ files_messages = [
@ -202,14 +189,31 @@ class Coder:
return files_messages return files_messages
def choose_files_listing(self): def get_map(self, chat_files, other_files):
res = self.choose_files_listing(other_files)
if res:
files_listing, ctags_msg = res
if chat_files:
other = "other "
else:
other = ""
repo_content = prompts.repo_content_prefix.format(
other=other,
ctags_msg=ctags_msg,
)
repo_content += files_listing
return repo_content
def choose_files_listing(self, other_files):
# 1/4 of gpt-4's context window # 1/4 of gpt-4's context window
max_map_tokens = 2048 max_map_tokens = 2048
if not self.repo: if not self.repo:
return return
other_files = set(self.get_all_abs_files()) - set(self.abs_fnames)
if not other_files: if not other_files:
return return