diff --git a/datalab_sdk/client.py b/datalab_sdk/client.py index f57e1bc..1883e58 100644 --- a/datalab_sdk/client.py +++ b/datalab_sdk/client.py @@ -426,6 +426,9 @@ def _build_conversion_result(self, result_data: Dict[str, Any], default_format: parse_quality_score=result_data.get("parse_quality_score"), runtime=result_data.get("runtime"), cost_breakdown=result_data.get("cost_breakdown"), + total_cost=result_data.get("total_cost"), + extraction_score_average=result_data.get("extraction_score_average"), + extraction_mode=result_data.get("extraction_mode"), evaluation=result_data.get("evaluation"), ) diff --git a/datalab_sdk/models.py b/datalab_sdk/models.py index 1757b7c..a64b032 100644 --- a/datalab_sdk/models.py +++ b/datalab_sdk/models.py @@ -15,6 +15,7 @@ class ProcessingOptions: max_pages: Optional[int] = None skip_cache: bool = False page_range: Optional[str] = None + processing_location: Optional[str] = None def to_form_data(self) -> Dict[str, Any]: """Convert to form data format for API requests""" @@ -201,6 +202,9 @@ class ConversionResult: parse_quality_score: Optional[float] = None runtime: Optional[float] = None cost_breakdown: Optional[Dict[str, Any]] = None + total_cost: Optional[int] = None # Total cost in credits + extraction_score_average: Optional[float] = None # Average extraction field score + extraction_mode: Optional[str] = None # Extraction mode used ("balanced" or "accurate") evaluation: Optional[Dict[str, Any]] = None # Evaluation results when run_eval=true def save_output( diff --git a/uv.lock b/uv.lock index 2145e97..aabfa7a 100644 --- a/uv.lock +++ b/uv.lock @@ -212,7 +212,7 @@ wheels = [ [[package]] name = "datalab-python-sdk" -version = "0.3.0" +version = "0.5.0" source = { editable = "." } dependencies = [ { name = "aiohttp" },