retab 0.0.36__py3-none-any.whl → 0.0.38__py3-none-any.whl

uiform/resources/documents/client.py

@@ -1,255 +0,0 @@
-from io import IOBase
-from pathlib import Path
-from typing import Any, Literal
-
-import PIL.Image
-from pydantic import HttpUrl
-
-from ..._resource import AsyncAPIResource, SyncAPIResource
-from ..._utils.mime import convert_mime_data_to_pil_image, prepare_mime_document
-from ...types.documents.create_messages import DocumentCreateMessageRequest, DocumentMessage, DocumentCreateInputRequest
-from ...types.mime import MIMEData
-from ...types.modalities import Modality
-from ...types.standards import PreparedRequest
-from .extractions import AsyncExtractions, Extractions
-from ..._utils.json_schema import load_json_schema
-
-
-class BaseDocumentsMixin:
-    def _prepare_create_messages(
-        self,
-        document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl,
-        modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
-        idempotency_key: str | None = None,
-    ) -> PreparedRequest:
-        mime_document = prepare_mime_document(document)
-        data: dict[str, Any] = {
-            "document": mime_document.model_dump(),
-            "modality": modality,
-        }
-        if image_resolution_dpi:
-            data["image_resolution_dpi"] = image_resolution_dpi
-        if browser_canvas:
-            data["browser_canvas"] = browser_canvas
-
-        loading_request = DocumentCreateMessageRequest.model_validate(data)
-        return PreparedRequest(method="POST", url="/v1/documents/create_messages", data=loading_request.model_dump(), idempotency_key=idempotency_key)
-
-    def _prepare_create_inputs(
-        self,
-        document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl,
-        json_schema: dict[str, Any] | Path | str,
-        modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
-        idempotency_key: str | None = None,
-    ) -> PreparedRequest:
-        mime_document = prepare_mime_document(document)
-        loaded_schema = load_json_schema(json_schema)
-
-        data: dict[str, Any] = {
-            "document": mime_document.model_dump(),
-            "modality": modality,
-            "json_schema": loaded_schema,
-        }
-        if image_resolution_dpi:
-            data["image_resolution_dpi"] = image_resolution_dpi
-        if browser_canvas:
-            data["browser_canvas"] = browser_canvas
-
-        loading_request = DocumentCreateInputRequest.model_validate(data)
-        return PreparedRequest(method="POST", url="/v1/documents/create_inputs", data=loading_request.model_dump(), idempotency_key=idempotency_key)
-
-    def _prepare_correct_image_orientation(self, document: Path | str | IOBase | MIMEData | PIL.Image.Image) -> PreparedRequest:
-        mime_document = prepare_mime_document(document)
-
-        if not mime_document.mime_type.startswith("image/"):
-            raise ValueError("Image is not a valid image")
-
-        return PreparedRequest(
-            method="POST",
-            url="/v1/documents/correct_image_orientation",
-            data={"document": mime_document.model_dump()},
-        )
-
-
-class Documents(SyncAPIResource, BaseDocumentsMixin):
-    """Documents API wrapper"""
-
-    def __init__(self, client: Any) -> None:
-        super().__init__(client=client)
-        self.extractions = Extractions(client=client)
-        # self.batch = Batch(client=client)
-
-    def correct_image_orientation(self, document: Path | str | IOBase | MIMEData | PIL.Image.Image) -> PIL.Image.Image:
-        """Corrects the orientation of an image using the UiForm API.
-
-        This method takes an image in various formats and returns a PIL Image with corrected orientation.
-        Useful for handling images from mobile devices or cameras that may have incorrect EXIF orientation.
-
-        Args:
-            image: The input image to correct. Can be:
-                - A file path (Path or str)
-                - A file-like object (IOBase)
-                - A MIMEData object
-                - A PIL Image object
-
-        Returns:
-            PIL.Image.Image: The orientation-corrected image as a PIL Image object
-
-        Raises:
-            ValueError: If the input is not a valid image
-            UiformAPIError: If the API request fails
-        """
-        request = self._prepare_correct_image_orientation(document)
-        response = self._client._prepared_request(request)
-        mime_response = MIMEData.model_validate(response['document'])
-        return convert_mime_data_to_pil_image(mime_response)
-
-    def create_messages(
-        self,
-        document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl,
-        modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
-        idempotency_key: str | None = None,
-    ) -> DocumentMessage:
-        """
-        Create document messages from a file using the UiForm API.
-
-        Args:
-            document: The document to process. Can be a file path (Path or str) or a file-like object.
-            modality: The processing modality to use. Defaults to "native".
-            image_resolution_dpi: Optional image resolution DPI.
-            browser_canvas: Optional browser canvas size.
-            idempotency_key: Optional idempotency key for the request
-        Returns:
-            DocumentMessage: The processed document message containing extracted content.
-
-        Raises:
-            UiformAPIError: If the API request fails.
-        """
-        request = self._prepare_create_messages(document, modality, image_resolution_dpi, browser_canvas, idempotency_key)
-        response = self._client._prepared_request(request)
-        return DocumentMessage.model_validate(response)
-
-    def create_inputs(
-        self,
-        document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl,
-        json_schema: dict[str, Any] | Path | str,
-        modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
-        idempotency_key: str | None = None,
-    ) -> DocumentMessage:
-        """
-        Create document inputs (messages with schema) from a file using the UiForm API.
-
-        Args:
-            document: The document to process. Can be a file path (Path or str), file-like object, MIMEData, PIL Image, or URL.
-            json_schema: The JSON schema to use for structuring the document content.
-            modality: The processing modality to use. Defaults to "native".
-            image_resolution_dpi: Optional image resolution DPI.
-            browser_canvas: Optional browser canvas size.
-            idempotency_key: Optional idempotency key for the request
-        Returns:
-            DocumentMessage: The processed document message containing extracted content with schema context.
-
-        Raises:
-            UiformAPIError: If the API request fails.
-        """
-        request = self._prepare_create_inputs(document, json_schema, modality, image_resolution_dpi, browser_canvas, idempotency_key)
-        response = self._client._prepared_request(request)
-        return DocumentMessage.model_validate(response)
-
-
-class AsyncDocuments(AsyncAPIResource, BaseDocumentsMixin):
-    """Documents API wrapper for asynchronous usage."""
-
-    def __init__(self, client: Any) -> None:
-        super().__init__(client=client)
-        self.extractions = AsyncExtractions(client=client)
-
-    async def create_messages(
-        self,
-        document: Path | str | IOBase | MIMEData | PIL.Image.Image,
-        modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
-        idempotency_key: str | None = None,
-    ) -> DocumentMessage:
-        """
-        Create document messages from a file using the UiForm API asynchronously.
-
-        Args:
-            document: The document to process. Can be a file path (Path or str) or a file-like object.
-            modality: The processing modality to use. Defaults to "native".
-            idempotency_key: Idempotency key for request
-        Returns:
-            DocumentMessage: The processed document message containing extracted content.
-
-        Raises:
-            UiformAPIError: If the API request fails.
-        """
-        request = self._prepare_create_messages(document, modality, image_resolution_dpi, browser_canvas, idempotency_key)
-        assert request.data is not None
-        print(request.data.keys())
-        print(request.data)
-        response = await self._client._prepared_request(request)
-        return DocumentMessage.model_validate(response)
-
-    async def create_inputs(
-        self,
-        document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl,
-        json_schema: dict[str, Any] | Path | str,
-        modality: Modality = "native",
-        image_resolution_dpi: int | None = None,
-        browser_canvas: Literal['A3', 'A4', 'A5'] | None = None,
-        idempotency_key: str | None = None,
-    ) -> DocumentMessage:
-        """
-        Create document inputs (messages with schema) from a file using the UiForm API asynchronously.
-
-        Args:
-            document: The document to process. Can be a file path (Path or str), file-like object, MIMEData, PIL Image, or URL.
-            json_schema: The JSON schema to use for structuring the document content.
-            modality: The processing modality to use. Defaults to "native".
-            image_resolution_dpi: Optional image resolution DPI.
-            browser_canvas: Optional browser canvas size.
-            idempotency_key: Idempotency key for request
-        Returns:
-            DocumentMessage: The processed document message containing extracted content with schema context.
-
-        Raises:
-            UiformAPIError: If the API request fails.
-        """
-        request = self._prepare_create_inputs(document, json_schema, modality, image_resolution_dpi, browser_canvas, idempotency_key)
-        response = await self._client._prepared_request(request)
-        return DocumentMessage.model_validate(response)
-
-    async def correct_image_orientation(self, document: Path | str | IOBase | MIMEData | PIL.Image.Image) -> PIL.Image.Image:
-        """Corrects the orientation of an image using the UiForm API asynchronously.
-
-        This method takes an image in various formats and returns a PIL Image with corrected orientation.
-        Useful for handling images from mobile devices or cameras that may have incorrect EXIF orientation.
-
-        Args:
-            image: The input image to correct. Can be:
-                - A file path (Path or str)
-                - A file-like object (IOBase)
-                - A MIMEData object
-                - A PIL Image object
-
-        Returns:
-            PIL.Image.Image: The orientation-corrected image as a PIL Image object
-
-        Raises:
-            ValueError: If the input is not a valid image
-            UiformAPIError: If the API request fails
-        """
-        request = self._prepare_correct_image_orientation(document)
-        response = await self._client._prepared_request(request)
-        mime_response = MIMEData.model_validate(response['document'])
-        return convert_mime_data_to_pil_image(mime_response)

