from datetime import datetime
from typing import Literal

from pydantic import BaseModel


class BaseWebhookTransport(BaseModel):
    method: Literal['webhook']
    callback: str


class BaseWebsocketTransport(BaseModel):
    method: Literal['websocket']
    session_id: str
    connected_at: datetime


class SubscribeWebhookTransport(BaseWebhookTransport):
    secret: str


Any = BaseWebhookTransport | BaseWebsocketTransport