retab 0.0.84__tar.gz → 0.0.85__tar.gz

{retab-0.0.84 → retab-0.0.85}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: retab
-Version: 0.0.84
+Version: 0.0.85
 Summary: Retab official python library
 Home-page: https://github.com/retab-dev/retab
 Author: Retab

retab-0.0.85/retab/resources/edit/agent/__init__.py

@@ -0,0 +1,4 @@
+from .client import Agent, AsyncAgent
+
+__all__ = ["Agent", "AsyncAgent"]
+

{retab-0.0.84/retab/resources/edit → retab-0.0.85/retab/resources/edit/agent}/client.py

@@ -1,5 +1,5 @@
 """
-Edit SDK client - Wrapper for document editing functionality.
+Agent Edit SDK client - Wrapper for agent-based document editing functionality.
 """
 
 from io import IOBase
@@ -9,26 +9,24 @@ from typing import Any
 import PIL.Image
 from pydantic import HttpUrl
 
-from ..._resource import AsyncAPIResource, SyncAPIResource
-from ...utils.mime import prepare_mime_document
-from ...types.documents.edit import (
+from ...._resource import AsyncAPIResource, SyncAPIResource
+from ....utils.mime import prepare_mime_document
+from ....types.documents.edit import (
     EditRequest,
     EditResponse,
 )
-from ...types.mime import MIMEData
-from ...types.standards import PreparedRequest, FieldUnset
-from .templates import Templates, AsyncTemplates
+from ....types.mime import MIMEData
+from ....types.standards import PreparedRequest, FieldUnset
 
 
-class BaseEditMixin:
-    """Shared methods for preparing edit API requests."""
+class BaseAgentMixin:
+    """Shared methods for preparing agent edit API requests."""
 
-    def _prepare_fill_document(
+    def _prepare_fill(
         self,
         instructions: str,
         document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl | None = None,
         model: str = FieldUnset,
-        template_id: str | None = FieldUnset,
         **extra_body: Any,
     ) -> PreparedRequest:
         request_dict: dict[str, Any] = {
@@ -41,8 +39,6 @@ class BaseEditMixin:
 
         if model is not FieldUnset:
             request_dict["model"] = model
-        if template_id is not FieldUnset:
-            request_dict["template_id"] = template_id
 
         # Merge any extra fields provided by the caller
         if extra_body:
@@ -51,24 +47,22 @@ class BaseEditMixin:
         edit_request = EditRequest(**request_dict)
         return PreparedRequest(
             method="POST",
-            url="/v1/edit/fill-document",
+            url="/v1/edit/agent/fill",
             data=edit_request.model_dump(mode="json", exclude_unset=True),
         )
 
 
-class Edit(SyncAPIResource, BaseEditMixin):
-    """Edit API wrapper for synchronous usage."""
+class Agent(SyncAPIResource, BaseAgentMixin):
+    """Agent Edit API wrapper for synchronous usage."""
 
     def __init__(self, client: Any) -> None:
         super().__init__(client=client)
-        self.templates = Templates(client=client)
 
-    def fill_document(
+    def fill(
         self,
         instructions: str,
         document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl | None = None,
         model: str = FieldUnset,
-        template_id: str | None = FieldUnset,
         **extra_body: Any,
     ) -> EditResponse:
         """
@@ -80,16 +74,11 @@ class Edit(SyncAPIResource, BaseEditMixin):
         3. LLM-based form filling using the provided instructions
         4. Returns the filled document with form field values populated
 
-        Either `document` OR `template_id` must be provided, but not both.
-
         Args:
             instructions: Instructions describing how to fill the form fields.
             document: The document to edit. Can be a file path (Path or str), file-like object,
-                MIMEData, PIL Image, or URL. Mutually exclusive with template_id.
+                MIMEData, PIL Image, or URL.
             model: The LLM model to use for inference. Defaults to "retab-small".
-            template_id: Template ID to use for filling. When provided, uses the template's
-                pre-defined form fields and empty PDF. Only works for PDF documents.
-                Mutually exclusive with document.
 
         Returns:
             EditResponse: Response containing:
@@ -105,30 +94,27 @@ class Edit(SyncAPIResource, BaseEditMixin):
             - PPTX/PPT: Native editing for presentations
             - XLSX/XLS: Native editing for spreadsheets
         """
-        request = self._prepare_fill_document(
+        request = self._prepare_fill(
             instructions=instructions,
             document=document,
             model=model,
-            template_id=template_id,
             **extra_body,
         )
         response = self._client._prepared_request(request)
         return EditResponse.model_validate(response)
 
 
-class AsyncEdit(AsyncAPIResource, BaseEditMixin):
-    """Edit API wrapper for asynchronous usage."""
+class AsyncAgent(AsyncAPIResource, BaseAgentMixin):
+    """Agent Edit API wrapper for asynchronous usage."""
 
     def __init__(self, client: Any) -> None:
         super().__init__(client=client)
-        self.templates = AsyncTemplates(client=client)
 
-    async def fill_document(
+    async def fill(
         self,
         instructions: str,
         document: Path | str | IOBase | MIMEData | PIL.Image.Image | HttpUrl | None = None,
         model: str = FieldUnset,
-        template_id: str | None = FieldUnset,
         **extra_body: Any,
     ) -> EditResponse:
         """
@@ -140,16 +126,11 @@ class AsyncEdit(AsyncAPIResource, BaseEditMixin):
         3. LLM-based form filling using the provided instructions
         4. Returns the filled document with form field values populated
 
-        Either `document` OR `template_id` must be provided, but not both.
-
         Args:
             instructions: Instructions describing how to fill the form fields.
             document: The document to edit. Can be a file path (Path or str), file-like object,
-                MIMEData, PIL Image, or URL. Mutually exclusive with template_id.
+                MIMEData, PIL Image, or URL.
             model: The LLM model to use for inference. Defaults to "retab-small".
-            template_id: Template ID to use for filling. When provided, uses the template's
-                pre-defined form fields and empty PDF. Only works for PDF documents.
-                Mutually exclusive with document.
 
         Returns:
             EditResponse: Response containing:
@@ -165,12 +146,12 @@ class AsyncEdit(AsyncAPIResource, BaseEditMixin):
             - PPTX/PPT: Native editing for presentations
             - XLSX/XLS: Native editing for spreadsheets
         """
-        request = self._prepare_fill_document(
+        request = self._prepare_fill(
             instructions=instructions,
             document=document,
             model=model,
-            template_id=template_id,
             **extra_body,
         )
         response = await self._client._prepared_request(request)
         return EditResponse.model_validate(response)
+

retab-0.0.85/retab/resources/edit/client.py

@@ -0,0 +1,41 @@
+"""
+Edit SDK client - Wrapper for document editing functionality.
+
+Provides access to:
+- edit.agent.fill() - Agent-based document editing (PDF, DOCX, PPTX, XLSX)
+- edit.templates.* - Template-based PDF form filling
+"""
+
+from typing import Any
+
+from ..._resource import AsyncAPIResource, SyncAPIResource
+from .templates import Templates, AsyncTemplates
+from .agent import Agent, AsyncAgent
+
+
+class Edit(SyncAPIResource):
+    """Edit API wrapper for synchronous usage.
+    
+    Sub-clients:
+        agent: Agent-based document editing (fill any document with AI)
+        templates: Template-based PDF form filling (for batch processing)
+    """
+
+    def __init__(self, client: Any) -> None:
+        super().__init__(client=client)
+        self.agent = Agent(client=client)
+        self.templates = Templates(client=client)
+
+
+class AsyncEdit(AsyncAPIResource):
+    """Edit API wrapper for asynchronous usage.
+    
+    Sub-clients:
+        agent: Agent-based document editing (fill any document with AI)
+        templates: Template-based PDF form filling (for batch processing)
+    """
+
+    def __init__(self, client: Any) -> None:
+        super().__init__(client=client)
+        self.agent = AsyncAgent(client=client)
+        self.templates = AsyncTemplates(client=client)

{retab-0.0.84 → retab-0.0.85}/retab/types/documents/edit.py

@@ -107,7 +107,7 @@ class OCRResult(BaseModel):
 class InferFormSchemaRequest(BaseModel):
     """Request to infer form schema from a PDF or DOCX document."""
     
-    document: MIMEData = Field(..., description="Input document (PDF or DOCX). DOCX files will be converted to PDF.")
+    document: MIMEData = Field(..., description="Input document (PDF, DOCX, XLSX or PPTX).")
     model: str = Field(default="retab-small", description="LLM model to use for inference")
 
 
@@ -118,7 +118,7 @@ class EditRequest(BaseModel):
     - When `document` is provided: OCR + LLM inference to detect and fill form fields
     - When `template_id` is provided: Uses pre-defined form fields from the template (PDF only)
     """
-    document: Optional[MIMEData] = Field(default=None, description="Input document (PDF or DOCX). DOCX files will be converted to PDF. Mutually exclusive with template_id.")
+    document: Optional[MIMEData] = Field(default=None, description="Input document (PDF, DOCX, XLSX or PPTX). Mutually exclusive with template_id.")
     model: str = Field(default="retab-small", description="LLM model to use for inference")
     instructions: str = Field(..., description="Instructions to fill the form")
     template_id: Optional[str] = Field(default=None, description="Template ID to use for filling. When provided, uses the template's pre-defined form fields and empty PDF. Only works for PDF documents. Mutually exclusive with document.")

{retab-0.0.84 → retab-0.0.85}/retab.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: retab
-Version: 0.0.84
+Version: 0.0.85
 Summary: Retab official python library
 Home-page: https://github.com/retab-dev/retab
 Author: Retab

{retab-0.0.84 → retab-0.0.85}/retab.egg-info/SOURCES.txt

@@ -18,6 +18,8 @@ retab/resources/documents/__init__.py
 retab/resources/documents/client.py
 retab/resources/edit/__init__.py
 retab/resources/edit/client.py
+retab/resources/edit/agent/__init__.py
+retab/resources/edit/agent/client.py
 retab/resources/edit/templates/__init__.py
 retab/resources/edit/templates/client.py
 retab/resources/extractions/__init__.py

{retab-0.0.84 → retab-0.0.85}/setup.py

@@ -6,7 +6,7 @@ with open("requirements.txt") as f:
 
 setup(
     name="retab",
-    version="0.0.84",
+    version="0.0.85",
     author="Retab",
     author_email="contact@retab.com",
     description="Retab official python library",