unique_toolkit 1.68.0__tar.gz → 1.68.2__tar.gz

{unique_toolkit-1.68.0 → unique_toolkit-1.68.2}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.68.0"
+version = "1.68.2"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -84,11 +84,15 @@ dev = [
 ]
 
 [tool.uv]
+exclude-newer = "2 weeks"
 constraint-dependencies = [
     "starlette>=0.49.1",
     "lxml>=5.0.0",
 ]
 
+[tool.uv.exclude-newer-package]
+"unique-sdk" = false
+
 [tool.poe.tasks]
 lint = "ruff check ."
 lint-fix = "ruff check . --fix"

{unique_toolkit-1.68.0 → unique_toolkit-1.68.2}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py

@@ -10,6 +10,12 @@ from openai.types.responses import ResponseCodeInterpreterToolCall
 from openai.types.responses.response_output_text import AnnotationContainerFileCitation
 from pydantic import BaseModel, Field, RootModel
 from pydantic.json_schema import SkipJsonSchema
+from tenacity import (
+    AsyncRetrying,
+    before_sleep_log,
+    stop_after_attempt,
+    wait_exponential,
+)
 
 from unique_toolkit import ChatService
 from unique_toolkit._common.execution import failsafe_async
@@ -54,6 +60,14 @@ class DisplayCodeInterpreterFilesPostProcessorConfig(BaseModel):
         default=10,
         description="The maximum number of concurrent file downloads.",
     )
+    max_download_retries: int = Field(
+        default=2,
+        description="Maximum number of additional download attempts per container file after the first try (0 = no retries).",
+    )
+    download_retry_base_delay: float = Field(
+        default=0.5,
+        description="Base delay in seconds for exponential backoff between download/upload retries.",
+    )
 
 
 class _ContentInfo(NamedTuple):
@@ -279,9 +293,18 @@ class DisplayCodeInterpreterFilesPostProcessor(
         semaphore: asyncio.Semaphore,
     ) -> _ContentInfo | None:
         async with semaphore:
-            logger.info("Fetching file content for %s", container_file.filename)
-            file_content = await self._client.containers.files.content.retrieve(
-                container_id=container_file.container_id, file_id=container_file.file_id
+            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,
+            )
+            file_content = await retry(
+                self._client.containers.files.content.retrieve,
+                container_id=container_file.container_id,
+                file_id=container_file.file_id,
             )
 
             logger.info(