feat: add Goal/Task data models, service layer, and API routes (Phase 1)
Main Agent 数字员工工厂基础设施： - 新增 Goal 和 Task SQLAlchemy 数据模型 - Agent 模型新增 agent_type / input_schema / output_schema - Execution 模型新增 goal_id 关联 - 新增 Goal/Task CRUD 服务层（含依赖检查、任务树、进度计算） - 新增 /api/v1/goals (9端点) + /api/v1/tasks (8端点) - 数据库迁移 013_add_goals_tasks Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
renjianbo
240
backend/app/api/tasks.py
Normal file
240
backend/app/api/tasks.py
Normal file
@@ -0,0 +1,240 @@
|
||||
"""
|
||||
Task API — 任务管理接口
|
||||
"""
|
||||
from fastapi import APIRouter, Depends, Query
|
||||
from sqlalchemy.orm import Session
|
||||
from pydantic import BaseModel, Field
|
||||
from typing import List, Optional, Dict, Any
|
||||
from datetime import datetime
|
||||
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 import goal_service
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
router = APIRouter(
|
||||
prefix="/api/v1/tasks",
|
||||
tags=["tasks"],
|
||||
responses={
|
||||
401: {"description": "未授权"},
|
||||
404: {"description": "资源不存在"},
|
||||
400: {"description": "请求参数错误"},
|
||||
},
|
||||
)
|
||||
|
||||
|
||||
# ──────────────────────────── Schemas ────────────────────────────
|
||||
|
||||
class TaskCreate(BaseModel):
|
||||
goal_id: str
|
||||
title: str
|
||||
description: str = ""
|
||||
priority: int = Field(default=5, ge=1, le=10)
|
||||
parent_task_id: Optional[str] = None
|
||||
depends_on: Optional[List[str]] = None
|
||||
assigned_agent_id: Optional[str] = None
|
||||
assigned_agent_name: Optional[str] = None
|
||||
task_config: Optional[Dict[str, Any]] = None
|
||||
deadline: Optional[datetime] = None
|
||||
requires_approval: bool = False
|
||||
approver_id: Optional[str] = None
|
||||
|
||||
|
||||
class TaskUpdate(BaseModel):
|
||||
title: Optional[str] = None
|
||||
description: Optional[str] = None
|
||||
status: Optional[str] = None
|
||||
priority: Optional[int] = Field(default=None, ge=1, le=10)
|
||||
task_config: Optional[Dict[str, Any]] = None
|
||||
depends_on: Optional[List[str]] = None
|
||||
assigned_agent_id: Optional[str] = None
|
||||
assigned_agent_name: Optional[str] = None
|
||||
result: Optional[Dict[str, Any]] = None
|
||||
error_message: Optional[str] = None
|
||||
execution_id: Optional[str] = None
|
||||
deadline: Optional[datetime] = None
|
||||
|
||||
|
||||
class TaskResponse(BaseModel):
|
||||
id: str
|
||||
goal_id: str
|
||||
title: str
|
||||
description: Optional[str]
|
||||
status: str
|
||||
priority: int
|
||||
task_config: Optional[Any]
|
||||
parent_task_id: Optional[str]
|
||||
depends_on: Optional[Any]
|
||||
result: Optional[Any]
|
||||
error_message: Optional[str]
|
||||
execution_id: Optional[str]
|
||||
assigned_agent_id: Optional[str]
|
||||
assigned_agent_name: Optional[str]
|
||||
requires_approval: bool
|
||||
approver_id: Optional[str]
|
||||
approval_status: Optional[str]
|
||||
started_at: Optional[datetime]
|
||||
completed_at: Optional[datetime]
|
||||
deadline: Optional[datetime]
|
||||
created_at: datetime
|
||||
updated_at: datetime
|
||||
|
||||
class Config:
|
||||
from_attributes = True
|
||||
|
||||
|
||||
class TaskDependencyCheck(BaseModel):
|
||||
task_id: str
|
||||
dependencies_met: bool
|
||||
pending_dependencies: List[str] = []
|
||||
|
||||
|
||||
# ──────────────────────────── Endpoints ────────────────────────────
|
||||
|
||||
@router.post("", response_model=TaskResponse, status_code=201)
|
||||
def create_task(
|
||||
data: TaskCreate,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""创建新任务"""
|
||||
return goal_service.create_task(
|
||||
db=db,
|
||||
goal_id=data.goal_id,
|
||||
title=data.title,
|
||||
description=data.description,
|
||||
priority=data.priority,
|
||||
parent_task_id=data.parent_task_id,
|
||||
depends_on=data.depends_on,
|
||||
assigned_agent_id=data.assigned_agent_id,
|
||||
assigned_agent_name=data.assigned_agent_name,
|
||||
task_config=data.task_config,
|
||||
deadline=data.deadline,
|
||||
requires_approval=data.requires_approval,
|
||||
approver_id=data.approver_id,
|
||||
)
|
||||
|
||||
|
||||
@router.get("", response_model=List[TaskResponse])
|
||||
def list_tasks(
|
||||
goal_id: Optional[str] = None,
|
||||
status: Optional[str] = None,
|
||||
assigned_agent_id: Optional[str] = None,
|
||||
parent_task_id: Optional[str] = None,
|
||||
skip: int = Query(default=0, ge=0),
|
||||
limit: int = Query(default=50, ge=1, le=200),
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""列出任务"""
|
||||
return goal_service.list_tasks(
|
||||
db=db,
|
||||
goal_id=goal_id,
|
||||
status=status,
|
||||
assigned_agent_id=assigned_agent_id,
|
||||
parent_task_id=parent_task_id,
|
||||
skip=skip,
|
||||
limit=limit,
|
||||
)
|
||||
|
||||
|
||||
@router.get("/{task_id}", response_model=TaskResponse)
|
||||
def get_task(
|
||||
task_id: str,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""获取任务详情"""
|
||||
return goal_service.get_task(db, task_id)
|
||||
|
||||
|
||||
@router.put("/{task_id}", response_model=TaskResponse)
|
||||
def update_task(
|
||||
task_id: str,
|
||||
data: TaskUpdate,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""更新任务"""
|
||||
return goal_service.update_task(
|
||||
db=db,
|
||||
task_id=task_id,
|
||||
title=data.title,
|
||||
description=data.description,
|
||||
status=data.status,
|
||||
priority=data.priority,
|
||||
task_config=data.task_config,
|
||||
depends_on=data.depends_on,
|
||||
assigned_agent_id=data.assigned_agent_id,
|
||||
assigned_agent_name=data.assigned_agent_name,
|
||||
result=data.result,
|
||||
error_message=data.error_message,
|
||||
execution_id=data.execution_id,
|
||||
deadline=data.deadline,
|
||||
)
|
||||
|
||||
|
||||
@router.delete("/{task_id}", status_code=204)
|
||||
def delete_task(
|
||||
task_id: str,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""删除任务"""
|
||||
goal_service.delete_task(db, task_id)
|
||||
return None
|
||||
|
||||
|
||||
@router.get("/{task_id}/check-dependencies", response_model=TaskDependencyCheck)
|
||||
def check_task_dependencies(
|
||||
task_id: str,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""检查任务的前置依赖是否满足"""
|
||||
met = goal_service.get_task_dependencies_met(db, task_id)
|
||||
task = goal_service.get_task(db, task_id)
|
||||
pending = []
|
||||
if not met:
|
||||
for dep_id in (task.depends_on or []):
|
||||
from app.models.task import Task
|
||||
dep = db.query(Task).filter(Task.id == dep_id).first()
|
||||
if dep and dep.status != "completed":
|
||||
pending.append(dep_id)
|
||||
return TaskDependencyCheck(
|
||||
task_id=task_id,
|
||||
dependencies_met=met,
|
||||
pending_dependencies=pending,
|
||||
)
|
||||
|
||||
|
||||
@router.post("/{task_id}/approve", response_model=TaskResponse)
|
||||
def approve_task(
|
||||
task_id: str,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""审批通过任务"""
|
||||
return goal_service.update_task(
|
||||
db=db,
|
||||
task_id=task_id,
|
||||
status="in_progress",
|
||||
)
|
||||
|
||||
|
||||
@router.post("/{task_id}/reject", response_model=TaskResponse)
|
||||
def reject_task(
|
||||
task_id: str,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""审批驳回任务"""
|
||||
return goal_service.update_task(
|
||||
db=db,
|
||||
task_id=task_id,
|
||||
status="failed",
|
||||
error_message="审批驳回",
|
||||
)
|
||||
Reference in New Issue
Block a user