unique-toolkit 2026.36.0.dev10__tar.gz → 2026.36.0.dev12__tar.gz

{unique_toolkit-2026.36.0.dev10 → unique_toolkit-2026.36.0.dev12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.36.0.dev10
+Version: 2026.36.0.dev12
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>
@@ -16,7 +16,7 @@ Requires-Dist: sseclient>=0.0.27,<0.0.28
 Requires-Dist: openai>=1.105.0,<3
 Requires-Dist: platformdirs>=4.0.0,<5
 Requires-Dist: pillow>=12.1.1
-Requires-Dist: unique-sdk>=2026.36.0.dev2,<2026.36.0rc0
+Requires-Dist: unique-sdk>=2026.36.0.dev5,<2026.36.0rc0
 Requires-Dist: jambo>=0.1.2,<0.2
 Requires-Dist: jinja2>=3.1.6,<4
 Requires-Dist: docxtpl>=0.20.1,<0.21

{unique_toolkit-2026.36.0.dev10 → unique_toolkit-2026.36.0.dev12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.36.0.dev10"
+version = "2026.36.0.dev12"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -28,7 +28,7 @@ dependencies = [
     "openai>=1.105.0,<3",
     "platformdirs>=4.0.0,<5",
     "pillow>=12.1.1",
-    "unique-sdk>=2026.36.0.dev2,<2026.36.0rc0",
+    "unique-sdk>=2026.36.0.dev5,<2026.36.0rc0",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",

{unique_toolkit-2026.36.0.dev10 → unique_toolkit-2026.36.0.dev12}/unique_toolkit/agentic_table/__init__.py

@@ -14,6 +14,8 @@ from .schemas import (
     AgreementStatus,
     ArtifactType,
     CreatedMagicTableSheet,
+    LibrarySheetRowVerifiedMetadata,
+    LibrarySheetRowVerifiedRow,
     LogEntry,
     MagicTableAction,
     MagicTableAddMetadataPayload,
@@ -26,11 +28,13 @@ from .schemas import (
     MagicTableGenerateArtifactPayload,
     MagicTableLibrarySheetRowVerifiedPayload,
     MagicTableRerunRowPayload,
+    MagicTableRerunRowsPayload,
     MagicTableSheet,
     MagicTableSheetCompletedPayload,
     MagicTableSheetCreatedPayload,
     MagicTableUpdateCellPayload,
     RerunRowMetadata,
+    RerunRowsMetadata,
     RowMetadataEntry,
     RowMetadataEntryInput,
     SelectionMethod,
@@ -74,11 +78,15 @@ __all__ = [
     "MagicTableGenerateArtifactPayload",
     "MagicTableLibrarySheetRowVerifiedPayload",
     "MagicTableRerunRowPayload",
+    "MagicTableRerunRowsPayload",
     "MagicTableSheet",
     "MagicTableSheetCompletedPayload",
     "MagicTableSheetCreatedPayload",
     "MagicTableUpdateCellPayload",
+    "LibrarySheetRowVerifiedMetadata",
+    "LibrarySheetRowVerifiedRow",
     "RerunRowMetadata",
+    "RerunRowsMetadata",
     "RowMetadataEntry",
     "RowMetadataEntryInput",
     "RowVerificationStatus",

{unique_toolkit-2026.36.0.dev10 → unique_toolkit-2026.36.0.dev12}/unique_toolkit/agentic_table/schemas.py

@@ -6,6 +6,7 @@ from pydantic import (
     BaseModel,
     Field,
     field_validator,
+    model_validator,
 )
 from unique_sdk import (
     AgenticTableSheetState,
@@ -22,7 +23,10 @@ from unique_sdk.api_resources._agentic_table import (
 )
 
 from unique_toolkit._common.exception import ConfigurationException
-from unique_toolkit._common.pydantic_helpers import get_configuration_dict
+from unique_toolkit._common.pydantic_helpers import (
+    NoneToDefault,
+    get_configuration_dict,
+)
 from unique_toolkit.app.schemas import (
     AssistantWebhookEvent,
     BaseEventPayload,
@@ -44,6 +48,7 @@ class MagicTableEventTypes(StrEnum):
     LIBRARY_SHEET_ROW_VERIFIED = "unique.magic-table.library-sheet-row.verified"
     SHEET_CREATED = "unique.magic-table.sheet-created"
     RERUN_ROW = "unique.magic-table.rerun-row"
+    RERUN_ROWS = "unique.magic-table.rerun-rows"
 
 
 class BaseMetadata(BaseModel):
@@ -211,9 +216,61 @@ class MagicTableSheetCreatedPayload(
 ########## Library Sheet Row Verified Payload ##########
 
 
+class LibrarySheetRowVerifiedRow(BaseModel):
+    model_config = get_configuration_dict()
+    row_order: int = Field(description="The row index of a verified row.")
+    row_id: str | None = Field(
+        default=None,
+        description="Optional pairing ID for a verified row.",
+    )
+
+
 class LibrarySheetRowVerifiedMetadata(BaseMetadata):
     model_config = get_configuration_dict()
-    row_order: int = Field(description="The row index of the row that was verified.")
+    row_order: int = Field(
+        description="The row index of the first (or only) verified row."
+    )
+    row_id: str | None = Field(
+        default=None,
+        description=(
+            "Optional pairing id for the first verified row. Ingest keys off "
+            "row_order (or rows[].row_order). Omitted or null stays None so "
+            "legacy payloads that only send rowOrder still parse."
+        ),
+    )
+    rows: Annotated[list[LibrarySheetRowVerifiedRow], NoneToDefault] = Field(
+        default_factory=list,
+        description="The full batch of verified rows, including the first row.",
+    )
+
+    @model_validator(mode="after")
+    def validate_first_row_matches_scalars(self) -> "LibrarySheetRowVerifiedMetadata":
+        """Keep old scalar consumers and new batch consumers on the same first row."""
+        if not self.rows:
+            return self
+
+        first_row = self.rows[0]
+        if first_row.row_order != self.row_order:
+            raise ValueError("rows[0].row_order must match row_order")
+        if (
+            self.row_id is not None
+            and first_row.row_id is not None
+            and first_row.row_id != self.row_id
+        ):
+            raise ValueError("rows[0].row_id must match row_id")
+        return self
+
+    @property
+    def verified_rows(self) -> list[LibrarySheetRowVerifiedRow]:
+        """Return one normalized row list for both legacy and bulk payloads."""
+        if self.rows:
+            return self.rows
+        return [
+            LibrarySheetRowVerifiedRow(
+                row_order=self.row_order,
+                row_id=self.row_id,
+            )
+        ]
 
 
 class MagicTableLibrarySheetRowVerifiedPayload(
@@ -248,6 +305,48 @@ class MagicTableRerunRowPayload(
 ): ...
 
 
+########## Rerun Rows Payload ##########
+
+
+class RerunRowsMetadata(BaseMetadata):
+    """Metadata of a bulk re-run: one event carrying N rows of the same sheet."""
+
+    model_config = get_configuration_dict()
+    source_file_ids: Annotated[list[str], NoneToDefault] = Field(
+        default_factory=list,
+        description=(
+            "Source file IDs for this rerun. Optional: a user-triggered bulk "
+            "rerun may omit sources when none changed. Null and omitted become []."
+        ),
+    )
+    row_orders: Annotated[list[int], NoneToDefault] = Field(
+        default_factory=list,
+        description=(
+            "1-based content row indices to rerun, in first-seen selection order. "
+            "Duplicates are dropped. Row 0 is the header and negatives address "
+            "nothing, so non-positive values are discarded. Empty means no work."
+        ),
+    )
+    context: str = Field(default="", description="The context text for the rerun.")
+
+    @field_validator("row_orders", mode="after")
+    @classmethod
+    def normalize_row_orders(cls, v: list[int]) -> list[int]:
+        return list(dict.fromkeys(row for row in v if row > 0))
+
+    @field_validator("context", mode="before")
+    @classmethod
+    def normalize_context(cls, v):
+        if v is None:
+            return ""
+        return v
+
+
+class MagicTableRerunRowsPayload(
+    MagicTableBasePayload[Literal[MagicTableAction.RERUN_ROWS], RerunRowsMetadata]
+): ...
+
+
 ########### Magic Table Event definition ###########
 
 
@@ -259,6 +358,7 @@ PayloadTypes = (
     | MagicTableLibrarySheetRowVerifiedPayload
     | MagicTableSheetCreatedPayload
     | MagicTableRerunRowPayload
+    | MagicTableRerunRowsPayload
 )
 
 MagicTablePayloadTypes = Annotated[PayloadTypes, Field(discriminator="action")]

{unique_toolkit-2026.36.0.dev10 → unique_toolkit-2026.36.0.dev12}/unique_toolkit/agentic_table/service.py

@@ -443,6 +443,7 @@ class AgenticTableService:
         include_log_history: bool = False,
         include_cell_meta_data: bool = False,
         include_row_metadata: bool = False,
+        row_orders: list[int] | None = None,
     ) -> MagicTableSheet:
         """
         Gets the sheet data from the Agentic Table paginated by batch_size.
@@ -454,9 +455,19 @@ class AgenticTableService:
             include_log_history (bool): Whether to include the log history.
             include_cell_meta_data (bool): Whether to include the cell metadata (renderer, selection, agreement status).
             include_row_metadata (bool): Whether to include the row metadata (key value pairs).
+            row_orders (list[int] | None): Sparse, 1-based content row indices to
+                fetch, using the same numbering as get_cell and rerun events. Row 0
+                is the header; non-positive values are discarded. An empty list
+                returns a header-only sheet. This cannot be combined with a
+                non-default start_row or with end_row.
         Returns:
             MagicTableSheet: The sheet data.
         """
+        if batch_size <= 0:
+            raise ValueError("Batch size must be greater than zero")
+        if row_orders is not None and (start_row != 0 or end_row is not None):
+            raise ValueError("row_orders cannot be combined with start_row or end_row")
+
         # Find the total number of rows
         sheet_info = await AgenticTable.get_sheet_data(
             user_id=self._user_id,
@@ -472,44 +483,77 @@ class AgenticTableService:
             raise RuntimeError(
                 "Expected magicTableRowCount in sheet response when includeRowCount=True"
             )
-        if end_row is None or end_row > total_rows:
-            end_row = total_rows
-        if start_row > end_row:
-            raise Exception("Start row is greater than end row")
-        if start_row < 0 or end_row < 0:
-            raise Exception("Start row or end row is negative")
 
         # Get the cells
         cells = []
-        for row in range(start_row, end_row, batch_size):
-            end_row_batch = min(row + batch_size, end_row)
-            if include_row_metadata:
-                sheet_partial = await AgenticTable.get_sheet_data(
-                    user_id=self._user_id,
-                    company_id=self._company_id,
-                    tableId=self.table_id,
-                    includeCells=True,
-                    includeLogHistory=include_log_history,
-                    includeRowCount=False,
-                    includeCellMetaData=include_cell_meta_data,  # renderer, selection, agreement status
-                    startRow=row,
-                    endRow=end_row_batch - 1,
-                    includeRowMetadata=True,
+        batch_params: list[AgenticTable.GetSheetData] = []
+        if row_orders is not None:
+            normalized_row_orders = list(
+                dict.fromkeys(row_order for row_order in row_orders if row_order > 0)
+            )
+            for offset in range(0, len(normalized_row_orders), batch_size):
+                batch_params.append(
+                    {
+                        "tableId": self.table_id,
+                        "rowOrders": normalized_row_orders[
+                            offset : offset + batch_size
+                        ],
+                    }
                 )
-            else:
-                sheet_partial = await AgenticTable.get_sheet_data(
-                    user_id=self._user_id,
-                    company_id=self._company_id,
-                    tableId=self.table_id,
-                    includeCells=True,
-                    includeLogHistory=include_log_history,
-                    includeRowCount=False,
-                    includeCellMetaData=include_cell_meta_data,  # renderer, selection, agreement status
-                    startRow=row,
-                    endRow=end_row_batch - 1,
+        else:
+            if end_row is None or end_row > total_rows:
+                end_row = total_rows
+            if start_row > end_row:
+                raise Exception("Start row is greater than end row")
+            if start_row < 0 or end_row < 0:
+                raise Exception("Start row or end row is negative")
+            for row in range(start_row, end_row, batch_size):
+                batch_params.append(
+                    {
+                        "tableId": self.table_id,
+                        "startRow": row,
+                        "endRow": min(row + batch_size, end_row) - 1,
+                    }
                 )
+
+        for request_params in batch_params:
+            request_params.update(
+                {
+                    "includeCells": True,
+                    "includeLogHistory": include_log_history,
+                    "includeRowCount": False,
+                    "includeCellMetaData": include_cell_meta_data,
+                }
+            )
+            if include_row_metadata:
+                request_params["includeRowMetadata"] = True
+            sheet_partial = await AgenticTable.get_sheet_data(
+                user_id=self._user_id,
+                company_id=self._company_id,
+                **request_params,
+            )
             if "magicTableCells" in sheet_partial:
                 batch_cells = sheet_partial["magicTableCells"]
+                requested_row_orders = request_params.get("rowOrders")
+                if requested_row_orders is not None:
+                    requested = set(requested_row_orders)
+                    returned = {cell["rowOrder"] for cell in batch_cells}
+                    unexpected = returned - requested
+                    missing = requested - returned
+                    if unexpected or missing:
+                        self.logger.warning(
+                            "Sparse magic-table sheet response did not match the "
+                            "requested rows; filtering response. requested=%s "
+                            "unexpected=%s missing=%s",
+                            sorted(requested),
+                            sorted(unexpected),
+                            sorted(missing),
+                        )
+                    batch_cells = [
+                        cell
+                        for cell in batch_cells
+                        if cell.get("rowOrder") in requested
+                    ]
                 if (
                     include_row_metadata
                     and not _sheet_batch_cells_include_row_metadata_from_api(