selfrelease
fad458b2a7
- UIUX 文档：填充 19 个缺口（多主体画像/健康度/AI+看板/增长域/洞察域/创始人端/OODA/助推/商密） - UIUX 文档：插入 6 个新章节（十四~十九），旧章节重编号为二十~三十一，更新目录和交叉引用 - 作业指导书 x5：导航改为 6 域分组，新增 Context Bar/工作模式/Insight Rail/决策线程/多工作区等 UI 概念 - 新建 docs/2-task-uiux.md：50 个代码落地开发任务，按 P0-P6 分优先级 + 8 Sprint 规划 - 后端/前端：大量新增模型、路由、组件（来自之前 Phase 开发）
59 lines
2.7 KiB
Python
59 lines
2.7 KiB
Python
"""Pre-mortem + Red Team 路由。"""
|
|
|
|
from fastapi import APIRouter, Depends, Query
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.database import get_db
|
|
from app.core.dependencies import get_current_user
|
|
from app.models.pre_mortem import PreMortemRecord, RedTeamRecord
|
|
from app.models.user import User
|
|
from app.schemas.common import ApiResponse, success
|
|
from app.services.pre_mortem_agent import run_pre_mortem
|
|
from app.services.red_team_agent import run_red_team
|
|
|
|
router_pre_mortem = APIRouter(prefix="/pre-mortems", tags=["pre-mortems"])
|
|
router_red_team = APIRouter(prefix="/red-teams", tags=["red-teams"])
|
|
|
|
|
|
@router_pre_mortem.get("", response_model=ApiResponse[list])
|
|
async def list_pre_mortems(company_id: str | None = Query(default=None), db: AsyncSession = Depends(get_db), user: User = Depends(get_current_user)):
|
|
"""获取 Pre-mortem 列表。"""
|
|
query = select(PreMortemRecord)
|
|
if company_id:
|
|
query = query.where(PreMortemRecord.company_id == company_id)
|
|
result = await db.execute(query.order_by(PreMortemRecord.created_at.desc()))
|
|
items = result.scalars().all()
|
|
return success(data=[
|
|
{"id": str(i.id), "company_id": str(i.company_id), "decision_context": i.decision_context, "failure_paths": i.failure_paths, "risk_checklist": i.risk_checklist, "mitigations": i.mitigations}
|
|
for i in items
|
|
])
|
|
|
|
|
|
@router_pre_mortem.post("/run", response_model=ApiResponse[dict])
|
|
async def run_pre_mortem_analysis(req: dict, user: User = Depends(get_current_user)):
|
|
"""AI Pre-mortem 失败推演。"""
|
|
result = await run_pre_mortem(req.get("decision_context", ""))
|
|
return success(data=result)
|
|
|
|
|
|
@router_red_team.get("", response_model=ApiResponse[list])
|
|
async def list_red_teams(company_id: str | None = Query(default=None), db: AsyncSession = Depends(get_db), user: User = Depends(get_current_user)):
|
|
"""获取 Red Team 列表。"""
|
|
query = select(RedTeamRecord)
|
|
if company_id:
|
|
query = query.where(RedTeamRecord.company_id == company_id)
|
|
result = await db.execute(query.order_by(RedTeamRecord.created_at.desc()))
|
|
items = result.scalars().all()
|
|
return success(data=[
|
|
{"id": str(i.id), "company_id": str(i.company_id), "perspective": i.perspective, "analysis": i.analysis, "vulnerabilities": i.vulnerabilities, "counterarguments": i.counterarguments}
|
|
for i in items
|
|
])
|
|
|
|
|
|
@router_red_team.post("/run", response_model=ApiResponse[dict])
|
|
async def run_red_team_analysis(req: dict, user: User = Depends(get_current_user)):
|
|
"""AI Red Team 对抗分析。"""
|
|
result = await run_red_team(req.get("company_context", ""), req.get("perspective", "competitor"))
|
|
return success(data=result)
|