from pydantic import BaseModel, Field

class UserRegister(BaseModel):
    username: str
    password: str

class QueryInput(BaseModel):
    query: str 

class TokenResponse(BaseModel):
    access_token: str
    token_type: str
    refresh_token: str

class RefreshTokenRequest(BaseModel):
    refresh_token: str