366 lines
11 KiB
Python
366 lines
11 KiB
Python
"""
|
||
Team API — 虚拟团队管理 + 项目执行
|
||
"""
|
||
from fastapi import APIRouter, Depends, Query, HTTPException
|
||
from fastapi.responses import StreamingResponse
|
||
from sqlalchemy.orm import Session
|
||
from pydantic import BaseModel, Field
|
||
from typing import List, Optional, Dict, Any
|
||
import json
|
||
import logging
|
||
|
||
from app.core.database import get_db
|
||
from app.api.auth import get_current_user
|
||
from app.models.user import User
|
||
from app.services.team_service import TeamService, get_preset_roles
|
||
from app.services.team_orchestrator import TeamOrchestrator
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
router = APIRouter(
|
||
prefix="/api/v1/teams",
|
||
tags=["teams"],
|
||
responses={
|
||
401: {"description": "未授权"},
|
||
404: {"description": "资源不存在"},
|
||
400: {"description": "请求参数错误"},
|
||
},
|
||
)
|
||
|
||
|
||
# ──────────────────────────── Schemas ────────────────────────────
|
||
|
||
class TeamCreate(BaseModel):
|
||
name: str
|
||
description: str = ""
|
||
workspace_id: Optional[str] = None
|
||
config: Optional[Dict[str, Any]] = None
|
||
|
||
|
||
class TeamUpdate(BaseModel):
|
||
name: Optional[str] = None
|
||
description: Optional[str] = None
|
||
config: Optional[Dict[str, Any]] = None
|
||
status: Optional[str] = None
|
||
|
||
|
||
class MemberAdd(BaseModel):
|
||
agent_id: str
|
||
role: str
|
||
position: int = 0
|
||
is_lead: bool = False
|
||
|
||
|
||
class ProjectExecute(BaseModel):
|
||
project_description: str = Field(..., min_length=1, description="项目描述")
|
||
auto_approve_files: bool = Field(default=True, description="是否自动批准文件写入（跳过确认弹窗）")
|
||
|
||
|
||
# ──────────────────────────── Endpoints ────────────────────────────
|
||
|
||
@router.get("/preset-roles")
|
||
def preset_roles():
|
||
"""获取预置角色定义（PM/设计师/开发/测试/DevOps）。"""
|
||
return {"roles": get_preset_roles()}
|
||
|
||
|
||
@router.post("")
|
||
def create_team(
|
||
body: TeamCreate,
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""创建团队。"""
|
||
svc = TeamService(db)
|
||
team = svc.create_team(
|
||
name=body.name,
|
||
user_id=current_user.id,
|
||
description=body.description,
|
||
workspace_id=body.workspace_id,
|
||
config=body.config,
|
||
)
|
||
return {"data": team.to_dict()}
|
||
|
||
|
||
@router.post("/template/software-company")
|
||
def create_software_company_template(
|
||
workspace_id: Optional[str] = Query(None),
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""一键创建「软件公司虚拟团队」模板。
|
||
|
||
自动创建 5 个角色 Agent（PM/设计师/开发/测试/DevOps）并组建团队。
|
||
若用户已有同名 Agent 则复用。
|
||
"""
|
||
svc = TeamService(db)
|
||
result = svc.create_software_company_template(
|
||
user_id=current_user.id,
|
||
workspace_id=workspace_id,
|
||
)
|
||
return {"data": result}
|
||
|
||
|
||
@router.post("/template/education-training")
|
||
def create_education_training_template(
|
||
workspace_id: Optional[str] = Query(None),
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""一键创建「教育培训团队」模板。
|
||
|
||
自动创建 4 个角色 Agent（课程设计/主讲讲师/作业助教/教务管理）并组建团队。
|
||
若用户已有同名 Agent 则复用。
|
||
"""
|
||
svc = TeamService(db)
|
||
result = svc.create_education_training_template(
|
||
user_id=current_user.id,
|
||
workspace_id=workspace_id,
|
||
)
|
||
return {"data": result}
|
||
|
||
|
||
@router.post("/template/platform-engineering")
|
||
def create_platform_engineering_template(
|
||
workspace_id: Optional[str] = Query(None),
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""一键创建「天工平台工程团队」模板。
|
||
|
||
自动创建 5 个角色 Agent（全栈开发/前端体验/平台运维/质量保障/产品负责人）并组建团队。
|
||
专用于天工智能体平台的自我维护与迭代升级。
|
||
若用户已有同名 Agent 则复用。
|
||
"""
|
||
svc = TeamService(db)
|
||
result = svc.create_platform_engineering_template(
|
||
user_id=current_user.id,
|
||
workspace_id=workspace_id,
|
||
)
|
||
return {"data": result}
|
||
|
||
|
||
@router.post("/template/tech-documentation")
|
||
def create_tech_doc_template(
|
||
workspace_id: Optional[str] = Query(None),
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""一键创建「技术文档团队」模板。
|
||
|
||
自动创建 5 个角色 Agent（文档架构师/技术写手/API文档专员/翻译审校/发布管理）并组建团队。
|
||
专用于软件项目的技术文档体系建设。
|
||
若用户已有同名 Agent 则复用。
|
||
"""
|
||
svc = TeamService(db)
|
||
result = svc.create_tech_doc_template(
|
||
user_id=current_user.id,
|
||
workspace_id=workspace_id,
|
||
)
|
||
return {"data": result}
|
||
|
||
|
||
@router.post("/template/health-management")
|
||
def create_health_management_template(
|
||
workspace_id: Optional[str] = Query(None),
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""一键创建「健康管理团队」模板。"""
|
||
svc = TeamService(db)
|
||
result = svc.create_health_management_template(
|
||
user_id=current_user.id, workspace_id=workspace_id,
|
||
)
|
||
return {"data": result}
|
||
|
||
|
||
@router.post("/template/medical-consultation")
|
||
def create_medical_consultation_template(
|
||
workspace_id: Optional[str] = Query(None),
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""一键创建「医疗咨询团队」模板。"""
|
||
svc = TeamService(db)
|
||
result = svc.create_medical_consultation_template(
|
||
user_id=current_user.id, workspace_id=workspace_id,
|
||
)
|
||
return {"data": result}
|
||
|
||
|
||
@router.post("/template/user-simulation-test")
|
||
def create_user_simulation_test_template(
|
||
workspace_id: Optional[str] = Query(None),
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""一键创建「系统应用测试团队」模板。
|
||
|
||
自动创建 5 个角色 Agent（测试规划师/功能测试员/体验审核员/边界探索员/性能评估员）并组建团队。
|
||
若用户已有同名 Agent 则复用。
|
||
"""
|
||
svc = TeamService(db)
|
||
result = svc.create_user_simulation_test_template(
|
||
user_id=current_user.id, workspace_id=workspace_id,
|
||
)
|
||
return {"data": result}
|
||
|
||
|
||
@router.get("")
|
||
def list_teams(
|
||
workspace_id: Optional[str] = Query(None),
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""列出当前用户/工作区的所有团队。"""
|
||
svc = TeamService(db)
|
||
teams = svc.list_teams(
|
||
user_id=current_user.id,
|
||
workspace_id=workspace_id,
|
||
)
|
||
return {"data": [t.to_dict() for t in teams]}
|
||
|
||
|
||
@router.get("/{team_id}")
|
||
def get_team(
|
||
team_id: str,
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""获取团队详情（含成员列表）。"""
|
||
svc = TeamService(db)
|
||
result = svc.get_team_with_members(team_id)
|
||
if not result:
|
||
raise HTTPException(status_code=404, detail="团队不存在")
|
||
return {"data": result}
|
||
|
||
|
||
@router.put("/{team_id}")
|
||
def update_team(
|
||
team_id: str,
|
||
body: TeamUpdate,
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""更新团队信息。"""
|
||
svc = TeamService(db)
|
||
team = svc.update_team(team_id, **body.model_dump(exclude_none=True))
|
||
if not team:
|
||
raise HTTPException(status_code=404, detail="团队不存在")
|
||
return {"data": team.to_dict()}
|
||
|
||
|
||
@router.delete("/{team_id}")
|
||
def delete_team(
|
||
team_id: str,
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""删除团队及其所有成员。"""
|
||
svc = TeamService(db)
|
||
ok = svc.delete_team(team_id)
|
||
if not ok:
|
||
raise HTTPException(status_code=404, detail="团队不存在")
|
||
return {"message": "团队已删除"}
|
||
|
||
|
||
@router.post("/{team_id}/members")
|
||
def add_member(
|
||
team_id: str,
|
||
body: MemberAdd,
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""为团队添加/替换角色成员。"""
|
||
svc = TeamService(db)
|
||
member = svc.add_member(
|
||
team_id=team_id,
|
||
agent_id=body.agent_id,
|
||
role=body.role,
|
||
position=body.position,
|
||
is_lead=body.is_lead,
|
||
)
|
||
if not member:
|
||
raise HTTPException(status_code=404, detail="团队不存在")
|
||
return {"data": member.to_dict()}
|
||
|
||
|
||
@router.delete("/{team_id}/members/{member_id}")
|
||
def remove_member(
|
||
team_id: str,
|
||
member_id: str,
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""从团队移除成员。"""
|
||
svc = TeamService(db)
|
||
ok = svc.remove_member(team_id, member_id)
|
||
if not ok:
|
||
raise HTTPException(status_code=404, detail="成员不存在")
|
||
return {"message": "成员已移除"}
|
||
|
||
|
||
@router.get("/{team_id}/members")
|
||
def list_members(
|
||
team_id: str,
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""列出团队成员。"""
|
||
svc = TeamService(db)
|
||
members = svc.get_members(team_id)
|
||
return {"data": [m.to_dict() for m in members]}
|
||
|
||
|
||
@router.post("/{team_id}/execute")
|
||
async def execute_project(
|
||
team_id: str,
|
||
body: ProjectExecute,
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""执行团队项目（非流式）。
|
||
|
||
PM 先分解计划，各角色按阶段顺序执行，QA 最终审查。
|
||
"""
|
||
orch = TeamOrchestrator(db, team_id, current_user.id)
|
||
try:
|
||
result = await orch.execute(body.project_description, auto_approve_files=body.auto_approve_files)
|
||
return {"data": result}
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
except Exception as e:
|
||
logger.error("团队执行失败: %s", e, exc_info=True)
|
||
raise HTTPException(status_code=500, detail=f"团队执行失败: {e}")
|
||
|
||
|
||
@router.post("/{team_id}/execute/stream")
|
||
async def execute_project_stream(
|
||
team_id: str,
|
||
body: ProjectExecute,
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user),
|
||
):
|
||
"""执行团队项目（SSE 流式）。"""
|
||
|
||
async def event_generator():
|
||
orch = TeamOrchestrator(db, team_id, current_user.id)
|
||
try:
|
||
async for event in orch.execute_stream(body.project_description, auto_approve_files=body.auto_approve_files):
|
||
yield f"data: {json.dumps(event, ensure_ascii=False)}\n\n"
|
||
except ValueError as e:
|
||
yield f"data: {json.dumps({'type': 'error', 'content': str(e)}, ensure_ascii=False)}\n\n"
|
||
except Exception as e:
|
||
logger.error("团队流式执行失败: %s", e, exc_info=True)
|
||
yield f"data: {json.dumps({'type': 'error', 'content': str(e)}, ensure_ascii=False)}\n\n"
|
||
|
||
return StreamingResponse(
|
||
event_generator(),
|
||
media_type="text/event-stream",
|
||
headers={
|
||
"Cache-Control": "no-cache",
|
||
"Connection": "keep-alive",
|
||
"X-Accel-Buffering": "no",
|
||
},
|
||
)
|