Skip to content

Commit 48514e5

Browse files
committed
#267: Fix /tokens on removed files
1 parent 0552c6a commit 48514e5

1 file changed

Lines changed: 13 additions & 1 deletion

File tree

aider/commands.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -581,6 +581,10 @@ def cmd_tokens(self, args):
581581

582582
relative_fname = self.coder.get_rel_fname(fname)
583583
content = self.io.read_text(fname)
584+
585+
if not content:
586+
continue
587+
584588
if is_image_file(relative_fname):
585589
tokens = self.coder.main_model.token_count_for_image(fname)
586590
else:
@@ -603,7 +607,11 @@ def cmd_tokens(self, args):
603607

604608
relative_fname = self.coder.get_rel_fname(fname)
605609
content = self.io.read_text(fname)
606-
if content is not None and not is_image_file(relative_fname):
610+
611+
if not content:
612+
continue
613+
614+
if not is_image_file(relative_fname):
607615
# approximate
608616
content = f"{relative_fname}\n{fence}\n" + content + f"{fence}\n"
609617
tokens = self.coder.main_model.token_count(content)
@@ -620,6 +628,10 @@ def cmd_tokens(self, args):
620628
relative_fname = self.coder.get_rel_fname(fname)
621629
if not is_image_file(relative_fname):
622630
stub = self.coder.get_file_stub(fname)
631+
632+
if not stub:
633+
continue
634+
623635
content = f"{relative_fname} (stub)\n{fence}\n" + stub + "{fence}\n"
624636
tokens = self.coder.main_model.token_count(content)
625637
res.append((tokens, f"{relative_fname} (read-only stub)", "/drop to remove"))

0 commit comments

Comments
 (0)