"""Analysis session Pydantic models."""

from typing import Any
from pydantic import BaseModel


class SessionCreate(BaseModel):
    """Create new analysis session."""

    environment: str
    api_mode: str
    request: list[Any]
    response: dict
    annotations: dict


class SessionResponse(BaseModel):
    """Analysis session response."""

    session_id: str
    user_id: str
    environment: str
    api_mode: str
    request: list[Any]
    response: dict
    annotations: dict
    created_at: str
    updated_at: str


class SessionListItem(BaseModel):
    """Session list item for listing sessions."""

    session_id: str
    environment: str
    created_at: str


class SessionList(BaseModel):
    """List of sessions with total count."""

    sessions: list[SessionListItem]
    total: int