unique_toolkit 1.68.5__tar.gz → 1.68.6__tar.gz

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

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

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

{unique_toolkit-1.68.5 → unique_toolkit-1.68.6}/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.6}/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.6}/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.6}/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(