unique_toolkit 2026.38.0__tar.gz → 2026.38.0.dev0__tar.gz

{unique_toolkit-2026.38.0 → unique_toolkit-2026.38.0.dev0}/PKG-INFO

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

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

{unique_toolkit-2026.38.0 → unique_toolkit-2026.38.0.dev0}/unique_toolkit/_common/utils/files.py

@@ -4,21 +4,6 @@ import mimetypes
 from enum import StrEnum
 from pathlib import Path
 
-# Takes precedence over `mimetypes` (which lacks `.msg` and varies by host)
-# so classification matches the platform's supported-mime-types catalog.
-_EXTENSION_MIME_OVERRIDES: dict[str, str] = {
-    ".msg": "application/vnd.ms-outlook",
-    ".eml": "message/rfc822",
-}
-
-
-def guess_mime_type(filename: str | Path) -> str | None:
-    suffix = Path(filename).suffix.lower()
-    if override := _EXTENSION_MIME_OVERRIDES.get(suffix):
-        return override
-    mime_type, _ = mimetypes.guess_type(str(filename))
-    return mime_type
-
 
 class FileMimeType(StrEnum):
     PDF = "application/pdf"
@@ -38,12 +23,10 @@ class FileMimeType(StrEnum):
     MD = "text/markdown"
     TXT = "text/plain"
     JSON = "application/json"
-    MSG = "application/vnd.ms-outlook"
-    EML = "message/rfc822"
 
     @classmethod
     def get_mime_from_file_path(cls, filepath: Path) -> "FileMimeType | None":
-        mime_type = guess_mime_type(filepath)
+        mime_type, _ = mimetypes.guess_type(filepath)
         if mime_type is None:
             return None
 
@@ -83,11 +66,6 @@ class FileMimeType(StrEnum):
     def is_csv(self) -> bool:
         return self == self.__class__.CSV
 
-    @property
-    def is_email(self) -> bool:
-        cls = self.__class__
-        return self in {cls.MSG, cls.EML}
-
     @classmethod
     def is_docx_mime(cls, filepath: Path) -> bool:
         mime_type = cls.get_mime_from_file_path(filepath)
@@ -144,22 +122,12 @@ def get_common_name(extension: FileMimeType) -> str:
             return "markdown"
         case FileMimeType.HTML:
             return "html"
-        case FileMimeType.MSG | FileMimeType.EML:
-            return "email"
         case _:  # pyright: ignore[reportUnnecessaryComparison]
             return "unknown"  # pyright: ignore[reportUnreachable]
 
 
-_MIME_EXTENSION_OVERRIDES: dict[str, str] = {
-    mime: ext for ext, mime in _EXTENSION_MIME_OVERRIDES.items()
-}
-
-
 def get_file_extensions(mimes: list[FileMimeType]) -> list[str]:
-    types = [
-        _MIME_EXTENSION_OVERRIDES.get(mime) or mimetypes.guess_extension(mime)
-        for mime in mimes
-    ]
+    types = [mimetypes.guess_extension(mime) for mime in mimes]
     return [t for t in types if t]
 
 
@@ -174,7 +142,7 @@ class ImageMimeType(StrEnum):
 
 
 def is_file_content(filename: str) -> bool:
-    mimetype = guess_mime_type(filename)
+    mimetype, _ = mimetypes.guess_type(filename)
 
     if not mimetype:
         return False
@@ -183,7 +151,7 @@ def is_file_content(filename: str) -> bool:
 
 
 def is_image_content(filename: str) -> bool:
-    mimetype = guess_mime_type(filename)
+    mimetype, _ = mimetypes.guess_type(filename)
 
     if not mimetype:
         return False

{unique_toolkit-2026.38.0 → unique_toolkit-2026.38.0.dev0}/unique_toolkit/agentic/tools/experimental/ask_user_tool/config.py

