unique_toolkit 1.33.3__tar.gz → 1.34.1__tar.gz

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

@@ -5,6 +5,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.1] - 2025-12-02
+- Update code interpreter tool instructions.
+
+## [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.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.33.3
+Version: 1.34.1
 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.1] - 2025-12-02
+- Update code interpreter tool instructions.
+
+## [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.1}/pyproject.toml

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

{unique_toolkit-1.33.3 → unique_toolkit-1.34.1}/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.1}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/config.py

@@ -11,27 +11,41 @@ DEFAULT_TOOL_DESCRIPTION = "Use this tool to run python code, e.g to generate pl
 
 DEFAULT_TOOL_DESCRIPTION_FOR_SYSTEM_PROMPT = """
 Use this tool to run python code, e.g to generate plots, process excel files, perform calculations, etc.
+
 Instructions:
-- All files uploaded to the chat are available in the code interpreter under the path `/mnt/data/<filename>
-- All files generated through code should be saved in the `/mnt/data` folder
 
-CRUCIAL Instructions for displaying images and files in the chat:
-Once files are generated in the `/mnt/data` folder you MUST reference them in the chat using markdown syntax in order to display them in the chat.
+Uploaded and generated files:
+- All files uploaded to the chat are available at the path `/mnt/data/<filename>`.
+- All files generated through code MUST be saved in the `/mnt/data` folder.
 
+CRUCIAL Instructions for displaying images and files in the chat:
+- Once files are generated in the `/mnt/data` folder you MUST reference them in the chat using markdown syntax in order to display them in the chat.
 WHENEVER you reference a generated file, you MUST use the following format:
 ```
-**Descriptive Title of Graph/Chart/File** (<- linebreak is important) (You must choose a good user friendly title, Other markdown syntax such as `#` can be used too)
+**Descriptive Title of Graph/Chart/File**  (linebreak is important. You must choose a good user friendly title, Other markdown syntax such as `#` can be used too)
 [*Generating your {Graph/Chart/File}…*](sandbox:/mnt/data/<filename>)
 ```
 IMPORTANT: Do NOT append a leading `!` even when displaying an image.
 Always use a line break between the title and the markdown!
-
 - Files with image file extensions are displayed directly in the chat, while other file extensions are shown as download links.
 - Not using syntax above will FAIL to show images to the user. 
 - YOU MUST use the syntax above to display images, otherwise the image will not be displayed in the chat.
 - Only the following file types are allowed to be uploaded to the platform, anything else will FAIL: PDF, DOCX, XLSX, PPTX, CSV, HTML, MD, TXT, PNG, JPG, JPEG.
+- You MUST always use this syntax, otherwise the files will not be displayed in the chat.
+
+
+# Displaying Dataframes/Tables:
+- Whenever asked to display a dataframe/table, it is CRITICAL to represent it faithfully as a markdown table in your response.
 
-You MUST always use this syntax, otherwise the files will not be displayed in the chat.
+Handling User Queries:
+- Whenever the user query requires using the python tool, you must always think first about the steps required.
+- Use the tool multiple times:
+    - You MUST NOT guess anything about the structure of the data / files uploaded. Rather, you MUST perform some data exploration first.
+        - Example: User uploads an excel files and asks a question about it. First Read the data, explore the columns, columns types, etc. Then use the tool to answer the user's query.
+        In this case, you can simply call the tool multiple times.
+        - REMEMBER that you can always read the content of text, csv files if needed. In this case, you MUST always limit the amount of data displayed.
+- If a tool step fails, you must recover as much as possible.
+- After exhausting all possible solutions without success, inform the user of what was tried and request clarification/help.
 """.strip()
 
 DEFAULT_TOOL_FORMAT_INFORMATION_FOR_SYSTEM_PROMPT = ""

{unique_toolkit-1.33.3 → unique_toolkit-1.34.1}/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,