megaproxy
d6118bac54
- Settings → AI: optional base URL and model name fields - Defaults to Anthropic/OpenAI public APIs when left blank - Custom URL enables Open WebUI, LM Studio, Ollama, and any OpenAI-compatible endpoint - Parse endpoint uses custom base URL and model if configured - Migration 0004: ai_base_url + ai_model columns on users - OpenAI provider label updated to "OpenAI-compatible" Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
38 lines
2.3 KiB
Python
38 lines
2.3 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import Boolean, DateTime, Integer, LargeBinary, String, Text
|
|
from sqlalchemy.dialects.postgresql import INET, UUID
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.db.base import Base
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
email: Mapped[str] = mapped_column(Text, unique=True, nullable=False, index=True)
|
|
password_hash: Mapped[str] = mapped_column(Text, nullable=False)
|
|
totp_secret_enc: Mapped[bytes | None] = mapped_column("totp_secret", type_=String, nullable=True)
|
|
totp_enabled: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)
|
|
totp_backup_codes_enc: Mapped[str | None] = mapped_column("totp_backup_codes", Text, nullable=True)
|
|
display_name: Mapped[str] = mapped_column(Text, nullable=False)
|
|
base_currency: Mapped[str] = mapped_column(String(10), default="GBP", nullable=False)
|
|
theme: Mapped[str] = mapped_column(String(20), default="dark", nullable=False)
|
|
locale: Mapped[str] = mapped_column(String(20), default="en-GB", nullable=False)
|
|
failed_login_attempts: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
|
|
locked_until: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
last_login_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
last_login_ip: Mapped[str | None] = mapped_column(INET, nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
|
|
deleted_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
|
|
ai_provider: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
ai_api_key_enc: Mapped[bytes | None] = mapped_column(LargeBinary, nullable=True)
|
|
ai_base_url: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
ai_model: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
accounts: Mapped[list["Account"]] = relationship(back_populates="user", lazy="noload") # type: ignore[name-defined]
|
|
sessions: Mapped[list["Session"]] = relationship(back_populates="user", lazy="noload") # type: ignore[name-defined]
|