from pydantic import BaseModel
from typing import List


# user registration parameters
class UserRegister(BaseModel):
    name: str
    userName: str
    email: str
    password: str


# user login parameters
class UserLogin(BaseModel):
    userName: str
    password: str


class ChatEntry(BaseModel):
    question: str
    answer: str


class ChatData(BaseModel):
    userID: str
    titleName: str
    titleUId: str
    dateRange: str
    chat: List[ChatEntry]


class UserIdModel(BaseModel):
    userID: str


class QueryModel(BaseModel):
    query: str
    logs: str


class DateRangeModel(BaseModel):
    duration: str