unique_toolkit 0.7.20__tar.gz → 0.7.22__tar.gz

{unique_toolkit-0.7.20 → unique_toolkit-0.7.22}/CHANGELOG.md

@@ -5,6 +5,11 @@ 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.22] - 2025-05-22
+- `messages` are now always serialized by alias. This affects `LanguageModelService.complete` and `LanguageModelService.complete_async`.
+
+## [0.7.21] - 2025-05-21
+- Extend the update the `ChatMessage` object to include the `Reference` object introduced in the public api
 
 ## [0.7.20] - 2025-05-21
 - Deprecate `LanguageModelTool` and associated models in favor of `LanguageModelToolDescription`

{unique_toolkit-0.7.20 → unique_toolkit-0.7.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.20
+Version: 0.7.22
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -111,6 +111,11 @@ 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.22] - 2025-05-22
+- `messages` are now always serialized by alias. This affects `LanguageModelService.complete` and `LanguageModelService.complete_async`.
+
+## [0.7.21] - 2025-05-21
+- Extend the update the `ChatMessage` object to include the `Reference` object introduced in the public api
 
 ## [0.7.20] - 2025-05-21
 - Deprecate `LanguageModelTool` and associated models in favor of `LanguageModelToolDescription`

{unique_toolkit-0.7.20 → unique_toolkit-0.7.22}/pyproject.toml

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

{unique_toolkit-0.7.20 → unique_toolkit-0.7.22}/unique_toolkit/chat/schemas.py

@@ -37,6 +37,15 @@ class ToolCall(BaseModel):
     function: Function
 
 
+class Reference(BaseModel):
+    model_config = model_config
+    name: str
+    url: str | None = None
+    sequence_number: int
+    source_id: str
+    source: str
+
+
 class ChatMessage(BaseModel):
     # This model should strictly meets https://github.com/Unique-AG/monorepo/blob/master/node/apps/node-chat/src/public-api/2023-12-06/dtos/message/public-message.dto.ts
     model_config = model_config
@@ -54,6 +63,7 @@ class ChatMessage(BaseModel):
     created_at: datetime | None = None
     completed_at: datetime | None = None
     updated_at: datetime | None = None
+    references: list[Reference] | None = None
 
     # TODO make sdk return role consistently in lowercase
     # Currently needed as sdk returns role in uppercase

{unique_toolkit-0.7.20 → unique_toolkit-0.7.22}/unique_toolkit/language_model/functions.py

@@ -222,10 +222,9 @@ def _prepare_completion_params_util(
         model_name.value if isinstance(model_name, LanguageModelName) else model_name
     )
 
-    # Different methods need different message dump parameters
     messages_dict = messages.model_dump(
         exclude_none=True,
-        by_alias=content_chunks is not None,  # Use by_alias for streaming methods
+        by_alias=True,
     )
 
     search_context = (