mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 00:35:00 +00:00
Use the Models enum instead of .value
This commit is contained in:
parent
130f73c864
commit
fcd8a6df73
3 changed files with 16 additions and 15 deletions
|
@ -69,7 +69,7 @@ class RepoMap:
|
|||
self,
|
||||
map_tokens=1024,
|
||||
root=None,
|
||||
main_model=Models.GPT4.value,
|
||||
main_model=Models.GPT4,
|
||||
io=None,
|
||||
repo_content_prefix=None,
|
||||
):
|
||||
|
@ -88,7 +88,7 @@ class RepoMap:
|
|||
else:
|
||||
self.has_ctags = False
|
||||
|
||||
self.tokenizer = tiktoken.encoding_for_model(main_model)
|
||||
self.tokenizer = tiktoken.encoding_for_model(main_model.value)
|
||||
self.repo_content_prefix = repo_content_prefix
|
||||
|
||||
def get_repo_map(self, chat_files, other_files):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue