unique_toolkit 1.68.5__tar.gz → 1.68.8__tar.gz

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

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

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.68.5"
+version = "1.68.8"
 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.5 → unique_toolkit-1.68.8}/unique_toolkit/agentic/feature_flags/feature_flags.py

@@ -87,6 +87,11 @@ class FeatureFlags(BaseSettings):
         description="Enable argument screening agent for web search tool calls (UN-18741). Can be 'true' or comma-separated company IDs.",
     )
 
+    enable_selected_uploaded_files_un_18470: FeatureFlag = Field(
+        default=FeatureFlag(False),
+        description="Use selected_uploaded_files from event payload instead of querying uploaded files from the backend. Can be 'true' or comma-separated company IDs.",
+    )
+
     model_config = SettingsConfigDict(
         extra="ignore",
         env_prefix="FEATURE_FLAG_",

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

@@ -211,14 +211,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 +219,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 +229,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(
@@ -732,8 +718,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 +811,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 "

{unique_toolkit-1.68.5 → unique_toolkit-1.68.8}/unique_toolkit/app/__init__.py

@@ -44,6 +44,9 @@ from .schemas import (
 from .schemas import (
     McpTool as McpTool,
 )
+from .schemas import (
+    UploadedFileInfo as UploadedFileInfo,
+)
 from .verification import (
     verify_signature_and_construct_event as verify_signature_and_construct_event,
 )

{unique_toolkit-1.68.5 → unique_toolkit-1.68.8}/unique_toolkit/app/schemas.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 from enum import StrEnum
 from logging import getLogger
@@ -46,7 +48,7 @@ class BaseEvent(BaseModel, Generic[FilterOptionsT]):
     company_id: str
 
     @classmethod
-    def from_json_file(cls, file_path: Path) -> "BaseEvent":
+    def from_json_file(cls, file_path: Path) -> BaseEvent:
         if not file_path.exists():
             raise FileNotFoundError(f"File not found: {file_path}")
         with file_path.open("r", encoding="utf-8") as f:
@@ -162,12 +164,32 @@ class EventAssistantMessage(ChatEventAssistantMessage):
     pass
 
 
+class UploadedFileInfo(BaseModel):
+    """Describes a file attached to the chat"""
+
+    model_config = model_config
+
+    id: str
+    title: str = ""
+    mime_type: str = ""
+
+
 class ChatEventAdditionalParameters(BaseModel):
     model_config = model_config
 
     translate_to_language: Optional[str] = None
     content_id_to_translate: Optional[str] = None
     user_space_instructions: str
+    uploaded_files: list[UploadedFileInfo] = Field(default_factory=list)
+    selected_uploaded_files: list[UploadedFileInfo] = Field(default_factory=list)
+
+    @property
+    def uploaded_file_ids(self) -> list[str]:
+        return [f.id for f in self.uploaded_files]
+
+    @property
+    def selected_uploaded_file_ids(self) -> list[str]:
+        return [f.id for f in self.selected_uploaded_files]
 
 
 @deprecated(
@@ -263,7 +285,7 @@ class ChatEvent(BaseEvent):
     version: Optional[str] = None
 
     @classmethod
-    def from_json_file(cls, file_path: Path) -> "ChatEvent":
+    def from_json_file(cls, file_path: Path) -> ChatEvent:
         if not file_path.exists():
             raise FileNotFoundError(f"File not found: {file_path}")
         with file_path.open("r", encoding="utf-8") as f:
@@ -326,7 +348,7 @@ class Event(ChatEvent):
     # payload: EventPayload
 
     @classmethod
-    def from_json_file(cls, file_path: Path) -> "Event":
+    def from_json_file(cls, file_path: Path) -> Event:
         if not file_path.exists():
             raise FileNotFoundError(f"File not found: {file_path}")
         with file_path.open("r", encoding="utf-8") as f:

{unique_toolkit-1.68.5 → unique_toolkit-1.68.8}/unique_toolkit/test_utilities/events.py

@@ -11,6 +11,7 @@ from unique_toolkit.app.schemas import (
     ChatEventPayload,
     ChatEventUserMessage,
     EventName,
+    UploadedFileInfo,
 )
 from unique_toolkit.app.unique_settings import UniqueSettings
 
@@ -92,11 +93,15 @@ class TestEventFactory:
         translate_to_language: str | None = None,
         content_id_to_translate: str | None = None,
         user_space_instructions: str = "",
+        uploaded_files: list[UploadedFileInfo] = [],
+        selected_uploaded_files: list[UploadedFileInfo] = [],
     ) -> ChatEventAdditionalParameters:
         return ChatEventAdditionalParameters(
             translate_to_language=translate_to_language,
             content_id_to_translate=content_id_to_translate,
             user_space_instructions=user_space_instructions,
+            uploaded_files=uploaded_files,
+            selected_uploaded_files=selected_uploaded_files,
         )
 
     def get_base_event(