unique_toolkit 0.5.52__py3-none-any.whl → 0.5.53__py3-none-any.whl

unique_toolkit/chat/schemas.py

@@ -87,11 +87,10 @@ class MessageAssessment(BaseModel):
     id: str
     object: str
     message_id: str
-    assistant_message_id: str
     status: MessageAssessmentStatus
-    explanation: str
-    label: MessageAssessmentLabel
     type: MessageAssessmentType
+    explanation: str | None = None
+    label: MessageAssessmentLabel | None = None
     is_visible: bool
-    created_at: datetime
-    updated_at: datetime
+    created_at: datetime | None = None
+    updated_at: datetime | None = None

unique_toolkit/chat/service.py

@@ -599,9 +599,9 @@ class ChatService(BaseService):
         self,
         assistant_message_id: str,
         status: MessageAssessmentStatus,
-        explanation: str,
-        label: MessageAssessmentLabel,
         type: MessageAssessmentType,
+        explanation: str | None = None,
+        label: MessageAssessmentLabel | None = None,
         is_visible: bool = True,
     ) -> MessageAssessment:
         """
@@ -610,9 +610,9 @@ class ChatService(BaseService):
         Args:
             assistant_message_id (str): The ID of the assistant message to assess
             status (MessageAssessmentStatus): The status of the assessment (e.g. "DONE")
-            explanation (str): Explanation of the assessment
-            label (MessageAssessmentLabel): The assessment label (e.g. "NEGATIVE")
             type (MessageAssessmentType): The type of assessment (e.g. "HALLUCINATION")
+            explanation (str | None): Explanation of the assessment
+            label (MessageAssessmentLabel | None): The assessment label (e.g. "NEGATIVE")
             is_visible (bool): Whether the assessment is visible to users. Defaults to True.
 
         Returns:
@@ -625,11 +625,11 @@ class ChatService(BaseService):
             assessment = unique_sdk.MessageAssessment.create(
                 user_id=self.event.user_id,
                 company_id=self.event.company_id,
-                assistant_message_id=assistant_message_id,
+                messageId=assistant_message_id,
                 status=status.name,
                 explanation=explanation,
-                label=label.name,
-                type=type.name,
+                label=label.name if label else None,
+                type=type.name if type else None,
                 isVisible=is_visible,
             )
             return MessageAssessment(**assessment)
@@ -641,9 +641,9 @@ class ChatService(BaseService):
         self,
         assistant_message_id: str,
         status: MessageAssessmentStatus,
-        explanation: str,
-        label: MessageAssessmentLabel,
         type: MessageAssessmentType,
+        explanation: str | None = None,
+        label: MessageAssessmentLabel | None = None,
         is_visible: bool = True,
     ) -> MessageAssessment:
         """
@@ -652,9 +652,9 @@ class ChatService(BaseService):
         Args:
             assistant_message_id (str): The ID of the assistant message to assess
             status (MessageAssessmentStatus): The status of the assessment (e.g. "DONE")
-            explanation (str): Explanation of the assessment
-            label (MessageAssessmentLabel): The assessment label (e.g. "NEGATIVE")
             type (MessageAssessmentType): The type of assessment (e.g. "HALLUCINATION")
+            explanation (str | None): Explanation of the assessment
+            label (MessageAssessmentLabel | None): The assessment label (e.g. "NEGATIVE")
             is_visible (bool): Whether the assessment is visible to users. Defaults to True.
 
         Returns:
@@ -667,11 +667,11 @@ class ChatService(BaseService):
             assessment = await unique_sdk.MessageAssessment.create_async(
                 user_id=self.event.user_id,
                 company_id=self.event.company_id,
-                assistant_message_id=assistant_message_id,
+                messageId=assistant_message_id,
                 status=status.name,
                 explanation=explanation,
-                label=label.name,
-                type=type.name,
+                label=label.name if label else None,
+                type=type.name if type else None,
                 isVisible=is_visible,
             )
             return MessageAssessment(**assessment)
@@ -683,9 +683,9 @@ class ChatService(BaseService):
         self,
         assistant_message_id: str,
         status: MessageAssessmentStatus,
-        explanation: str,
-        label: MessageAssessmentLabel,
         type: MessageAssessmentType,