@@ -11,32 +11,26 @@ from unique_toolkit.elicitation import (
 )
 
 _DEFAULT_TOOL_DESCRIPTION = """
-Ask the user a question through a structured form and block until they answer. Use whenever you need information, a decision, or confirmation before continuing correctly: clarifying questions, confirmation before destructive or irreversible work, missing parameters, or choosing among plausible options. Do not ask the user in normal chat text while this tool is available—call AskUser, wait for the result, then continue. If unsure which interpretation, scope, or value applies, call AskUser instead of guessing. The form's Confirm/Decline buttons are the user's consent: for destructive actions state the consequence in `message` and proceed only when the user confirms; use an empty-properties schema when there is nothing else to ask. Form and chat anchoring are handled for you; the call blocks until the user responds, declines, cancels, or times out.
+Ask the user a question through a structured form and block until they answer. Use whenever you need information, a decision, or confirmation before continuing correctly: clarifying questions, confirmation before destructive or irreversible work, missing parameters, or choosing among plausible options. Do not ask the user in normal chat text while this tool is available—call AskUser, wait for the result, then continue. If unsure which interpretation, scope, or value applies, call AskUser instead of guessing. Destructive actions must include an explicit confirmation field. Form and chat anchoring are handled for you; the call blocks until the user responds, declines, cancels, or times out.
 """.strip()
 
 _DEFAULT_TOOL_DESCRIPTION_FOR_SYSTEM_PROMPT = """
 - If your next message would ask the user for information, a decision, or confirmation, call AskUser instead of writing the question in chat.
 - If more than one reasonable reading of the request exists, call AskUser before acting—do not silently pick a default.
-- Destructive or irreversible actions need the user's confirmation: state the consequence in `message` and proceed only on an accepted result. The Confirm button is the consent on any form, so never model it as a boolean `confirm` field; if there is nothing else to ask, use `response_schema` `{"type":"object","properties":{}}`.
+- Destructive or irreversible actions require an explicit confirmation field (e.g. boolean `confirm`) in `response_schema`.
 """.strip()
 
 _DEFAULT_MESSAGE_DESCRIPTION = "The question or prompt shown to the user"
 
 _DEFAULT_RESPONSE_SCHEMA_DESCRIPTION = """
-JSON Schema object (root type "object") for the form shown to the user; you construct it. The Confirm/Decline buttons are the user's consent on any form, so a confirmation needs no field of its own: empty properties for a plain yes/no; single string property for free-text; string + enum for one choice (dropdown); array with items.enum for choose-many; boolean only for a genuine data field where unchecked (false) is a valid answer. Mark needed fields required and add short property descriptions (shown as help text). A required boolean means "must be answered", not "must be ticked": the user can submit it unchecked. Give every boolean a title, it is the checkbox label. Prefer one small form over many calls.
+JSON Schema object (root type "object") for the form shown to the user; you construct it. Single string property for free-text; boolean for a confirmation; string + enum for one choice (dropdown); array with items.enum for choose-many. Mark needed fields required and add short property descriptions (shown as help text). Prefer one small form over many calls.
 Examples:
-    - confirm: {"type":"object","properties":{}};
     - free-text: {"type":"object","properties":{"answer":{"type":"string"}},"required":["answer"]};
+    - confirm: {"type":"object","properties":{"confirm":{"type":"boolean"}},"required":["confirm"]};
     - choice: {"type":"object","properties":{"region":{"type":"string","enum":["EU","US"]}},"required":["region"]};
-    - multi: {"type":"object","properties":{"topics":{"type":"array","items":{"type":"string","enum":["a","b"]}}}};
-    - yes/no data: {"type":"object","properties":{"include_appendix":{"type":"boolean","title":"Include appendix"}}}.
+    - multi: {"type":"object","properties":{"topics":{"type":"array","items":{"type":"string","enum":["a","b"]}}}}.
 """.strip()
 
-_DEFAULT_ACCEPTED_MESSAGE = (
-    "The user confirmed the request. The form had no fields, so there is no "
-    "response content. Proceed with the confirmed action."
-)
-
 _TIMEOUT_SECONDS_MIN = 1
 _TIMOUT_SECONDS_MAX = 3600
 
@@ -87,19 +81,6 @@ class AskUserToolConfig(BaseToolConfig):
         title="`response_schema` tool input field description",
     )
 
