Auto-continue feature for free models

Dockerfile

@@ -14,6 +14,9 @@ RUN python3 /tmp/patch_use_agent_chat_full.py
 COPY patch_chat_input.py /tmp/patch_chat_input.py
 RUN python3 /tmp/patch_chat_input.py
 
+COPY patch_sse_transport.py /tmp/patch_sse_transport.py
+RUN python3 /tmp/patch_sse_transport.py
+
 WORKDIR /source/frontend
 RUN npm config set fetch-timeout 120000 && \
     npm config set fetch-retries 3 && \
@@ -43,8 +46,8 @@ COPY configs/frontend_agent_config.json ./configs/frontend_agent_config.json
 COPY patch_models.py /tmp/patch_models.py
 RUN python /tmp/patch_models.py
 
-COPY patch_auto_continue.py /tmp/patch_auto_continue.py
-RUN python /tmp/patch_auto_continue.py
+COPY patch_agent_loop.py /tmp/patch_agent_loop.py
+RUN python /tmp/patch_agent_loop.py || echo "agent_loop patch not applied (may need adjustment)"
 
 RUN mkdir -p /app/session_logs && chown -R user:user /app
 
@@ -56,4 +59,5 @@ ENV HOME=/home/user \
 
 EXPOSE 7860
 WORKDIR /app/backend
+# python:slim does not include bash by default; start.sh is POSIX-compatible.
 CMD ["sh", "start.sh"]

patch_agent_loop.py

@@ -11,28 +11,18 @@ def patch():
     with open(AGENT_LOOP, "r", encoding="utf-8") as f:
         content = f.read()
     
-    # Add import re if needed for detection
     # Add _check_task_incomplete function before class Handlers
-    
     check_func = '''
 
 def _check_task_incomplete(session: Session, llm_result: LLMResult) -> bool:
-    """Check if model stopped streaming but task is incomplete.
-    
-    Returns True when:
-    - LLM finished without tool calls
-    - But there are unfinished plan items
-    """
+    """Check if model stopped streaming but task is incomplete."""
     if llm_result.tool_calls_acc:
         return False  # Has tool calls, task continues via tool execution
     
     plan = getattr(session, "current_plan", None) or []
     unfinished = [item for item in plan if item.get("status") in ("pending", "in_progress")]
     
-    # Also check if there are any pending actions in the loop
-    pending_actions = getattr(session, "pending_actions", None)
-    
-    return len(unfinished) > 0 or bool(pending_actions)
+    return len(unfinished) > 0
 
 
 def _unfinished_plan_items(session: Session) -> list[dict[str, str]]:
@@ -42,19 +32,16 @@ def _unfinished_plan_items(session: Session) -> list[dict[str, str]]:
 
 '''
     
-    # Insert after LLMResult dataclass definition and before Handlers class
     if "_check_task_incomplete" not in content:
         content = content.replace(
             "class Handlers:",
             check_func + "\n\nclass Handlers:"
         )
     
-    # Now we need to add the call in the run_agent method
-    # Find where LLM result is processed and check for incomplete task
+    # Add the check after LLM response processing
     check_call = '''
             # === Check for incomplete task after LLM response ===
             if not llm_result.tool_calls_acc and llm_result.content:
-                # Model finished without tool calls - check if task is done
                 unfinished = _unfinished_plan_items(session)
                 if unfinished:
                     await session.send_event(
@@ -68,24 +55,16 @@ def _unfinished_plan_items(session: Session) -> list[dict[str, str]]:
                     )
 '''
     
-    # Insert this check after processing tool calls but before final_response check
-    # Look for pattern: iteration increment and session updates
     if "task_incomplete" not in content:
-        # Find location after tool_calls processing
+        # Find a good insertion point - after tool calls processing
         insert_marker = "  # -- End of turn --"
         if insert_marker in content:
-            content = content.replace(
-                insert_marker,
-                check_call + "\n" + insert_marker
-            )
+            content = content.replace(insert_marker, check_call + "\n" + insert_marker)
         else:
-            # Try another location - after llm_result processing
+            # Try another location
             result_marker = "final_response = llm_result.content or None"
             if result_marker in content:
-                content = content.replace(
-                    result_marker,
-                    result_marker + "\n" + check_call
-                )
+                content = content.replace(result_marker, result_marker + "\n" + check_call)
     
     with open(AGENT_LOOP, "w", encoding="utf-8") as f:
         f.write(content)

patch_chat_input.py

@@ -1,10 +1,9 @@
-"""Patch ChatInput.tsx - Fixed version with all required props."""
+"""Patch ChatInput.tsx for auto-continue buttons."""
 
 CHAT_INPUT = "/source/frontend/src/components/Chat/ChatInput.tsx"
 
 def patch():
-    import os
-    import re
+    import os, re
     if not os.path.exists(CHAT_INPUT):
         print(f"SKIP: {CHAT_INPUT} not found")
         return
@@ -12,22 +11,24 @@ def patch():
     with open(CHAT_INPUT, "r", encoding="utf-8") as f:
         content = f.read()
     
-    # Add Button to MUI imports
-    if "Button" not in content:
+    # 1. Add Button to MUI imports
+    if "Button" not in content and "@mui/material" in content:
         content = content.replace(
             "import { Box, IconButton, Stack, Tooltip } from '@mui/material';",
             "import { Box, IconButton, Stack, Tooltip, Button } from '@mui/material';"
         )
     
-    # Add import for DatasetUploadResponse if needed
+    # 2. Add DatasetUploadResponse import if needed
     if "DatasetUploadResponse" not in content:
         content = content.replace(
             "import { apiFetch } from '@/utils/api';",
             "import { apiFetch } from '@/utils/api';\nimport type { DatasetUploadResponse } from '@/types/agent';"
         )
     
-    # Replace interface ChatInputProps with complete version including auto-continue props
-    new_props = '''interface ChatInputProps {
+    # 3. Add props to interface
+    if "showAutoContinue" not in content:
+        props_code = '''
+interface ChatInputProps {
   sessionId: string;
   initialModelPath: string | null | undefined;
   onSend: (text: string) => Promise<void>;
@@ -40,19 +41,21 @@ def patch():
   taskIncompleteInfo?: { incompletePlan: Array<{ id: string; content: string; status: string }> } | null;
   startAutoContinue?: () => void;
   cancelAutoContinue?: () => void;
-}'''
-    
-    if "showAutoContinue" not in content:
-        # Find existing interface
-        match = re.search(r'interface ChatInputProps\s*\{[^}]*\}', content, re.DOTALL)
+}
+'''
+        # Find and replace existing interface
+        match = re.search(r'interface ChatInputProps[^}]*[}][^}]*[}]', content, re.DOTALL)
         if match:
-            content = content.replace(match.group(0), new_props)
-            print("OK: Replaced ChatInputProps interface")
+            content = content.replace(match.group(0), props_code)
         else:
-            print("WARN: Could not find ChatInputProps interface")
+            content = content.replace(
+                "export default function ChatInput",
+                props_code + "\nexport default function ChatInput"
+            )
     
-    # Add auto-continue buttons before the flex:1 Box
-    buttons_code = '''      {/* Auto-continue controls for free models */}
+    # 4. Add auto-continue buttons before the flex box
+    buttons_code = '''
+      {/* Auto-continue controls for free models */}
       {showAutoContinue && taskIncompleteInfo && (
         <Stack direction="row" spacing={1} sx={{ mb: 1, px: 1 }}>
           <Button
@@ -74,15 +77,15 @@ def patch():
             Tạm dừng
           </Button>
         </Stack>
-      )}'''
+      )}
+'''
     
     if "Tự động tiếp tục" not in content:
         # Insert before <Box sx={{ flex: 1 }}
         content = content.replace(
-            "<Box sx={{ flex: 1 }} />",
-            buttons_code + "\n      <Box sx={{ flex: 1 }} />"
+            "<Box sx={{ flex: 1 ",
+            buttons_code + "        <Box sx={{ flex: 1 "
         )
