unique_toolkit 0.7.20__tar.gz → 0.7.21__tar.gz

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

@@ -5,6 +5,8 @@ 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.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.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.20
+Version: 0.7.21
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -111,6 +111,8 @@ 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.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.21}/pyproject.toml

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

{unique_toolkit-0.7.20 → unique_toolkit-0.7.21}/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