unique_toolkit 0.5.11__tar.gz → 0.5.13__tar.gz

{unique_toolkit-0.5.11 → unique_toolkit-0.5.13}/CHANGELOG.md

@@ -5,6 +5,15 @@ 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.13] - 2024-08-19
+- Added `items` to `LanguageModelToolParameterProperty` schema to add support for parameters with list types.
+- Added `returns` to `LanguageModelTool` schema to describe the return types of tool calls.
+
+
+## [0.5.12] - 2024-08-7
+- added `completedAt` datetime to `unique_sdk.Message.modify` and `unique_sdk.Message.modify_async`
+- added `original_text` and `language` to `EventUserMessage`
+
 ## [0.5.11] - 2024-08-6
 - made all domain specific functions and classes directly importable from `unique_toolkit.[DOMAIN_NAME]`
 - renamed `RerankerConfig` to `ContentRerankerConfig`

{unique_toolkit-0.5.11 → unique_toolkit-0.5.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.11
+Version: 0.5.13
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: numpy (>=1.26.4,<2.0.0)
 Requires-Dist: pydantic (>=2.8.2,<3.0.0)
 Requires-Dist: pyhumps (>=3.8.0,<4.0.0)
+Requires-Dist: pytest-mock (>=3.14.0,<4.0.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: regex (>=2024.5.15,<2025.0.0)
 Requires-Dist: tiktoken (>=0.7.0,<0.8.0)
@@ -100,6 +101,15 @@ 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.13] - 2024-08-19
+- Added `items` to `LanguageModelToolParameterProperty` schema to add support for parameters with list types.
+- Added `returns` to `LanguageModelTool` schema to describe the return types of tool calls.
+
+
+## [0.5.12] - 2024-08-7
+- added `completedAt` datetime to `unique_sdk.Message.modify` and `unique_sdk.Message.modify_async`
+- added `original_text` and `language` to `EventUserMessage`
+
 ## [0.5.11] - 2024-08-6
 - made all domain specific functions and classes directly importable from `unique_toolkit.[DOMAIN_NAME]`
 - renamed `RerankerConfig` to `ContentRerankerConfig`

{unique_toolkit-0.5.11 → unique_toolkit-0.5.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.5.11"
+version = "0.5.13"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",
@@ -21,6 +21,7 @@ python-dotenv = "^1.0.1"
 regex = "^2024.5.15"
 tiktoken = "^0.7.0"
 unique-sdk = "^0.9.4"
+pytest-mock = "^3.14.0"
 
 
 [tool.poetry.group.dev.dependencies]

unique_toolkit-0.5.13/unique_toolkit/_common/_time_utils.py

@@ -0,0 +1,5 @@
+from datetime import datetime
+
+
+def get_datetime_now(format: str = "%Y-%m-%d %H:%M:%S.%f"):
+    return datetime.now().strftime(format)

{unique_toolkit-0.5.11 → unique_toolkit-0.5.13}/unique_toolkit/app/schemas.py

@@ -21,7 +21,9 @@ class EventUserMessage(BaseModel):
 
     id: str
     text: str
+    original_text: str
     created_at: str
+    language: str
 
 
 class EventAssistantMessage(BaseModel):

{unique_toolkit-0.5.11 → unique_toolkit-0.5.13}/unique_toolkit/chat/service.py

@@ -5,6 +5,7 @@ from typing import Optional
 import unique_sdk
 from unique_sdk._list_object import ListObject
 
+from unique_toolkit._common import _time_utils
 from unique_toolkit._common._base_service import BaseService
 from unique_toolkit.app.schemas import Event
 from unique_toolkit.chat.schemas import ChatMessage, ChatMessageRole
@@ -55,11 +56,12 @@ class ChatService(BaseService):
             message = unique_sdk.Message.modify(
                 user_id=self.event.user_id,
                 company_id=self.event.company_id,
-                id=message_id,  # type: ignore
+                id=message_id,
                 chatId=self.event.payload.chat_id,
                 text=content,
-                references=self._map_references(references),  # type: ignore
+                references=self._map_references(references),
                 debugInfo=debug_info or {},
+                completedAt=_time_utils.get_datetime_now(),  # type: ignore
             )
         except Exception as e:
             self.logger.error(f"Failed to modify assistant message: {e}")
@@ -94,11 +96,12 @@ class ChatService(BaseService):
             message = await unique_sdk.Message.modify_async(
                 user_id=self.event.user_id,
                 company_id=self.event.company_id,
-                id=message_id,  # type: ignore
+                id=message_id,
                 chatId=self.event.payload.chat_id,
                 text=content,
-                references=self._map_references(references),  # type: ignore
+                references=self._map_references(references),
                 debugInfo=debug_info or {},
+                completedAt=_time_utils.get_datetime_now(),  # type: ignore
             )
         except Exception as e:
             self.logger.error(f"Failed to modify assistant message: {e}")

{unique_toolkit-0.5.11 → unique_toolkit-0.5.13}/unique_toolkit/language_model/schemas.py

@@ -1,6 +1,6 @@
 import json
 from enum import StrEnum
-from typing import Any, Optional
+from typing import Any, Optional, Self
 
 from humps import camelize
 from pydantic import BaseModel, ConfigDict, RootModel, field_validator, model_validator
@@ -154,6 +154,7 @@ class LanguageModelToolParameterProperty(BaseModel):
     type: str
     description: str
     enum: Optional[list[Any]] = None
+    items: Optional[Self] = None
 
 
 class LanguageModelToolParameters(BaseModel):
@@ -166,3 +167,6 @@ class LanguageModelTool(BaseModel):
     name: str
     description: str
     parameters: LanguageModelToolParameters
+    returns: LanguageModelToolParameterProperty | LanguageModelToolParameters | None = (
+        None
+    )