unique_toolkit 1.64.0__tar.gz → 1.64.1__tar.gz

{unique_toolkit-1.64.0 → unique_toolkit-1.64.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.64.0
+Version: 1.64.1
 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-1.64.0 → unique_toolkit-1.64.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.64.0"
+version = "1.64.1"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

{unique_toolkit-1.64.0 → unique_toolkit-1.64.1}/unique_toolkit/agentic/feature_flags/feature_flags.py

@@ -82,6 +82,11 @@ class FeatureFlags(BaseSettings):
         description="Emit codeExecution fences in message.text for code interpreter outputs (UN-17972). When disabled, inline file refs are kept as-is. Can be 'true' or comma-separated company IDs.",
     )
 
+    enable_tool_call_persistence_un_15977: FeatureFlag = Field(
+        default=FeatureFlag(False),
+        description="Persist tool calls and reconstruct tool call history across turns (UN-15977). Can be 'true' or comma-separated company IDs.",
+    )
+
     model_config = SettingsConfigDict(
         extra="ignore",
         env_prefix="FEATURE_FLAG_",

{unique_toolkit-1.64.0 → unique_toolkit-1.64.1}/unique_toolkit/agentic/history_manager/history_manager.py

@@ -77,6 +77,11 @@ class HistoryManagerConfig(BaseModel):
             * self.percent_of_max_tokens_for_history,
         )
 
+    enable_tool_call_persistence: bool = Field(
+        default=False,
+        description="When True, tool calls are persisted to the database and reconstructed from prior turns.",
+    )
+
     uploaded_content_config: (
         Annotated[
             UploadedContentConfig,
@@ -125,6 +130,7 @@ class HistoryManager:
             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,
         )
         self._reference_manager = reference_manager
         self._tool_call_result_history: list[ToolCallResponse] = []

{unique_toolkit-1.64.0 → unique_toolkit-1.64.1}/unique_toolkit/agentic/history_manager/loop_token_reducer.py

@@ -10,6 +10,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,
+    get_full_history_with_contents,
     get_full_history_with_contents_and_tool_calls,
 )
 from unique_toolkit.agentic.history_manager.utils import serialize_tool_content_json
@@ -54,6 +55,7 @@ class LoopTokenReducer:
         has_uploaded_content_config: bool,
         reference_manager: ReferenceManager,
         language_model: LMI,
+        enable_tool_call_persistence: bool = False,
     ):
         self._max_history_tokens = max_history_tokens
         self._has_uploaded_content_config = has_uploaded_content_config
@@ -73,6 +75,7 @@ class LoopTokenReducer:
         )
         self._max_db_source_number: int = -1
         self._db_source_map: dict[int, ContentChunk] = {}
+        self._enable_tool_call_persistence = enable_tool_call_persistence
 
     @property
     def max_db_source_number(self) -> int:
@@ -303,19 +306,31 @@ class LoopTokenReducer:
         Returns:
             list[LanguageModelMessage]: The history
         """
-        full_history, max_src, src_map = get_full_history_with_contents_and_tool_calls(
-            user_message=self._user_message,
-            chat_id=self._chat_id,
-            chat_service=self._chat_service,
-            content_service=self._content_service,
-            file_content_serialization_type=(
-                FileContentSerialization.NONE
-                if self._has_uploaded_content_config
-                else FileContentSerialization.FILE_NAME
-            ),
+        file_content_serialization_type = (
+            FileContentSerialization.NONE
+            if self._has_uploaded_content_config
+            else FileContentSerialization.FILE_NAME
         )
-        self._max_db_source_number = max_src
-        self._db_source_map = src_map
+        if self._enable_tool_call_persistence:
+            full_history, max_src, src_map = (
+                get_full_history_with_contents_and_tool_calls(
+                    user_message=self._user_message,
+                    chat_id=self._chat_id,
+                    chat_service=self._chat_service,
+                    content_service=self._content_service,
+                    file_content_serialization_type=file_content_serialization_type,
+                )
+            )
+            self._max_db_source_number = max_src
+            self._db_source_map = src_map
+        else:
+            full_history = get_full_history_with_contents(
+                user_message=self._user_message,
+                chat_id=self._chat_id,
+                chat_service=self._chat_service,
+                content_service=self._content_service,
+                file_content_serialization_type=file_content_serialization_type,
+            )
 
         if remove_from_text is not None:
             full_history.root = await self._clean_messages(