unique_toolkit 1.33.2__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/agentic/tools/config.py

@@ -1,12 +1,12 @@
-import json
 from enum import StrEnum
-from typing import Annotated, Any, Dict
+from typing import Annotated, Any
 
 from pydantic import (
     BaseModel,
     BeforeValidator,
     Field,
     ValidationInfo,
+    field_serializer,
     model_validator,
 )
 
@@ -123,46 +123,6 @@ class ToolBuildConfig(BaseModel):
         value["configuration"] = config
         return value
 
-    def model_dump(self) -> Dict[str, Any]:
-        """
-        Returns a dict representation of the tool config that preserves
-        subclass fields from `configuration` by delegating to its own
-        model_dump. This prevents `{}` when `configuration` is typed
-        as `BaseToolConfig` but holds a subclass instance.
-        """
-        data: Dict[str, Any] = {
-            "name": self.name,
-            "configuration": self.configuration.model_dump()
-            if self.configuration
-            else None,
-            "display_name": self.display_name,
-            "icon": self.icon,
-            "selection_policy": self.selection_policy,
-            "is_exclusive": self.is_exclusive,
-            "is_sub_agent": self.is_sub_agent,
-            "is_enabled": self.is_enabled,
-        }
-        return data
-
-    def model_dump_json(self) -> str:
-        """
-        Returns a JSON string representation of the tool config.
-        Ensures `configuration` is fully serialized by using the
-        subclass's `model_dump_json()` when available.
-        """
-        config_json = (
-            self.configuration.model_dump_json() if self.configuration else None
-        )
-        config = json.loads(config_json) if config_json else None
-
-        data: Dict[str, Any] = {
-            "name": self.name,
-            "configuration": config,
-            "display_name": self.display_name,
-            "icon": self.icon,
-            "selection_policy": self.selection_policy,
-            "is_exclusive": self.is_exclusive,
-            "is_sub_agent": self.is_sub_agent,
-            "is_enabled": self.is_enabled,
-        }
-        return json.dumps(data)
+    @field_serializer("configuration")
+    def serialize_config(self, value: BaseToolConfig) -> dict[str, Any]:
+        return value.__class__.model_dump(value)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.33.2
+Version: 1.34.0
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -121,6 +121,12 @@ 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.
+
 ## [1.33.2] - 2025-12-01
 - Upgrade numpy to >2.1.0 to ensure compatibility with langchain library
 

{unique_toolkit-1.33.2.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
@@ -99,7 +99,7 @@ unique_toolkit/agentic/tools/a2a/tool/_schema.py,sha256=wMwyunViTnxaURvenkATEvyf
 unique_toolkit/agentic/tools/a2a/tool/config.py,sha256=zK9onAMYlYrPc0MCz0gBocd22jm0H5ar1ihNJ7CyjJU,5780
 unique_toolkit/agentic/tools/a2a/tool/service.py,sha256=OJx1zc1roPgt4aa1fDw4XTF9VOCCF72dhEJaNMEMOSU,13865
 unique_toolkit/agentic/tools/agent_chunks_hanlder.py,sha256=x32Dp1Z8cVW5i-XzXbaMwX2KHPcNGmqEU-FB4AV9ZGo,1909
-unique_toolkit/agentic/tools/config.py,sha256=N7IVtMecCCf8hxGUf-1GEk6ldIoEZfXi4fo5GMpHKus,5418
+unique_toolkit/agentic/tools/config.py,sha256=71PSq8LoR80irk2iJfy_AgXOPi4Nh3luW5z5cDvY8i0,3920
 unique_toolkit/agentic/tools/factory.py,sha256=A1Aliwx037UAk9ADiDsg0zjCWWnvzV_PxwJNoPTvW6c,1434
 unique_toolkit/agentic/tools/mcp/__init__.py,sha256=RLF_p-LDRC7GhiB3fdCi4u3bh6V9PY_w26fg61BLyco,122
 unique_toolkit/agentic/tools/mcp/manager.py,sha256=DPYwwDe6RSZyuPaxn-je49fP_qOOs0ZV46EM6GZcV4c,2748
@@ -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.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.33.2.dist-info/METADATA,sha256=i3ioZ4szjn5j4MiVzeO2-LlbLi5H9MeCXllSwYn0gTc,45339
-unique_toolkit-1.33.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.33.2.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,,