flaxflow 1.0.0__py3-none-any.whl

flaxflow/__init__.py

@@ -0,0 +1,43 @@
+"""SDK oficial em Python para a FlaxFlow Engine API (processamento de documentos com IA).
+
+Quickstart:
+    >>> from flaxflow import FlaxFlow
+    >>> fx = FlaxFlow(api_key="sk_...")
+    >>> job = fx.processing.process_document("invoice.pdf", document_id="<doc_id>")
+    >>> print(job.result)
+"""
+
+from __future__ import annotations
+
+from ._files import SUPPORTED_DOCUMENT_TYPES
+from .client import FlaxFlow
+from .errors import (
+    APIConnectionError,
+    APIStatusError,
+    AuthenticationError,
+    ConflictError,
+    FlaxFlowError,
+    NotFoundError,
+    PermissionDeniedError,
+    RateLimitError,
+    ServerError,
+    UnprocessableEntityError,
+)
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "FlaxFlow",
+    "FlaxFlowError",
+    "APIConnectionError",
+    "APIStatusError",
+    "AuthenticationError",
+    "PermissionDeniedError",
+    "NotFoundError",
+    "ConflictError",
+    "UnprocessableEntityError",
+    "RateLimitError",
+    "ServerError",
+    "SUPPORTED_DOCUMENT_TYPES",
+    "__version__",
+]

flaxflow/_files.py

@@ -0,0 +1,78 @@
+"""Helpers para transformar arquivos em payloads base64 que a Engine API espera."""
+
+from __future__ import annotations
+
+import base64
+import os
+from pathlib import Path
+from typing import IO, Union
+
+# Mapeia extensões de arquivo para o `documentType` aceito pela API.
+_EXT_TO_DOCUMENT_TYPE = {
+    ".pdf": "pdf",
+    ".png": "png",
+    ".jpg": "jpeg",
+    ".jpeg": "jpeg",
+    ".webp": "webp",
+    ".gif": "gif",
+}
+
+#: Tipos de arquivo aceitos pela Engine API.
+SUPPORTED_DOCUMENT_TYPES = ("pdf", "png", "jpeg", "webp", "gif")
+
+FileInput = Union[str, "os.PathLike[str]", bytes, IO[bytes]]
+
+
+def detect_document_type(path: Union[str, "os.PathLike[str]"]) -> str:
+    """Infere o `documentType` a partir da extensão do caminho.
+
+    Raises:
+        ValueError: se a extensão não for suportada.
+    """
+    ext = Path(os.fspath(path)).suffix.lower()
+    try:
+        return _EXT_TO_DOCUMENT_TYPE[ext]
+    except KeyError as exc:
+        raise ValueError(
+            f"Could not infer documentType from extension {ext!r}. "
+            f"Pass document_type explicitly (one of {SUPPORTED_DOCUMENT_TYPES})."
+        ) from exc
+
+
+def _read_bytes(file: FileInput) -> bytes:
+    if isinstance(file, bytes):
+        return file
+    if hasattr(file, "read"):
+        return file.read()  # type: ignore[union-attr]
+    with open(os.fspath(file), "rb") as fh:  # type: ignore[arg-type]
+        return fh.read()
+
+
+def resolve_file(
+    file: FileInput,
+    document_type: str | None = None,
+) -> tuple[str, str]:
+    """Resolve uma entrada de arquivo para ``(base64_document, document_type)``.
+
+    ``file`` pode ser um caminho, ``bytes`` ou um objeto file-like. Quando
+    ``document_type`` não é informado, tenta inferir pela extensão (só funciona
+    para caminhos).
+    """
+    resolved_type = document_type
+    if resolved_type is None:
+        if isinstance(file, (str, os.PathLike)):
+            resolved_type = detect_document_type(file)
+        else:
+            raise ValueError(
+                "document_type is required when file is not a path "
+                f"(one of {SUPPORTED_DOCUMENT_TYPES})."
+            )
+
+    if resolved_type not in SUPPORTED_DOCUMENT_TYPES:
+        raise ValueError(
+            f"Unsupported document_type {resolved_type!r}; "
+            f"expected one of {SUPPORTED_DOCUMENT_TYPES}."
+        )
+
+    encoded = base64.b64encode(_read_bytes(file)).decode("ascii")
+    return encoded, resolved_type

flaxflow/_generated/__init__.py

@@ -0,0 +1,5 @@
+"""Modelos gerados a partir da spec OpenAPI (`scripts/generate.sh`). Não editar à mão."""
+
+from . import models
+
+__all__ = ["models"]