WIN-JHFT4D3SIVT\caoxiaozhu
5667190abe
- Phase 1: Infrastructure (state, prompts, registry) - Phase 2: Execution engine (AI adapters, security classifier, executors) - Phase 3: Agent integration (graph nodes, routing) - Phase 4: Streaming interaction (PTY terminal, WebSocket) - Phase 5: Frontend integration (Vue components)
134 lines
4.4 KiB
Python
134 lines
4.4 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime, timezone
|
|
from enum import Enum
|
|
from typing import Any, Literal
|
|
from uuid import uuid4
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
TaskLifecycleStatus = Literal["pending", "in_progress", "completed", "failed", "blocked"]
|
|
VerificationStatus = Literal["passed", "failed", "skipped"]
|
|
TaskResultStatus = Literal["completed", "failed", "blocked", "passed", "skipped"]
|
|
InterruptStatus = Literal["requested", "acknowledged", "resolved"]
|
|
BudgetMode = Literal["direct", "collaboration"]
|
|
|
|
|
|
class CodeProviderType(str, Enum):
|
|
CLAUDE = "claude"
|
|
GEMINI = "gemini"
|
|
CODEX = "codex"
|
|
OPENCODE = "opencode"
|
|
|
|
|
|
class RiskLevelType(str, Enum):
|
|
LOW = "low"
|
|
HIGH = "high"
|
|
|
|
|
|
class InterruptRecord(BaseModel):
|
|
interrupt_id: str
|
|
reason: str
|
|
status: InterruptStatus = "requested"
|
|
requested_by: str | None = None
|
|
source_event_id: str | None = None
|
|
requested_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
|
payload: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
|
|
class RecoveryRecord(BaseModel):
|
|
recovery_id: str
|
|
source_interrupt_id: str | None = None
|
|
strategy: str | None = None
|
|
resumed_from_task_id: str | None = None
|
|
resumed_from_thread_id: str | None = None
|
|
recovered_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
|
payload: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
|
|
class CollaborationBudget(BaseModel):
|
|
mode: BudgetMode = "direct"
|
|
max_parallel_tasks: int | None = None
|
|
remaining_parallel_tasks: int | None = None
|
|
max_tool_calls: int | None = None
|
|
remaining_tool_calls: int | None = None
|
|
max_iterations: int | None = None
|
|
remaining_iterations: int | None = None
|
|
escalation_threshold: int | None = None
|
|
metadata: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
|
|
class AgentTask(BaseModel):
|
|
task_id: str
|
|
title: str
|
|
status: TaskLifecycleStatus = "pending"
|
|
owner_agent_id: str | None = None
|
|
role: str | None = None
|
|
goal: str | None = None
|
|
parent_task_id: str | None = None
|
|
child_task_ids: list[str] = Field(default_factory=list)
|
|
thread_id: str | None = None
|
|
message_id: str | None = None
|
|
message_index: int | None = None
|
|
expected_evidence: list[dict[str, Any]] = Field(default_factory=list)
|
|
evidence: list[dict[str, Any]] = Field(default_factory=list)
|
|
interrupt_records: list[InterruptRecord | dict[str, Any]] = Field(default_factory=list)
|
|
recovery_records: list[RecoveryRecord | dict[str, Any]] = Field(default_factory=list)
|
|
collaboration_budget: CollaborationBudget | dict[str, Any] | None = None
|
|
result_summary: str | None = None
|
|
created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
|
updated_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class TaskResult(BaseModel):
|
|
task_id: str
|
|
status: TaskResultStatus
|
|
summary: str | None = None
|
|
evidence: list[dict[str, Any]] = Field(default_factory=list)
|
|
owner_agent_id: str | None = None
|
|
parent_task_id: str | None = None
|
|
child_task_ids: list[str] = Field(default_factory=list)
|
|
thread_id: str | None = None
|
|
message_id: str | None = None
|
|
message_index: int | None = None
|
|
interrupt_records: list[InterruptRecord | dict[str, Any]] = Field(default_factory=list)
|
|
recovery_records: list[RecoveryRecord | dict[str, Any]] = Field(default_factory=list)
|
|
budget_snapshot: CollaborationBudget | dict[str, Any] | None = None
|
|
next_action: str | None = None
|
|
output_data: dict[str, Any] | None = None
|
|
|
|
|
|
class CodeTaskType(str, Enum):
|
|
DEMO = "demo"
|
|
PROJECT = "project"
|
|
MODIFICATION = "modification"
|
|
|
|
|
|
class CodeTask(BaseModel):
|
|
"""代码任务请求模型"""
|
|
|
|
task_id: str = Field(default_factory=lambda: str(uuid4()))
|
|
task_type: CodeTaskType
|
|
ai_provider: CodeProviderType
|
|
sandbox_mode: bool = False
|
|
workspace_path: str | None = None
|
|
user_prompt: str
|
|
parent_task_id: str | None = None
|
|
thread_id: str | None = None
|
|
message_id: str | None = None
|
|
created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class CodeExecutionResultSchema(BaseModel):
|
|
"""代码执行结果模型 (API 响应用)"""
|
|
|
|
success: bool
|
|
message: str
|
|
files_created: list[str] = Field(default_factory=list)
|
|
output: str = ""
|
|
error: str | None = None
|
|
exit_code: int = 0
|
|
execution_time: float | None = None
|
|
sandbox_session_id: str | None = None
|