28 lines
497 B
Python
Executable File
28 lines
497 B
Python
Executable File
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] |