File size: 1,100 Bytes
8ba64a4 |
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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
from typing import Any, Dict, Optional
from pydantic import BaseModel
class SubmitResponse(BaseModel):
task_id: str
class ResultResponse(BaseModel):
request_meta: Optional[Dict[str, str]] = None
task_id: str
result: dict
status_code: int
detail: str
attempt: int
class HealthCheckResponse(BaseModel):
status: str
class APIResponse(BaseModel):
detail: str
data: Dict[str, Any]
attempts: int
class APIErrorResponse(BaseModel):
detail: str
HEALTH_CHECK_RESPONSES = {}
SUBMIT_EXTRACT_RESPONSES = {
400: {
"model": APIErrorResponse,
},
500: {"model": APIErrorResponse},
}
SUBMIT_FOLLOW_RESPONSES = {
400: {
"model": APIErrorResponse,
},
500: {"model": APIErrorResponse},
}
RESULT_RESPONSES = {
400: {
"model": APIErrorResponse,
},
404: {
"model": APIErrorResponse,
},
500: {"model": APIErrorResponse},
}
RESPONSES = {
400: {
"model": APIErrorResponse,
},
404: {
"model": APIErrorResponse,
},
500: {"model": APIErrorResponse},
}
|