diff --git a/datalab_sdk/client.py b/datalab_sdk/client.py index 7b86994..b22dfd6 100644 --- a/datalab_sdk/client.py +++ b/datalab_sdk/client.py @@ -270,7 +270,9 @@ async def convert( images=result_data.get("images"), metadata=result_data.get("metadata"), error=result_data.get("error"), + error_in=result_data.get("error_in"), page_count=result_data.get("page_count"), + total_pages=result_data.get("total_pages"), status=result_data.get("status", "complete"), ) diff --git a/datalab_sdk/models.py b/datalab_sdk/models.py index c1aeba9..7f0ca34 100644 --- a/datalab_sdk/models.py +++ b/datalab_sdk/models.py @@ -72,7 +72,9 @@ class ConversionResult: images: Optional[Dict[str, str]] = None metadata: Optional[Dict[str, Any]] = None error: Optional[str] = None - page_count: Optional[int] = None + error_in: Optional[str] = None # Where the error occurred (e.g., "VALIDATION") + page_count: Optional[int] = None # Number of pages processed + total_pages: Optional[int] = None # Total number of pages in the document status: str = "complete" def save_output(