wahnsinn vibe

backend/apps/ai_admin/__init__.py

@@ -0,0 +1,72 @@
+from fastapi import APIRouter, Depends, HTTPException
+from pydantic import BaseModel
+from sqlalchemy.orm import Session
+
+from core.db import get_db
+from core.security import require_admin
+
+from apps.ai_core.models import AIAuditLog
+from apps.ai_core.tools import get_tool, validate_args
+
+from .planner import build_plan
+from .tool_defs import register_all
+
+router = APIRouter(dependencies=[Depends(require_admin)])
+
+
+class PlanIn(BaseModel):
+    prompt: str
+
+
+class ExecuteCardIn(BaseModel):
+    tool: str
+    args: dict = {}
+
+
+class ExecuteIn(BaseModel):
+    cards: list[ExecuteCardIn]
+
+
+@router.post("/plan")
+def plan_endpoint(body: PlanIn):
+    if not body.prompt.strip():
+        raise HTTPException(400, "Empty prompt")
+    cards = build_plan(body.prompt)
+    return {"cards": cards}
+
+
+@router.post("/execute")
+def execute_endpoint(
+    body: ExecuteIn,
+    claims: dict = Depends(require_admin),
+    db: Session = Depends(get_db),
+):
+    user_id = int(claims["sub"])
+    results = []
+    for card in body.cards:
+        spec = get_tool(card.tool)
+        if not spec:
+            results.append({"tool": card.tool, "ok": False, "error": "unknown tool"})
+            db.add(AIAuditLog(user_id=user_id, tool=card.tool, args=card.args, result={"error": "unknown tool"}, ok=False))
+            db.commit()
+            continue
+        missing = validate_args(spec, card.args)
+        if missing:
+            results.append({"tool": card.tool, "ok": False, "error": f"missing: {missing}"})
+            db.add(AIAuditLog(user_id=user_id, tool=card.tool, args=card.args, result={"missing": missing}, ok=False))
+            db.commit()
+            continue
+        try:
+            res = spec.handler(card.args, db)
+            results.append({"tool": card.tool, "ok": True, "result": res})
+            db.add(AIAuditLog(user_id=user_id, tool=card.tool, args=card.args, result=res, ok=True))
+            db.commit()
+        except Exception as e:  # noqa: BLE001
+            results.append({"tool": card.tool, "ok": False, "error": str(e)})
+            db.add(AIAuditLog(user_id=user_id, tool=card.tool, args=card.args, result={"error": str(e)}, ok=False))
+            db.commit()
+    return {"results": results}
+
+
+def on_load() -> None:
+    register_all()