unique_toolkit 2026.34.2__tar.gz → 2026.36.0__tar.gz

{unique_toolkit-2026.34.2 → unique_toolkit-2026.36.0}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.34.2"
+version = "2026.36.0"
 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.34.2",
+    "unique-sdk>=2026.36.0",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",

{unique_toolkit-2026.34.2 → unique_toolkit-2026.36.0}/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.34.2 → unique_toolkit-2026.36.0}/unique_toolkit/agentic/history_manager/history_construction_with_contents.py

@@ -196,7 +196,9 @@ async def download_encoded_images_async(
                 image_string = f"data:{mime_type};base64," + encoded_string
                 base64_encoded_images.append(image_string)
             except Exception as e:
-                _LOGGER.warning("Failed to download image %s: %s", im.key, e)
+                _LOGGER.warning(
+                    "Failed to download image %s: %s", im.key, e, exc_info=True
+                )
     return base64_encoded_images
 
 

{unique_toolkit-2026.34.2 → unique_toolkit-2026.36.0}/unique_toolkit/agentic/tools/mcp/tool_wrapper.py

@@ -4,9 +4,10 @@ import json
 import logging
 import mimetypes
 import uuid
-from typing import Any, Dict
+from typing import Any
 
 import unique_sdk
+from pydantic import ValidationError
 
 from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
 from unique_toolkit.agentic.tools.mcp.models import MCPToolConfig
@@ -17,6 +18,7 @@ from unique_toolkit.app.schemas import ChatEvent, McpServer, McpTool
 from unique_toolkit.chat.schemas import MessageLog, MessageLogStatus
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.content.functions import upload_content_from_bytes
+from unique_toolkit.content.schemas import ContentChunk, ContentReference
 from unique_toolkit.language_model import LanguageModelService
 from unique_toolkit.language_model.schemas import (
     LanguageModelFunction,
@@ -25,6 +27,8 @@ from unique_toolkit.language_model.schemas import (
 
 logger = logging.getLogger(__name__)
 
+REFERENCE_META_KEY = "unique.app/reference"
+
 
 class MCPToolWrapper(Tool[MCPToolConfig]):
     """Wrapper class for MCP tools that implements the Tool interface"""
@@ -67,7 +71,7 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
             parameters=self._mcp_tool.input_schema,
         )
 
-    def _json_schema_to_python_type(self, schema: Dict[str, Any]) -> type:
+    def _json_schema_to_python_type(self, schema: dict[str, Any]) -> type:
         """Convert JSON schema type to Python type"""
         json_type = schema.get("type", "string")
 
@@ -135,9 +139,14 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
             # Use SDK to call the public API
             result = await self._call_mcp_tool_via_sdk(arguments)
 
+            content_chunks, references = self._extract_structured_references(result)
             content_str, image_data_urls = await asyncio.to_thread(
                 self._process_mcp_result, result
             )
+            if content_chunks:
+                # Structured references already carry the text; keep any
+                # extracted images but drop the redundant flat text.
+                content_str = ""
 
             # TODO: Why result here not applied directly to the body of the tool_response? like so how does it know the results in the history?
             tool_response = ToolCallResponse(
@@ -150,6 +159,7 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
                 },
                 error_message="",
                 content=content_str,
+                content_chunks=content_chunks or None,
                 image_data_urls=image_data_urls,
             )
 
@@ -158,6 +168,7 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
                 progress_message="_Completed MCP tool_",
                 status=MessageLogStatus.COMPLETED,
                 active_message_log=active_message_log,
+                references=references,
             )
 
             return tool_response
@@ -189,17 +200,19 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
         progress_message: str | None = None,
         status: MessageLogStatus = MessageLogStatus.RUNNING,
         active_message_log: MessageLog | None = None,
