unique_toolkit 0.8.55__tar.gz → 0.8.57__tar.gz

{unique_toolkit-0.8.55 → unique_toolkit-0.8.57}/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).
 
+## [0.8.57] - 2025-09-14
+- Added more utils to commons
+
+## [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.57}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.8.55
+Version: 0.8.57
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -119,6 +119,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).
 
+## [0.8.57] - 2025-09-14
+- Added more utils to commons
+
+## [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.57}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.8.55"
+version = "0.8.57"
 description = ""
 authors = [
     "Cedric Klinkert <cedric.klinkert@unique.ch>",
@@ -8,7 +8,7 @@ authors = [
     "Sadique Sheik <sadique@unique.ch>",
     "Fabian Schläpfer <fabian@unique.ch>",
     "Pascal Hauri <pascal@unique.ch>",
-    "Ejiroghene Oki-Peter<oki-peter.ejiroghene@unique.ai>"
+    "Ejiroghene Oki-Peter<oki-peter.ejiroghene@unique.ai>",
 ]
 readme = ["README.md", "CHANGELOG.md"]
 license = "Proprietary"

unique_toolkit-0.8.57/unique_toolkit/_common/utils/write_configuration.py

@@ -0,0 +1,51 @@
+import json
+from pathlib import Path
+
+from humps import kebabize, pascalize
+from pydantic import BaseModel
+
+
+def write_module_configuration_schema(
+    service_folderpath: Path,
+    write_folderpath: Path,
+    config: BaseModel,
+    sub_name: str = "",
+):
+    filename_prefix = pascalize(service_folderpath.name)
+
+    filepath = (
+        write_folderpath
+        / f"{filename_prefix}{f'-{sub_name}' if sub_name else ''}Schema.json"
+    )
+
+    with open(filepath, "w") as f:
+        json.dump(config.model_json_schema(by_alias=True), f, indent=4)
+
+
+def write_service_configuration(
+    service_folderpath: Path,
+    write_folderpath: Path,
+    config: BaseModel,
+    sub_name: str = "",
+):
+    filename_prefix = kebabize(service_folderpath.name)
+
+    filepath = (
+        write_folderpath
+        / f"{filename_prefix}{f'-{sub_name}' if sub_name else ''}-configuration-schema.json"
+    )
+
+    with open(filepath, "w") as f:
+        json.dump(config.model_json_schema(by_alias=True), f, indent=4)
+    filepath = (
+        write_folderpath
+        / f"{filename_prefix}{f'-{sub_name}' if sub_name else ''}-default-configuration.json"
+    )
+
+    # We exclude language_model_info as it is infered from language_model_name
+    with open(filepath, "w") as f:
+        f.write(
+            config.model_dump_json(
+                by_alias=True, indent=4, exclude=set(["language_model_info"])
+            )
+        )

{unique_toolkit-0.8.55 → unique_toolkit-0.8.57}/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.57}/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(