26 lines
538 B
Python
26 lines
538 B
Python
from pydantic import BaseModel
|
|
from typing import Optional
|
|
|
|
|
|
class QuestionRequest(BaseModel):
|
|
"""Request model for questions."""
|
|
question: str
|
|
mode: str = "mix"
|
|
response_type: str = "Multiple Paragraphs"
|
|
|
|
|
|
class QuestionResponse(BaseModel):
|
|
"""Response model for questions."""
|
|
question: str
|
|
answer: str
|
|
|
|
|
|
class HealthResponse(BaseModel):
|
|
"""Response model for health check."""
|
|
status: str
|
|
rag_initialized: bool
|
|
|
|
|
|
class ErrorResponse(BaseModel):
|
|
"""Response model for errors."""
|
|
detail: str |