Merge pull request #551 from cecli-dev/v0.100.4

V0.100.4

Author: dwash96

cecli/__init__.py

@@ -1,6 +1,6 @@
 from packaging import version
 
-__version__ = "0.100.2.dev"
+__version__ = "0.100.4.dev"
 safe_version = __version__
 
 try:

cecli/coders/base_coder.py

@@ -42,6 +42,7 @@
 from cecli.exceptions import LiteLLMExceptions
 from cecli.helpers import command_parser, coroutines, nested, responses
 from cecli.helpers.conversation import ConversationService, MessageTag
+from cecli.helpers.file_system import FileSystemService
 from cecli.helpers.io_proxy import IOProxy
 from cecli.helpers.observations.service import ObservationService
 from cecli.helpers.profiler import TokenProfiler
@@ -459,9 +460,7 @@ def __init__(
 
         self.context_compaction_max_tokens = context_compaction_max_tokens
         self.context_compaction_summary_tokens = context_compaction_summary_tokens
-        self.max_reflections = (
-            3 if self.edit_format == "agent" else nested.getter(self.args, "max_reflections", 3)
-        )
+        self.max_reflections = nested.getter(self.args, "max_reflections", 3)
 
         if not fnames:
             fnames = []
@@ -547,7 +546,6 @@ def __init__(
 
         self.data_cache = {
             "repo": {"last_key": "", "read_only_count": None},
-            "relative_files": None,
         }
 
         self.repo = repo
@@ -594,6 +592,12 @@ def __init__(
         if not self.repo:
             self.root = utils.find_common_root(self.abs_fnames)
 
+        # Initialize the FileSystemService singleton for all agents
+        FileSystemService.get_instance(
+            root=self.root if hasattr(self, "root") else ".",
+            repo=self.repo if hasattr(self, "repo") else None,
+        )
+
         if read_only_fnames:
             self.abs_read_only_fnames = set()
             for fname in read_only_fnames:
@@ -3226,7 +3230,7 @@ async def check_for_file_mentions(self, content):
         group = ConfirmGroup(new_mentions)
         for rel_fname in sorted(new_mentions):
             message = "Add file to the chat?"
-            if self.args.tui:
+            if self.args and self.args.tui:
                 message = f"Add file to the chat? ({rel_fname})"
 
             if await self.io.confirm_ask(
@@ -3909,37 +3913,16 @@ def is_file_safe(self, fname):
             return
 
     def get_all_relative_files(self):
-        if self.repo_map and self.repo:
-            try:
-                staged_files_hash = hash(
-                    str([item.a_path for item in self.repo.repo.index.diff("HEAD")])
-                )
-                if (
-                    staged_files_hash == self.data_cache["repo"]["last_key"]
-                    and self.data_cache["relative_files"]
-                ):
-                    return self.data_cache["relative_files"]
-            except ANY_GIT_ERROR as err:
-                # Handle git errors gracefully - fall back to getting tracked files
-                if self.verbose:
-                    self.io.tool_warning(f"Git error while checking staged files: {err}")
-                # Continue to get tracked files normally
-
-        if self.repo:
-            if hasattr(self.repo, "workspace_path") and self.repo.workspace_path:
-                files = self.repo.get_workspace_files()
-            elif not self.repo.cecli_ignore_file or not self.repo.cecli_ignore_file.is_file():
-                files = self.repo.get_tracked_files()
-            else:
-                files = self.repo.get_non_ignored_files_from_root()
-        else:
-            files = self.get_inchat_relative_files()
-        # This is quite slow in large repos
-        # files = [fname for fname in files if self.is_file_safe(fname)]
-
-        self.data_cache["relative_files"] = sorted(set(files))
-
-        return self.data_cache["relative_files"]
+        """Get all files known to the file service singleton."""
+        fs = FileSystemService.get_instance()
+        if fs.trie:
+            # Auto-rebuild if the repository state has changed
+            # (e.g., new commits, staged files, or HEAD change)
+            if fs.needs_rebuild():
+                fs.rebuild()
+            files = fs.list_all()
+            return files
+        return self.get_inchat_relative_files()
 
     def get_all_abs_files(self):
         files = self.get_all_relative_files()

cecli/commands/agent.py

@@ -9,6 +9,7 @@ class AgentCommand(BaseCommand):
         "Enter agent mode to autonomously discover and manage relevant files. If no prompt"
         " provided, switches to agent mode."
     )
+    show_completion_notification = True
 
     @classmethod
     async def execute(cls, io, coder, args, **kwargs):

cecli/commands/agent_model.py

@@ -9,7 +9,7 @@
 class AgentModelCommand(BaseCommand):
     NORM_NAME = "agent-model"
     DESCRIPTION = "Switch the Agent Model to a new LLM"
-    show_completion_notification = False
+    show_completion_notification = True
 
     @classmethod
     async def execute(cls, io, coder, args, **kwargs):

cecli/commands/architect.py

@@ -9,6 +9,7 @@ class ArchitectCommand(BaseCommand):
         "Enter architect/editor mode using 2 different models. If no prompt provided, switches to"
         " architect/editor mode."
     )
+    show_completion_notification = True
 
     @classmethod
     async def execute(cls, io, coder, args, **kwargs):

cecli/commands/ask.py

@@ -9,6 +9,7 @@ class AskCommand(BaseCommand):
         "Ask questions about the code base without editing any files. If no prompt provided,"
         " switches to ask mode."
     )
+    show_completion_notification = True
 
     @classmethod
     async def execute(cls, io, coder, args, **kwargs):

cecli/commands/code.py

@@ -6,6 +6,7 @@
 class CodeCommand(BaseCommand):
     NORM_NAME = "code"
     DESCRIPTION = "Ask for changes to your code. If no prompt provided, switches to code mode."
+    show_completion_notification = True
 
     @classmethod
     async def execute(cls, io, coder, args, **kwargs):

cecli/commands/command_prefix.py

@@ -7,6 +7,7 @@
 class CommandPrefixCommand(BaseCommand):
     NORM_NAME = "command-prefix"
     DESCRIPTION = "Change command prefix for all running commands"
+    show_completion_notification = True
 
     @classmethod
     async def execute(cls, io, coder, args, **kwargs):

cecli/commands/commit.py

@@ -8,6 +8,7 @@
 class CommitCommand(BaseCommand):
     NORM_NAME = "commit"
     DESCRIPTION = "Commit edits to the repo made outside the chat (commit message optional)"
+    show_completion_notification = True
 
     @classmethod
     async def execute(cls, io, coder, args, **kwargs):

cecli/commands/compact.py

@@ -4,6 +4,7 @@
 class CompactCommand(BaseCommand):
     NORM_NAME = "compact"
     DESCRIPTION = "Force compaction of the chat history context"
+    show_completion_notification = True
 
     @classmethod
     async def execute(cls, io, coder, args, **kwargs):