from pydantic import BaseModel
from typing import List


class Title(BaseModel):
    title: str

class Query(BaseModel):
    query: str

class ChatObject(BaseModel):
    title: str
    user_query: str
    model_key: str
    retrived_passage: str
    retrived_ref_ids: str
    model_answer: str
    status:str='success'
    prompt_type: str= "question-answer"


class LLMOutput(BaseModel):
    text : str
    source : List[str]

class LLMInput(BaseModel):
    query : str
    knowledge : List[dict]