+        explanation: str | None = None,
+        label: MessageAssessmentLabel | None = None,
     ) -> MessageAssessment:
         """
         Modifies a message assessment for an assistant message synchronously.
@@ -693,8 +693,8 @@ class ChatService(BaseService):
         Args:
             assistant_message_id (str): The ID of the assistant message to assess
             status (MessageAssessmentStatus): The status of the assessment (e.g. "DONE")
-            explanation (str): Explanation of the assessment
-            label (MessageAssessmentLabel): The assessment label (e.g. "NEGATIVE")
+            explanation (str | None): Explanation of the assessment
+            label (MessageAssessmentLabel | None): The assessment label (e.g. "NEGATIVE")
             type (MessageAssessmentType): The type of assessment (e.g. "HALLUCINATION")
 
         Returns:
@@ -707,10 +707,10 @@ class ChatService(BaseService):
             assessment = unique_sdk.MessageAssessment.modify(
                 user_id=self.event.user_id,
                 company_id=self.event.company_id,
-                assistant_message_id=assistant_message_id,
+                messageId=assistant_message_id,
                 status=status.name,
                 explanation=explanation,
-                label=label.name,
+                label=label.name if label else None,
                 type=type.name,
             )
             return MessageAssessment(**assessment)
@@ -721,10 +721,10 @@ class ChatService(BaseService):
     async def modify_message_assessment_async(
         self,
         assistant_message_id: str,
-        status: MessageAssessmentStatus,
-        explanation: str,
-        label: MessageAssessmentLabel,
         type: MessageAssessmentType,
+        status: MessageAssessmentStatus | None = None,
+        explanation: str | None = None,
+        label: MessageAssessmentLabel | None = None,
     ) -> MessageAssessment:
         """
         Modifies a message assessment for an assistant message asynchronously.
@@ -732,8 +732,8 @@ class ChatService(BaseService):
         Args:
             assistant_message_id (str): The ID of the assistant message to assess
             status (MessageAssessmentStatus): The status of the assessment (e.g. "DONE")
-            explanation (str): Explanation of the assessment
-            label (MessageAssessmentLabel): The assessment label (e.g. "NEGATIVE")
+            explanation (str | None): Explanation of the assessment
+            label (MessageAssessmentLabel | None): The assessment label (e.g. "NEGATIVE")
             type (MessageAssessmentType): The type of assessment (e.g. "HALLUCINATION")
 
         Returns:
@@ -746,10 +746,10 @@ class ChatService(BaseService):
             assessment = await unique_sdk.MessageAssessment.modify_async(
                 user_id=self.event.user_id,
                 company_id=self.event.company_id,
-                assistant_message_id=assistant_message_id,
-                status=status.name,
+                messageId=assistant_message_id,
+                status=status.name if status else None,
                 explanation=explanation,
-                label=label.name,
+                label=label.name if label else None,
                 type=type.name,
             )
             return MessageAssessment(**assessment)

{unique_toolkit-0.5.52.dist-info → unique_toolkit-0.5.53.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.52
+Version: 0.5.53
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -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.53] - 2025-02-01
+- Correct `MessageAssessment` schemas
+
 ## [0.5.52] - 2025-02-01
 - Add `MessageAssessment` schemas and functions to `ChatService` to handle message assessments.
 - Fix `LanguageModelService.complete_async_util` to use the correct async method.

{unique_toolkit-0.5.52.dist-info → unique_toolkit-0.5.53.dist-info}/RECORD

@@ -11,8 +11,8 @@ unique_toolkit/app/performance/async_wrapper.py,sha256=yVVcRDkcdyfjsxro-N29SBvi-
 unique_toolkit/app/schemas.py,sha256=6RY7Ex-B3pOnFILlitHEi9sqJvupbpdxlN9xt33qRsM,1571
 unique_toolkit/app/verification.py,sha256=mffa6wm0i4hJbwzofePrkaia46xumMzECwQ0T3eKAx0,1929
 unique_toolkit/chat/__init__.py,sha256=4xS-Mcv7Oqdhprw1JEqD3nwGFflla4R2o7RNJyA5Wek,537
-unique_toolkit/chat/schemas.py,sha256=pWEOkLgZn4LUriFAQeKFHDAqUe1vjl20LPsoM1tdIic,2227
-unique_toolkit/chat/service.py,sha256=AQ61nrUBLTomBCzSmyzZ9257pRH666zKNDrtQHoPkrE,27851
+unique_toolkit/chat/schemas.py,sha256=LOGO-yyQZ6ifb4lDEOSIQl_6-di8KXEB7AKtkjA3xSU,2253
+unique_toolkit/chat/service.py,sha256=9sQbfwgb2GoUuDL3xoG-7P9oOEj-TAssZleXwSZ3HiY,28121
 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
@@ -44,7 +44,7 @@ unique_toolkit/language_model/service.py,sha256=m4B4YD4wxfU8HNo_stqbfnlKXziYBAwq
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJs8FEZXcgQTNenw,1406
 unique_toolkit/short_term_memory/service.py,sha256=Jd9P72-VvJy7hnqNrjmrmB5BHmsKuOpTiT0Jr-dBbsQ,1682
-unique_toolkit-0.5.52.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.5.52.dist-info/METADATA,sha256=M7kVSW4aRX-04EGrhWt136ICM75ok4vFPkjBJyvkbZc,16293
-unique_toolkit-0.5.52.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.52.dist-info/RECORD,,
+unique_toolkit-0.5.53.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.5.53.dist-info/METADATA,sha256=BUtdCWRnnsrlHOl6XJtWxvXrZbhNPPR00g4R09QGHpI,16357
+unique_toolkit-0.5.53.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.53.dist-info/RECORD,,