mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 00:35:00 +00:00
refactor tokenizer
This commit is contained in:
parent
6f1cebc4c2
commit
547ae142ba
3 changed files with 4 additions and 9 deletions
|
@ -52,7 +52,7 @@ class RepoMap:
|
|||
|
||||
self.max_map_tokens = map_tokens
|
||||
|
||||
self.tokenizer = main_model.tokenizer
|
||||
self.token_count = main_model.token_count
|
||||
self.repo_content_prefix = repo_content_prefix
|
||||
|
||||
def get_repo_map(self, chat_files, other_files):
|
||||
|
@ -89,9 +89,6 @@ class RepoMap:
|
|||
|
||||
return repo_content
|
||||
|
||||
def token_count(self, string):
|
||||
return len(self.tokenizer(string))
|
||||
|
||||
def get_rel_fname(self, fname):
|
||||
return os.path.relpath(fname, self.root)
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue