vendimia-compose/backend/schemas/users.py

from pydantic import BaseModel


class UserBase(BaseModel):
    username: str
    company_id: int

    class Config:
        orm_mode = True


class User(UserBase):
    password: str
    token: str

    class Config:
        orm_mode = True