|
|
from pydantic import BaseModel, Field, validator |
|
|
from typing import List, Dict, Any, Optional |
|
|
|
|
|
class QueryRequest(BaseModel): |
|
|
query: str = Field(..., min_length=1, max_length=1000, description="The query text to search for") |
|
|
|
|
|
@validator('query') |
|
|
def validate_query(cls, v): |
|
|
if not v.strip(): |
|
|
raise ValueError('Query cannot be empty or contain only whitespace') |
|
|
return v.strip() |
|
|
|
|
|
class SourceDocument(BaseModel): |
|
|
source: str |
|
|
text: str |
|
|
score: float |
|
|
|
|
|
class QueryResponse(BaseModel): |
|
|
answer: str |
|
|
source_documents: List[Dict[str, Any]] |
|
|
|
|
|
class ErrorResponse(BaseModel): |
|
|
error: str |
|
|
detail: str |
|
|
status_code: int |
|
|
timestamp: Optional[str] = None |
|
|
|
|
|
class UploadResponse(BaseModel): |
|
|
filename: str |
|
|
message: str |
|
|
num_chunks_stored: int |
|
|
|
