unique_toolkit 0.5.22__py3-none-any.whl → 0.5.23__py3-none-any.whl

unique_toolkit/chat/service.py

@@ -35,7 +35,9 @@ class ChatService(BaseService):
         Args:
             debug_info (dict): The new debug information.
         """
-        params = self._construct_message_params(assistant=False, debug_info=debug_info)
+        params = self._construct_message_modify_params(
+            assistant=False, debug_info=debug_info
+        )
         try:
             await unique_sdk.Message.modify_async(**params)
         except Exception as e:
@@ -49,7 +51,9 @@ class ChatService(BaseService):
         Args:
             debug_info (dict): The new debug information.
         """
-        params = self._construct_message_params(assistant=False, debug_info=debug_info)
+        params = self._construct_message_modify_params(
+            assistant=False, debug_info=debug_info
+        )
         try:
             unique_sdk.Message.modify(**params)
         except Exception as e:
@@ -62,6 +66,7 @@ class ChatService(BaseService):
         references: Optional[list[ContentReference]] = None,
         debug_info: Optional[dict] = None,
         message_id: Optional[str] = None,
+        set_completed_at: Optional[bool] = False,
     ) -> ChatMessage:
         """
         Modifies a message in the chat session synchronously.
@@ -71,6 +76,7 @@ class ChatService(BaseService):
             references (list[ContentReference]): list of ContentReference objects. Defaults to [].
             debug_info (dict[str, Any]]]): Debug information. Defaults to {}.
             message_id (str, optional): The message ID. Defaults to None, then the ChatState user message id is used.
+            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
 
         Returns:
             ChatMessage: The modified message.
@@ -79,12 +85,13 @@ class ChatService(BaseService):
             Exception: If the modification fails.
         """
         try:
-            params = self._construct_message_params(
+            params = self._construct_message_modify_params(
                 assistant=False,
                 content=content,
                 references=references,
                 debug_info=debug_info,
                 message_id=message_id,
+                set_completed_at=set_completed_at,
             )
             message = unique_sdk.Message.modify(**params)
         except Exception as e:
@@ -98,6 +105,7 @@ class ChatService(BaseService):
         references: list[ContentReference] = [],
         debug_info: dict = {},
         message_id: Optional[str] = None,
+        set_completed_at: Optional[bool] = False,
     ) -> ChatMessage:
         """
         Modifies a message in the chat session asynchronously.
@@ -107,6 +115,7 @@ class ChatService(BaseService):
             message_id (str, optional): The message ID. Defaults to None, then the ChatState user message id is used.
             references (list[ContentReference]): list of ContentReference objects. Defaults to None.
             debug_info (Optional[dict[str, Any]]], optional): Debug information. Defaults to None.
+            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
 
         Returns:
             ChatMessage: The modified message.
@@ -115,12 +124,13 @@ class ChatService(BaseService):
             Exception: If the modification fails.
         """
         try:
-            params = self._construct_message_params(
+            params = self._construct_message_modify_params(
                 assistant=False,
                 content=content,
                 references=references,
                 debug_info=debug_info,
                 message_id=message_id,
+                set_completed_at=set_completed_at,
             )
             message = await unique_sdk.Message.modify_async(**params)
         except Exception as e:
@@ -134,6 +144,7 @@ class ChatService(BaseService):
         references: list[ContentReference] = [],
         debug_info: dict = {},
         message_id: Optional[str] = None,
+        set_completed_at: Optional[bool] = False,
     ) -> ChatMessage:
         """
         Modifies a message in the chat session synchronously.
@@ -143,6 +154,7 @@ class ChatService(BaseService):
             references (list[ContentReference]): list of ContentReference objects. Defaults to [].
             debug_info (dict[str, Any]]]): Debug information. Defaults to {}.
             message_id (Optional[str]): The message ID. Defaults to None.
+            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
 
         Returns:
             ChatMessage: The modified message.
@@ -151,12 +163,13 @@ class ChatService(BaseService):
             Exception: If the modification fails.
         """
         try:
-            params = self._construct_message_params(
+            params = self._construct_message_modify_params(
                 assistant=True,
                 content=content,
                 references=references,
                 debug_info=debug_info,
                 message_id=message_id,
+                set_completed_at=set_completed_at,
             )
             message = unique_sdk.Message.modify(**params)
         except Exception as e:
@@ -170,6 +183,7 @@ class ChatService(BaseService):
         references: list[ContentReference] = [],
         debug_info: dict = {},
         message_id: Optional[str] = None,
+        set_completed_at: Optional[bool] = False,
     ) -> ChatMessage:
         """
         Modifies a message in the chat session asynchronously.
@@ -179,6 +193,7 @@ class ChatService(BaseService):
             message_id (str, optional): The message ID. Defaults to None, then the ChatState assistant message id is used.
             references (list[ContentReference]): list of ContentReference objects. Defaults to None.
             debug_info (Optional[dict[str, Any]]], optional): Debug information. Defaults to None.
+            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
 
         Returns:
             ChatMessage: The modified message.
@@ -187,12 +202,13 @@ class ChatService(BaseService):
             Exception: If the modification fails.
         """
         try:
