File size: 373 Bytes
a122f91
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
from pydantic import BaseModel, Field

class UserRequest(BaseModel):
    question: str = Field(..., description="The user's question or request.")
class OutputResponse(BaseModel):
    answer: str = Field(..., description="The answer generated by the model.")
    justification: str = Field(
        ..., description="Why this query is relevant to the user's request."
    )