unique-toolkit 2026.36.0.dev11__tar.gz → 2026.36.0.dev13__tar.gz

{unique_toolkit-2026.36.0.dev11 → unique_toolkit-2026.36.0.dev13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.36.0.dev11
+Version: 2026.36.0.dev13
 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.dev3,<2026.36.0rc0
+Requires-Dist: unique-sdk>=2026.36.0.dev6,<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.dev11 → unique_toolkit-2026.36.0.dev13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.36.0.dev11"
+version = "2026.36.0.dev13"
 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.dev3,<2026.36.0rc0",
+    "unique-sdk>=2026.36.0.dev6,<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.dev11 → unique_toolkit-2026.36.0.dev13}/unique_toolkit/_common/pydantic/rjsf_tags.py

@@ -739,6 +739,20 @@ class RJSFMetaTag:
         return RJSFMetaTag({**union_attrs, "anyOf": any_of_branches})
 
 
+class DynamicRJSFTag:
+    """Defer RJSF metadata until ``ui_schema_for_model`` runs.
+
+    Use for env-driven admin locks (``ui:disabled``) that must reflect the
+    deployment environment at schema-generation time, not at module import.
+    """
+
+    def __init__(self, factory: Callable[[], RJSFMetaTag]) -> None:
+        self._factory = factory
+
+    def resolve_rjsf_tag(self) -> RJSFMetaTag:
+        return self._factory()
+
+
 # --------- Helpers ----------
 _NONE_TYPES = {type(None)}
 """
@@ -785,6 +799,12 @@ def _collect_metatags(extras: list[Any]) -> dict[str, Any]:
     """
     out: dict[str, Any] = {}
     for x in extras:
+        resolve = getattr(x, "resolve_rjsf_tag", None)
+        if callable(resolve):
+            tag = resolve()
+            if isinstance(tag, RJSFMetaTag):
+                out.update(tag.attrs)
+            continue
         if isinstance(x, RJSFMetaTag):
             out.update(x.attrs)
     return out
@@ -938,6 +958,9 @@ def ui_schema_for_model(
         # Re-process the unwrapped type to extract metadata from the Annotated branch.
         if not meta and get_origin(base) is Annotated:
             base, meta = _walk_annotated_chain(base)
+        field_meta = _collect_metatags(list(field_info.metadata))
+        if field_meta:
+            meta = {**meta, **field_meta}
         # Start with field-level metadata (flat dict)
         if meta:
             node.update(meta)

{unique_toolkit-2026.36.0.dev11 → unique_toolkit-2026.36.0.dev13}/unique_toolkit/agentic_table/__init__.py

@@ -14,6 +14,8 @@ from .schemas import (
     AgreementStatus,
     ArtifactType,
     CreatedMagicTableSheet,
+    LibrarySheetRowVerifiedMetadata,
+    LibrarySheetRowVerifiedRow,
     LogEntry,
     MagicTableAction,
     MagicTableAddMetadataPayload,
@@ -81,6 +83,8 @@ __all__ = [
     "MagicTableSheetCompletedPayload",
     "MagicTableSheetCreatedPayload",
     "MagicTableUpdateCellPayload",
+    "LibrarySheetRowVerifiedMetadata",
+    "LibrarySheetRowVerifiedRow",
     "RerunRowMetadata",
     "RerunRowsMetadata",
     "RowMetadataEntry",

{unique_toolkit-2026.36.0.dev11 → unique_toolkit-2026.36.0.dev13}/unique_toolkit/agentic_table/schemas.py

@@ -6,6 +6,7 @@ from pydantic import (
     BaseModel,
     Field,
     field_validator,
+    model_validator,
 )
 from unique_sdk import (
     AgenticTableSheetState,
@@ -215,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(

{unique_toolkit-2026.36.0.dev11 → unique_toolkit-2026.36.0.dev13}/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(