unique_toolkit 2026.30.0.dev12__py3-none-any.whl → 2026.30.0.dev13__py3-none-any.whl

unique_toolkit/agentic/history_manager/history_construction_with_contents.py

@@ -6,7 +6,7 @@ import mimetypes
 from datetime import datetime
 from enum import StrEnum
 from itertools import groupby
-from typing import TYPE_CHECKING, Iterator
+from typing import TYPE_CHECKING, Callable, Iterator
 
 if TYPE_CHECKING:
     from unique_toolkit.content.schemas import ContentChunk
@@ -200,9 +200,7 @@ async def download_encoded_images_async(
     return base64_encoded_images
 
 
-class FileContentSerialization(StrEnum):
-    NONE = "none"
-    FILE_NAME = "file_name"
+FileContentSerializer = Callable[[Content], str | None]
 
 
 class ImageContentInclusion(StrEnum):
@@ -210,23 +208,17 @@ class ImageContentInclusion(StrEnum):
     ALL = "all"
 
 
-def file_content_serialization(
+def _serialize_file_contents(
+    text: str,
     file_contents: list[Content],
-    file_content_serialization: FileContentSerialization,
+    file_content_serializer: FileContentSerializer | None,
 ) -> str:
-    match file_content_serialization:
-        case FileContentSerialization.NONE:
-            return ""
-        case FileContentSerialization.FILE_NAME:
-            file_names = [
-                f"- Uploaded file: {f.key} at {f.created_at}" for f in file_contents
-            ]
-            return "\n".join(
-                [
-                    "Files Uploaded to Chat can be accessed by internal search tool if available:\n",
-                ]
-                + file_names,
-            )
+    serialized_files: list[str] = []
+    if file_content_serializer is not None:
+        for file_content in file_contents:
+            if serialized_file := file_content_serializer(file_content):
+                serialized_files.append(serialized_file)
+    return (text + "\n\n" + "\n".join(serialized_files)).strip()
 
 
 def _append_element_to_builder(
@@ -234,9 +226,9 @@ def _append_element_to_builder(
     c: ChatMessageWithContents,
     text: str,
     include_images: ImageContentInclusion,
-    file_content_serialization_type: FileContentSerialization,
     content_service: ContentService,
     chat_id: str,
+    file_content_serializer: FileContentSerializer | None = None,
     selected_content_ids: set[str] | None = None,
 ) -> None:
     if len(c.contents) > 0:
@@ -249,15 +241,12 @@ def _append_element_to_builder(
             image_contents = [
                 co for co in image_contents if co.id in selected_content_ids
             ]
-        content = (
-            text
-            + "\n\n"
-            + file_content_serialization(
-                file_contents,
-                file_content_serialization_type,
-            )
-        ).strip()
-        if include_images and len(image_contents) > 0:
+        content = _serialize_file_contents(
+            text,
+            file_contents,
+            file_content_serializer,
+        )
+        if include_images and image_contents:
             builder.image_message_append(
                 content=content,
                 images=download_encoded_images(
@@ -285,9 +274,9 @@ async def _append_element_to_builder_async(
     c: ChatMessageWithContents,
     text: str,
     include_images: ImageContentInclusion,
-    file_content_serialization_type: FileContentSerialization,
     content_service: ContentService,
     chat_id: str,
+    file_content_serializer: FileContentSerializer | None = None,
     selected_content_ids: set[str] | None = None,
 ) -> None:
     if len(c.contents) > 0:
@@ -300,15 +289,12 @@ async def _append_element_to_builder_async(
             image_contents = [
                 co for co in image_contents if co.id in selected_content_ids
             ]
-        content = (
-            text
-            + "\n\n"
-            + file_content_serialization(
-                file_contents,
-                file_content_serialization_type,
-            )
-        ).strip()
-        if include_images and len(image_contents) > 0:
+        content = _serialize_file_contents(
+            text,
+            file_contents,
+            file_content_serializer,
+        )
+        if include_images and image_contents:
             builder.image_message_append(
                 content=content,
                 images=await download_encoded_images_async(
@@ -336,7 +322,7 @@ def get_full_history_with_contents(
     chat_service: ChatService,
     content_service: ContentService,
     include_images: ImageContentInclusion = ImageContentInclusion.ALL,
-    file_content_serialization_type: FileContentSerialization = FileContentSerialization.FILE_NAME,
+    file_content_serializer: FileContentSerializer | None = None,
     selected_content_ids: set[str] | None = None,
 ) -> LanguageModelMessages:
     grouped_elements = get_chat_history_with_contents(
@@ -361,7 +347,7 @@ def get_full_history_with_contents(
             c=c,
             text=text,
             include_images=include_images,
-            file_content_serialization_type=file_content_serialization_type,
+            file_content_serializer=file_content_serializer,
             content_service=content_service,
             chat_id=chat_id,
             selected_content_ids=selected_content_ids,
@@ -376,7 +362,7 @@ async def get_full_history_with_contents_async(
     chat_service: ChatService,
     content_service: ContentService,
     include_images: ImageContentInclusion = ImageContentInclusion.ALL,
-    file_content_serialization_type: FileContentSerialization = FileContentSerialization.FILE_NAME,
+    file_content_serializer: FileContentSerializer | None = None,
     selected_content_ids: set[str] | None = None,
 ) -> LanguageModelMessages:
     grouped_elements = await get_chat_history_with_contents_async(
@@ -400,7 +386,7 @@ async def get_full_history_with_contents_async(
             c=c,
             text=text,
             include_images=include_images,
-            file_content_serialization_type=file_content_serialization_type,
+            file_content_serializer=file_content_serializer,
             content_service=content_service,
             chat_id=chat_id,
             selected_content_ids=selected_content_ids,
@@ -415,7 +401,7 @@ def get_full_history_with_contents_and_tool_calls(
     chat_service: ChatService,
     content_service: ContentService,
     include_images: ImageContentInclusion = ImageContentInclusion.ALL,
-    file_content_serialization_type: FileContentSerialization = FileContentSerialization.FILE_NAME,
+    file_content_serializer: FileContentSerializer | None = None,
     selected_content_ids: set[str] | None = None,
 ) -> tuple[LanguageModelMessages, int, dict[int, "ContentChunk"]]:
     """Build the full LLM message history, including persisted tool call rounds.
@@ -544,7 +530,7 @@ def get_full_history_with_contents_and_tool_calls(
             c=c,
             text=text,
             include_images=include_images,
-            file_content_serialization_type=file_content_serialization_type,
+            file_content_serializer=file_content_serializer,
             content_service=content_service,
             chat_id=chat_id,
             selected_content_ids=selected_content_ids,
@@ -559,7 +545,7 @@ async def get_full_history_with_contents_and_tool_calls_async(
     chat_service: ChatService,
     content_service: ContentService,
     include_images: ImageContentInclusion = ImageContentInclusion.ALL,
-    file_content_serialization_type: FileContentSerialization = FileContentSerialization.FILE_NAME,
+    file_content_serializer: FileContentSerializer | None = None,
     selected_content_ids: set[str] | None = None,
 ) -> tuple[LanguageModelMessages, int, dict[int, "ContentChunk"]]:
     """Async version of get_full_history_with_contents_and_tool_calls."""
@@ -662,7 +648,7 @@ async def get_full_history_with_contents_and_tool_calls_async(
             c=c,
             text=text,
             include_images=include_images,
-            file_content_serialization_type=file_content_serialization_type,
+            file_content_serializer=file_content_serializer,
             content_service=content_service,
             chat_id=chat_id,
             selected_content_ids=selected_content_ids,

unique_toolkit/agentic/history_manager/history_manager.py

@@ -10,6 +10,9 @@ from unique_toolkit._common.feature_flags.schema import (
 )
 from unique_toolkit._common.pydantic_helpers import DeactivatedNone
 from unique_toolkit._common.validators import LMI
+from unique_toolkit.agentic.history_manager.history_construction_with_contents import (
+    FileContentSerializer,
+)
 from unique_toolkit.agentic.history_manager.loop_token_reducer import LoopTokenReducer
 from unique_toolkit.agentic.history_manager.utils import (
     serialize_tool_content_json,
@@ -115,6 +118,7 @@ class HistoryManager:
         config: HistoryManagerConfig,
         language_model: LMI,
         reference_manager: ReferenceManager,
+        file_content_serializer: FileContentSerializer | None = None,
     ):
         self._config = config
         self._logger = logger
@@ -123,10 +127,10 @@ class HistoryManager:
             logger=self._logger,
             event=event,
             max_history_tokens=self._config.max_history_tokens,
-            has_uploaded_content_config=bool(self._config.uploaded_content_config),
             language_model=self._language_model,
             reference_manager=reference_manager,
             enable_tool_call_persistence=self._config.enable_tool_call_persistence,
+            file_content_serializer=file_content_serializer,
         )
         self._reference_manager = reference_manager
         self._tool_call_result_history: list[ToolCallResponse] = []

unique_toolkit/agentic/history_manager/loop_token_reducer.py

@@ -9,7 +9,7 @@ from unique_toolkit._common.token.token_counting import (
 )
 from unique_toolkit._common.validators import LMI
 from unique_toolkit.agentic.history_manager.history_construction_with_contents import (
-    FileContentSerialization,
+    FileContentSerializer,
     get_full_history_with_contents_and_tool_calls_async,
     get_full_history_with_contents_async,
 )
@@ -54,13 +54,13 @@ class LoopTokenReducer:
         logger: Logger,
         event: ChatEvent,
         max_history_tokens: int,
-        has_uploaded_content_config: bool,
         reference_manager: ReferenceManager,
         language_model: LMI,
         enable_tool_call_persistence: bool = False,
+        file_content_serializer: FileContentSerializer | None = None,
     ):
         self._max_history_tokens = max_history_tokens
-        self._has_uploaded_content_config = has_uploaded_content_config
+        self._file_content_serializer = file_content_serializer
         self._logger = logger
         self._reference_manager = reference_manager
         self._language_model = language_model
@@ -309,11 +309,6 @@ class LoopTokenReducer:
         Returns:
             list[LanguageModelMessage]: The history
         """
-        file_content_serialization_type = (
-            FileContentSerialization.NONE
-            if self._has_uploaded_content_config
-            else FileContentSerialization.FILE_NAME
-        )
         if self._enable_tool_call_persistence:
             (
                 full_history,
@@ -324,7 +319,7 @@ class LoopTokenReducer:
                 chat_id=self._chat_id,
                 chat_service=self._chat_service,
                 content_service=self._content_service,
-                file_content_serialization_type=file_content_serialization_type,
+                file_content_serializer=self._file_content_serializer,
                 selected_content_ids=self._selected_content_ids,
             )
             self._max_db_source_number = max_src
@@ -335,7 +330,7 @@ class LoopTokenReducer:
                 chat_id=self._chat_id,
                 chat_service=self._chat_service,
                 content_service=self._content_service,
-                file_content_serialization_type=file_content_serialization_type,
+                file_content_serializer=self._file_content_serializer,
                 selected_content_ids=self._selected_content_ids,
             )
 

{unique_toolkit-2026.30.0.dev12.dist-info → unique_toolkit-2026.30.0.dev13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.30.0.dev12
+Version: 2026.30.0.dev13
 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-2026.30.0.dev12.dist-info → unique_toolkit-2026.30.0.dev13.dist-info}/RECORD

@@ -113,9 +113,9 @@ unique_toolkit/agentic/evaluation/tests/test_prompt_loaders.py,sha256=zBREdlKf5t
 unique_toolkit/agentic/evaluation/utils.py,sha256=HmyPaDV8wdW-_gOjjW-wDaMKgdrsP5-SHP7OqTmGI_A,264
 unique_toolkit/agentic/feature_flags/__init__.py,sha256=LhE2cHoa9AYBOR7TjiIToOn46sttm9paKcrzE7gnDPM,149
 unique_toolkit/agentic/feature_flags/feature_flags.py,sha256=ul9-StpKO-4Kzq0GxTR4UlM-fRQsOD4BGr5aNmAiyBo,5315
-unique_toolkit/agentic/history_manager/history_construction_with_contents.py,sha256=yqaG_tqyijgW6DEpfulg1zKtzQLewgWEegWnvZvE4P4,25849
-unique_toolkit/agentic/history_manager/history_manager.py,sha256=1W9908BoKzBRwxaLx_AVQE4QsMHj4mq3rdeNunTpHfY,15739
-unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=egXlsiN6QxoWM3R4DAjsuhSap7RU3DqBJeDGfw9Ndc0,26846
+unique_toolkit/agentic/history_manager/history_construction_with_contents.py,sha256=70lm8pYmQpb0HRpqzPTb4ikh8vVbvhfGR5S-D4Ftbbc,25306
+unique_toolkit/agentic/history_manager/history_manager.py,sha256=RIVeQxlCq25JwUY6WWQzz5zoOCouAK7shbxUztpgZPs,15903
+unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=v9K4TbEVK9HBLgbvax9KZmAqkCkowk6v391AJeZdVts,26645
 unique_toolkit/agentic/history_manager/settings.py,sha256=Lrx3a8yXeU0MXrWlV6RZSHsgYWF9Zrs8LyR96OpqQ_k,1216
 unique_toolkit/agentic/history_manager/utils.py,sha256=hYgaahQq9gKm7V2Q5qnyZtqw5opO-MKH7wTGq9-iEgE,6455
 unique_toolkit/agentic/loop_runner/__init__.py,sha256=KPZLJ9fCzR9mwnvAgrA_dPsDJlcQS8o0r0amNdmBljY,1352
@@ -407,6 +407,6 @@ unique_toolkit/short_term_memory/service.py,sha256=2JMBEBn6GoyK-a0xvdcWpPft-3C74
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=vRe0aMJWuVIK-VsqomgxQImzke7MDmwgSxxJfYCa7zw,1386
 unique_toolkit/test_utilities/events.py,sha256=FLrhMh0A_mQutAUjoD0FdMcVQ4Gq31r35OVGzRjvLro,6760
-unique_toolkit-2026.30.0.dev12.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.30.0.dev12.dist-info/METADATA,sha256=XpaKCUPzzdREkJVfO1q4QF-G5BBdQkh2DG8_NTupKGY,7460
-unique_toolkit-2026.30.0.dev12.dist-info/RECORD,,
+unique_toolkit-2026.30.0.dev13.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.30.0.dev13.dist-info/METADATA,sha256=ySP5zJPYH87Q1z_K649-D8vyfa7Xg1lma3hVMblb4qg,7460
+unique_toolkit-2026.30.0.dev13.dist-info/RECORD,,