unique_toolkit 1.14.8__tar.gz → 1.14.10__tar.gz

{unique_toolkit-1.14.8 → unique_toolkit-1.14.10}/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.14.10] - 2025-10-13
+- Fix token counter if images in history
+
+## [1.14.9] - 2025-10-13
+- Fix removal of metadata
+
 ## [1.14.8] - 2025-10-13
 - Use default tool icon on validation error
 

{unique_toolkit-1.14.8 → unique_toolkit-1.14.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.14.8
+Version: 1.14.10
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,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.14.10] - 2025-10-13
+- Fix token counter if images in history
+
+## [1.14.9] - 2025-10-13
+- Fix removal of metadata
+
 ## [1.14.8] - 2025-10-13
 - Use default tool icon on validation error
 

{unique_toolkit-1.14.8 → unique_toolkit-1.14.10}/pyproject.toml

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

{unique_toolkit-1.14.8 → unique_toolkit-1.14.10}/unique_toolkit/agentic/history_manager/loop_token_reducer.py

@@ -266,7 +266,9 @@ class LoopTokenReducer:
         selected_messages = []
         token_count = 0
         for msg in messages[::-1]:
-            msg_token_count = self._count_tokens(str(msg.content))
+            msg_token_count = self._count_message_tokens(
+                LanguageModelMessages(root=[msg])
+            )
             if token_count + msg_token_count > token_limit:
                 break
             selected_messages.append(msg)
@@ -293,9 +295,6 @@ class LoopTokenReducer:
                 )
         return messages
 
-    def _count_tokens(self, text: str) -> int:
-        return len(self._encoder.encode(text))
-
     def ensure_last_message_is_user_message(self, limited_history_messages):
         """
         As the token limit can be reached in the middle of a gpt_request,

{unique_toolkit-1.14.8 → unique_toolkit-1.14.10}/unique_toolkit/services/knowledge_base.py

@@ -637,6 +637,8 @@ class KnowledgeBaseService:
     ) -> ContentInfo:
         """
         Removes the specified keys irreversibly from the content metadata.
+
+        Note: Keys are camelized before being removed as metadata keys are stored in camelCase.
         """
 
         if content_info.metadata is None:
@@ -644,7 +646,7 @@ class KnowledgeBaseService:
             return content_info
 
         for key in keys_to_remove:
-            content_info.metadata.pop(key, None)
+            content_info.metadata[humps.camelize(key)] = None
 
         return update_content(
             user_id=self._user_id,
@@ -673,6 +675,11 @@ class KnowledgeBaseService:
         metadata_filter: dict[str, Any] | None = None,
         content_infos: list[ContentInfo] | None = None,
     ) -> list[ContentInfo]:
+        """Update the metadata of the contents matching the metadata filter.
+
+        Note: Keys are camelized before being updated as metadata keys are stored in camelCase.
+        """
+
         additional_metadata_camelized = humps.camelize(additional_metadata)
         additional_metadata_camelized = self._pop_forbidden_metadata_keys(
             additional_metadata_camelized
@@ -710,6 +717,12 @@ class KnowledgeBaseService:
         metadata_filter: dict[str, Any] | None = None,
         content_infos: list[ContentInfo] | None = None,
     ) -> list[ContentInfo]:
+        """Remove the specified keys irreversibly from the content metadata.
+
+        Note: Keys are camelized before being removed as metadata keys are stored in camelCase.
+
+        """
+
         if content_infos is None:
             content_infos = self.get_paginated_content_infos(
                 metadata_filter=metadata_filter,