"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""

from __future__ import annotations
from .toolcallstatus import ToolCallStatus
from openrouter.types import BaseModel, UNSET_SENTINEL
from pydantic import model_serializer
from typing import List, Literal, Optional
from typing_extensions import NotRequired, TypedDict


OutputWebSearchServerToolItemTypeURL = Literal["url",]


class OutputWebSearchServerToolItemSourceTypedDict(TypedDict):
    type: OutputWebSearchServerToolItemTypeURL
    url: str


class OutputWebSearchServerToolItemSource(BaseModel):
    type: OutputWebSearchServerToolItemTypeURL

    url: str


OutputWebSearchServerToolItemTypeSearch = Literal["search",]


class OutputWebSearchServerToolItemActionTypedDict(TypedDict):
    r"""The search action performed, matching OpenAI web_search_call.action shape. Includes the query the model issued and optional source URLs returned by the search provider."""

    query: str
    type: OutputWebSearchServerToolItemTypeSearch
    sources: NotRequired[List[OutputWebSearchServerToolItemSourceTypedDict]]


class OutputWebSearchServerToolItemAction(BaseModel):
    r"""The search action performed, matching OpenAI web_search_call.action shape. Includes the query the model issued and optional source URLs returned by the search provider."""

    query: str

    type: OutputWebSearchServerToolItemTypeSearch

    sources: Optional[List[OutputWebSearchServerToolItemSource]] = None

    @model_serializer(mode="wrap")
    def serialize_model(self, handler):
        optional_fields = set(["sources"])
        serialized = handler(self)
        m = {}

        for n, f in type(self).model_fields.items():
            k = f.alias or n
            val = serialized.get(k, serialized.get(n))

            if val != UNSET_SENTINEL:
                if val is not None or k not in optional_fields:
                    m[k] = val

        return m


OutputWebSearchServerToolItemTypeOpenrouterWebSearch = Literal["openrouter:web_search",]


class OutputWebSearchServerToolItemTypedDict(TypedDict):
    r"""An openrouter:web_search server tool output item"""

    status: ToolCallStatus
    type: OutputWebSearchServerToolItemTypeOpenrouterWebSearch
    action: NotRequired[OutputWebSearchServerToolItemActionTypedDict]
    r"""The search action performed, matching OpenAI web_search_call.action shape. Includes the query the model issued and optional source URLs returned by the search provider."""
    id: NotRequired[str]


class OutputWebSearchServerToolItem(BaseModel):
    r"""An openrouter:web_search server tool output item"""

    status: ToolCallStatus

    type: OutputWebSearchServerToolItemTypeOpenrouterWebSearch

    action: Optional[OutputWebSearchServerToolItemAction] = None
    r"""The search action performed, matching OpenAI web_search_call.action shape. Includes the query the model issued and optional source URLs returned by the search provider."""

    id: Optional[str] = None

    @model_serializer(mode="wrap")
    def serialize_model(self, handler):
        optional_fields = set(["action", "id"])
        serialized = handler(self)
        m = {}

        for n, f in type(self).model_fields.items():
            k = f.alias or n
            val = serialized.get(k, serialized.get(n))

            if val != UNSET_SENTINEL:
                if val is not None or k not in optional_fields:
                    m[k] = val

        return m