-    accepted_message: Annotated[
-        str,
-        RJSFMetaTag.StringWidget.textarea(rows=2),
-    ] = Field(
-        default=_DEFAULT_ACCEPTED_MESSAGE,
-        description=(
-            "Message returned to the model when the user accepts a form without "
-            "fields (empty-properties confirmation schema). Forms with fields "
-            "return the submitted values as JSON instead."
-        ),
-        title="Accepted response (empty form)",
-    )
-
     declined_message: Annotated[
         str,
         RJSFMetaTag.StringWidget.textarea(

{unique_toolkit-2026.38.0 → unique_toolkit-2026.38.0.dev0}/unique_toolkit/agentic/tools/experimental/ask_user_tool/tool.py

@@ -158,17 +158,8 @@ class AskUserTool(Tool[AskUserToolConfig]):
 
         _LOGGER.info("User responded to elicitation request")
 
-        # Decide on the schema, not the payload: a form with optional fields left
-        # blank also comes back empty, and that is an answer ("{}"), not consent.
-        if not params.response_schema.get("properties"):
-            return ToolCallResponse(
-                id=tool_call.id,
-                name=self.name,
-                content=self.config.accepted_message,
-            )
-
         return ToolCallResponse(
             id=tool_call.id,
             name=self.name,
-            content=json.dumps(elicitation.response_content or {}, ensure_ascii=False),
+            content=json.dumps(elicitation.response_content, ensure_ascii=False),
         )

{unique_toolkit-2026.38.0 → unique_toolkit-2026.38.0.dev0}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/artifacts.py

@@ -6,6 +6,7 @@ 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
@@ -15,9 +16,6 @@ from unique_toolkit.agentic.tools.openai_builtin.code_interpreter.postprocessors
     convert_office_bytes_to_preview_pdf,
     office_extension,
 )
-from unique_toolkit.agentic.tools.openai_builtin.code_interpreter.schemas import (
-    CodeInterpreterContainerFile,
-)
 from unique_toolkit.content.schemas import Content
 
 _CODE_EXECUTION_ARTIFACT_METADATA_KEY: str = "codeExecutionArtifactMetadata"
@@ -63,7 +61,7 @@ def _kb_safe_mime(mime: str) -> str:
 
 
 def _artifact_metadata(
-    file: CodeInterpreterContainerFile,
+    file: AnnotationContainerFileCitation,
 ) -> dict[str, Any]:
     return {
         _CODE_EXECUTION_ARTIFACT_METADATA_KEY: CodeExecutionArtifactMetadata(
@@ -76,7 +74,7 @@ def _artifact_metadata(
 
 async def _upload_artifact_bytes(
     chat_service: ChatService,
-    file: CodeInterpreterContainerFile,
+    file: AnnotationContainerFileCitation,
     file_bytes: bytes,
     mime: str,
 ) -> Content:
@@ -99,7 +97,7 @@ async def _upload_artifact_bytes(
 
 async def _upload_office_artifact_with_preview(
     chat_service: ChatService,
-    file: CodeInterpreterContainerFile,
+    file: AnnotationContainerFileCitation,
     file_bytes: bytes,
     mime: str,
 ) -> Content | None:
@@ -148,7 +146,7 @@ async def _upload_office_artifact_with_preview(
 
 async def save_code_execution_artifact(
     chat_service: ChatService,
-    file: CodeInterpreterContainerFile,
+    file: AnnotationContainerFileCitation,
     file_bytes: bytes,
     *,
     attach_office_preview: bool = False,

{unique_toolkit-2026.38.0 → unique_toolkit-2026.38.0.dev0}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py

@@ -6,10 +6,10 @@ import time
 from dataclasses import dataclass
 from mimetypes import guess_type
 from typing import NamedTuple, TypedDict, override
-from urllib.parse import quote, unquote
 
 import httpx
 from openai import AsyncOpenAI
+from openai.types.responses.response_output_text import AnnotationContainerFileCitation
 from pydantic import BaseModel, Field, RootModel
 from pydantic.json_schema import SkipJsonSchema
 from tenacity import (
@@ -34,7 +34,6 @@ from unique_toolkit.agentic.tools.openai_builtin.code_interpreter.postprocessors
 )
 from unique_toolkit.agentic.tools.openai_builtin.code_interpreter.schemas import (
     CodeInterpreterBlock,
-    CodeInterpreterContainerFile,
     CodeInterpreterFile,
     CodeInterpreterFileType,
 )
@@ -312,7 +311,7 @@ class _FileProgressTracker:
             if state.phase == "pending":
                 continue
             progress = self._format_inline(filename, state)
-            pattern = _sandbox_link_pattern(filename)
+            pattern = rf"!?\[.*?\]\(sandbox:/mnt/data/{re.escape(filename)}\)"
             text = re.sub(pattern, progress, text)
 
         active = {
@@ -402,7 +401,6 @@ class DisplayCodeInterpreterFilesPostProcessor(
 
         # Cleared at the start of each run(); holds this-turn download sizes.
         self._file_size_map: dict[str, int] = {}
-        self._container_files: list[CodeInterpreterContainerFile] = []
 
         # Resolved in run() (before apply_postprocessing_to_response) since flag evaluation is async.
         self._fence_ff_on = False
@@ -421,89 +419,6 @@ class DisplayCodeInterpreterFilesPostProcessor(
             reraise=True,
         )
 
-    async def _resolve_container_files(
-        self, loop_response: ResponsesLanguageModelStreamResponse
-    ) -> list[CodeInterpreterContainerFile]:
-        """
-        Matches files cited by the agent to their id.
-        First tries to match files using the `citations` in the llm response.
-        If that fails, uses `container.files.list` in order to find the correct file
-        """
-        container_files = [
-            CodeInterpreterContainerFile(
-                container_id=annotation.container_id,
-                file_id=annotation.file_id,
-                filename=annotation.filename,
-            )
-            for annotation in loop_response.container_files
-        ]
-
-        text = loop_response.message.text or ""
-        cited_files = set(_extract_container_file_citations_from_text(text))
-
-        unresolved_files = cited_files - {file.filename for file in container_files}
-
-        if len(unresolved_files) == 0:
-            return container_files
-
-        code_interpreter_calls = loop_response.code_interpreter_calls
-
-        if len(code_interpreter_calls) == 0:
-            return container_files
-
-        container_id = code_interpreter_calls[0].container_id
-
-        self._log.info(
-            "Recovering %d unresolved sandbox link(s) from container %s: %s",
-            len(unresolved_files),
-            container_id,
-            list(unresolved_files),
-        )
-
-        try:
-            async for container_file in self._client.containers.files.list(
-                container_id=container_id,
-                order="desc",
-            ):
-                if container_file.source != "assistant":
-                    continue
-
-                path = container_file.path
-                filename = (
-                    path.removeprefix("/mnt/data/")
-                    if path.startswith("/mnt/data/")
-                    else path
-                )
-
-                if filename in unresolved_files:
-                    unresolved_files.remove(filename)
-                    container_files.append(
-                        CodeInterpreterContainerFile(
-                            container_id=container_file.container_id,
-                            file_id=container_file.id,
-                            filename=filename,
-                        )
-                    )
-
-                if len(unresolved_files) == 0:
-                    break
-
-        except Exception:
-            self._log.exception(
-                "Failed to list files for container %s while recovering "
-                "missing citations; dangling-link handling will continue.",
-                container_id,
-            )
-
-        if len(unresolved_files) > 0:
-            self._log.info(
-                "%d sandbox link(s) left unresolved after recovery: %s",
-                len(unresolved_files),
-                list(unresolved_files),
-            )
-
-        return container_files
-
     @override
     async def run(
         self, loop_response: ResponsesLanguageModelStreamResponse
@@ -522,10 +437,9 @@ class DisplayCodeInterpreterFilesPostProcessor(
             company_id=self._company_id or COMPANY_ID_PLACEHOLDER,
         )
 
-        container_files = await self._resolve_container_files(loop_response)
-        self._container_files = container_files
+        container_files = loop_response.container_files
         self._log.info(
-            "run() resolved %d container file(s): %s",
+            "run() found %d container file annotation(s): %s",
             len(container_files),
             [cf.filename for cf in container_files],
         )
@@ -621,12 +535,10 @@ class DisplayCodeInterpreterFilesPostProcessor(
             save_ms,
         )
 
-        return self._compute_artifacts_debug_info(loop_response, container_files)
+        return self._compute_artifacts_debug_info(loop_response)
 
     def _compute_artifacts_debug_info(
-        self,
-        loop_response: ResponsesLanguageModelStreamResponse,
-        container_files: list[CodeInterpreterContainerFile],
+        self, loop_response: ResponsesLanguageModelStreamResponse
     ) -> ArtifactsDebugInfo | None:
         """Compute this turn's successfully-created artifacts.
 
@@ -650,7 +562,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
 
         created_filenames = {
             cf.filename
-            for cf in container_files
+            for cf in loop_response.container_files
             if self._content_map.get(cf.filename) is not None
         }
         total_bytes = sum(
@@ -767,10 +679,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
 
         if self._fence_ff_on:
             code_blocks = _build_code_blocks(
-                loop_response,
-                self._content_map,
-                self._container_files,
-                include_html=self._html_fence_ff_on,
+                loop_response, self._content_map, include_html=self._html_fence_ff_on
             )
             self._log.info(
                 "Fence injection — %d code block(s), files: %s",
@@ -824,7 +733,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
     @failsafe_async(failure_return_value=None, logger=logger)
     async def _download_and_upload_container_files_to_knowledge_base(
         self,
-        container_file: CodeInterpreterContainerFile,
+        container_file: AnnotationContainerFileCitation,
         semaphore: asyncio.Semaphore,
         tracker: _FileProgressTracker | None = None,
     ) -> _ContentInfo | None:
@@ -895,7 +804,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
 
     async def _download_file_bytes_with_progress(
         self,
-        container_file: CodeInterpreterContainerFile,
+        container_file: AnnotationContainerFileCitation,
         tracker: _FileProgressTracker | None,
     ) -> bytes:
         """Download container file with streaming progress and manual retry loop.
@@ -978,7 +887,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
 
     async def _stream_download_bytes(
         self,
-        container_file: CodeInterpreterContainerFile,
+        container_file: AnnotationContainerFileCitation,
         tracker: _FileProgressTracker | None,
         retry_attempt: int,
         download_start: float,
@@ -1286,7 +1195,6 @@ def _inject_code_execution_fences(
 def _build_code_blocks(
     loop_response: ResponsesLanguageModelStreamResponse,
     content_map: dict[str, str | None],
-    container_files: list[CodeInterpreterContainerFile],
     include_html: bool = False,
 ) -> list[CodeInterpreterBlock]:
     """Map each code interpreter call to the files it produced via /mnt/data/ path matching.
@@ -1317,7 +1225,7 @@ def _build_code_blocks(
     for idx, call in enumerate(calls):
         if not call.code:
             continue
-        for annotation in container_files:
+        for annotation in loop_response.container_files:
             if (
                 content_map.get(annotation.filename) is not None
                 and f"/mnt/data/{annotation.filename}" in call.code
@@ -1346,7 +1254,7 @@ def _build_code_blocks(
     for idx, call in enumerate(calls):
         if not call.code:
             continue
-        for annotation in container_files:
+        for annotation in loop_response.container_files:
             if annotation.filename in primary_matched_filenames:
                 continue
             if content_map.get(annotation.filename) is None:
@@ -1373,7 +1281,7 @@ def _build_code_blocks(
         if call.code:
             last_block_idx = idx
     if last_block_idx is not None:
-        for annotation in container_files:
+        for annotation in loop_response.container_files:
             if annotation.filename in file_to_block_idx:
                 continue
             if content_map.get(annotation.filename) is None:
@@ -1386,7 +1294,7 @@ def _build_code_blocks(
     # file appears exactly once per block (last annotation wins, consistent with
     # the last-writer-wins rule applied in step 1).
     block_file_map: dict[int, dict[str, CodeInterpreterFile]] = {}
-    for annotation in container_files:
+    for annotation in loop_response.container_files:
         if not include_html and _get_file_type(annotation.filename) == "html":
             continue
         content_id = content_map.get(annotation.filename)
@@ -1437,24 +1345,10 @@ def _warn_missing_content_ids(text: str, content_map: dict[str, str | None]) ->
 # so the entire `[label](sandbox://...)` token can be replaced rather than just the URL.
 _SANDBOX_MARKDOWN_LINK_RE = re.compile(r"!?\[.*?\]\(sandbox:/mnt/data/\S+?\)")
 
-
-def _sandbox_link_pattern(filename: str) -> str:
-    """Return a pattern matching the markdown sandbox link for `filename`."""
-    spellings = "|".join(
-        dict.fromkeys([re.escape(filename), re.escape(quote(filename))])
-    )
-    return rf"!?\[.*?\]\(sandbox:/mnt/data/(?:{spellings})\)"
-
-
 # Extracts the bare filename from a sandbox URL like `sandbox:/mnt/data/foo.csv`.
 _SANDBOX_FILENAME_RE = re.compile(r"sandbox:/mnt/data/([^)\s]+)")
 
 
-def _extract_container_file_citations_from_text(text: str) -> list[str]:
-    """Extract container filenames from sandbox citations in their text order."""
-    return [unquote(filename) for filename in _SANDBOX_FILENAME_RE.findall(text)]
-
-
 def _replace_dangling_sandbox_links(
     text: str, *, ci_ran: bool = True
 ) -> tuple[str, bool]:
@@ -1477,8 +1371,8 @@ def _replace_dangling_sandbox_links(
         return text, False
 
     def _replacement(m: re.Match[str]) -> str:
-        filenames = _extract_container_file_citations_from_text(m.group())
-        filename = filenames[0] if filenames else "the file"
+        filename_match = _SANDBOX_FILENAME_RE.search(m.group())
+        filename = filename_match.group(1) if filename_match else "the file"
         logger.warning(
             "Dangling sandbox link found in final text: '%s'. ci_ran=%s. "
             "The file was either never generated or the link format did not match "
@@ -1539,7 +1433,7 @@ def _get_next_ref_number(references: list[ContentReference] | None) -> int:
 def _replace_container_file_error(
     text: str, filename: str, error_message: str
 ) -> tuple[str, bool]:
-    image_markdown = _sandbox_link_pattern(filename)
+    image_markdown = rf"!?\[.*?\]\(sandbox:/mnt/data/{re.escape(filename)}\)"
 
     if not re.search(image_markdown, text):
         logger.warning(
@@ -1560,7 +1454,7 @@ def _replace_container_file_error(
 def _replace_container_image_citation(
     text: str, filename: str, content_id: str
 ) -> tuple[str, bool]:
-    image_markdown = _sandbox_link_pattern(filename)
+    image_markdown = rf"!?\[.*?\]\(sandbox:/mnt/data/{re.escape(filename)}\)"
 
     if not re.search(image_markdown, text):
         logger.warning(
@@ -1582,7 +1476,7 @@ def _replace_container_image_citation(
 def _replace_container_html_citation(
     text: str, filename: str, content_id: str
 ) -> tuple[str, bool]:
-    link_core = _sandbox_link_pattern(filename)
+    link_core = rf"!?\[.*?\]\(sandbox:/mnt/data/{re.escape(filename)}\)"
     html_markdown = link_core
 
     if not re.search(html_markdown, text):
@@ -1640,7 +1534,7 @@ def _replace_container_file_citation(
     original pre-fence behaviour is restored: the link is replaced with <sup>N</sup>
     and the file remains accessible via the references panel.
     """
-    file_markdown = _sandbox_link_pattern(filename)
+    file_markdown = rf"!?\[.*?\]\(sandbox:/mnt/data/{re.escape(filename)}\)"
 
     if not re.search(file_markdown, text):
         logger.warning(

{unique_toolkit-2026.38.0 → unique_toolkit-2026.38.0.dev0}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/schemas.py

@@ -1,6 +1,5 @@
 """Code interpreter types used by the generated-files postprocessor."""
 
-from dataclasses import dataclass
 from typing import Literal
 
 from pydantic import BaseModel
@@ -8,15 +7,6 @@ from pydantic import BaseModel
 CodeInterpreterFileType = Literal["image", "document", "html"]
 
 
-@dataclass(frozen=True)
-class CodeInterpreterContainerFile:
-    """Provider-independent identity of a file in a code interpreter container."""
-
-    container_id: str
-    file_id: str
-    filename: str
-
-
 class CodeInterpreterFile(BaseModel):
     """A single file produced by a code interpreter execution."""
 

{unique_toolkit-2026.38.0 → unique_toolkit-2026.38.0.dev0}/unique_toolkit/agentic_table/schemas.py

@@ -1,4 +1,3 @@
-from datetime import datetime
 from enum import StrEnum
 from typing import Annotated, Any, Generic, Literal, TypeVar, override
 
@@ -502,15 +501,6 @@ class MagicTableSheet(BaseModel):
     created_by: str
     company_id: str
     created_at: str
-    user_aborted_at: datetime | None = Field(
-        default=None,
-        description=(
-            "When the user stopped an in-progress refresh. Set by "
-            "stopMagicTableRefresh and returned on GET sheet/library so the "
-            "agent can poll and exit. None when the run was not aborted or "
-            "when an older backend omits the field."
-        ),
-    )
     magic_table_cells: list[MagicTableCell] = Field(
         default_factory=list, description="The cells in the sheet"
     )