diff --git a/aider/repomap.py b/aider/repomap.py index 0b438114c..c484bfb4f 100644 --- a/aider/repomap.py +++ b/aider/repomap.py @@ -106,7 +106,8 @@ class RepoMap: if self.use_ctags: files_listing = self.get_tags_map(other_files) num_tokens = self.token_count(files_listing) - self.io.tool_output(f"ctags map: {num_tokens/1024:.1f} k-tokens") + if self.io: + self.io.tool_output(f"ctags map: {num_tokens/1024:.1f} k-tokens") if num_tokens < max_map_tokens: ctags_msg = " with selected ctags info" return files_listing, ctags_msg @@ -114,7 +115,8 @@ class RepoMap: files_listing = self.get_simple_files_map(other_files) ctags_msg = "" num_tokens = self.token_count(files_listing) - self.io.tool_output(f"simple map: {num_tokens/1024:.1f} k-tokens") + if self.io: + self.io.tool_output(f"simple map: {num_tokens/1024:.1f} k-tokens") if num_tokens < max_map_tokens: return files_listing, ctags_msg