rag / app /models.py
Kareman's picture
feat(ContextAI)
14faba3
raw
history blame contribute delete
480 Bytes
from sqlmodel import SQLModel, Field
from pydantic import BaseModel
# Database model
class User(SQLModel, table=True):
id: int | None = Field(default=None, primary_key=True)
username: str
hashed_password: str
# Pydantic models for API
class UserCreate(BaseModel):
username: str
password: str
class UserLogin(BaseModel):
username: str
password: str
class Token(BaseModel):
access_token: str
refresh_token: str
token_type: str = "bearer"