Only clear invocation cache on success

Author: Your Name

cecli/tools/command_interactive.py

@@ -93,6 +93,8 @@ def _run_interactive():
                     " large_file_token_threshold)"
                 )
 
+            cls.clear_invocation_cache()
+
             if exit_status == 0:
                 return (
                     "Interactive command finished successfully (exit code 0)."

cecli/tools/delete_text.py

@@ -118,6 +118,8 @@ def execute(
             )
 
             coder.files_edited_by_tools.add(rel_path)
+            cls.clear_invocation_cache()
+
             # 5. Format and return result
             success_message = f"Deleted lines {start_line} to {end_line} in {file_path}"
             return format_tool_result(
@@ -130,10 +132,8 @@ def execute(
         except ToolError as e:
             # Handle errors raised by utility functions (expected errors)
             coder.edit_allowed = False
-            cls.clear_invocation_cache()
             return handle_tool_error(coder, tool_name, e, add_traceback=False)
         except Exception as e:
             # Handle unexpected errors
             coder.edit_allowed = False
-            cls.clear_invocation_cache()
             return handle_tool_error(coder, tool_name, e)

cecli/tools/finished.py

@@ -26,6 +26,7 @@ def execute(cls, coder, **kwargs):
 
         This gives the LLM explicit control over when it can stop looping
         """
+        cls.clear_invocation_cache()
 
         if coder:
             coder.agent_finished = True

cecli/tools/insert_text.py

@@ -123,6 +123,7 @@ def execute(
             )
 
             coder.files_edited_by_tools.add(rel_path)
+            cls.clear_invocation_cache()
 
             # 5. Format and return result
             success_message = f"Inserted content at {start_line} in {file_path}"
@@ -134,14 +135,12 @@ def execute(
             )
 
         except ToolError as e:
-            coder.edit_allowed = False
-            cls.clear_invocation_cache()
             # Handle errors raised by utility functions (expected errors)
+            coder.edit_allowed = False
             return handle_tool_error(coder, tool_name, e, add_traceback=False)
 
         except Exception as e:
             coder.edit_allowed = False
-            cls.clear_invocation_cache()
             return handle_tool_error(coder, tool_name, e)
 
     @classmethod

cecli/tools/replace_text.py

@@ -262,6 +262,8 @@ def execute(
                     success_message += "\nFailed edits:\n" + "\n".join(all_failed_edits)
                 change_id_to_return = None  # Multiple change IDs, can't return single one
 
+            cls.clear_invocation_cache()
+
             return format_tool_result(
                 coder,
                 tool_name,
@@ -272,12 +274,10 @@ def execute(
         except ToolError as e:
             # Handle errors raised by utility functions or explicitly raised here
             coder.edit_allowed = False
-            cls.clear_invocation_cache()
             return handle_tool_error(coder, tool_name, e, add_traceback=False)
         except Exception as e:
             # Handle unexpected errors
             coder.edit_allowed = False
-            cls.clear_invocation_cache()
             return handle_tool_error(coder, tool_name, e)
 
     @classmethod

cecli/tools/utils/base_tool.py

@@ -88,10 +88,7 @@ def process_response(cls, coder, params):
             cls._invocations[tool_name].append((current_params_tuple, params))
             if len(cls._invocations[tool_name]) > 3:
                 cls._invocations[tool_name] = cls._invocations[tool_name][-3:]
-        else:
-            # When TRACK_INVOCATIONS is False, clear all invocation history
-            # This indicates the job is making progress, so reset tracking for all tools
-            cls.clear_invocation_cache()
+
         try:
             return cls.execute(coder, **params)
         except Exception as e: