unique_toolkit 1.5.1__py3-none-any.whl → 1.7.0__py3-none-any.whl

unique_toolkit/agentic/history_manager/history_manager.py

@@ -217,18 +217,21 @@ class HistoryManager:
         return messages
 
     async def get_user_visible_chat_history(
-        self, assistant_message_text: str | None = None
+        self,
+        assistant_message_text: str | None = None,
+        remove_from_text: Callable[[str], Awaitable[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.
+            remove_from_text (Callable[[str], Awaitable[str]] | None): A function to remove text from the history before returning it.
 
         Returns:
             LanguageModelMessages: The user visible chat history.
         """
-        history = await self._token_reducer._get_history_from_db()
+        history = await self._token_reducer.get_history_from_db(remove_from_text)
         if assistant_message_text:
             history.append(
                 LanguageModelAssistantMessage(content=assistant_message_text)

unique_toolkit/agentic/history_manager/loop_token_reducer.py

@@ -113,21 +113,6 @@ 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
@@ -165,7 +150,7 @@ class LoopTokenReducer:
         rendered_system_message_string: str,
         remove_from_text: Callable[[str], Awaitable[str]],
     ) -> list[LanguageModelMessage]:
-        history_from_db = await self._get_history_from_db(remove_from_text)
+        history_from_db = await self.get_history_from_db(remove_from_text)
         history_from_db = self._replace_user_message(
             history_from_db, original_user_message, rendered_user_message_string
         )
@@ -237,7 +222,7 @@ class LoopTokenReducer:
             ]
         return history
 
-    async def _get_history_from_db(
+    async def get_history_from_db(
         self, remove_from_text: Callable[[str], Awaitable[str]] | None = None
     ) -> list[LanguageModelMessage]:
         """

{unique_toolkit-1.5.1.dist-info → unique_toolkit-1.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.5.1
+Version: 1.7.0
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,11 @@ 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.7.0] - 2025-10-01
+- Add functionality to remove text in `get_user_visible_chat_history`
+
+## [1.6.0] - 2025-10-01
+- revert and simplify 1.5.0 
 
 ## [1.5.1] - 2025-10-01
 - Fix filtering logic to raise a ConfigurationException if chat event filters are not specified

{unique_toolkit-1.5.1.dist-info → unique_toolkit-1.7.0.dist-info}/RECORD

@@ -42,8 +42,8 @@ unique_toolkit/agentic/evaluation/schemas.py,sha256=m9JMCUmeqP8KhsJOVEzsz6dRXUe1
 unique_toolkit/agentic/evaluation/tests/test_context_relevancy_service.py,sha256=4tDxHTApbaTMxN1sNS8WCqj2BweRk6YqZ5_zHP45jto,7977
 unique_toolkit/agentic/evaluation/tests/test_output_parser.py,sha256=RN_HcBbU6qy_e_PoYyUFcjWnp3ymJ6-gLj6TgEOupAI,3107
 unique_toolkit/agentic/history_manager/history_construction_with_contents.py,sha256=c8Zy3erSbHGT8AdICRRlSK91T_FN6tNpTznvUzpLbWk,9023
-unique_toolkit/agentic/history_manager/history_manager.py,sha256=klUtDtcddUI7pdZIhIedl7YIOaZJBRp0N5O_Yh-nrqk,9217
-unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=RH97HV2EiShsnw1C5USqAZ3vQxcoRjzxDZf_CibQ7GU,19109
+unique_toolkit/agentic/history_manager/history_manager.py,sha256=1MSFEQtw7jYrcFVEgnTIe6LrGo36WVlutzqTwcFJq5w,9449
+unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=3QSDXZ9M12-cDhYr7-UgDYGEMySkXENt1OkfD0CruQ8,18538
 unique_toolkit/agentic/history_manager/utils.py,sha256=NDSSz0Jp3oVJU3iKlVScmM1AOe-6hTiVjLr16DUPsV0,5656
 unique_toolkit/agentic/postprocessor/postprocessor_manager.py,sha256=GDzJhaoOUwxZ37IINkQ7au4CHmAOFS5miP2lqv8ZwZA,4277
 unique_toolkit/agentic/reference_manager/reference_manager.py,sha256=1GeoFX1-RLdTcns1358GJADDSAcTAM2J0jJJpln08qo,4005
@@ -138,7 +138,7 @@ unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJ
 unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBuE9sI2o9Aajqjxg,8884
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=cxWjb2dxEI2HGsakKdVCkSNi7VK9mr08w5sDcFCQyWI,9553
-unique_toolkit-1.5.1.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.5.1.dist-info/METADATA,sha256=1jL7EK6jc2i-l7G1fRUBjmFTxsrp_I43P3d6Yx_zF0w,34514
-unique_toolkit-1.5.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.5.1.dist-info/RECORD,,
+unique_toolkit-1.7.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.7.0.dist-info/METADATA,sha256=EayzMLYA56Chqh20xTaaWz3cSLOY5lk6dkvvxa9tHZA,34662
+unique_toolkit-1.7.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.7.0.dist-info/RECORD,,