unique_toolkit 0.5.21__tar.gz → 0.5.22__tar.gz

{unique_toolkit-0.5.21 → unique_toolkit-0.5.22}/CHANGELOG.md

@@ -5,8 +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.5.22] - 2024-09-17
+- Add `LanguageModelToolMessage` as additional `LanguageModelMessage`
+
 ## [0.5.21] - 2024-09-16
-- Add `tool` as new role `ChatMessage`, as well as `tool_calls` and `tool_call_id` as additional parameters
+- Add `tool` as new role to `ChatMessage`, as well as `tool_calls` and `tool_call_id` as additional parameters
 
 ## [0.5.20] - 2024-09-16
 - `LanguageModelService` now supports complete_util_async that can be called without instantiating the class, currently being used in the Hallucination service and evaluation API

{unique_toolkit-0.5.21 → unique_toolkit-0.5.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.21
+Version: 0.5.22
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -100,8 +100,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.5.22] - 2024-09-17
+- Add `LanguageModelToolMessage` as additional `LanguageModelMessage`
+
 ## [0.5.21] - 2024-09-16
-- Add `tool` as new role `ChatMessage`, as well as `tool_calls` and `tool_call_id` as additional parameters
+- Add `tool` as new role to `ChatMessage`, as well as `tool_calls` and `tool_call_id` as additional parameters
 
 ## [0.5.20] - 2024-09-16
 - `LanguageModelService` now supports complete_util_async that can be called without instantiating the class, currently being used in the Hallucination service and evaluation API

{unique_toolkit-0.5.21 → unique_toolkit-0.5.22}/pyproject.toml

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

{unique_toolkit-0.5.21 → unique_toolkit-0.5.22}/unique_toolkit/language_model/__init__.py

@@ -39,6 +39,9 @@ from .schemas import (
 from .schemas import (
     LanguageModelTool as LanguageModelTool,
 )
+from .schemas import (
+    LanguageModelToolMessage as LanguageModelToolMessage,
+)
 from .schemas import (
     LanguageModelToolParameterProperty as LanguageModelToolParameterProperty,
 )

{unique_toolkit-0.5.21 → unique_toolkit-0.5.22}/unique_toolkit/language_model/schemas.py

@@ -17,6 +17,7 @@ class LanguageModelMessageRole(StrEnum):
     USER = "user"
     SYSTEM = "system"
     ASSISTANT = "assistant"
+    TOOL = "tool"
 
 
 class LanguageModelFunction(BaseModel):
@@ -72,6 +73,14 @@ class LanguageModelAssistantMessage(LanguageModelMessage):
         return LanguageModelMessageRole.ASSISTANT
 
 
+class LanguageModelToolMessage(LanguageModelMessage):
+    role: LanguageModelMessageRole = LanguageModelMessageRole.TOOL
+
+    @field_validator("role", mode="before")
+    def set_role(cls, value):
+        return LanguageModelMessageRole.TOOL
+
+
 class LanguageModelMessages(RootModel):
     root: list[LanguageModelMessage]