unique_toolkit 0.8.55__tar.gz → 0.8.56__tar.gz

{unique_toolkit-0.8.55 → unique_toolkit-0.8.56}/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).
 
+## [0.8.56] - 2025-09-12
+- Fixed token counter in utils
+
 ## [0.8.55] - 2025-09-10
 - Update documentation with agentic managers
 

{unique_toolkit-0.8.55 → unique_toolkit-0.8.56}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.8.55
+Version: 0.8.56
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -119,6 +119,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).
 
+## [0.8.56] - 2025-09-12
+- Fixed token counter in utils
+
 ## [0.8.55] - 2025-09-10
 - Update documentation with agentic managers
 

{unique_toolkit-0.8.55 → unique_toolkit-0.8.56}/pyproject.toml

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

{unique_toolkit-0.8.55 → unique_toolkit-0.8.56}/unique_toolkit/history_manager/history_manager.py

@@ -170,7 +170,7 @@ class HistoryManager:
         )  # it can be that the tool response does not have content chunks
 
         # Transform content chunks into sources to be appended to tool result
-        sources = transform_chunks_to_string(
+        stringified_sources, sources = transform_chunks_to_string(
             content_chunks,
             self._source_enumerator,
             None,  # Use None for SourceFormatConfig
@@ -183,7 +183,7 @@ class HistoryManager:
 
         # Append the result to the history
         return LanguageModelToolMessage(
-            content=sources,
+            content=stringified_sources,
             tool_call_id=tool_response.id,  # type: ignore
             name=tool_response.name,
         )

{unique_toolkit-0.8.55 → unique_toolkit-0.8.56}/unique_toolkit/history_manager/utils.py

@@ -64,7 +64,7 @@ def transform_chunks_to_string(
     max_source_number: int,
     cfg: SourceFormatConfig | None,
     full_sources_serialize_dump: bool = False,
-) -> str:
+) -> tuple[str, list[Source]]:
     """Transform content chunks into a string of sources.
 
     Args:
@@ -76,7 +76,7 @@ def transform_chunks_to_string(
         str: String for the tool call response
     """
     if not content_chunks:
-        return "No relevant sources found."
+        return "No relevant sources found.", []
     if full_sources_serialize_dump:
         sources = [
             Source(
@@ -114,7 +114,7 @@ def transform_chunks_to_string(
             }
             for i, chunk in enumerate(content_chunks)
         ]
-    return json.dumps(sources)
+    return json.dumps(sources), sources
 
 
 def load_sources_from_string(