Merge branch 'main' into sitter-map

This commit is contained in:
Paul Gauthier 2023-09-01 12:00:48 -07:00
commit fa6ae80653
13 changed files with 163 additions and 42 deletions

View file

@ -6,7 +6,6 @@ from collections import Counter, defaultdict, namedtuple
from pathlib import Path
import networkx as nx
import tiktoken
from diskcache import Cache
from grep_ast import TreeContext
from tqdm import tqdm
@ -87,7 +86,7 @@ class RepoMap:
self,
map_tokens=1024,
root=None,
main_model=models.GPT4,
main_model=models.Model.strong_model(),
io=None,
repo_content_prefix=None,
verbose=False,
@ -103,7 +102,7 @@ class RepoMap:
self.max_map_tokens = map_tokens
self.tokenizer = tiktoken.encoding_for_model(main_model.name)
self.tokenizer = main_model.tokenizer
self.repo_content_prefix = repo_content_prefix
def get_repo_map(self, chat_files, other_files):