| from pydantic import BaseModel, HttpUrl, Field |
| from typing import List |
|
|
| class QueryRequest(BaseModel): |
| documents: HttpUrl |
| questions: List[str] |
|
|
| class QueryResponse(BaseModel): |
| answers: List[str] |
|
|
| class Question(BaseModel): |
| question: str |
|
|
| class FinalAnswer(BaseModel): |
| answer: str = Field(description="The complete and detailed answer to the user's question, based strictly on the provided context.") |
|
|
| class FinalAnswerList(BaseModel): |
| answers: List[FinalAnswer] = Field(description="A list of final answers for each corresponding input question.") |
|
|
| class GeneratedQueriesForEachQuestion(BaseModel): |
| queries: List[str] = Field(description="A list of 3 distinct, self-contained search queries based on the original question.") |
|
|
| class GeneratedQueries(BaseModel): |
| lst: List[GeneratedQueriesForEachQuestion] |
