deer-flow/backend/src/config/model_config.py

from pydantic import BaseModel, ConfigDict, Field


class ModelConfig(BaseModel):
    """Config section for a model"""

    name: str = Field(..., description="Unique name for the model")
    display_name: str | None = Field(
        ..., default_factory=lambda: None, description="Display name for the model"
    )
    description: str | None = Field(
        ..., default_factory=lambda: None, description="Description for the model"
    )
    use: str = Field(
        ...,
        description="Class path of the model provider(e.g. langchain_openai.ChatOpenAI)",
    )
    model: str = Field(..., description="Model name")
    model_config = ConfigDict(extra="allow")
    supports_thinking: bool = Field(
        default_factory=lambda: False, description="Whether the model supports thinking"
    )
    when_thinking_enabled: dict | None = Field(
        default_factory=lambda: None,
        description="Extra settings to be passed to the model when thinking is enabled",
    )