"""V8: Safe auto-continue via useEffect pattern (no hoisting issues).""" import os events_ts = "/source/frontend/src/types/events.ts" sse_ts = "/source/frontend/src/lib/sse-chat-transport.ts" hook_ts = "/source/frontend/src/hooks/useAgentChat.ts" input_tsx = "/source/frontend/src/components/Chat/ChatInput.tsx" session_tsx = "/source/frontend/src/components/SessionChat.tsx" agent_py = "/app/agent/core/agent_loop.py" def patch_events(): with open(events_ts) as f: c = f.read() if "'task_incomplete'" not in c: c = c.replace( " | 'interrupted'", " | 'interrupted'\n | 'task_incomplete'" ) with open(events_ts, 'w') as f: f.write(c) print("OK: events.ts") def patch_sse(): with open(sse_ts) as f: c = f.read() if "onTaskIncomplete:" not in c: c = c.replace( " onInterrupted: () => void;", " onInterrupted: () => void;\n onTaskIncomplete: (incompletePlan: Array<{ id: string; content: string; status: string }>) => void;" ) if "case 'task_incomplete'" not in c: case_handler = """ case 'task_incomplete': sideChannel.onTaskIncomplete( (event.data?.incomplete_plan as Array<{ id: string; content: string; status: string }>) || [], ); break; default:""" c = c.replace("\n default:", "\n" + case_handler) with open(sse_ts, 'w') as f: f.write(c) print("OK: sse.ts") def patch_hook(): with open(hook_ts) as f: c = f.read() # 1. Import useState if "useState" not in c: c = c.replace( "import { useCallback, useEffect, useMemo, useRef } from 'react';", "import { useCallback, useEffect, useMemo, useRef, useState } from 'react';" ) # 2. State after callbacksRef state = """ // Auto-continue state for free models const [_showAc, _setShowAc] = useState(false); const _acPlanRef = useRef>([]);""" if "_showAc" not in c: c = c.replace( "callbacksRef.current = { onReady, onError, onSessionDead };", "callbacksRef.current = { onReady, onError, onSessionDead };" + state ) # 3. onTaskIncomplete in interface if "onTaskIncomplete:" not in c: c = c.replace( " onInterrupted: () => void;", " onInterrupted: () => void;\n onTaskIncomplete: (incompletePlan: Array<{ id: string; content: string; status: string }>) => void;" ) # 4. Handler in sideChannel (only set state + ref, no chat.sendMessage) handler = """ onTaskIncomplete: (incompletePlan: Array<{ id: string; content: string; status: string }>) => { _acPlanRef.current = incompletePlan; _setShowAc(true); }, onInterrupted: () => { /* no-op */ },""" if "onTaskIncomplete: (incompletePlan" not in c: old = ' onInterrupted: () => { /* no-op \u2014 handled by stop() caller */ },' if old in c: c = c.replace(old, handler) else: c = c.replace(' onInterrupted: () => { /* no-op */ },', handler) # 5. useEffect to watch _showAc - runs timer, calls chat.sendMessage via chatActionsRef effect = """ // Auto-continue effect: when _showAc becomes true, start 10s timer useEffect(() => { if (!_showAc) return; const timer = setTimeout(() => { const plan = _acPlanRef.current; const setMsgs = chatActionsRef.current.setMessages; const msgs = chatActionsRef.current.messages; if (setMsgs && plan.length > 0) { const planStr = plan.map(i => `- ${i.content}`).join('\\\\n'); const newMsg = { id: 'ac-' + Date.now(), role: 'user' as const, parts: [{ type: 'text' as const, text: `[TIẾP TỤC] Task chưa hoàn thành:\\\\n${planStr}` }], content: '' }; setMsgs([...msgs, newMsg]); } _setShowAc(false); }, 10000); return () => clearTimeout(timer); }, [_showAc]); """ if "Auto-continue effect" not in c: # Insert before return c = c.replace("\n return {\n messages: chat.messages,", effect + "\n return {\n messages: chat.messages,") # 6. Cancel + return cancel = """ const _acCancel = useCallback(() => { _setShowAc(false); }, []);""" if "_acCancel" not in c: c = c.replace("\n return {\n messages: chat.messages,", cancel + "\n return {\n messages: chat.messages,") if "_showAc," not in c: c = c.replace( "refreshMessages,\n };", "refreshMessages,\n _showAc,\n _acCancel,\n };" ) with open(hook_ts, 'w') as f: f.write(c) print("OK: useAgentChat.ts") def patch_chat_input(): with open(input_tsx) as f: c = f.read() if "Button" not in c: c = c.replace(" Tooltip,", " Tooltip,\n Button,") if "_showAc" not in c: old_iface = """interface ChatInputProps { sessionId?: string; initialModelPath?: string | null; onSend: (text: string) => void; onStop?: () => void; onDatasetUploaded?: () => Promise | boolean; isProcessing?: boolean; disabled?: boolean; placeholder?: string; }""" new_iface = """interface ChatInputProps { sessionId?: string; initialModelPath?: string | null; onSend: (text: string) => void; onStop?: () => void; onDatasetUploaded?: () => Promise | boolean; isProcessing?: boolean; disabled?: boolean; placeholder?: string; _showAc?: boolean; _acCancel?: () => void; }""" c = c.replace(old_iface, new_iface) old_props = "placeholder = 'Ask anything...' }: ChatInputProps) {" new_props = "placeholder = 'Ask anything...', _showAc = false, _acCancel }: ChatInputProps) {" if old_props not in c: old_props2 = "placeholder = 'Ask anything...' }: ChatInputProps) {" new_props2 = "placeholder = 'Ask anything...', _showAc = false, _acCancel }: ChatInputProps) {" c = c.replace(old_props2, new_props2) else: c = c.replace(old_props, new_props) if "T\u1ea1m d\u1eebng" not in c: btn = """ {_showAc && _acCancel && ( )} """ if " list[dict[str, str]]: p = getattr(s, "current_plan", None) or [] return [it for it in p if it.get("status") in ("pending", "in_progress")] """ c = c.replace("class Handlers:", fn + "\n\nclass Handlers:") if "task_incomplete" not in c: check = """ # Auto-continue detection if not llm_result.tool_calls_acc and llm_result.content: unfinished = _unfinished_plan(session) if unfinished: await session.send_event(Event(event_type="task_incomplete", data={"incomplete_plan": unfinished})) """ marker = " # -- End of turn --" if marker in c: c = c.replace(marker, check + "\n" + marker) with open(agent_py, 'w') as f: f.write(c) print("OK: agent_loop.py") if __name__ == "__main__": patch_events() patch_sse() patch_hook() patch_chat_input() patch_session() patch_agent() print("\nDONE: V8 - safe useEffect pattern")