Skip to content

Commit 86175a1

Browse files
Merge pull request #3127 from FeepingCreature/fix/sort-file-tokens
refactor: Sort files by token count in cmd_tokens
2 parents 5402ed1 + 6d6e25d commit 86175a1

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

aider/commands.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -404,6 +404,7 @@ def cmd_tokens(self, args):
404404

405405
fence = "`" * 3
406406

407+
file_res = []
407408
# files
408409
for fname in self.coder.abs_fnames:
409410
relative_fname = self.coder.get_rel_fname(fname)
@@ -414,7 +415,7 @@ def cmd_tokens(self, args):
414415
# approximate
415416
content = f"{relative_fname}\n{fence}\n" + content + "{fence}\n"
416417
tokens = self.coder.main_model.token_count(content)
417-
res.append((tokens, f"{relative_fname}", "/drop to remove"))
418+
file_res.append((tokens, f"{relative_fname}", "/drop to remove"))
418419

419420
# read-only files
420421
for fname in self.coder.abs_read_only_fnames:
@@ -424,7 +425,10 @@ def cmd_tokens(self, args):
424425
# approximate
425426
content = f"{relative_fname}\n{fence}\n" + content + "{fence}\n"
426427
tokens = self.coder.main_model.token_count(content)
427-
res.append((tokens, f"{relative_fname} (read-only)", "/drop to remove"))
428+
file_res.append((tokens, f"{relative_fname} (read-only)", "/drop to remove"))
429+
430+
file_res.sort()
431+
res.extend(file_res)
428432

429433
self.io.tool_output(
430434
f"Approximate context window usage for {self.coder.main_model.name}, in tokens:"

0 commit comments

Comments
 (0)