-        print("OK: Added auto-continue buttons")
     
     with open(CHAT_INPUT, "w", encoding="utf-8") as f:
         f.write(content)

patch_sse_transport.py

@@ -0,0 +1,32 @@
+"""Patch sse-chat-transport.ts - handle task_incomplete event."""
+
+SSE_TRANSPORT = "/source/frontend/src/lib/sse-chat-transport.ts"
+
+def patch():
+    import os
+    if not os.path.exists(SSE_TRANSPORT):
+        print(f"SKIP: {SSE_TRANSPORT} not found")
+        return
+        
+    with open(SSE_TRANSPORT, "r", encoding="utf-8") as f:
+        content = f.read()
+    
+    # Add task_incomplete case in createEventToChunkStream
+    if "case 'task_incomplete'" not in content:
+        task_incomplete_case = '''      case 'task_incomplete':
+          sideChannel.onTaskIncomplete(
+            (event.data?.incomplete_plan as Array<{ id: string; content: string; status: string }>) || [],
+          );
+          break;
+'''
+        content = content.replace(
+            "case 'turn_complete':",
+            task_incomplete_case + "\n        case 'turn_complete':"
+        )
+    
+    with open(SSE_TRANSPORT, "w", encoding="utf-8") as f:
+        f.write(content)
+    print(f"OK: Patched {SSE_TRANSPORT}")
+
+if __name__ == "__main__":
+    patch()

patch_use_agent_chat_full.py

@@ -1,4 +1,4 @@
-"""Complete patch for useAgentChat.ts - auto-continue feature (fixed TS)."""
+"""Patch useAgentChat.ts - auto-continue feature."""
 
 USE_AGENT_CHAT = "/source/frontend/src/hooks/useAgentChat.ts"
 
@@ -25,12 +25,13 @@ def patch():
   const [showAutoContinue, setShowAutoContinue] = useState(false);
   const [taskIncompleteInfo, setTaskIncompleteInfo] = useState<{
     incompletePlan: Array<{ id: string; content: string; status: string }>;
-  } | null>(null);'''
+  } | null>(null);
+'''
     
     if "autoContinueTimerRef" not in content:
         content = content.replace(
             "callbacksRef.current = { onReady, onError, onSessionDead };",
-            "callbacksRef.current = { onReady, onError, onSessionDead };\n" + state_decl
+            "callbacksRef.current = { onReady, onError, onSessionDead };" + state_decl
         )
     
     # 3. Add onTaskIncomplete to SideChannelCallbacks interface
@@ -41,7 +42,7 @@ def patch():
         )
     
     # 4. Add onTaskIncomplete handler in sideChannel
-    sidechannel_handler = '''      onTaskIncomplete: (incompletePlan: Array<{ id: string; content: string; status: string }>) => {
+    sidechannel_handler = '''      onTaskIncomplete: (incompletePlan) => {
         setTaskIncompleteInfo({ incompletePlan });
         setShowAutoContinue(true);
         // Auto-start 10s timer for free models
@@ -53,7 +54,7 @@ def patch():
         }, 10000);
       },'''
     
-    if "onTaskIncomplete: (incompletePlan:" not in content:
+    if "onTaskIncomplete: (incompletePlan)" not in content:
         content = content.replace(
             "      onInterrupted: () => { /* no-op — handled by stop() caller */ },",
             sidechannel_handler + "\n      onInterrupted: () => { /* no-op — handled by stop() caller */ },"
@@ -96,7 +97,8 @@ def patch():
         clearTimeout(autoContinueTimerRef.current);
       }
     };
-  }, []);'''
+  }, []);
+'''
     
     if "startAutoContinue" not in content:
         content = content.replace(