retab 0.0.66__py3-none-any.whl → 0.0.68__py3-none-any.whl

{retab-0.0.66.dist-info → retab-0.0.68.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: retab
-Version: 0.0.66
+Version: 0.0.68
 Summary: Retab official python library
 Home-page: https://github.com/retab-dev/retab
 Author: Retab
@@ -29,6 +29,8 @@ Requires-Dist: email_validator
 Requires-Dist: python-stdnum
 Requires-Dist: nanoid
 Requires-Dist: openai
+Requires-Dist: anthropic
+Requires-Dist: google-genai
 Requires-Dist: tiktoken
 Requires-Dist: truststore
 Dynamic: author

{retab-0.0.66.dist-info → retab-0.0.68.dist-info}/RECORD

@@ -1,8 +1,8 @@
-retab/__init__.py,sha256=CfWHrp7ci-WqA6dZXzP7gFkpXAVro4nxeCOiCV83BIc,130
+retab/__init__.py,sha256=s4GawWTRBYz4VY-CyAV5-ZdFtdw8V5oopGIYm9GgdSo,188
 retab/_resource.py,sha256=JfAU4UTa05ugWfbrpO7fsVr_pFewht99NkoIfK6kBQM,577
 retab/client copy.py,sha256=RqKYBvLB8Bp6QpS3DodOXffWQFdr36bobQRSU09vES0,27883
-retab/client.py,sha256=JisT6NGHgPEEOqIPygeC_tR0wNfZM_xq1sRzhbaW8Nw,27031
-retab/generate_types.py,sha256=Yj0ZgYftTYWQeX-RPW02ROfvanymDOpLpi_P1AVYOLA,8365
+retab/client.py,sha256=wOiTAzGw6Hi7FVbrs-TwSuPR6tmgOLMYjY1eBgrgBBk,29929
+retab/generate_types.py,sha256=cUu1IX65uU__MHivmEb_PZtzAi8DYsvppZvcY30hj90,8425
 retab/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 retab/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 retab/resources/models.py,sha256=4WidFBnTGZEA65DSn2pLP2SRnCVXkMTw7o_m8xVCFC4,2469
@@ -10,14 +10,12 @@ retab/resources/schemas.py,sha256=qWkOAZrimtbTHzhDof3z4PG72F7MIPrb3A495f6mrHE,50
 retab/resources/documents/__init__.py,sha256=OjXmngFN0RKqO4SI-mJBNzr6Ex6rMxfq0DxaqzP0RQs,89
 retab/resources/documents/client.py,sha256=7-BaveXUlwZqtMTvM3dNiltZS83jJVXDRuyfFMg_eGw,37446
 retab/resources/projects/__init__.py,sha256=tPR3_3tr7bsoYd618qmGjnYN2R23PmF5oCFd7Z5_HGY,85
-retab/resources/projects/client.py,sha256=8MygpRCBWqK5IXEKr3vvdLZhs2Quy4MUuPx78m-sgcI,15945
-retab/resources/projects/documents.py,sha256=pUgANQbfm1W-iXsa_4ffRAr3kVNt7KAy8AKb09IMH2w,10098
-retab/resources/projects/iterations.py,sha256=I5caQgotAGFijojr6KzOE3NWC8CTIwUSOzN2aivTnX4,16945
+retab/resources/projects/client.py,sha256=-T5iSRp-jODNVg-cgQiQ3S6D53cwCTwNQsI0Rr9vqsU,14654
 retab/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 retab/types/browser_canvas.py,sha256=LdW1SnaWb4WdFgFBCo8p9k78K2fmzV5YtloQtIR7hk0,82
 retab/types/chat.py,sha256=x9VbtPMa4w6Gc0HrFC3ILl6cCnfEn5ytDnwJtZmlcys,1436
 retab/types/inference_settings.py,sha256=co0C5ynGm7z7Bvi7gP3CFwhkRJUd-bMNBVbavFu7Iuo,939
-retab/types/mime.py,sha256=6TytTxtimrTCRutCFDMEnwFoV4EOQlwNASLefmmfbNo,10998
+retab/types/mime.py,sha256=1k3REXEVJ4dlFErs27Xx05sIWHWr3KjLzuKSDXEMNQQ,11973
 retab/types/modality.py,sha256=4B8LctdUBZVgIjtS2FjrJpljn2Eyse0XE1bpFsGb9O4,131
 retab/types/pagination.py,sha256=-XrKILKX_5isTHTfShLiK3Kwp21Y6Wqy0Jci8lIFQig,109
 retab/types/standards.py,sha256=7aGtuvzzkKidvqY8JB2Cjfn43V80FeKwrTtp162kjKc,1477
@@ -26,19 +24,15 @@ retab/types/documents/correct_orientation.py,sha256=e-ivsslI6L6Gl0YkcslXw_DH620x
 retab/types/documents/create_messages.py,sha256=fibThBlJg1iKQfVkvIVL1QqZxrraIl9qeO0lJHLo_nQ,7509
 retab/types/documents/extract.py,sha256=KeEDnbvXTtgamrGvpBINsg5MJAue-Xf_lNi1a71M7oE,20067
 retab/types/documents/parse.py,sha256=ekXnjnFEx7adZ8dJKivo5QRtYnKBEUFzJiIM9WY-LPY,1454
-retab/types/projects/Untitled-2.py,sha256=bKFsPNfY6RloyZ1A1tX-3pOzvZ2OY_S-wle43BuAn98,1205610
-retab/types/projects/__init__.py,sha256=oIOBN_6-bpFxeZpvyPdXP-LHDNERreSS7TvjdNkVmIA,920
-retab/types/projects/documents.py,sha256=0N9HJIghObrHX6GWgFP4w7VlvCaYZqBnExH2ULRLo8I,2029
-retab/types/projects/iterations.py,sha256=2V4WFrLcuQklCuMx0rD0LUtYTZOvCaCnTnu2O5MH3QU,5782
+retab/types/projects/__init__.py,sha256=I7P_dems5_LOLgYQ-4Bzt9B6P6jRlQwP-D_9GxRDhVk,155
 retab/types/projects/metrics.py,sha256=J8aZdVbqlszfxosAZyTB7l6lp9WgdL5QgLMlLrckN7k,1946
-retab/types/projects/model.py,sha256=Lwn1KUzjBZl-kqugWRcmcUbEoRPxnH0Mn4ycrEJ-7i4,2380
+retab/types/projects/model.py,sha256=0He-lMiT4jX1SwukNowq41LFPeozWDnTVsbjlmCNWJw,4374
 retab/types/projects/predictions.py,sha256=GeQX-nCmbzKL50aaSFlSyn0sgnbzKserPK6DSQT8ACk,1114
-retab/types/projects/v2.py,sha256=8fNV5lH4Zn5D97AQK5pZVLjYFi-2MSBN9Su2eNoGVpM,6610
 retab/types/schemas/__init__.py,sha256=9ODWiC_4pUVKxoIKglYZjvRjRyd1ZCVxG8GBdQgHNbU,57
-retab/types/schemas/chat.py,sha256=U-zYbHtJ6HHxU2t56sfLkhLttoJpng0mjVN2tlccnig,21225
+retab/types/schemas/chat.py,sha256=ppTidxsNslTKE5aBva04i9IxeARMqYpXYLjtR7V6pBc,21219
 retab/types/schemas/generate.py,sha256=DN4pMJihoADyAZgA2RW7BddFzhbNOoaEnfEjjjUyMKM,1007
 retab/types/schemas/layout.py,sha256=JLPwQGIWfPBoe1Y5r-MhiNDJigzZ-yKZnVGgox0uqMk,1487
-retab/types/schemas/model.py,sha256=lOaSEabXQUDJO7bQc_c4fDZp2o86zbJSRF2QfINM02s,69594
+retab/types/schemas/model.py,sha256=MwiSrck_5NCDvxjDSqgiDtmenGowXnfoOZ7KTINKW34,69588
 retab/types/schemas/templates.py,sha256=XihWTHi6t_6QjxN07n_1dee5KdhHiuoHAYfmKwI7gQg,1708
 retab/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 retab/utils/display.py,sha256=ZFPbiBnwEWGR-suS8e9Xilz9OqyYRDwsKYWfbFSJPJM,18868
@@ -48,7 +42,7 @@ retab/utils/mime.py,sha256=mTP_lqSPttOP5DYJxopiWaeFXrUCPjhwd7y53nCVGO4,6189
 retab/utils/stream_context_managers.py,sha256=gI1gVQSj3nWz6Mvjz7Ix5AiY0g6vSL-c2tPfuP04izo,2314
 retab/utils/usage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 retab/utils/usage/json_schema.py,sha256=kRbL5E5OStlhlNlBXlxHNwaVHKd6MVhyqIb0y4aj8JA,84322
-retab-0.0.66.dist-info/METADATA,sha256=vGQMlgjUh8uANjfSUz041GSKUV00pmAjtjdXvqbeQ8A,4479
-retab-0.0.66.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-retab-0.0.66.dist-info/top_level.txt,sha256=waQR0EGdhLIQtztoE3AXg7ik5ONQ9q_bsKVpyFuJdq0,6
-retab-0.0.66.dist-info/RECORD,,
+retab-0.0.68.dist-info/METADATA,sha256=ZqKChL45ZWbRxN1B7QMPk-VWdz0qKezYOQOgPRhDFc0,4532
+retab-0.0.68.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+retab-0.0.68.dist-info/top_level.txt,sha256=waQR0EGdhLIQtztoE3AXg7ik5ONQ9q_bsKVpyFuJdq0,6
+retab-0.0.68.dist-info/RECORD,,

retab/resources/projects/documents.py

@@ -1,257 +0,0 @@
-from io import IOBase
-from pathlib import Path
-from typing import Any, Dict, List, Union
-
-import PIL.Image
-from pydantic import HttpUrl
-
-from ..._resource import AsyncAPIResource, SyncAPIResource
-from ...utils.mime import prepare_mime_document
-from ...types.projects import DocumentItem, ProjectDocument, PatchProjectDocumentRequest
-from ...types.projects.predictions import PredictionMetadata
-from ...types.mime import MIMEData
-from ...types.standards import PreparedRequest, DeleteResponse
-from ...types.documents.extract import RetabParsedChatCompletion
-
-
-class DocumentsMixin:
-    def prepare_get(self, project_id: str, document_id: str) -> PreparedRequest:
-        return PreparedRequest(method="GET", url=f"/v1/projects/{project_id}/documents/{document_id}")
-
-    def prepare_create(self, project_id: str, document: MIMEData, annotation: dict[str, Any], annotation_metadata: dict[str, Any] | None = None, **extra_body: Any) -> PreparedRequest:
-        # Serialize the MIMEData
-        # Build known payload and merge extras
-        payload: dict[str, Any] = {
-            "mime_data": document,
-            "annotation": annotation,
-            "annotation_metadata": PredictionMetadata(**annotation_metadata) if annotation_metadata else None,
-        }
-        if extra_body:
-            payload.update(extra_body)
-        document_item = DocumentItem(**payload)
-        return PreparedRequest(method="POST", url=f"/v1/projects/{project_id}/documents", data=document_item.model_dump(mode="json"))
-
-    def prepare_list(self, project_id: str) -> PreparedRequest:
-        return PreparedRequest(method="GET", url=f"/v1/projects/{project_id}/documents")
-
-    def prepare_update(self, project_id: str, document_id: str, annotation: dict[str, Any], **extra_body: Any) -> PreparedRequest:
-        body: dict[str, Any] = {"annotation": annotation}
-        if extra_body:
-            body.update(extra_body)
-        update_request = PatchProjectDocumentRequest(**body)
-        return PreparedRequest(method="PATCH", url=f"/v1/projects/{project_id}/documents/{document_id}", data=update_request.model_dump(mode="json", exclude_unset=True))
-
-    def prepare_delete(self, project_id: str, document_id: str) -> PreparedRequest:
-        return PreparedRequest(method="DELETE", url=f"/v1/projects/{project_id}/documents/{document_id}")
-
-    def prepare_llm_annotate(self, project_id: str, document_id: str) -> PreparedRequest:
-        return PreparedRequest(method="POST", url=f"/v1/projects/{project_id}/documents/{document_id}/llm-annotate", data={"stream": False})
-
-
-class Documents(SyncAPIResource, DocumentsMixin):
-    """Documents API wrapper for projects"""
-
-    def create(
-        self,
-        project_id: str,
-        document: Union[Path, str, IOBase, MIMEData, PIL.Image.Image, HttpUrl],
-        annotation: Dict[str, Any],
-        annotation_metadata: Dict[str, Any] | None = None,
-        **extra_body: Any,
-    ) -> ProjectDocument:
-        """
-        Create a document for an project.
-
-        Args:
-            project_id: The ID of the project
-            document: The document to process. Can be:
-                - A file path (Path or str)
-                - A file-like object (IOBase)
-                - A MIMEData object
-                - A PIL Image object
-                - A URL (HttpUrl)
-            annotation: The ground truth for the document
-
-        Returns:
-            ProjectDocument: The created document
-        Raises:
-            HTTPException if the request fails
-        """
-        # Convert document to MIME data format
-        mime_document: MIMEData = prepare_mime_document(document)
-
-        # Let prepare_create handle the serialization
-        request = self.prepare_create(project_id, mime_document, annotation, annotation_metadata, **extra_body)
-        response = self._client._prepared_request(request)
-        return ProjectDocument(**response)
-
-    def list(self, project_id: str) -> List[ProjectDocument]:
-        """
-        List documents for an project.
-
-        Args:
-            project_id: The ID of the project
-
-        Returns:
-            List[ProjectDocument]: List of documents
-        Raises:
-            HTTPException if the request fails
-        """
-        request = self.prepare_list(project_id)
-        response = self._client._prepared_request(request)
-        return [ProjectDocument(**item) for item in response.get("data", [])]
-
-    def get(self, project_id: str, document_id: str) -> ProjectDocument:
-        """
-        Get a document by ID.
-
-        Args:
-            project_id: The ID of the project
-            document_id: The ID of the document
-
-        Returns:
-            ProjectDocument: The document
-        Raises:
-            HTTPException if the request fails
-        """
-        request = self.prepare_get(project_id, document_id)
-        response = self._client._prepared_request(request)
-        return ProjectDocument(**response)
-
-    def update(self, project_id: str, document_id: str, annotation: dict[str, Any], **extra_body: Any) -> ProjectDocument:
-        """
-        Update a document.
-
-        Args:
-            project_id: The ID of the project
-            document_id: The ID of the document
-            annotation: The ground truth for the document
-        Returns:
-            ProjectDocument: The updated document
-        Raises:
-            HTTPException if the request fails
-        """
-        request = self.prepare_update(project_id, document_id, annotation=annotation, **extra_body)
-        response = self._client._prepared_request(request)
-        return ProjectDocument(**response)
-
-    def delete(self, project_id: str, document_id: str) -> DeleteResponse:
-        """
-        Delete a document.
-
-        Args:
-            project_id: The ID of the project
-            document_id: The ID of the document
-
-        Returns:
-            DeleteResponse: The response containing success status and ID
-        Raises:
-            HTTPException if the request fails
-        """
-        request = self.prepare_delete(project_id, document_id)
-        return self._client._prepared_request(request)
-
-    def llm_annotate(self, project_id: str, document_id: str) -> RetabParsedChatCompletion:
-        """
-        Annotate a document with an LLM. This method updates the document (within the project) with the latest extraction.
-        """
-        request = self.prepare_llm_annotate(project_id, document_id)
-        response = self._client._prepared_request(request)
-        return RetabParsedChatCompletion(**response)
-
-
-class AsyncDocuments(AsyncAPIResource, DocumentsMixin):
-    """Async Documents API wrapper for projects"""
-
-    async def create(
-        self,
-        project_id: str,
-        document: Union[Path, str, IOBase, MIMEData, PIL.Image.Image, HttpUrl],
-        annotation: Dict[str, Any],
-        annotation_metadata: Dict[str, Any] | None = None,
-        **extra_body: Any,
-    ) -> ProjectDocument:
-        """
-        Create a document for an project.
-
-        Args:
-            project_id: The ID of the project
-            document: The document to process. Can be:
-                - A file path (Path or str)
-                - A file-like object (IOBase)
-                - A MIMEData object
-                - A PIL Image object
-                - A URL (HttpUrl)
-            annotation: The ground truth for the document
-            annotation_metadata: The metadata of the annotation
-        Returns:
-            ProjectDocument: The created document
-        Raises:
-            HTTPException if the request fails
-        """
-        # Convert document to MIME data format
-        mime_document: MIMEData = prepare_mime_document(document)
-
-        # Let prepare_create handle the serialization
-        request = self.prepare_create(project_id, mime_document, annotation, annotation_metadata, **extra_body)
-        response = await self._client._prepared_request(request)
-        return ProjectDocument(**response)
-
-    async def list(self, project_id: str) -> List[ProjectDocument]:
-        """
-        List documents for an project.
-
-        Args:
-            project_id: The ID of the project
-
-        Returns:
-            List[ProjectDocument]: List of documents
-        Raises:
-            HTTPException if the request fails
-        """
-        request = self.prepare_list(project_id)
-        response = await self._client._prepared_request(request)
-        return [ProjectDocument(**item) for item in response.get("data", [])]
-
-    async def update(self, project_id: str, document_id: str, annotation: dict[str, Any], **extra_body: Any) -> ProjectDocument:
-        """
-        Update a document.
-
-        Args:
-            project_id: The ID of the project
-            document_id: The ID of the document
-            annotation: The ground truth for the document
-
-        Returns:
-            ProjectDocument: The updated document
-        Raises:
-            HTTPException if the request fails
-        """
-        request = self.prepare_update(project_id, document_id, annotation, **extra_body)
-        response = await self._client._prepared_request(request)
-        return ProjectDocument(**response)
-
-    async def delete(self, project_id: str, document_id: str) -> DeleteResponse:
-        """
-        Delete a document.
-
-        Args:
-            project_id: The ID of the project
-            document_id: The ID of the document
-
-        Returns:
-            DeleteResponse: The response containing success status and ID
-        Raises:
-            HTTPException if the request fails
-        """
-        request = self.prepare_delete(project_id, document_id)
-        return await self._client._prepared_request(request)
-
-    async def llm_annotate(self, project_id: str, document_id: str) -> RetabParsedChatCompletion:
-        """
-        Annotate a document with an LLM.
-        This method updates the document (within the project) with the latest extraction.
-        """
-        request = self.prepare_llm_annotate(project_id, document_id)
-        response = await self._client._prepared_request(request)
-        return RetabParsedChatCompletion(**response)