-            params = self._construct_message_params(
+            params = self._construct_message_modify_params(
                 assistant=True,
                 content=content,
                 references=references,
                 debug_info=debug_info,
                 message_id=message_id,
+                set_completed_at=set_completed_at,
             )
             message = await unique_sdk.Message.modify_async(**params)
         except Exception as e:
@@ -287,6 +303,7 @@ class ChatService(BaseService):
         content: str,
         references: list[ContentReference] = [],
         debug_info: dict = {},
+        set_completed_at: Optional[bool] = False,
     ):
         """
         Creates a message in the chat session synchronously.
@@ -295,6 +312,7 @@ class ChatService(BaseService):
             content (str): The content for the message.
             references (list[ContentReference]): list of ContentReference objects. Defaults to None.
             debug_info (dict[str, Any]]): Debug information. Defaults to None.
+            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
 
         Returns:
             ChatMessage: The created message.
@@ -304,16 +322,14 @@ class ChatService(BaseService):
         """
 
         try:
-            message = unique_sdk.Message.create(
-                user_id=self.event.user_id,
-                company_id=self.event.company_id,
-                chatId=self.event.payload.chat_id,
-                assistantId=self.event.payload.assistant_id,
-                text=content,
-                role=ChatMessageRole.ASSISTANT.name,
-                references=self._map_references(references),  # type: ignore
-                debugInfo=debug_info,
+            params = self._construct_message_create_params(
+                content=content,
+                references=references,
+                debug_info=debug_info,
+                set_completed_at=set_completed_at,
             )
+
+            message = unique_sdk.Message.create(**params)
         except Exception as e:
             self.logger.error(f"Failed to create assistant message: {e}")
             raise e
@@ -324,6 +340,7 @@ class ChatService(BaseService):
         content: str,
         references: list[ContentReference] = [],
         debug_info: dict = {},
+        set_completed_at: Optional[bool] = False,
     ):
         """
         Creates a message in the chat session asynchronously.
@@ -332,6 +349,7 @@ class ChatService(BaseService):
             content (str): The content for the message.
             references (list[ContentReference]): list of references. Defaults to None.
             debug_info (dict[str, Any]]): Debug information. Defaults to None.
+            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
 
         Returns:
             ChatMessage: The created message.
@@ -340,16 +358,14 @@ class ChatService(BaseService):
             Exception: If the creation fails.
         """
         try:
