unique_toolkit 1.68.6__tar.gz → 1.68.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.68.6
+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>
@@ -35,9 +35,9 @@ Requires-Dist: fastapi>=0.121.0,<0.122 ; extra == 'fastapi'
 Requires-Dist: starlette>=0.49.1 ; extra == 'fastapi'
 Requires-Dist: uvicorn>=0.37.0,<0.38 ; extra == 'fastapi'
 Requires-Dist: openai>=1.105.0,<2 ; extra == 'langchain'
-Requires-Dist: langchain>=0.3.27,<0.4 ; extra == 'langchain'
-Requires-Dist: langchain-openai>=0.3.28,<0.4 ; extra == 'langchain'
-Requires-Dist: langchain-core>=0.3.0,<0.4 ; extra == 'langchain'
+Requires-Dist: langchain>=1.0.0,<2 ; extra == 'langchain'
+Requires-Dist: langchain-openai>=0.3.28 ; extra == 'langchain'
+Requires-Dist: langchain-core>=1.0.0,<2 ; extra == 'langchain'
 Requires-Python: >=3.12
 Provides-Extra: fastapi
 Provides-Extra: langchain

{unique_toolkit-1.68.6 → unique_toolkit-1.68.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.68.6"
+version = "1.68.9"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -48,9 +48,9 @@ dependencies = [
 [project.optional-dependencies]
 langchain = [
     "openai>=1.105.0,<2",
-    "langchain>=0.3.27,<0.4",
-    "langchain-openai>=0.3.28,<0.4",
-    "langchain-core>=0.3.0,<0.4",
+    "langchain>=1.0.0,<2",
+    "langchain-openai>=0.3.28",
+    "langchain-core>=1.0.0,<2",
 ]
 fastapi = [
     "fastapi>=0.121.0,<0.122",
@@ -66,22 +66,7 @@ build-backend = "uv_build"
 module-root = ""
 
 [dependency-groups]
-dev = [
-    "ruff==0.12.10",
-    "pytest>=7.4.3,<8",
-    "tox>=4.11.4,<5",
-    "pyright>=1.1.341,<2",
-    "basedpyright>=1.33.0,<2",
-    "pytest-cov>=4.1.0,<5",
-    "pre-commit>=3.7.1,<4",
-    "pytest-asyncio>=0.23.8,<0.24",
-    "pytest-mock>=3.14.0,<4",
-    "ipykernel>=6.29.5,<7",
-    "deptry>=0.24.0,<0.25",
-    "poethepoet>=0.32.0,<0.33",
-    "isort>=6.0.1,<7",
-    "entangled-cli>=2.4.0,<3",
-]
+dev = []
 
 [tool.uv]
 exclude-newer = "2 weeks"
@@ -93,6 +78,9 @@ constraint-dependencies = [
 [tool.uv.exclude-newer-package]
 "unique-sdk" = false
 
+[tool.uv.sources]
+unique-sdk = { workspace = true }
+
 [tool.poe.tasks]
 lint = "ruff check ."
 lint-fix = "ruff check . --fix"

{unique_toolkit-1.68.6 → 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
@@ -211,14 +244,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 )
                 changed |= replaced
 
-            # HTML rendered as legacy HtmlRendering block — only when fence FF is off
-            elif (
-                is_html
-                and not fence_ff_on
-                and feature_flags.enable_html_rendering_un_15131.is_enabled(
-                    self._company_id
-                )
-            ):
+            elif is_html:
                 loop_response.message.text, replaced = _replace_container_html_citation(
                     text=loop_response.message.text,
                     filename=filename,
@@ -226,7 +252,6 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 )
                 changed |= replaced
 
-            # Files and HTML (fence FF on: HTML → htmlWithSource; others → fileWithSource)
             else:
                 loop_response.message.text, replaced = _replace_container_file_citation(
                     text=loop_response.message.text,
@@ -237,13 +262,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 )
                 changed |= replaced
 
-            # htmlWithSource / HtmlRendering blocks carry contentId directly — no separate reference needed
-            is_html_rendered = is_html and (
-                fence_ff_on
-                or feature_flags.enable_html_rendering_un_15131.is_enabled(
-                    self._company_id
-                )
-            )
+            is_html_rendered = is_html
             if replaced and not (is_image or is_html_rendered or fence_ff_on):
                 loop_response.message.references.append(
                     ContentReference(
@@ -293,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,
@@ -313,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",
@@ -392,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",
@@ -732,8 +746,12 @@ def _build_code_blocks(
     # overwritten across executions. Using a dict keyed by filename ensures each
     # file appears exactly once per block (last annotation wins, consistent with
     # the last-writer-wins rule applied in step 1).
+    # HTML files are excluded: they are always rendered via HtmlRendering blocks
+    # and never participate in fence injection.
     block_file_map: dict[int, dict[str, CodeInterpreterFile]] = {}
     for annotation in loop_response.container_files:
+        if _get_file_type(annotation.filename) == "html":
+            continue
         content_id = content_map.get(annotation.filename)
         if content_id is None:
             continue
@@ -821,6 +839,8 @@ def _warn_unmatched_code_blocks(
     for filename, content_id in content_map.items():
         if content_id is None:
             continue
+        if _get_file_type(filename) == "html":
+            continue
         if filename not in fenced_filenames:
             logger.warning(
                 "File '%s' (content_id=%s) could not be matched to any code block "
@@ -946,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(