unique_toolkit 1.68.8__tar.gz → 1.68.9__tar.gz

{unique_toolkit-1.68.8 → unique_toolkit-1.68.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.68.8
+Version: 1.68.9
 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.68.8 → unique_toolkit-1.68.9}/pyproject.toml

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

{unique_toolkit-1.68.8 → unique_toolkit-1.68.9}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py

@@ -131,6 +131,19 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 chat_id=chat_id,
             )
 
+    def _build_retry(self) -> AsyncRetrying:
+        """Build a tenacity retry policy from the current config.
+
+        Reused for both container-file downloads and knowledge-base uploads so
+        that every I/O call gets the same exponential-backoff behaviour.
+        """
+        return AsyncRetrying(
+            stop=stop_after_attempt(1 + self._config.max_download_retries),
+            wait=wait_exponential(multiplier=self._config.download_retry_base_delay),
+            before_sleep=before_sleep_log(logger, logging.WARNING),
+            reraise=True,
+        )
+
     @override
     async def run(self, loop_response: ResponsesLanguageModelStreamResponse) -> None:
         logger.info("Fetching and adding code interpreter files to the response")
@@ -138,7 +151,14 @@ class DisplayCodeInterpreterFilesPostProcessor(
         container_files = loop_response.container_files
         logger.info("Found %s container files", len(container_files))
 
-        self._content_map: dict[str, str | None] = await self._load_previous_files()  # type: ignore
+        try:
+            self._content_map: dict[str, str | None] = await self._load_previous_files()  # type: ignore
+        except Exception:
+            logger.exception(
+                "Failed to load previous code interpreter files from short-term "
+                "memory; proceeding without previous file context."
+            )
+            self._content_map = {}
 
         semaphore = asyncio.Semaphore(self._config.max_concurrent_file_downloads)
         tasks = [
@@ -155,21 +175,34 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 result.content_id if result is not None else None
             )
 
-        await self._save_generated_files(
-            [
-                _ContentInfo(filename=filename, content_id=content_id)
-                for filename, content_id in self._content_map.items()
-                if content_id is not None
-            ]
-        )
+        try:
+            await self._save_generated_files(
+                [
+                    _ContentInfo(filename=filename, content_id=content_id)
+                    for filename, content_id in self._content_map.items()
+                    if content_id is not None
+                ]
+            )
+        except Exception:
+            logger.exception(
+                "Failed to save generated code interpreter files to short-term "
+                "memory; file replacement will still proceed."
+            )
 
-        if feature_flags.enable_code_execution_fence_un_17972.is_enabled(
-            self._company_id
-        ):
-            self._orphan_code_blocks = await self._upload_orphan_code_as_txt(
-                loop_response
+        try:
+            if feature_flags.enable_code_execution_fence_un_17972.is_enabled(
+                self._company_id
+            ):
+                self._orphan_code_blocks = await self._upload_orphan_code_as_txt(
+                    loop_response
+                )
+            else:
+                self._orphan_code_blocks = []
+        except Exception:
+            logger.exception(
+                "Failed to process orphan code blocks; "
+                "file replacement will still proceed."
             )
-        else:
             self._orphan_code_blocks = []
 
     @override
@@ -279,14 +312,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
         semaphore: asyncio.Semaphore,
     ) -> _ContentInfo | None:
         async with semaphore:
-            retry = AsyncRetrying(
-                stop=stop_after_attempt(1 + self._config.max_download_retries),
-                wait=wait_exponential(
-                    multiplier=self._config.download_retry_base_delay
-                ),
-                before_sleep=before_sleep_log(logger, logging.WARNING),
-                reraise=True,
-            )
+            retry = self._build_retry()
             file_content = await retry(
                 self._client.containers.files.content.retrieve,
                 container_id=container_file.container_id,
@@ -299,7 +325,8 @@ class DisplayCodeInterpreterFilesPostProcessor(
             )
 
             assert self._chat_service is not None  # Checked in __init__
-            content = await self._chat_service.upload_to_chat_from_bytes_async(
+            content = await self._build_retry()(
+                self._chat_service.upload_to_chat_from_bytes_async,
                 content=file_content.content,
                 content_name=container_file.filename,
                 mime_type=guess_type(container_file.filename)[0] or "text/plain",
@@ -378,7 +405,8 @@ class DisplayCodeInterpreterFilesPostProcessor(
             filename = f"code_{i + 1}.txt" if use_numeric_suffix else "code.txt"
 
             try:
-                content = await self._chat_service.upload_to_chat_from_bytes_async(
+                content = await self._build_retry()(
+                    self._chat_service.upload_to_chat_from_bytes_async,
                     content=txt_content.encode("utf-8"),
                     content_name=filename,
                     mime_type="text/plain",
@@ -938,7 +966,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 = rf"\[.*?\]\(sandbox:/mnt/data/{re.escape(filename)}\)"
+    file_markdown = rf"!?\[.*?\]\(sandbox:/mnt/data/{re.escape(filename)}\)"
 
     if not re.search(file_markdown, text):
         logger.warning(