unique_toolkit 0.5.11__py3-none-any.whl → 0.5.13__py3-none-any.whl

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/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/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/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
+    )

{unique_toolkit-0.5.11.dist-info → unique_toolkit-0.5.13.dist-info}/METADATA

@@ -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.dist-info → unique_toolkit-0.5.13.dist-info}/RECORD

@@ -1,15 +1,16 @@
 unique_toolkit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/_common/_base_service.py,sha256=S8H0rAebx7GsOldA7xInLp3aQJt9yEPDQdsGSFRJsGg,276
+unique_toolkit/_common/_time_utils.py,sha256=ztmTovTvr-3w71Ns2VwXC65OKUUh-sQlzbHdKTQWm-w,135
 unique_toolkit/app/__init__.py,sha256=sZyGrz74jBlAjv6OcHgcp6VtP6-AKKpaVYjakr1Xk60,735
 unique_toolkit/app/init_logging.py,sha256=Sh26SRxOj8i8dzobKhYha2lLrkrMTHfB1V4jR3h23gQ,678
 unique_toolkit/app/init_sdk.py,sha256=Nv4Now4pMfM0AgRhbtatLpm_39rKxn0WmRLwmPhRl-8,1285
 unique_toolkit/app/performance/async_tasks.py,sha256=H0l3OAcosLwNHZ8d2pd-Di4wHIXfclEvagi5kfqLFPA,1941
 unique_toolkit/app/performance/async_wrapper.py,sha256=yVVcRDkcdyfjsxro-N29SBvi-7773wnfDplef6-y8xw,1077
-unique_toolkit/app/schemas.py,sha256=oNZBLyzQR4jRAIXtrcfJ3N0euoLuTVLTPL6wFhIKLRA,1323
+unique_toolkit/app/schemas.py,sha256=_PIROOUtdKvZFZdvkCORhj-MVWvfkopIQW7VIFmguRg,1364
 unique_toolkit/app/verification.py,sha256=UZqTHg3PX_QxMjeLH_BVBYoMVqMnMpeMoqvyTBKDqj8,1996
 unique_toolkit/chat/__init__.py,sha256=1prdTVfLOf6NgU-Aa1VIO-XiR6OYuRm51LaVRfKDCqc,267
 unique_toolkit/chat/schemas.py,sha256=ff4M-XMagF0Evn3FcKHHP5xzDEyufZgq9Dmit3i8r_E,802
-unique_toolkit/chat/service.py,sha256=gIj5tNkwveQ3Rv9-B83Hw1lZRx2mYEmK7megI5paJNc,13409
+unique_toolkit/chat/service.py,sha256=mjzj0GCpIbGMX_k1aei8BY-132DCPv53FI7TqbXfaKk,13544
 unique_toolkit/chat/state.py,sha256=Cjgwv_2vhDFbV69xxsn7SefhaoIAEqLx3ferdVFCnOg,1445
 unique_toolkit/chat/utils.py,sha256=ihm-wQykBWhB4liR3LnwPVPt_qGW6ETq21Mw4HY0THE,854
 unique_toolkit/content/__init__.py,sha256=MSH2sxjQyKD2Sef92fzE5Dt9SihdzivB6yliSwJfTmQ,890
@@ -22,10 +23,10 @@ unique_toolkit/embedding/service.py,sha256=Iiw-sbdkjuWlWMfLM9qyC4GNTJOotQAaVjkYv
 unique_toolkit/embedding/utils.py,sha256=v86lo__bCJbxZBQ3OcLu5SuwT6NbFfWlcq8iyk6BuzQ,279
 unique_toolkit/language_model/__init__.py,sha256=QgU_uwpVh1URQyVs6l-6Am4UwmEEhuGXNic3dUZ0FCc,1701
 unique_toolkit/language_model/infos.py,sha256=W74PiBkAIEdTdgygdVoq_EWlziVOOfhSuETq4dKX7LM,8934
-unique_toolkit/language_model/schemas.py,sha256=6PySftHNX1e00ncieRn4h35T4VneOO8j-mfpkC8Cvk8,4447
+unique_toolkit/language_model/schemas.py,sha256=h5zjZNk7O-wLKtRuiNtMCIbp5hEVXrAOviKonQcjFuI,4594
 unique_toolkit/language_model/service.py,sha256=JjsOOcGDcR7db3yF3_oDXclEGfxqmwWpL5jor7Q42cU,10470
 unique_toolkit/language_model/utils.py,sha256=WBPj1XKkDgxy_-T8HCZvsfkkSzj_1w4UZzNmyvdbBLY,1081
-unique_toolkit-0.5.11.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
-unique_toolkit-0.5.11.dist-info/METADATA,sha256=NlvoVokeI_H7krp1oJVXHs6IJmMrkI2pHoLBWPNmuCo,9581
-unique_toolkit-0.5.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.11.dist-info/RECORD,,
+unique_toolkit-0.5.13.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
+unique_toolkit-0.5.13.dist-info/METADATA,sha256=7SZ9_BbfL1M0Esja2QmEpsKvyx9Bva04cmgGeRkFaSw,10041
+unique_toolkit-0.5.13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.13.dist-info/RECORD,,