"""设置 API schema（生成偏好）。

前端只见生成参数偏好；模型网关状态由神识承接（用户面见神识阁）。
"""

from pydantic import BaseModel, ConfigDict, Field


class GenerationPreferencesResponse(BaseModel):
    """生成参数偏好（业务层，不含密钥）。"""

    model_config = ConfigDict(
        json_schema_extra={
            "examples": [
                {
                    "default_model_hint": "default-chat",
                    "temperature": 0.7,
                    "top_p": 1.0,
                    "max_output_tokens": 4096,
                    "streaming": True,
                }
            ]
        }
    )

    default_model_hint: str = Field(description="默认对话逻辑模型名。")
    temperature: float = Field(description="采样温度。")
    top_p: float = Field(description="核采样概率。")
    max_output_tokens: int = Field(description="最大输出 token 数。")
    streaming: bool = Field(description="是否启用流式输出。")


class UpdateGenerationPreferencesRequest(BaseModel):
    """更新生成偏好；至少传一个字段。"""

    default_model_hint: str | None = Field(
        default=None, max_length=120, description="默认对话逻辑模型名。"
    )
    temperature: float | None = Field(default=None, ge=0.0, le=2.0, description="采样温度。")
    top_p: float | None = Field(default=None, ge=0.0, le=1.0, description="核采样概率。")
    max_output_tokens: int | None = Field(
        default=None, ge=1, le=200000, description="最大输出 token 数。"
    )
    streaming: bool | None = Field(default=None, description="是否启用流式输出。")