Update backend/main.py

backend/main.py

@@ -33,7 +33,7 @@ async def startup_index():
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    await refresh_model_cache()   # fetch all models on startup
+    await refresh_model_cache()
     await startup_index()
     yield
 
@@ -44,7 +44,7 @@ agents = {}
 STATIC_DIR = "static"
 
 class ChatRequest(BaseModel):
-    model: str = "groq-llama-3.1-8b-instant"
+    model: str = "groq-llama-3.3-70b-versatile"
     messages: List[Dict[str, str]]
     context_files: Optional[List[str]] = None
     project_path: str = "/app/sandbox"
@@ -75,7 +75,11 @@ class ReviewRequest(BaseModel):
     pr_diff: str
     project_path: str
 
-# ---------- API ----------
+class ApplyRequest(BaseModel):
+    task_id: str
+    approved_indices: List[int]
+
+# ---------- API Routes ----------
 @app.get("/api/models")
 async def get_models():
     return get_available_models_sync()
@@ -113,17 +117,26 @@ async def chat(req: ChatRequest):
             yield "data: [DONE]\n\n"
     return StreamingResponse(event_stream(), media_type="text/event-stream")
 
+# ---------- Agent endpoints with error handling ----------
 @app.post("/api/agent/run")
 async def run_agent(req: AgentTaskRequest, background_tasks: BackgroundTasks):
-    agent = CodingAgent(project_path=req.project_path, model=req.model)
-    task_id = str(uuid.uuid4())
-    agents[task_id] = agent
-    if req.background:
-        background_tasks.add_task(agent.run_task, req.prompt)
-        return {"task_id": task_id, "status": "started"}
-    else:
-        result = await agent.run_task(req.prompt)
-        return {"task_id": task_id, "status": agent.status, "logs": agent.logs, "pending_changes": result["pending_changes"]}
+    try:
+        agent = CodingAgent(project_path=req.project_path, model=req.model)
+        task_id = str(uuid.uuid4())
+        agents[task_id] = agent
+        if req.background:
+            background_tasks.add_task(agent.run_task, req.prompt)
+            return {"task_id": task_id, "status": "started"}
+        else:
+            result = await agent.run_task(req.prompt)
+            return {
+                "task_id": task_id,
+                "status": agent.status,
+                "logs": agent.logs,
+                "pending_changes": result["pending_changes"]
+            }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/api/agent/approve/{task_id}")
 async def approve_agent_changes(task_id: str):
@@ -132,20 +145,28 @@ async def approve_agent_changes(task_id: str):
         raise HTTPException(404, "Agent not found")
     if agent.status != "awaiting_approval":
         raise HTTPException(400, "Agent not awaiting approval")
-    await agent.approve_and_apply()
-    return {"status": agent.status, "logs": agent.logs}
+    try:
+        await agent.approve_and_apply()
+        return {"status": agent.status, "logs": agent.logs}
+    except Exception as e:
+        raise HTTPException(500, detail=str(e))
 
 @app.get("/api/agent/task/{task_id}")
 async def get_agent_status(task_id: str):
     agent = agents.get(task_id)
     if not agent:
         raise HTTPException(404, "Task not found")
-    return {"status": agent.status, "logs": agent.logs, "pending_changes": agent.pending_changes}
+    return {
+        "status": agent.status,
+        "logs": agent.logs,
+        "pending_changes": agent.pending_changes
+    }
 
 @app.get("/api/agent/tasks")
 async def list_agent_tasks():
     return {"tasks": list(agents.keys())}
 
+# ---------- Other endpoints ----------
 @app.post("/api/autocomplete")
 async def autocomplete(req: AutocompleteRequest):
     suggestions = await generate_autocomplete(req)
@@ -164,8 +185,10 @@ async def search_codebase(query: str, project_path: str = "/app/sandbox", top_k:
     if project_path not in indexers:
         CBI = get_codebase_index()
         idx = CBI(project_path)
-        try: idx.build_index()
-        except Exception as e: raise HTTPException(503, f"Indexing failed: {str(e)}")
+        try:
+            idx.build_index()
+        except Exception as e:
+            raise HTTPException(503, f"Indexing failed: {str(e)}")
         indexers[project_path] = idx
     idx = indexers[project_path]
     results = idx.search(query, top_k)
@@ -214,9 +237,15 @@ async def terminal_ws(websocket: WebSocket):
             cmd = await websocket.receive_text()
             if cmd.startswith("exec:"):
                 command = cmd[5:]
-                proc = await asyncio.create_subprocess_shell(command, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE, cwd="/app/sandbox")
+                proc = await asyncio.create_subprocess_shell(
+                    command,
+                    stdout=asyncio.subprocess.PIPE,
+                    stderr=asyncio.subprocess.PIPE,
+                    cwd="/app/sandbox"
+                )
                 stdout, stderr = await proc.communicate()
-                await websocket.send_text(stdout.decode() + stderr.decode())
+                output = stdout.decode() + stderr.decode()
+                await websocket.send_text(output)
             elif cmd == "ctrl-c" and proc:
                 proc.terminate()
     except WebSocketDisconnect:
@@ -227,10 +256,16 @@ async def list_files(project_path: str = "/app/sandbox"):
     def build_tree(path):
         tree = []
         for item in sorted(os.listdir(path)):
-            if item.startswith('.') or item == '__pycache__': continue
+            if item.startswith('.') or item == '__pycache__':
+                continue
             full = os.path.join(path, item)
-            node = {"name": item, "path": full.replace(project_path, "").lstrip("/"), "type": "folder" if os.path.isdir(full) else "file"}
-            if os.path.isdir(full): node["children"] = build_tree(full)
+            node = {
+                "name": item,
+                "path": full.replace(project_path, "").lstrip("/"),
+                "type": "folder" if os.path.isdir(full) else "file"
+            }
+            if os.path.isdir(full):
+                node["children"] = build_tree(full)
             tree.append(node)
         return tree
     try:
@@ -253,13 +288,17 @@ async def upload_file(file: UploadFile, project_path: str = "/app/sandbox"):
     dest.write_bytes(content)
     return {"filename": file.filename, "size": len(content)}
 
+# ---------- Serve frontend ----------
 @app.get("/{full_path:path}")
 async def serve_frontend(full_path: str):
-    if full_path.startswith("api"): raise HTTPException(404)
+    if full_path.startswith("api"):
+        raise HTTPException(404)
     file_path = os.path.join(STATIC_DIR, full_path)
-    if os.path.isfile(file_path): return FileResponse(file_path)
+    if os.path.isfile(file_path):
+        return FileResponse(file_path)
     index_path = os.path.join(STATIC_DIR, "index.html")
-    if os.path.isfile(index_path): return FileResponse(index_path)
+    if os.path.isfile(index_path):
+        return FileResponse(index_path)
     raise HTTPException(404, detail="Not found")
 
 if __name__ == "__main__":