unique_toolkit 0.7.2__tar.gz → 0.7.3__tar.gz

{unique_toolkit-0.7.2 → unique_toolkit-0.7.3}/CHANGELOG.md

@@ -6,6 +6,9 @@ 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.3] - 2025-03-20
+- Enable handling tool calls in message builder
+
 ## [0.7.2] - 2025-03-17
 - HotFix `ContentService.search_content_chunks` to use `chat_id` from event if provided.
 

{unique_toolkit-0.7.2 → unique_toolkit-0.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.2
+Version: 0.7.3
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -112,6 +112,9 @@ 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.3] - 2025-03-20
+- Enable handling tool calls in message builder
+
 ## [0.7.2] - 2025-03-17
 - HotFix `ContentService.search_content_chunks` to use `chat_id` from event if provided.
 

{unique_toolkit-0.7.2 → unique_toolkit-0.7.3}/pyproject.toml

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

{unique_toolkit-0.7.2 → unique_toolkit-0.7.3}/unique_toolkit/language_model/builder.py

@@ -2,6 +2,8 @@ from typing_extensions import Self
 
 from unique_toolkit.language_model import (
     LanguageModelAssistantMessage,
+    LanguageModelFunction,
+    LanguageModelFunctionCall,
     LanguageModelMessage,
     LanguageModelMessageRole,
     LanguageModelMessages,
@@ -48,11 +50,22 @@ class MessagesBuilder:
         self.messages.append(message)
         return self
 
-    def assistant_message_append(self, content: str) -> Self:
+    def assistant_message_append(
+        self, content: str, tool_calls: list[LanguageModelFunction] | None = None
+    ) -> Self:
         """Appends an assistant message to the messages list."""
         message = LanguageModelAssistantMessage(content=content)
+        if tool_calls:
+            message.tool_calls = [
+                LanguageModelFunctionCall(
+                    id=tool_call.id,
+                    type="function",
+                    function=tool_call,
+                )
+                for tool_call in tool_calls
+            ]
         self.messages.append(message)
-        return self  # Return self to allow method chaining
+        return self
 
     def tool_message_append(self, name: str, tool_call_id: str, content: str) -> Self:
         """Appends a tool message to the messages list."""
@@ -60,7 +73,7 @@ class MessagesBuilder:
             name=name, tool_call_id=tool_call_id, content=content
         )
         self.messages.append(message)
-        return self  # Return self to allow method chaining
+        return self
 
     def build(self, reset: bool = True) -> LanguageModelMessages:
         """Returns the list of messages and resets the builder"""