from datetime import datetime

from pydantic import BaseModel, ConfigDict


class AIGCMetadata(BaseModel):
    """生成内容来源标识。"""

    model_config = ConfigDict(strict=True, extra="forbid")

    Label: str
    ContentProducer: str
    ProduceID: str
    ReservedCode1: str
    ContentPropagator: str
    PropagateID: str
    ReservedCode2: str


class TextTrace(BaseModel):
    """生成文本及其来源标识。"""

    content_key: str
    text: str
    reasoning_content: str | None = None
    metadata: AIGCMetadataEnvelope
    model_id: str | None = None
    served_model: str | None = None
    generated_at: datetime | None = None