unique_toolkit 0.7.31__tar.gz → 0.7.32__tar.gz

{unique_toolkit-0.7.31 → unique_toolkit-0.7.32}/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.7.32] - 2025-06-24
+- Create `classmethod` for `LanguageModelMessages` to load raw messages to root
+
 ## [0.7.31] - 2025-06-19
 - Add typings to references in payload from `LanguageModelStreamResponseMessage` 
 - Add `original_index` to the base reference to reflect updated api

{unique_toolkit-0.7.31 → unique_toolkit-0.7.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.31
+Version: 0.7.32
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -17,7 +17,7 @@ Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: regex (>=2024.5.15,<2025.0.0)
 Requires-Dist: tiktoken (>=0.7.0,<0.8.0)
 Requires-Dist: typing-extensions (>=4.9.0,<5.0.0)
-Requires-Dist: unique-sdk (>=0.9.33,<0.10.0)
+Requires-Dist: unique-sdk (>=0.9.36,<0.10.0)
 Description-Content-Type: text/markdown
 
 # Unique Toolkit
@@ -111,6 +111,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.7.32] - 2025-06-24
+- Create `classmethod` for `LanguageModelMessages` to load raw messages to root
+
 ## [0.7.31] - 2025-06-19
 - Add typings to references in payload from `LanguageModelStreamResponseMessage` 
 - Add `original_index` to the base reference to reflect updated api

{unique_toolkit-0.7.31 → unique_toolkit-0.7.32}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.7.31"
+version = "0.7.32"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",
@@ -20,7 +20,7 @@ numpy = "^1.26.4"
 python-dotenv = "^1.0.1"
 regex = "^2024.5.15"
 tiktoken = "^0.7.0"
-unique-sdk = "^0.9.33"
+unique-sdk = "^0.9.36"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "0.11.7"

{unique_toolkit-0.7.31 → unique_toolkit-0.7.32}/unique_toolkit/language_model/schemas.py

@@ -243,6 +243,40 @@ class LanguageModelMessages(RootModel):
         | LanguageModelUserMessage
     ]
 
+    @classmethod
+    def load_messages_to_root(cls, data: list[dict] | dict) -> Self:
+        """Convert list of dictionaries to appropriate message objects based on role."""
+        # Handle case where data is already wrapped in root
+        if isinstance(data, dict) and "root" in data:
+            messages_list = data["root"]
+        elif isinstance(data, list):
+            messages_list = data
+        else:
+            raise ValueError("Invalid data type")
+
+        # Convert the messages list
+        converted_messages = []
+        for item in messages_list:
+            if isinstance(item, dict):
+                role = item.get("role", "").lower()
+
+                # Map dictionary to appropriate message class based on role
+                if role == "system":
+                    converted_messages.append(LanguageModelSystemMessage(**item))
+                elif role == "user":
+                    converted_messages.append(LanguageModelUserMessage(**item))
+                elif role == "assistant":
+                    converted_messages.append(LanguageModelAssistantMessage(**item))
+                elif role == "tool":
+                    converted_messages.append(LanguageModelToolMessage(**item))
+                else:
+                    # Fallback to base LanguageModelMessage
+                    converted_messages.append(LanguageModelMessage(**item))
+            else:
+                # If it's already a message object, keep it as is
+                converted_messages.append(item)
+        return cls(root=converted_messages)
+
     def __str__(self):
         return "\n\n".join([str(message) for message in self.root])