fixed commands to deal with optional repo map

This commit is contained in:
Paul Gauthier 2023-06-05 22:07:19 -07:00
parent c8beac9a2d
commit cf3bf998dd

View file

@ -109,10 +109,11 @@ class Commands:
# repo map # repo map
other_files = set(self.coder.get_all_abs_files()) - set(self.coder.abs_fnames) other_files = set(self.coder.get_all_abs_files()) - set(self.coder.abs_fnames)
repo_content = self.coder.repo_map.get_repo_map(self.coder.abs_fnames, other_files) if self.coder.repo_map:
if repo_content: repo_content = self.coder.repo_map.get_repo_map(self.coder.abs_fnames, other_files)
tokens = len(self.tokenizer.encode(repo_content)) if repo_content:
res.append((tokens, "repository map")) tokens = len(self.tokenizer.encode(repo_content))
res.append((tokens, "repository map"))
# files # files
for fname in self.coder.abs_fnames: for fname in self.coder.abs_fnames: