unique_toolkit 0.5.46__tar.gz → 0.5.47__tar.gz

{unique_toolkit-0.5.46 → unique_toolkit-0.5.47}/CHANGELOG.md

@@ -5,6 +5,10 @@ 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.5.47] - 2025-01-07
+- Added a message builder to build language model messages conveniently without importing all different messages.
+- Move tool_calls to assistant message as not needed anywhere else.
+
 ## [0.5.46] - 2025-01-07
 - Added `AZURE_GPT_35_TURBO_0125` as new model into toolkit.
 

{unique_toolkit-0.5.46 → unique_toolkit-0.5.47}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.46
+Version: 0.5.47
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -100,6 +100,10 @@ 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.5.47] - 2025-01-07
+- Added a message builder to build language model messages conveniently without importing all different messages.
+- Move tool_calls to assistant message as not needed anywhere else.
+
 ## [0.5.46] - 2025-01-07
 - Added `AZURE_GPT_35_TURBO_0125` as new model into toolkit.
 

{unique_toolkit-0.5.46 → unique_toolkit-0.5.47}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.5.46"
+version = "0.5.47"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

unique_toolkit-0.5.47/unique_toolkit/language_model/builder.py

@@ -0,0 +1,50 @@
+from unique_toolkit.language_model import (
+    LanguageModelAssistantMessage,
+    LanguageModelMessage,
+    LanguageModelMessages,
+    LanguageModelSystemMessage,
+    LanguageModelToolMessage,
+    LanguageModelUserMessage,
+)
+
+
+class MessagesBuilder:
+    def __init__(self):
+        self.messages: list[LanguageModelMessage] = []
+
+    def system_message_append(self, content):
+        """Appends a user message to the messages list."""
+        message = LanguageModelSystemMessage(content=content)
+        self.messages.append(message)
+        return self  # Return self to allow method chaining
+
+    def user_message_append(self, content):
+        """Appends a user message to the messages list."""
+        message = LanguageModelUserMessage(content=content)
+        self.messages.append(message)
+        return self  # Return self to allow method chaining
+
+    def assistant_message_append(self, content):
+        """Appends an assistant message to the messages list."""
+        message = LanguageModelAssistantMessage(content=content)
+        self.messages.append(message)
+        return self  # Return self to allow method chaining
+
+    def tool_message_append(self, name: str, tool_call_id: str, content: str):
+        """Appends a tool message to the messages list."""
+        message = LanguageModelToolMessage(
+            name=name, tool_call_id=tool_call_id, content=content
+        )
+        self.messages.append(message)
+        return self  # Return self to allow method chaining
+
+    def build(self, reset: bool = True) -> LanguageModelMessages:
+        """Returns the list of messages and resets the builder"""
+        messages = LanguageModelMessages(root=self.messages)
+        if reset:
+            self.messages = []
+        return messages
+
+    def model_dump(self, **kwargs):
+        """Dumps the LanguageModelMessages model"""
+        return self.build().model_dump(**kwargs)

{unique_toolkit-0.5.46 → unique_toolkit-0.5.47}/unique_toolkit/language_model/schemas.py

@@ -89,7 +89,6 @@ class LanguageModelMessage(BaseModel):
     model_config = model_config
     role: LanguageModelMessageRole
     content: Optional[str | list[dict]] = None
-    tool_calls: Optional[list[LanguageModelFunctionCall]] = None
 
     def __str__(self):
         return format_message(self.role.capitalize(), message=self.content, num_tabs=1)
@@ -113,6 +112,7 @@ class LanguageModelUserMessage(LanguageModelMessage):
 
 class LanguageModelAssistantMessage(LanguageModelMessage):
     role: LanguageModelMessageRole = LanguageModelMessageRole.ASSISTANT
+    tool_calls: Optional[list[LanguageModelFunctionCall]] = None
 
     @field_validator("role", mode="before")
     def set_role(cls, value):
@@ -159,7 +159,7 @@ class LanguageModelCompletionChoice(BaseModel):
     model_config = model_config
 
     index: int
-    message: LanguageModelMessage
+    message: LanguageModelAssistantMessage
     finish_reason: str