unique_toolkit 1.4.4__tar.gz → 1.5.0__tar.gz

{unique_toolkit-1.4.4 → unique_toolkit-1.5.0}/CHANGELOG.md

@@ -5,6 +5,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.5.0] - 2025-10-01
+- Allow history manager to fetch only ui visible text
+
 ## [1.4.4] - 2025-09-30
 - Fix bugs with display of sub-agent answers and evaluations when multiple sub-agent from the same assistant run concurrently.
 

{unique_toolkit-1.4.4 → unique_toolkit-1.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.4.4
+Version: 1.5.0
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,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.5.0] - 2025-10-01
+- Allow history manager to fetch only ui visible text
+
 ## [1.4.4] - 2025-09-30
 - Fix bugs with display of sub-agent answers and evaluations when multiple sub-agent from the same assistant run concurrently.
 

{unique_toolkit-1.4.4 → unique_toolkit-1.5.0}/pyproject.toml

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

{unique_toolkit-1.4.4 → unique_toolkit-1.5.0}/unique_toolkit/agentic/history_manager/history_manager.py

@@ -215,3 +215,22 @@ class HistoryManager:
             remove_from_text=remove_from_text,
         )
         return messages
+
+    async def get_user_visible_chat_history(
+        self, assistant_message_text: str | None = None
+    ) -> LanguageModelMessages:
+        """Get the user visible chat history.
+
+        Args:
+            assistant_message_text (str | None): The latest assistant message to append to the history, as this is not extracted from the history.
+            If None, the history will be returned without the latest assistant message.
+
+        Returns:
+            LanguageModelMessages: The user visible chat history.
+        """
+        history = await self._token_reducer._get_history_from_db()
+        if assistant_message_text:
+            history.append(
+                LanguageModelAssistantMessage(content=assistant_message_text)
+            )
+        return LanguageModelMessages(history)

{unique_toolkit-1.4.4 → unique_toolkit-1.5.0}/unique_toolkit/agentic/history_manager/loop_token_reducer.py

@@ -113,6 +113,21 @@ class LoopTokenReducer:
 
         return messages
 
+    async def get_user_visible_chat_history(
+        self, message: LanguageModelAssistantMessage
+    ) -> LanguageModelMessages:
+        """Get the user visible chat history.
+
+        Args:
+            message (LanguageModelAssistantMessage): The latest assistant message to append to the history, as this is not extracted from the history.
+
+        Returns:
+            LanguageModelMessages: The user visible chat history.
+        """
+        history = await self._get_history_from_db()
+        history.append(message)
+        return LanguageModelMessages(history)
+
     def _exceeds_token_limit(self, token_count: int) -> bool:
         """Check if token count exceeds the maximum allowed limit and if at least one tool call has more than one source."""
         # At least one tool call should have more than one chunk as answer
@@ -223,7 +238,7 @@ class LoopTokenReducer:
         return history
 
     async def _get_history_from_db(
-        self, remove_from_text: Callable[[str], Awaitable[str]]
+        self, remove_from_text: Callable[[str], Awaitable[str]] | None = None
     ) -> list[LanguageModelMessage]:
         """
         Get the history of the conversation. The function will retrieve a subset of the full history based on the configuration.
@@ -242,10 +257,10 @@ class LoopTokenReducer:
                 else FileContentSerialization.FILE_NAME
             ),
         )
-
-        full_history.root = await self._clean_messages(
-            full_history.root, remove_from_text
-        )
+        if remove_from_text is not None:
+            full_history.root = await self._clean_messages(
+                full_history.root, remove_from_text
+            )
 
         limited_history_messages = self._limit_to_token_window(
             full_history.root, self._max_history_tokens