unique_toolkit 1.53.3__tar.gz → 1.53.4__tar.gz

{unique_toolkit-1.53.3 → unique_toolkit-1.53.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.53.3
+Version: 1.53.4
 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>

{unique_toolkit-1.53.3 → unique_toolkit-1.53.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.53.3"
+version = "1.53.4"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

{unique_toolkit-1.53.3 → unique_toolkit-1.53.4}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py

@@ -194,10 +194,17 @@ class DisplayCodeInterpreterFilesPostProcessor(
 
             # Files (including HTML when feature flag is disabled)
             else:
+                fence_ff_on = (
+                    feature_flags.enable_code_execution_fence_un_17972.is_enabled(
+                        self._company_id
+                    )
+                )
                 loop_response.message.text, replaced = _replace_container_file_citation(
                     text=loop_response.message.text,
                     filename=filename,
                     content_id=content_id,
+                    ref_number=ref_number,
+                    use_content_link=fence_ff_on,
                 )
                 changed |= replaced
 
@@ -547,8 +554,20 @@ unique://content/{content_id}
 
 
 def _replace_container_file_citation(
-    text: str, filename: str, content_id: str
+    text: str,
+    filename: str,
+    content_id: str,
+    ref_number: int,
+    use_content_link: bool,
 ) -> tuple[str, bool]:
+    """Replace a sandbox file link with either an inline content link or a superscript ref.
+
+    When the fence feature flag is on (use_content_link=True), the sandbox link is
+    replaced with [filename](unique://content/{id}) so the subsequent fence injection
+    step can locate and wrap it. When the flag is off (use_content_link=False), the
+    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 = rf"\[.*?\]\(sandbox:/mnt/data/{re.escape(filename)}\)"
 
     if not re.search(file_markdown, text):
@@ -556,8 +575,9 @@ def _replace_container_file_citation(
         return text, False
 
     logger.info("Displaying file %s", filename)
-    return re.sub(
-        file_markdown,
-        f"[{filename}](unique://content/{content_id})",
-        text,
-    ), True
+    replacement = (
+        f"[{filename}](unique://content/{content_id})"
+        if use_content_link
+        else f"<sup>{ref_number}</sup>"
+    )
+    return re.sub(file_markdown, replacement, text), True