feat: AI Worker 平台 MVP v1.0.0 - 多租户/项目管理/AI Worker/任务编排/HITL审核/成本治理

2026-08-12 13:30:47 +08:00
parent bf5a7622d2
commit 9c5d39ef25
1 changed files with 57 additions and 0 deletions
@@ -0,0 +1,57 @@
+"""Task model."""
+from sqlalchemy import (
+    Column, String, Integer, Text, DateTime, ForeignKey, Boolean, JSON
+)
+from app.models.base import Base, TimestampMixin, TenantMixin
+
+
+class Task(Base, TimestampMixin, TenantMixin):
+    """A task is a work unit within a project.
+
+    Can be assigned to a human user or an AI Worker.
+    Status: pending -> assigned -> in_progress -> review -> done -> rejected -> cancelled
+    Priority: low, medium, high, urgent
+    """
+    __tablename__ = "tasks"
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    project_id = Column(Integer, ForeignKey("projects.id"), nullable=False, index=True)
+
+    title = Column(String(500), nullable=False)
+    description = Column(Text, default="")
+    status = Column(String(50), nullable=False, default="pending")
+    priority = Column(String(20), nullable=False, default="medium")
+
+    # Assignment: assignee is a User (human or AI-linked)
+    assignee_id = Column(Integer, ForeignKey("users.id"), nullable=True)
+
+    # If assigned to AI, which worker profile
+    worker_id = Column(Integer, ForeignKey("ai_workers.id"), nullable=True)
+
+    # Task type: single_call, agent_loop, multi_agent (MVP: single_call + agent_loop)
+    task_type = Column(String(50), default="single_call")
+
+    # Task input (prompt, context, parameters)
+    input_data = Column(Text, default="")  # JSON string with prompt/context
+    output_data = Column(Text, nullable=True)  # JSON string with result
+
+    # HITL: does this task require human review?
+    requires_review = Column(Boolean, default=True)
+    review_status = Column(String(50), nullable=True)  # pending, approved, rejected
+
+    # Cost tracking
+    token_cost_estimate = Column(Integer, nullable=True)
+
+    # Dependencies (list of task IDs that must complete first)
+    depends_on = Column(Text, default="[]")  # JSON array
+
+    # Timeline
+    due_date = Column(DateTime, nullable=True)
+    started_at = Column(DateTime, nullable=True)
+    completed_at = Column(DateTime, nullable=True)
+
+    # Execution metadata (retries, duration, etc.)
+    execution_meta = Column(Text, default="{}")  # JSON
+
+    def __repr__(self):
+        return f"<Task {self.title} [{self.status}]>"