-            message = await unique_sdk.Message.create_async(
-                user_id=self.event.user_id,
-                company_id=self.event.company_id,
-                chatId=self.event.payload.chat_id,
-                assistantId=self.event.payload.assistant_id,
-                text=content,
-                role=ChatMessageRole.ASSISTANT.name,
-                references=self._map_references(references),  # type: ignore
-                debugInfo=debug_info,
+            params = self._construct_message_create_params(
+                content=content,
+                references=references,
+                debug_info=debug_info,
+                set_completed_at=set_completed_at,
             )
+
+            message = await unique_sdk.Message.create_async(**params)
         except Exception as e:
             self.logger.error(f"Failed to create assistant message: {e}")
             raise e
@@ -477,13 +493,14 @@ class ChatService(BaseService):
         last_index = max(0, last_index)
         return messages[last_index:]
 
-    def _construct_message_params(
+    def _construct_message_modify_params(
         self,
         assistant: bool = True,
         content: Optional[str] = None,
         references: Optional[list[ContentReference]] = None,
         debug_info: Optional[dict] = None,
         message_id: Optional[str] = None,
+        set_completed_at: Optional[bool] = False,
     ):
         if message_id:
             # Message ID specified. No need to guess
@@ -497,6 +514,11 @@ class ChatService(BaseService):
             if content is None:
                 content = self.event.payload.user_message.text
 
+        if set_completed_at:
+            completed_at_datetime = _time_utils.get_datetime_now()
+        else:
+            completed_at_datetime = None
+
         params = {
             "user_id": self.event.user_id,
             "company_id": self.event.company_id,
@@ -505,6 +527,39 @@ class ChatService(BaseService):
             "text": content,
             "references": self._map_references(references) if references else [],
             "debugInfo": debug_info,
-            "completedAt": _time_utils.get_datetime_now(),
+            "completedAt": completed_at_datetime,
+        }
+        return params
+
+    def _construct_message_create_params(
+        self,
+        role: ChatMessageRole = ChatMessageRole.ASSISTANT,
+        content: Optional[str] = None,
+        references: Optional[list[ContentReference]] = None,
+        debug_info: Optional[dict] = None,
+        assistantId: Optional[str] = None,
+        set_completed_at: Optional[bool] = False,
+    ):
+        if assistantId:
+            # Assistant ID specified. No need to guess
+            assistantId = assistantId
+        else:
+            assistantId = self.event.payload.assistant_id
+
+        if set_completed_at:
+            completed_at_datetime = _time_utils.get_datetime_now()
+        else:
+            completed_at_datetime = None
+
+        params = {
+            "user_id": self.event.user_id,
+            "company_id": self.event.company_id,
+            "assistantId": assistantId,
+            "role": role.value.upper(),
+            "chatId": self.event.payload.chat_id,
+            "text": content,
+            "references": self._map_references(references) if references else [],
+            "debugInfo": debug_info,
+            "completedAt": completed_at_datetime,
         }
         return params

unique_toolkit-0.5.23.dist-info/LICENSE

@@ -0,0 +1 @@
+`unique_toolkit` is covered by the [`Unique License v1`](https://github.com/Unique-AG/license/releases/tag/unique-license.v1), unless the/a header or a nested LICENSE specifies another license.

{unique_toolkit-0.5.22.dist-info → unique_toolkit-0.5.23.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.22
+Version: 0.5.23
 Summary: 
-License: MIT
+License: Proprietary
 Author: Martin Fadler
 Author-email: martin.fadler@unique.ch
 Requires-Python: >=3.11,<4.0
-Classifier: License :: OSI Approved :: MIT License
+Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
@@ -17,7 +17,7 @@ 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)
 Requires-Dist: typing-extensions (>=4.9.0,<5.0.0)
-Requires-Dist: unique-sdk (>=0.9.6,<0.10.0)
+Requires-Dist: unique-sdk (>=0.9.7,<0.10.0)
 Description-Content-Type: text/markdown
 
 # Unique Toolkit
@@ -100,6 +100,9 @@ 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.23] - 2024-09-23
+- Add `set_completed_at` as a boolen parameter to the following functions: `modify_user_message`, `modify_user_message_async`, `modify_assistant_message`, `modify_assistant_message_async`, `create_assistant_message` and `create_assistant_message`. This parameter allows the `completedAt` timestamp on the database to be updated when set to True.
+
 ## [0.5.22] - 2024-09-17
 - Add `LanguageModelToolMessage` as additional `LanguageModelMessage`
 

{unique_toolkit-0.5.22.dist-info → unique_toolkit-0.5.23.dist-info}/RECORD

@@ -10,7 +10,7 @@ unique_toolkit/app/schemas.py,sha256=tzrmUFKZUdC1P3LxZ7DrElpkMtekUDoClb7jCRzGqNQ
 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=IHOnb3pWlRlSPoEUWBTl6LK8YeMdlg2iXi9ghyBeiLw,1495
-unique_toolkit/chat/service.py,sha256=nVZ8agpEsXI2-5bS61e1Eawd0abioEQPov2xahiJtfk,17848
+unique_toolkit/chat/service.py,sha256=5BurWe0skZ7O3DVztAcU_fRJeO8bbD_Xv7sOlMQYTmI,20252
 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
@@ -26,7 +26,7 @@ unique_toolkit/language_model/infos.py,sha256=ETAUV0YTs6BjwuiTdhKz247CtL0W8Jwo3-
 unique_toolkit/language_model/schemas.py,sha256=sLpE29Ks0zEfhZUQrYOt1Cak2xzQcr9fpTXFDHkfURA,4868
 unique_toolkit/language_model/service.py,sha256=CvVo5CBa5Ia_fQD3DtJRsVChybuUfGFV5ml2_78_p1I,13395
 unique_toolkit/language_model/utils.py,sha256=WBPj1XKkDgxy_-T8HCZvsfkkSzj_1w4UZzNmyvdbBLY,1081
-unique_toolkit-0.5.22.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
-unique_toolkit-0.5.22.dist-info/METADATA,sha256=Ic20qPiV5WMEWyvebx9GXvJCIkSZjJOax2EhwqACo6o,11341
-unique_toolkit-0.5.22.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.22.dist-info/RECORD,,
+unique_toolkit-0.5.23.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.5.23.dist-info/METADATA,sha256=hSBE8S9sPqcS7ZvvaKEZh-PUL1tvYauGjq4_XbXWYGE,11719
+unique_toolkit-0.5.23.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.23.dist-info/RECORD,,

unique_toolkit-0.5.22.dist-info/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2024 Unique-AG
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.