uiform/resources/models.py

@@ -1,45 +0,0 @@
-from openai.types.model import Model
-
-from .._resource import AsyncAPIResource, SyncAPIResource
-from ..types.standards import PreparedRequest
-
-
-class ModelsMixin:
-    def prepare_list(self) -> PreparedRequest:
-        return PreparedRequest(method="GET", url="/v1/models")
-
-
-class Models(SyncAPIResource, ModelsMixin):
-    """Models API wrapper"""
-
-    def list(self) -> list[Model]:
-        """
-        List all available models.
-
-        Returns:
-            list[str]: List of available models
-        Raises:
-            HTTPException if the request fails
-        """
-
-        request = self.prepare_list()
-        output = self._client._prepared_request(request)
-        return [Model(**model) for model in output["data"]]
-
-
-class AsyncModels(AsyncAPIResource, ModelsMixin):
-    """Models Asyncronous API wrapper"""
-
-    async def list(self) -> list[Model]:
-        """
-        List all available models.
-
-        Returns:
-            list[str]: List of available models
-        Raises:
-            HTTPException if the request fails
-        """
-
-        request = self.prepare_list()
-        output = await self._client._prepared_request(request)
-        return [Model(**model) for model in output["data"]]

uiform/resources/processors/automations/client.py

@@ -1,78 +0,0 @@
-import hashlib
-import hmac
-import json
-from typing import Any
-
-from ...._resource import AsyncAPIResource, SyncAPIResource
-from .endpoints import AsyncEndpoints, Endpoints
-from .links import AsyncLinks, Links
-from .mailboxes import AsyncMailboxes, Mailboxes
-from .outlook import AsyncOutlooks, Outlooks
-from .tests import AsyncTests, Tests
-from .logs import AsyncLogs, Logs
-
-class SignatureVerificationError(Exception):
-    """Raised when webhook signature verification fails."""
-
-    pass
-
-
-class AutomationsMixin:
-    def _verify_event(self, event_body: bytes, event_signature: str, secret: str) -> Any:
-        """
-        Verify the signature of a webhook event.
-
-        Args:
-            body: The raw request body
-            signature: The signature header
-            secret: The secret key used for signing
-
-        Returns:
-            The parsed event payload
-
-        Raises:
-            SignatureVerificationError: If the signature verification fails
-        """
-        expected_signature = hmac.new(secret.encode(), event_body, hashlib.sha256).hexdigest()
-
-        if not hmac.compare_digest(event_signature, expected_signature):
-            raise SignatureVerificationError("Invalid signature")
-
-        return json.loads(event_body.decode('utf-8'))
-
-
-class Automations(SyncAPIResource, AutomationsMixin):
-    """Automations API wrapper"""
-
-    def __init__(self, client: Any) -> None:
-        super().__init__(client=client)
-        self.mailboxes = Mailboxes(client=client)
-        self.links = Links(client=client)
-        self.outlook = Outlooks(client=client)
-        self.endpoints = Endpoints(client=client)
-        self.tests = Tests(client=client)
-        self.logs = Logs(client=client)
-    def verify_event(self, event_body: bytes, event_signature: str, secret: str) -> Any:
-        """
-        Verify the signature of a webhook event.
-        """
-        return self._verify_event(event_body, event_signature, secret)
-
-
-class AsyncAutomations(AsyncAPIResource, AutomationsMixin):
-    """Async Automations API wrapper"""
-
-    def __init__(self, client: Any) -> None:
-        super().__init__(client=client)
-        self.mailboxes = AsyncMailboxes(client=client)
-        self.links = AsyncLinks(client=client)
-        self.outlook = AsyncOutlooks(client=client)
-        self.endpoints = AsyncEndpoints(client=client)
-        self.tests = AsyncTests(client=client)
-        self.logs = AsyncLogs(client=client)
-
-    async def verify_event(self, event_body: bytes, event_signature: str, secret: str) -> Any:
-        """
-        Verify the signature of a webhook event.
-        """
-        return self._verify_event(event_body, event_signature, secret)