"""内驱力 API schema（database-schema §22；/drive 端点）。"""

from datetime import datetime

from pydantic import Field

from app.schemas.common import OrmModel


class CuriosityResponse(OrmModel):
    """好奇问题响应（related_entities 已从 JSON 文本解析）。"""

    id: str
    character_id: str
    cycle_id: str
    question: str
    source: str
    related_entities: list[str] = Field(default_factory=list)
    expected_value: float
    urgency: float
    risk_level: str
    created_at: datetime


class SubgoalResponse(OrmModel):
    """探索性次生目标响应。"""

    id: str
    character_id: str
    cycle_id: str
    parent_goal_id: str | None = None
    curiosity_question_id: str
    title: str
    description: str
    expected_information_gain: float
    interruption_cost: float
    risk_level: str
    status: str
    reviewed_at: datetime | None = None
    reviewed_by: str
    created_at: datetime