Update backend/agent.py

backend/agent.py

@@ -9,12 +9,12 @@ class CodingAgent:
         self.model = model
         self.status = "idle"
         self.logs = []
-        self.proposed_changes = []  # list of {file, search, replace}
+        self.proposed_changes = []
 
     async def plan_task(self, prompt: str) -> List[dict]:
         """Generate a plan with file edits (diffs) but do not apply them."""
         self.status = "planning"
-        system = """You are a coding agent. For the given task, you must output a JSON array of file changes.
+        system = """You are a coding agent. For the given task, output a JSON array of file changes.
         Each change is an object with:
         - "filepath": relative path
         - "action": "create" or "edit"
@@ -27,7 +27,6 @@ class CodingAgent:
         ]
         response = await self._generate(messages)
         try:
-            # Extract JSON array from response (may be surrounded by markdown)
             json_match = re.search(r"\[.*\]", response, re.DOTALL)
             if json_match:
                 changes = json.loads(json_match.group())
@@ -64,6 +63,24 @@ class CodingAgent:
                     self.logs.append(f"Error applying change {idx}: {e}")
         self.status = "completed"
 
+    async def execute_task(self, prompt: str, auto_approve: bool = False):
+        """Old execute_task interface; kept for backward compatibility."""
+        changes = await self.plan_task(prompt)
+        if auto_approve:
+            await self.apply_changes(list(range(len(changes))))
+        return "\n".join(self.logs)
+
+    async def execute_task_stream(self, prompt: str, auto_approve: bool = False):
+        """Stream wrapper for the old agent run endpoint."""
+        yield {"type": "status", "content": "Agent planning..."}
+        changes = await self.plan_task(prompt)
+        if auto_approve:
+            await self.apply_changes(list(range(len(changes))))
+            yield {"type": "result", "content": "Changes applied."}
+        else:
+            yield {"type": "plan", "changes": changes}
+        yield {"type": "done", "content": ""}
+
     async def fix_error(self, file_path: str, error_output: str):
         p = self.project_path / file_path
         code = p.read_text() if p.exists() else ""
@@ -78,15 +95,4 @@ class CodingAgent:
         full = ""
         async for token in route_chat(self.model, messages):
             full += token
-        return full
-
-    async def execute_task_stream(self, prompt: str, auto_approve: bool = False):
-        """Stream results for the old execute_task (we'll keep for backward compat)."""
-        yield {"type": "status", "content": "Agent planning..."}
-        changes = await self.plan_task(prompt)
-        if auto_approve:
-            await self.apply_changes(list(range(len(changes))))
-            yield {"type": "result", "content": "Changes applied."}
-        else:
-            yield {"type": "plan", "changes": changes}
-        yield {"type": "done", "content": ""}
+        return full