patches: agent_loop detection + auto-approve toggle

patch_agent_backend.py

@@ -0,0 +1,89 @@
+"""Patch: agent_loop.py task detection + auto-approve for free models."""
+
+import os
+import ast
+
+AGENT_LOOP = "/app/agent/core/agent_loop.py"
+AGENT_FILE = "/app/backend/routes/agent.py"
+SESSION_FILE = "/app/agent/core/session.py"
+
+# ===== PATCH 1: agent_loop.py - detect model stopped mid-task =====
+with open(AGENT_LOOP) as f:
+    c = f.read()
+
+# Add helper function before class Handlers
+if "_unfinished_plan" not in c:
+    fn = (
+        "\n\ndef _unfinished_plan(s):\n"
+        "    p = getattr(s, 'current_plan', None) or []\n"
+        "    return [it for it in p if it.get('status') in ('pending', 'in_progress')]\n\n"
+    )
+    c = c.replace("class Handlers:", fn + "class Handlers:")
+    print("OK: Added _unfinished_plan()")
+
+# Add detection: if model produced no tool calls but content still exists,
+# and plan items are still pending/in_progress, send session_update event
+if "ac_plan" not in c:
+    check = (
+        "\n            # Auto-continue: detect model stopped mid-task\n"
+        "            if not llm_result.tool_calls_acc and llm_result.content:\n"
+        "                unfinished = _unfinished_plan(session)\n"
+        "                if unfinished:\n"
+        "                    await session.send_event(Event(event_type='session_update', data={\n"
+        "                        'ac_plan': unfinished,\n"
+        "                    }))\n"
+    )
+    c = c.replace("  # -- End of turn --", check + "  # -- End of turn --")
+    print("OK: Added mid-task detection")
+
+try:
+    ast.parse(c)
+    with open(AGENT_LOOP, "w") as f:
+        f.write(c)
+    print("OK: agent_loop.py patched")
+except SyntaxError as e:
+    print(f"FAIL: agent_loop.py syntax: {e}")
+
+# ===== PATCH 2: Enable auto_approval for ALL sessions by default =====
+# So free models don't block on user approval
+with open(SESSION_FILE) as f:
+    c = f.read()
+
+if "AUTO_APPROVE" not in c:
+    # Find where auto_approval is set on the session object
+    # Patch: when creating a new session, set auto_approval.enabled = True
+    auto_patch = (
+        "\n        # === PATCH: Auto-approve for free models ===\n"
+        "        self.auto_approval = {\n"
+        '            "enabled": True,\n'
+        '            "cost_cap_usd": None,\n'
+        '            "estimated_spend_usd": 0.0,\n'
+        '            "remaining_usd": None,\n'
+        "        }\n"
+    )
+    
+    # Find session init method
+    init_match = "class Session:"
+    if init_match in c:
+        # Insert after the class definition, find __init__
+        init_found = "def __init__"
+        if init_found in c:
+            # Find first line inside __init__ that's an assignment
+            init_start = c.find(init_found)
+            body_start = c.find("\n", init_start) + 1
+            body_start = c.find("        ", body_start)  # first indented line
+            c = c[:body_start] + auto_patch + c[body_start:]
+            print("OK: Added auto_approve default")
+    
+    try:
+        ast.parse(c)
+        with open(SESSION_FILE, "w") as f:
+            f.write(c)
+        print("OK: session.py patched")
+    except SyntaxError as e:
+        print(f"FAIL: session.py syntax: {e}")
+        # Don't block on this patch
+else:
+    print("OK: session.py already has auto_approve")
+
+print("DONE: Backend patches applied")