brief-rags-bench/app/models/settings.py

31 lines
711 B
Python

"""User settings Pydantic models."""
from pydantic import BaseModel
class EnvironmentSettings(BaseModel):
"""Settings for a specific environment (IFT/PSI/PROD)."""
apiMode: str = "bench"
bearerToken: str = ""
systemPlatform: str = ""
systemPlatformUser: str = ""
platformUserId: str = ""
platformId: str = ""
withClassify: bool = False
resetSessionMode: bool = True
class UserSettings(BaseModel):
"""User settings for all environments."""
user_id: str
settings: dict[str, EnvironmentSettings] # ift, psi, prod
updated_at: str
class UserSettingsUpdate(BaseModel):
"""Update user settings request."""
settings: dict[str, EnvironmentSettings]