Spaces:
Sleeping
Sleeping
File size: 799 Bytes
e34edc7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
"""
Pydantic models for the ingestion pipeline API
"""
from typing import List, Dict, Any, Optional
from pydantic import BaseModel
# Response models (same as main system)
class UploadResponse(BaseModel):
job_id: str
status: str
total_files: Optional[int] = None
class JobStatusResponse(BaseModel):
job_id: str
status: str
total: int
completed: int
progress: float
last_error: Optional[str] = None
created_at: float
class HealthResponse(BaseModel):
ok: bool
mongodb_connected: bool
service: str = "ingestion_pipeline"
class FileResponse(BaseModel):
filename: str
summary: str
class FilesListResponse(BaseModel):
files: List[FileResponse]
filenames: List[str]
class ChunksResponse(BaseModel):
chunks: List[Dict[str, Any]]
|