From 5ffdf208a661a14a33051c3db748d825e81abe54 Mon Sep 17 00:00:00 2001
From: giveen <ajaxx20020@gmail.com>
Date: Thu, 2 Apr 2026 14:37:19 -0600
Subject: [PATCH] fix: resume agent task after auto-compact instead of dropping
 to prompt

After auto-compact fired the main run loop fell back to get_user_input()
and waited for the human, so the agent stopped working mid-task.

Root cause: no mechanism existed to replay the current task into the next
iteration after message_history was cleared.

Fix:
- Add _post_compact_input (str | None) variable initialised to None at
  session start.
- Capture _last_user_input from user_input just before turn_count += 1.
- After a successful auto-compact set _post_compact_input to
  _last_user_input (or 'Continue the current task.' if it was blank).
- At the top of the input-gathering block, consume _post_compact_input
  before calling get_user_input() so the agent immediately re-runs with
  its previous task prompt and the fresh compacted context.
---
 src/cai/cli.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/src/cai/cli.py b/src/cai/cli.py
index abe48d90..a4ce9400 100644
--- a/src/cai/cli.py
+++ b/src/cai/cli.py
@@ -442,6 +442,9 @@ def run_cai_cli(
     agent = starting_agent
     turn_count = 0
     idle_time = 0
+    # Holds a user message to replay on the next iteration without prompting
+    # the user — set by auto-compact so the agent continues its current task.
+    _post_compact_input: str | None = None
     console = Console()
     last_model = os.getenv("CAI_MODEL", "alias1")
     last_agent_type = os.getenv("CAI_AGENT_TYPE", "one_tool_agent")
@@ -690,6 +693,11 @@ def run_cai_cli(
                 if use_initial_prompt:
                     user_input = initial_prompt
                     use_initial_prompt = False  # Only use it once
+                elif _post_compact_input is not None:
+                    # Auto-compact just ran — replay the last task so the agent
+                    # continues working without waiting for human input.
+                    user_input = _post_compact_input
+                    _post_compact_input = None
                 else:
                     # Get user input with command completion and history
                     user_input = get_user_input(
@@ -1711,6 +1719,9 @@ def run_cai_cli(
                 agent.model.message_history[:] = fix_message_list(agent.model.message_history)
             turn_count += 1
 
+            # Capture user_input here so auto-compact can replay it after clearing history.
+            _last_user_input = user_input if isinstance(user_input, str) else ""
+
             # Auto-compact: when CAI_SUPPORT_MODEL + CAI_SUPPORT_INTERVAL are both set,
             # compact the conversation every N turns using the support model so the
             # main model's context window is kept small.  After compaction the
@@ -1738,9 +1749,16 @@ def run_cai_cli(
                         _reloaded = _AM.get_active_agent()
                         if _reloaded is not None:
                             agent = _reloaded
+                        # Queue the last user task to be replayed on the next
+                        # iteration so the agent continues without human input.
+                        _post_compact_input = (
+                            _last_user_input
+                            if _last_user_input.strip()
+                            else "Continue the current task."
+                        )
                         console.print(
                             "[bold green]✓ Memory summary applied to agent system prompt — "
-                            "context window reset[/bold green]\n"
+                            "context window reset — continuing task[/bold green]\n"
                         )
                 except (ValueError, Exception) as _e:
                     if os.getenv("CAI_DEBUG", "1") == "2":