"""模型能力矩阵相关 Pydantic 模型"""

from typing import Literal, Optional

from pydantic import BaseModel, Field


class ValidateMatchRequest(BaseModel):
    """模型-需求匹配校验请求"""

    truth_node_id: int = Field(..., description="宪章节点 ID")
    model_version_id: int = Field(..., description="模型版本 ID")
    project_id: Optional[int] = Field(default=None, description="项目 ID（用于交叉校验）")


class CostTrackingQuery(BaseModel):
    """成本跟踪查询参数"""

    period: str = Field(default="weekly", description="周期: daily/weekly/monthly/project")


class CostOptimizationQuery(BaseModel):
    """成本优化查询参数"""

    period: str = Field(default="weekly", description="周期: daily/weekly/monthly/project")


class LifecycleImpactRequest(BaseModel):
    """模型生命周期影响评估请求"""

    old_model_version_id: int = Field(..., description="旧模型版本 ID")
    new_model_version_id: int = Field(..., description="新模型版本 ID")


class GenerateMigrationTasksRequest(BaseModel):
    """生成迁移任务请求"""

    max_tasks: int = Field(default=5, ge=1, le=20, description="最多生成任务数")


class ExperimentSyncRequest(BaseModel):
    """实验结果同步请求"""

    experiment_id: int = Field(..., description="实验 ID")
    truth_node_id: int = Field(..., description="目标宪章 ID")
    project_id: Optional[int] = Field(default=None, description="项目 ID（用于交叉校验）")
    sync_type: Literal["result_sync", "metric_update", "conclusion_sync"] = Field(
        default="result_sync",
        description="同步类型",
    )
    auto_apply_threshold: float = Field(
        default=0.8, ge=0.0, le=1.0, description="自动应用阈值"
    )


class ExperimentSyncReviewRequest(BaseModel):
    """实验同步审核请求"""

    action: Literal["approve", "reject"] = Field(..., description="审核动作")
    review_notes: Optional[str] = Field(default=None, description="审核备注")


class ExperimentSyncRollbackRequest(BaseModel):
    """实验同步回滚请求"""

    rollback_notes: Optional[str] = Field(default=None, description="回滚备注")


class CostOptimizationAnalyzeRequest(BaseModel):
    """成本优化分析请求"""

    truth_node_id: int = Field(..., description="宪章 ID")
    project_id: Optional[int] = Field(default=None, description="项目 ID（用于交叉校验）")
    period: str = Field(default="weekly", description="周期: daily/weekly/monthly/project")


class ProjectTruthNodeContextItem(BaseModel):
    """项目上下文中的宪章项"""

    id: int
    title: str
    status: str
    node_type: str


class ProjectExperimentContextItem(BaseModel):
    """项目上下文中的实验项"""

    id: int
    title: str
    status: str
    target_metric: Optional[str] = None