dograh/api/schemas/user_configuration.py

from datetime import datetime

from pydantic import BaseModel

from api.services.configuration.registry import (
    EmbeddingsConfig,
    LLMConfig,
    STTConfig,
    TTSConfig,
)


class UserConfiguration(BaseModel):
    llm: LLMConfig | None = None
    stt: STTConfig | None = None
    tts: TTSConfig | None = None
    embeddings: EmbeddingsConfig | None = None
    test_phone_number: str | None = None
    timezone: str | None = None
    last_validated_at: datetime | None = None