mirror of
https://github.com/docling-project/docling-serve.git
synced 2025-11-29 08:33:50 +00:00
57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
import enum
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from docling.datamodel.document import ConversionStatus, ErrorItem
|
|
from docling.utils.profiling import ProfilingItem
|
|
from docling_jobkit.datamodel.result import ExportDocumentResponse
|
|
from docling_jobkit.datamodel.task_meta import TaskProcessingMeta
|
|
|
|
|
|
# Status
|
|
class HealthCheckResponse(BaseModel):
|
|
status: str = "ok"
|
|
|
|
|
|
class ClearResponse(BaseModel):
|
|
status: str = "ok"
|
|
|
|
|
|
class ConvertDocumentResponse(BaseModel):
|
|
document: ExportDocumentResponse
|
|
status: ConversionStatus
|
|
errors: list[ErrorItem] = []
|
|
processing_time: float
|
|
timings: dict[str, ProfilingItem] = {}
|
|
|
|
|
|
class PresignedUrlConvertDocumentResponse(BaseModel):
|
|
processing_time: float
|
|
num_converted: int
|
|
num_succeeded: int
|
|
num_failed: int
|
|
|
|
|
|
class ConvertDocumentErrorResponse(BaseModel):
|
|
status: ConversionStatus
|
|
|
|
|
|
class TaskStatusResponse(BaseModel):
|
|
task_id: str
|
|
task_status: str
|
|
task_position: Optional[int] = None
|
|
task_meta: Optional[TaskProcessingMeta] = None
|
|
|
|
|
|
class MessageKind(str, enum.Enum):
|
|
CONNECTION = "connection"
|
|
UPDATE = "update"
|
|
ERROR = "error"
|
|
|
|
|
|
class WebsocketMessage(BaseModel):
|
|
message: MessageKind
|
|
task: Optional[TaskStatusResponse] = None
|
|
error: Optional[str] = None
|