Ayomide
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from typing import List, Optional, Dict, Any
|
|
from pydantic import BaseModel, Field
|
|
from ..models.document import EntityType, AccessLevel
|
|
|
|
|
|
class SearchRequest(BaseModel):
|
|
"""Search request schema."""
|
|
query: str = Field(..., description="Search query string", min_length=1)
|
|
limit: int = Field(10, ge=1, le=100, description="Maximum number of results")
|
|
entity_types: Optional[List[EntityType]] = Field(None, description="Filter by entity types")
|
|
similarity_threshold: Optional[float] = Field(None, ge=0.0, le=1.0, description="Minimum similarity score")
|
|
|
|
|
|
class DocumentUploadResponse(BaseModel):
|
|
"""Document upload response schema."""
|
|
document_id: str
|
|
filename: str
|
|
status: str
|
|
entities_found: int
|
|
access_level: str
|
|
message: str
|
|
|
|
|
|
class StatsResponse(BaseModel):
|
|
"""System statistics response schema."""
|
|
total_documents: int
|
|
document_types: Dict[str, int]
|
|
access_levels: Dict[str, int]
|
|
languages: Dict[str, int]
|
|
total_entities: int
|
|
average_entities_per_doc: float
|
|
vector_store_stats: Dict[str, Any]
|
|
system_status: str
|
|
|
|
|
|
class ErrorResponse(BaseModel):
|
|
"""Error response schema."""
|
|
error: str
|
|
detail: str
|
|
timestamp: float |