app/models/admin.py

"""Admin API models."""

from typing import List, Optional

from pydantic import BaseModel


class SessionInfo(BaseModel):
    """Session information."""

    id: str
    path: str
    created_at: float
    size: int


class SessionList(BaseModel):
    """List of sessions."""

    sessions: List[SessionInfo]


class ConfigUpdate(BaseModel):
    """Configuration update model."""

    openai_api_key: Optional[str] = None
    aider_model: Optional[str] = None
    temp_dir_prefix: Optional[str] = None


class ConfigResponse(BaseModel):
    """Configuration response model."""

    openai_api_key: Optional[str]
    aider_model: str
    temp_dir_prefix: str