unique_toolkit 1.33.3__tar.gz → 1.34.0__tar.gz

{unique_toolkit-1.33.3 → unique_toolkit-1.34.0}/CHANGELOG.md

@@ -5,6 +5,9 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.34.0] - 2025-12-02
+- Add option to upload code interpreter generated files to the chat.
+
 ## [1.33.3] - 2025-12-02
 - Fix serialization of ToolBuildConfig `configuration` field.
 

{unique_toolkit-1.33.3 → unique_toolkit-1.34.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.33.3
+Version: 1.34.0
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -121,6 +121,9 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.34.0] - 2025-12-02
+- Add option to upload code interpreter generated files to the chat.
+
 ## [1.33.3] - 2025-12-02
 - Fix serialization of ToolBuildConfig `configuration` field.
 

{unique_toolkit-1.33.3 → unique_toolkit-1.34.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "1.33.3"
+version = "1.34.0"
 description = ""
 authors = [
     "Cedric Klinkert <cedric.klinkert@unique.ch>",

{unique_toolkit-1.33.3 → unique_toolkit-1.34.0}/unique_toolkit/agentic/responses_api/postprocessors/generated_files.py

@@ -8,6 +8,7 @@ from openai import AsyncOpenAI
 from openai.types.responses.response_output_text import AnnotationContainerFileCitation
 from pydantic import BaseModel, Field, RootModel
 
+from unique_toolkit import ChatService
 from unique_toolkit.agentic.postprocessor.postprocessor_manager import (
     ResponsesApiPostprocessor,
 )
@@ -27,9 +28,13 @@ logger = logging.getLogger(__name__)
 
 class DisplayCodeInterpreterFilesPostProcessorConfig(BaseModel):
     model_config = get_configuration_dict()
+    upload_to_chat: bool = Field(
+        default=False,
+        description="Whether to upload the generated files to the chat.",
+    )
     upload_scope_id: str = Field(
         default="<SCOPE_ID_PLACEHOLDER>",
-        description="The scope ID where the generated files will be uploaded.",
+        description="The scope ID where the generated files will be uploaded. Ignored if `uploadToChat` is set.",
     )
 
     file_download_failed_message: str = Field(
@@ -75,6 +80,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
         client: AsyncOpenAI,
         content_service: ContentService | KnowledgeBaseService,
         config: DisplayCodeInterpreterFilesPostProcessorConfig,
+        chat_service: ChatService | None = None,
         # Short term memory arguments, we prefer to explicitely pass the required auth variables
         # as it is crucial that we use chat-level short term memory not to leak user files to other chats.
         # Technically, short term memory can be scoped company-level, we would like to ensure this case is avoided.
@@ -83,9 +89,14 @@ class DisplayCodeInterpreterFilesPostProcessor(
         chat_id: str | None = None,
     ) -> None:
         super().__init__(self.__class__.__name__)
+
         self._content_service = content_service
-        self._config = config
+        self._chat_service = chat_service
         self._client = client
+        self._config = config
+
+        if self._config.upload_to_chat and self._chat_service is None:
+            raise ValueError("ChatService is required if uploadToChat is True")
 
         self._short_term_memory_manager = None
         if chat_id is not None and user_id is not None and company_id is not None:
@@ -199,13 +210,23 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 "Uploading file content for %s to knowledge base",
                 container_file.filename,
             )
-            content = await self._content_service.upload_content_from_bytes_async(
-                content=file_content.content,
-                content_name=container_file.filename,
-                skip_ingestion=True,
-                mime_type=guess_type(container_file.filename)[0] or "text/plain",
-                scope_id=self._config.upload_scope_id,
-            )
+
+            if self._config.upload_to_chat:
+                assert self._chat_service is not None  # Checked in __init__
+                content = await 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",
+                    skip_ingestion=True,
+                )
+            else:
+                content = await self._content_service.upload_content_from_bytes_async(
+                    content=file_content.content,
+                    content_name=container_file.filename,
+                    mime_type=guess_type(container_file.filename)[0] or "text/plain",
+                    scope_id=self._config.upload_scope_id,
+                    skip_ingestion=True,
+                )
 
             return _ContentInfo(filename=container_file.filename, content_id=content.id)
 

{unique_toolkit-1.33.3 → unique_toolkit-1.34.0}/unique_toolkit/services/chat_service.py

@@ -70,6 +70,7 @@ from unique_toolkit.content.functions import (
     download_content_to_bytes,
     search_contents,
     upload_content_from_bytes,
+    upload_content_from_bytes_async,
 )
 from unique_toolkit.content.schemas import (
     Content,
@@ -1607,6 +1608,30 @@ class ChatService(ChatServiceDeprecated):
             metadata=metadata,
         )
 
+    async def upload_to_chat_from_bytes_async(
+        self,
+        *,
+        content: bytes,
+        content_name: str,
+        mime_type: str,
+        scope_id: str | None = None,
+        skip_ingestion: bool = False,
+        ingestion_config: unique_sdk.Content.IngestionConfig | None = None,
+        metadata: dict[str, Any] | None = None,
+    ) -> Content:
+        return await upload_content_from_bytes_async(
+            user_id=self._user_id,
+            company_id=self._company_id,
+            content=content,
+            content_name=content_name,
+            mime_type=mime_type,
+            scope_id=scope_id,
+            chat_id=self._chat_id,
+            skip_ingestion=skip_ingestion,
+            ingestion_config=ingestion_config,
+            metadata=metadata,
+        )
+
     def download_chat_content_to_bytes(self, *, content_id: str) -> bytes:
         return download_content_to_bytes(
             user_id=self._user_id,