+        references: list[ContentReference] | None = None,
     ) -> MessageLog | None:
         return self._message_step_logger.create_or_update_message_log(
             active_message_log=active_message_log,
             header=self.display_name(),
             progress_message=progress_message,
             status=status,
+            references=references or [],
         )
 
     def _extract_and_validate_arguments(
         self, tool_call: LanguageModelFunction
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """
         Extract and validate arguments from tool call, handling various formats robustly.
 
@@ -246,7 +259,39 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
             f"Unexpected arguments type for MCP tool {self.name}: {type(raw_arguments)}"
         )
 
-    def _process_mcp_result(self, result: Dict[str, Any]) -> tuple[str, list[str]]:
+    def _extract_structured_references(
+        self, result: dict[str, Any]
+    ) -> tuple[list[ContentChunk], list[ContentReference]]:
+        """Opt-in: content items with valid ``unique.app/reference`` + chunk ``_meta``."""
+        chunks: list[ContentChunk] = []
+        references: list[ContentReference] = []
+        content_items = self._extract_content_items(result)
+        for item in content_items:
+            if not isinstance(item, dict) or item.get("type") != "text":
+                continue
+            meta = item.get("_meta")
+            if not isinstance(meta, dict):
+                continue
+            ref_payload = meta.get(REFERENCE_META_KEY)
+            if not isinstance(ref_payload, dict):
+                continue
+            try:
+                reference = ContentReference.model_validate(ref_payload)
+                chunk = ContentChunk.model_validate(
+                    {k: v for k, v in meta.items() if k != REFERENCE_META_KEY}
+                )
+            except ValidationError:
+                self.logger.warning(
+                    "MCP tool %s: ignoring invalid %s metadata",
+                    self.name,
+                    REFERENCE_META_KEY,
+                )
+                continue
+            references.append(reference)
+            chunks.append(chunk)
+        return chunks, references
+
+    def _process_mcp_result(self, result: dict[str, Any]) -> tuple[str, list[str]]:
         """Parse MCP result content array, upload images, return content string and data URLs."""
         content_items = self._extract_content_items(result)
         if not content_items:
@@ -286,7 +331,7 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
             content_str = json.dumps(result)
         return content_str, image_data_urls
 
-    def _extract_content_items(self, result: Dict[str, Any]) -> list[Any]:
+    def _extract_content_items(self, result: dict[str, Any]) -> list[Any]:
         """Extract content array from MCP result (handles various response shapes)."""
         if "content" in result and isinstance(result["content"], list):
             return result["content"]
@@ -337,7 +382,7 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
 
         return data_url, content_id
 
-    async def _call_mcp_tool_via_sdk(self, arguments: Dict[str, Any]) -> Dict[str, Any]:
+    async def _call_mcp_tool_via_sdk(self, arguments: dict[str, Any]) -> dict[str, Any]:
         """Call MCP tool via SDK to public API"""
         try:
             self.logger.info(

unique_toolkit-2026.36.0/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/artifacts.py

@@ -0,0 +1,213 @@
+import asyncio
+import logging
+import tempfile
+from mimetypes import guess_type
+from pathlib import Path
+from typing import Any
+
+import unique_sdk.utils.file_io as file_io
+from openai.types.responses.response_output_text import AnnotationContainerFileCitation
+from pydantic import BaseModel
+
+from unique_toolkit import ChatService
+from unique_toolkit._common.utils.files import FileMimeType, ImageMimeType
+from unique_toolkit.agentic.tools.openai_builtin.code_interpreter.postprocessors.office_preview import (
+    OFFICE_PREVIEW_MIME_TYPES,
+    convert_office_bytes_to_preview_pdf,
+    office_extension,
+)
+from unique_toolkit.content.schemas import Content
+
+_CODE_EXECUTION_ARTIFACT_METADATA_KEY: str = "codeExecutionArtifactMetadata"
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class CodeExecutionArtifactMetadata(BaseModel):
+    container_id: str
+    file_id: str
+    filepath: str
+
+
+def _kb_safe_mime(mime: str) -> str:
+    """Return a MIME type the Unique KB will accept.
+
+    Membership is defined by ``FileMimeType`` and ``ImageMimeType`` in
+    ``unique_toolkit._common.utils.files`` (same catalog as path-based helpers
+    like ``FileMimeType.is_valid_mime``, but here we already have a resolved
+    MIME string from ``mimetypes.guess_type``, so we use StrEnum value lookup
+    instead of re-parsing a path).
+
+    The KB GraphQL API rejects many code-file MIME types (e.g. ``text/x-python``
+    for ``.py`` files).  Anything not in those enums is coerced to
+    ``text/plain`` so the file can be stored and downloaded without changing
+    its bytes.
+
+    Other ``image/*`` subtypes (e.g. ``image/jpg``) still pass through unchanged.
+    """
+    try:
+        FileMimeType(mime)
+        return mime
+    except ValueError:
+        pass
+    try:
+        ImageMimeType(mime)
+        return mime
+    except ValueError:
+        pass
+    if mime.startswith("image/"):
+        return mime
+    return "text/plain"
+
+
+def _artifact_metadata(
+    file: AnnotationContainerFileCitation,
+) -> dict[str, Any]:
+    return {
+        _CODE_EXECUTION_ARTIFACT_METADATA_KEY: CodeExecutionArtifactMetadata(
+            container_id=file.container_id,
+            file_id=file.file_id,
+            filepath=f"/mnt/data/{file.filename}",
+        ).model_dump()
+    }
+
+
+async def _upload_artifact_bytes(
+    chat_service: ChatService,
+    file: AnnotationContainerFileCitation,
+    file_bytes: bytes,
+    mime: str,
+) -> Content:
+    _LOGGER.info(
+        "Uploading '%s' to knowledge base (%d bytes, mime type %s)",
+        file.filename,
+        len(file_bytes),
+        mime,
+    )
+
+    return await chat_service.upload_to_chat_from_bytes_async(
+        content=file_bytes,
+        content_name=file.filename,
+        mime_type=mime,
+        skip_ingestion=True,
+        hide_in_chat=True,
+        metadata=_artifact_metadata(file),
+    )
+
+
+async def _upload_office_artifact_with_preview(
+    chat_service: ChatService,
+    file: AnnotationContainerFileCitation,
+    file_bytes: bytes,
+    mime: str,
+) -> Content | None:
+    """Upload an Office artifact with a PDF preview, or ``None`` to fall back."""
+    with tempfile.TemporaryDirectory(prefix="code-interpreter-preview-") as tmp:
+        output_dir = Path(tmp)
+        preview_path = await convert_office_bytes_to_preview_pdf(
+            filename=file.filename,
+            file_bytes=file_bytes,
+            tmp_dir=output_dir,
+            logger=_LOGGER,
+        )
+        if preview_path is None:
+            _LOGGER.warning(
+                "Code interpreter Office artifact '%s' could not be converted "
+                "to PDF; uploading without preview.",
+                file.filename,
+            )
+            return None
+
+        source_path = output_dir / Path(file.filename).name
+        _LOGGER.info(
+            "Uploading code interpreter Office artifact '%s' with PDF preview '%s'",
+            file.filename,
+            preview_path.name,
+        )
+
+        uploaded = await asyncio.to_thread(
+            file_io.upload_file,
+            chat_service._user_id,
+            chat_service._company_id,
+            str(source_path),
+            file.filename,
+            mime,
+            chat_id=chat_service._chat_id,
+            ingestion_config={
+                "uniqueIngestionMode": "SKIP_INGESTION",
+                "hideInChat": True,
+            },
+            metadata=_artifact_metadata(file),
+            preview_pdf_path=str(preview_path),
+        )
+
+    return Content.model_validate(uploaded, by_alias=True, by_name=True)
+
+
+async def save_code_execution_artifact(
+    chat_service: ChatService,
+    file: AnnotationContainerFileCitation,
+    file_bytes: bytes,
+    *,
+    attach_office_preview: bool = False,
+) -> Content:
+    """Upload a code-interpreter artifact to the chat knowledge base.
+
+    ``attach_office_preview`` is opt-in (default ``False``) so existing
+    Unique AI callers keep the historical bytes-only upload. When ``True``,
+    Word / PowerPoint / Excel files are converted to a sibling PDF and
+    uploaded with ``preview_pdf_path``. Conversion is best-effort: if
+    LibreOffice is missing or conversion fails, the original bytes are
+    uploaded instead. Upload failures are not retried via the bytes path,
+    because ``file_io.upload_file`` may already have created a content row.
+    """
+    if attach_office_preview:
+        extension = office_extension(file.filename)
+        if extension is not None:
+            mime = _kb_safe_mime(OFFICE_PREVIEW_MIME_TYPES[extension])
+            content = await _upload_office_artifact_with_preview(
+                chat_service=chat_service,
+                file=file,
+                file_bytes=file_bytes,
+                mime=mime,
+            )
+            if content is not None:
+                return content
+            return await _upload_artifact_bytes(
+                chat_service=chat_service,
+                file=file,
+                file_bytes=file_bytes,
+                mime=mime,
+            )
+
+    raw_mime = guess_type(file.filename)[0] or "text/plain"
+    mime = _kb_safe_mime(raw_mime)
+
+    if mime != raw_mime:
+        _LOGGER.info(
+            "MIME type '%s' is not supported by the Unique KB; "
+            "uploading '%s' as 'text/plain' so the file can be stored and downloaded.",
+            raw_mime,
+            file.filename,
+        )
+
+    return await _upload_artifact_bytes(
+        chat_service=chat_service,
+        file=file,
+        file_bytes=file_bytes,
+        mime=mime,
+    )
+
+
+def load_code_execution_metadata(
+    content: Content,
+) -> CodeExecutionArtifactMetadata | None:
+    if (
+        content.metadata is None
+        or _CODE_EXECUTION_ARTIFACT_METADATA_KEY not in content.metadata
+    ):
+        return None
+
+    return CodeExecutionArtifactMetadata.model_validate(
+        content.metadata[_CODE_EXECUTION_ARTIFACT_METADATA_KEY]
+    )

{unique_toolkit-2026.34.2 → unique_toolkit-2026.36.0}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py

@@ -117,6 +117,15 @@ class DisplayCodeInterpreterFilesPostProcessorConfig(BaseModel):
         "Applies per SDK attempt. The OpenAI SDK default of 600s is too generous "
         "for small generated files; a shorter timeout lets us retry sooner.",
     )
+    attach_office_preview: SkipJsonSchema[bool] = Field(
+        default=False,
+        description=(
+            "When True, convert Word/PowerPoint/Excel artifacts to a sibling "
+            "PDF and attach it as the chat side-panel preview. Default False "
+            "keeps the historical bytes-only upload. Conversion is best-effort "
+            "and falls back to the original file."
+        ),
+    )
 
 
 class _ContentInfo(NamedTuple):
@@ -762,6 +771,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 chat_service=self._chat_service,
                 file=container_file,
                 file_bytes=file_bytes,
+                attach_office_preview=self._config.attach_office_preview,
             )
             upload_ms = (time.monotonic() - upload_t0) * 1000
             if upload_ms > 2000:

unique_toolkit-2026.36.0/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/office_preview.py

@@ -0,0 +1,154 @@
+"""LibreOffice PDF previews for code-interpreter Office artifacts.
+
+When the OpenAI code interpreter emits a Word, PowerPoint, or Excel file,
+the chat UI can render a sibling PDF in the side panel far more reliably
+than the Office binary. Conversion is best-effort: if ``soffice`` is
+missing, times out, or rejects the file, callers upload the original
+without a preview.
+"""
+
+from __future__ import annotations
+
+import asyncio
+import logging
+import shutil
+import subprocess
+import tempfile
+from pathlib import Path
+
+_SOFFICE_TIMEOUT_SECONDS = 180
+
+# Canonical MIME types so slim containers without ``/etc/mime.types`` still
+# upload Office files with a KB-accepted type instead of falling back to
+# ``text/plain`` via ``mimetypes.guess_type``.
+OFFICE_PREVIEW_MIME_TYPES: dict[str, str] = {
+    ".doc": "application/msword",
+    ".docx": (
+        "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+    ),
+    ".ppt": "application/vnd.ms-powerpoint",
+    ".pptx": (
+        "application/vnd.openxmlformats-officedocument.presentationml.presentation"
+    ),
+    ".xls": "application/vnd.ms-excel",
+    ".xlsx": ("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"),
+}
+
+
+def office_extension(filename: str) -> str | None:
+    """Return the lowercased Office suffix, or ``None`` if not previewable."""
+    suffix = Path(filename).suffix.lower()
+    return suffix if suffix in OFFICE_PREVIEW_MIME_TYPES else None
+
+
+def _resolve_soffice_binary() -> str | None:
+    return shutil.which("soffice") or shutil.which("libreoffice")
+
+
+def _run_soffice(
+    soffice: str,
+    source: Path,
+    output_dir: Path,
+    logger: logging.Logger,
+) -> None:
+    """Invoke headless ``soffice`` to convert *source* into *output_dir*.
+
+    Uses an isolated ``UserInstallation`` profile so concurrent LibreOffice
+    invocations do not collide on ``$HOME/.config/libreoffice``.
+    """
+    with tempfile.TemporaryDirectory(
+        prefix=".libreoffice-profile-",
+        dir=output_dir,
+    ) as profile_dir:
+        proc = subprocess.run(
+            [
+                soffice,
+                f"-env:UserInstallation={Path(profile_dir).resolve().as_uri()}",
+                "--headless",
+                "--convert-to",
+                "pdf",
+                "--outdir",
+                str(output_dir),
+                str(source),
+            ],
+            capture_output=True,
+            text=True,
+            timeout=_SOFFICE_TIMEOUT_SECONDS,
+        )
+    if proc.returncode != 0:
+        logger.warning(
+            "office_preview: soffice exited %d for '%s' "
+            "(stdout=%d bytes, stderr=%d bytes)",
+            proc.returncode,
+            source.name,
+            len((proc.stdout or "").strip()),
+            len((proc.stderr or "").strip()),
+        )
+        raise subprocess.CalledProcessError(
+            proc.returncode, proc.args, proc.stdout, proc.stderr
+        )
+
+
+async def convert_office_bytes_to_preview_pdf(
+    filename: str,
+    file_bytes: bytes,
+    tmp_dir: Path,
+    logger: logging.Logger,
+) -> Path | None:
+    """Write *file_bytes* into *tmp_dir* and convert to a sibling preview PDF.
+
+    The original file is written as ``tmp_dir / Path(filename).name`` so the
+    caller can pass that path to the SDK uploader. The preview is rendered
+    into ``tmp_dir / .previews / <stem>.pdf`` so it cannot collide with a
+    same-named PDF the interpreter also produced.
+
+    Returns the preview path, or ``None`` when conversion is unavailable or
+    fails. Never raises for expected conversion problems.
+    """
+    safe_filename = Path(filename).name
+    source_path = tmp_dir / safe_filename
+    source_path.write_bytes(file_bytes)
+
+    soffice = _resolve_soffice_binary()
+    if soffice is None:
+        logger.warning(
+            "office_preview: soffice/libreoffice not found on PATH — "
+            "uploading '%s' without a PDF preview.",
+            safe_filename,
+        )
+        return None
+
+    previews_dir = tmp_dir / ".previews"
+    previews_dir.mkdir(exist_ok=True)
+    target = previews_dir / f"{source_path.stem}.pdf"
+
+    try:
+        await asyncio.to_thread(
+            _run_soffice, soffice, source_path, previews_dir, logger
+        )
+    except subprocess.TimeoutExpired:
+        logger.warning(
+            "office_preview: soffice timed out after %ds for '%s' — "
+            "uploading original only",
+            _SOFFICE_TIMEOUT_SECONDS,
+            safe_filename,
+        )
+        return None
+    except subprocess.CalledProcessError:
+        return None
+    except Exception:
+        logger.exception(
+            "office_preview: unexpected error for '%s' — uploading original only",
+            safe_filename,
+        )
+        return None
+
+    if target.is_file():
+        return target
+
+    logger.warning(
+        "office_preview: soffice returned 0 but '%s' is missing for '%s'",
+        target.name,
+        safe_filename,
+    )
+    return None

{unique_toolkit-2026.34.2 → unique_toolkit-2026.36.0}/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",