Upload patch_agent_loop.py

patch_agent_loop.py

@@ -0,0 +1,95 @@
+"""Patch agent_loop.py - detect task incomplete and send event."""
+
+AGENT_LOOP = "/app/agent/core/agent_loop.py"
+
+def patch():
+    import os
+    if not os.path.exists(AGENT_LOOP):
+        print(f"SKIP: {AGENT_LOOP} not found")
+        return
+        
+    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
+    """
+    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)
+
+
+def _unfinished_plan_items(session: Session) -> list[dict[str, str]]:
+    """Helper to get unfinished plan items."""
+    plan = getattr(session, "current_plan", None) or []
+    return [item for item in plan if item.get("status") in ("pending", "in_progress")]
+
+'''
+    
+    # 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
+    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(
+                        Event(
+                            event_type="task_incomplete",
+                            data={
+                                "incomplete_plan": unfinished,
+                                "message": "Model phản hồi đã dừng nhưng nhiệm vụ chưa hoàn thành."
+                            }
+                        )
+                    )
+'''
+    
+    # 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
+        insert_marker = "  # -- End of turn --"
+        if insert_marker in content:
+            content = content.replace(
+                insert_marker,
+                check_call + "\n" + insert_marker
+            )
+        else:
+            # Try another location - after llm_result processing
+            result_marker = "final_response = llm_result.content or None"
+            if result_marker in content:
+                content = content.replace(
+                    result_marker,
+                    result_marker + "\n" + check_call
+                )
+    
+    with open(AGENT_LOOP, "w", encoding="utf-8") as f:
+        f.write(content)
+    print(f"OK: Patched {AGENT_LOOP}")
+
+if __name__ == "__main__":
+    patch()