from enum import StrEnum

from pydantic import BaseModel, ConfigDict, Field


class WriteFileTransport(StrEnum):
    """How a model supplies the body of a ``write_file`` call."""

    NATIVE = "native"
    BUFFERED_GATEWAY = "buffered_gateway"
    ARTIFACT_RENDERER = "artifact_renderer"
    # Kept for existing deployments that already opted into the first bridge
    # implementation. New buffered-gateway configurations should use
    # ``buffered_gateway``; unlike the legacy free-form body bridge it keeps
    # the original writer contract and enforces a typed transaction boundary.
    TEXT_BRIDGE = "text_bridge"


class ToolCallTransport(StrEnum):
    """How provider tool calls reach DeerFlow.

    ``deepseek_dsml`` is an application-side compatibility fallback for
    OpenAI-compatible gateways that leave DeepSeek V4/V4.1 DSML in assistant
    ``content`` instead of returning standard ``message.tool_calls``.
    """

    NATIVE = "native"
    DEEPSEEK_DSML = "deepseek_dsml"


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")
    max_input_tokens: int | None = Field(
        default=None,
        ge=1,
        description="Maximum input context tokens DeerFlow may send to the model",
    )
    model_config = ConfigDict(extra="allow")
    use_responses_api: bool | None = Field(
        default=None,
        description="Whether to route OpenAI ChatOpenAI calls through the /v1/responses API",
    )
    output_version: str | None = Field(
        default=None,
        description="Structured output version for OpenAI responses content, e.g. responses/v1",
    )
    write_file_transport: WriteFileTransport = Field(
        default=WriteFileTransport.NATIVE,
        description=(
            "How DeerFlow receives write_file content. Use buffered_gateway to keep the "
            "original write_file contract while obtaining its body from a second ordinary-text "
            "stream when an OpenAI-compatible gateway buffers tool arguments. "
            "artifact_renderer is the earlier internal-handshake compatibility mode. "
            "text_bridge remains available only for backward compatibility."
        ),
    )
    tool_call_transport: ToolCallTransport = Field(
        default=ToolCallTransport.NATIVE,
        description=(
            "How DeerFlow receives provider tool calls. Use deepseek_dsml only "
            "for a DeepSeek V4/V4.1 OpenAI-compatible endpoint that leaks raw "
            "DSML through assistant content instead of message.tool_calls."
        ),
    )
    supports_thinking: bool = Field(default_factory=lambda: False, description="Whether the model supports thinking")
    supports_reasoning_effort: bool = Field(
        default_factory=lambda: False,
        description=(
            "Legacy parser compatibility only. Gateway reasoning-effort capability is stored in the "
            "model_capabilities database table."
        ),
    )
    when_thinking_enabled: dict | None = Field(
        default_factory=lambda: None,
        description="Extra settings to be passed to the model when thinking is enabled",
    )
    when_thinking_disabled: dict | None = Field(
        default_factory=lambda: None,
        description="Extra settings to be passed to the model when thinking is disabled",
    )
    supports_vision: bool = Field(default_factory=lambda: False, description="Whether the model supports vision/image inputs")
    thinking: dict | None = Field(
        default_factory=lambda: None,
        description=(
            "Thinking settings for the model. If provided, these settings will be passed to the model when thinking is enabled. "
            "This is a shortcut for `when_thinking_enabled` and will be merged with `when_thinking_enabled` if both are provided."
        ),
    )