unique_toolkit 1.33.3__py3-none-any.whl → 1.34.0__py3-none-any.whl

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/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,

{unique_toolkit-1.33.3.dist-info → unique_toolkit-1.34.0.dist-info}/METADATA

@@ -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.dist-info → unique_toolkit-1.34.0.dist-info}/RECORD

@@ -68,7 +68,7 @@ unique_toolkit/agentic/postprocessor/postprocessor_manager.py,sha256=s6HFhA61TE0
 unique_toolkit/agentic/reference_manager/reference_manager.py,sha256=x51CT0D8HHu2LzgXdHGy0leOYpjnsxVbPZ2nc28G9mA,4005
 unique_toolkit/agentic/responses_api/__init__.py,sha256=9WTO-ef7fGE9Y1QtZJFm8Q_jkwK8Srtl-HWvpAD2Wxs,668
 unique_toolkit/agentic/responses_api/postprocessors/code_display.py,sha256=h6ZqPR0kPQnxM0ynshYQTa1BrcN8XGbUz9p03m8rOj0,2339
-unique_toolkit/agentic/responses_api/postprocessors/generated_files.py,sha256=9KfuLdhRvsW4K7dO6Q1fkdpyzKUqmMbIwVCO5eu3fDw,10732
+unique_toolkit/agentic/responses_api/postprocessors/generated_files.py,sha256=ZdxJQQlD5K2Arn_NJVQNzIUwp-q5SmIE17o-UquTa0A,11691
 unique_toolkit/agentic/responses_api/stream_handler.py,sha256=Y1IM0uiPBdlab5UuOTCsHTaVX-fd9MxfS3xkwhdFie4,647
 unique_toolkit/agentic/short_term_memory_manager/persistent_short_term_memory_manager.py,sha256=uF3HSoZF0hBfuNhIE9N8KRtuwDfpoeXUFVrv_cyZ3Sw,5839
 unique_toolkit/agentic/thinking_manager/thinking_manager.py,sha256=41QWFsdRrbWlQHBfYCFv726UDom4WbcvaRfjCmoUOQI,4183
@@ -189,7 +189,7 @@ unique_toolkit/language_model/service.py,sha256=rt5LoQnDifIa5TnykGkJNl03lj0QhCMd
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/protocols/support.py,sha256=ZEnbQL5w2-T_1AeM8OHycZJ3qbdfVI1nXe0nL9esQEw,5544
 unique_toolkit/services/__init__.py,sha256=90-IT5FjMcnlqxjp5kme9Fqgp_on46rggctIqHMdqsw,195
-unique_toolkit/services/chat_service.py,sha256=vA97DO4iiXMeTNuwS-bYeskX0DnqO_8vL4yzJAfsGtw,60111
+unique_toolkit/services/chat_service.py,sha256=tHFmpbDiooYIvM5vZePsnwHdYUEcIBeD3db4lcsMJ_s,60941
 unique_toolkit/services/knowledge_base.py,sha256=uc89GL_NZXeFkJKkdHSSh2y1Wx0tmgasWk6uyGi4G_M,36210
 unique_toolkit/short_term_memory/__init__.py,sha256=2mI3AUrffgH7Yt-xS57EGqnHf7jnn6xquoKEhJqk3Wg,185
 unique_toolkit/short_term_memory/constants.py,sha256=698CL6-wjup2MvU19RxSmQk3gX7aqW_OOpZB7sbz_Xg,34
@@ -199,7 +199,7 @@ unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBu
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=Ozhh70qCn2yOzRWr9d8WmJeTo7AQurwd3tStgBMPFLA,1246
 unique_toolkit/test_utilities/events.py,sha256=_mwV2bs5iLjxS1ynDCjaIq-gjjKhXYCK-iy3dRfvO3g,6410
-unique_toolkit-1.33.3.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.33.3.dist-info/METADATA,sha256=X_ddHgP_CSpLGtfeB9_wRZ-ZO6ESULNu_oRPRj2Xpog,45428
-unique_toolkit-1.33.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.33.3.dist-info/RECORD,,
+unique_toolkit-1.34.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.34.0.dist-info/METADATA,sha256=DNDLgni0ur55-wGblbTjZqzJpFMKfx_IsWIfwWD5Lvk,45523
+unique_toolkit-1.34.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.34.0.dist-info/RECORD,,