unique_sdk 0.9.17__py3-none-any.whl → 0.9.19__py3-none-any.whl

unique_sdk/__init__.py

@@ -82,6 +82,9 @@ from unique_sdk.api_resources._short_term_memory import (
 )
 from unique_sdk.api_resources._embedding import Embeddings as Embeddings
 from unique_sdk.api_resources._acronyms import Acronyms as Acronyms
+from unique_sdk.api_resources._message_assessment import (
+    MessageAssessment as MessageAssessment,
+)
 
 # Unique QL
 

unique_sdk/_api_resource.py

@@ -22,7 +22,6 @@ retry_dict = {
     "error_messages": [
         "problem proxying the request",
         "Upstream service reached a hard timeout",
-        "Invalid response body from API",
         "You can retry your request",
         "Internal server error",
         "Connection aborted",

unique_sdk/_util.py

@@ -210,6 +210,7 @@ def retry_on_error(
                 try:
                     return await func(*args, **kwargs)
                 except Exception as e:
+                    logger.error(f"Retrying because of {e}")
                     should_retry = any(
                         err_msg.lower() in str(e).lower() for err_msg in error_messages
                     )
@@ -235,6 +236,8 @@ def retry_on_error(
                 try:
                     return func(*args, **kwargs)
                 except Exception as e:
+                    logger.error(f"Retrying because of {e}")
+
                     should_retry = any(
                         err_msg.lower() in str(e).lower() for err_msg in error_messages
                     )

unique_sdk/_version.py

@@ -1 +1 @@
-VERSION = "0.9.15"
+VERSION = "0.9.18"

unique_sdk/api_resources/_message_assessment.py

@@ -0,0 +1,72 @@
+from typing import (
+    ClassVar,
+    Literal,
+    Unpack,
+)
+
+from unique_sdk._api_resource import APIResource
+from unique_sdk._request_options import RequestOptions
+
+
+class MessageAssessment(APIResource["MessageAssessment"]):
+    OBJECT_NAME: ClassVar[Literal["message_assessment"]] = "message_assessment"
+    RESOURCE_URL = "/message-assessment"
+
+    class CreateParams(RequestOptions):
+        assistant_message_id: str
+        status: Literal["PENDING", "DONE", "ERROR"]
+        explanation: str
+        label: Literal["POSITIVE", "NEGATIVE", "VERIFIED", "UNVERIFIED"]
+        type: Literal["HALLUCINATION", "COMPLIANCE"]
+        isVisible: bool
+
+    class ModifyParams(RequestOptions):
+        assistant_message_id: str
+        status: Literal["PENDING", "DONE", "ERROR"]
+        explanation: str
+        label: Literal["POSITIVE", "NEGATIVE", "VERIFIED", "UNVERIFIED"]
+        type: Literal["HALLUCINATION", "COMPLIANCE"]
+
+    @classmethod
+    def create(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["MessageAssessment.CreateParams"],
+    ) -> "MessageAssessment":
+        return cls._static_request(
+            "post", cls.RESOURCE_URL, user_id, company_id, params=params
+        )
+
+    @classmethod
+    async def create_async(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["MessageAssessment.CreateParams"],
+    ) -> "MessageAssessment":
+        return cls._static_request_async(
+            "post", cls.RESOURCE_URL, user_id, company_id, params=params
+        )
+
+    @classmethod
+    def modify(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["MessageAssessment.ModifyParams"],
+    ) -> "MessageAssessment":
+        return cls._static_request(
+            "patch", cls.RESOURCE_URL, user_id, company_id, params=params
+        )
+
+    @classmethod
+    async def modify_async(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["MessageAssessment.ModifyParams"],
+    ) -> "MessageAssessment":
+        return cls._static_request_async(
+            "patch", cls.RESOURCE_URL, user_id, company_id, params=params
+        )

{unique_sdk-0.9.17.dist-info → unique_sdk-0.9.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.9.17
+Version: 0.9.19
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -34,6 +34,7 @@ The Unique Python SDK provides access to the public API of Unique FinanceGPT. It
    - [Search](#search)
    - [Search String](#search-string)
    - [Short Term Memory](#short-term-memory)
+   - [Message Assessment](#message-assessment)
 6. [UniqueQL](#uniqueql)
    - [Query Structure](#uniqueql-query-structure)
    - [Metadata Filtering](#metadata-filtering)
@@ -587,6 +588,7 @@ you need to provide an `memoryName` as an identifier.
 
 you can then save it and retreive it live defined below.
 
+
 #### `unique_sdk.ShortTermMemory.create`
 
 ```python
@@ -614,6 +616,43 @@ m = unique_sdk.ShortTermMemory.find_latest(
 print(m)
 ```
 
+### Message Assessment
+
+Used to create and modify message assessments for tracking hallucinations and compliance evaluations of assistant messages.
+
+#### `unique_sdk.MessageAssessment.create`
+
+Create a new message assessment for an assistant message.
+
+```python
+assessment = unique_sdk.MessageAssessment.create(
+    user_id=user_id,
+    company_id=company_id,
+    assistant_message_id="msg_...",
+    status="DONE",
+    explanation="This response contains incorrect information about...",
+    label="NEGATIVE",
+    type="HALLUCINATION",
+    isVisible=True
+)
+```
+
+#### `unique_sdk.MessageAssessment.modify` 
+
+Modify an existing message assessment.
+
+```python
+assessment = unique_sdk.MessageAssessment.modify(
+    user_id=user_id,
+    company_id=company_id,
+    assistant_message_id="msg_...",
+    status="DONE",
+    explanation="Updated explanation...",
+    label="NEGATIVE",
+    type="HALLUCINATION"
+)
+```
+
 ## UniqueQL
 
 [UniqueQL](https://unique-ch.atlassian.net/wiki/x/coAXHQ) is an advanced query language designed to enhance search capabilities within various search modes such as Vector, Full-Text Search (FTS), and Combined. This query language enables users to perform detailed searches by filtering through metadata attributes like filenames, URLs, dates, and more. UniqueQL is versatile and can be translated into different query formats for various database systems, including PostgreSQL and Qdrant.
@@ -899,6 +938,12 @@ 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.9.19] - 2025-01-31
+- Add `MessageAssessment` resource
+
+## [0.9.18] - 2025-01-22
+- Removed `Invalid response body from API` from `retry_dict` as it's our own artificail error.
+
 ## [0.9.17] - 2025-01-03
 - BREAKING CHANGE!! Removed unused `id` from `ShortTermMemory` create and find methods.
 

{unique_sdk-0.9.17.dist-info → unique_sdk-0.9.19.dist-info}/RECORD

@@ -1,6 +1,6 @@
-unique_sdk/__init__.py,sha256=Cv42hZ3dlsNACMIza7LDARXiY-w7a_ijlSuY_jADJD8,3072
+unique_sdk/__init__.py,sha256=_bFNx6aDyXx_P8VO3RQISQH0PNIqpY3Zev3tsVNUeps,3177
 unique_sdk/_api_requestor.py,sha256=O6iF_d_jc1ceTYVa3kOQHXjYKdmBp0kGW_pfMw1ClwE,14543
-unique_sdk/_api_resource.py,sha256=tfnQCmIXoBLiMIjqte_ZHDepQ-8usG3ZJ_PIqmBtneY,6286
+unique_sdk/_api_resource.py,sha256=ytjomI-IVJwsbvdPyuZCfF-bl-Abgf66bu1D34YxCu8,6244
 unique_sdk/_api_version.py,sha256=Ku4JPdeyJtnX5eJJvRCEc1_u44UObdVrvrL1T-WwWCs,46
 unique_sdk/_error.py,sha256=j-deT0PJ-exLCwUkqORRaxaLLrGunDag9bKJSmBBKZI,3343
 unique_sdk/_http_client.py,sha256=CSc2gduhQ42v6G1oh4Jyg8XL7Sa3YQ7WdTpc6kAyTBI,10369
@@ -10,8 +10,8 @@ unique_sdk/_request_options.py,sha256=oHh2AKka6j9pO53Htur3Wj0VJSusEjq8zkXYY179B_
 unique_sdk/_unique_object.py,sha256=PLaIzb6NPYghfMKHqDx_ZyojcDcuQzTrATXrktLqMa0,11442
 unique_sdk/_unique_ql.py,sha256=aUYXhYPA-_kPfImTapilCyDvB_53YpLd51kxILe7fDA,1707
 unique_sdk/_unique_response.py,sha256=q19hIxlsrkLAQylfHA0webp6bsCCHXkDJDwaC69Iev8,576
-unique_sdk/_util.py,sha256=J8L4ynMM6_zzN5uaIxriShr9lYpYa_5RMQAWKuqIcC0,8806
-unique_sdk/_version.py,sha256=vvDEh1A9sxOePPyj9r8ujoRVR1sAvFJpJuKrx-Zzxqs,19
+unique_sdk/_util.py,sha256=PlWU8ao74iK1T_On6hravH7O6MXUGSaCHrhkIAmwoag,8929
+unique_sdk/_version.py,sha256=j4_tPC6t3enRds7LqiRuWSyfrYHfEo6CXIDARAWW98I,19
 unique_sdk/_webhook.py,sha256=GYxbUibQN_W4XlNTHaMIksT9FQJk4LJmlKcxOu3jqiU,2855
 unique_sdk/api_resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_sdk/api_resources/_acronyms.py,sha256=GIU1XH1flGWQYcpsFqTYwg4ioIGxVmb15tux84nmhEg,891
@@ -21,6 +21,7 @@ unique_sdk/api_resources/_embedding.py,sha256=C6qak7cCUBMBINfPhgH8taCJZ9n6w1MUEl
 unique_sdk/api_resources/_event.py,sha256=bpWF9vstdoAWbUzr-iiGP713ceP0zPk77GJXiImf9zg,374
 unique_sdk/api_resources/_integrated.py,sha256=R5a5oSVQXkh5xyeFwor-7sfePsBdtUyh8SbTMS6AMz4,3036
 unique_sdk/api_resources/_message.py,sha256=gEDIzg3METZU2k7m69meAuf0IWmZxnYOjbBKPRMwPYE,7688
+unique_sdk/api_resources/_message_assessment.py,sha256=6BDofQrZXWEO08SBb1fWLI2YoH9PQ-uPLwNMKyjUMO4,2179
 unique_sdk/api_resources/_search.py,sha256=m9gL-Cv347r6cM9MwGWgddORGHzPF3jK5dzL3plaeNI,1883
 unique_sdk/api_resources/_search_string.py,sha256=I43XqKfYwWSwhZ4r0UHx5fs1MUGbLszWHrxqBvo6Tx8,1785
 unique_sdk/api_resources/_short_term_memory.py,sha256=vPRN-Y0WPx74E6y-A3LocGc0TxJdzT-xGL66WzZwKRg,2820
@@ -28,7 +29,7 @@ unique_sdk/utils/chat_history.py,sha256=5UqL9hF1O9pV7skbNOlEibF5rHdYsmG3m5-YEPUo
 unique_sdk/utils/file_io.py,sha256=tcS-5NA97AyiJPhKpWs3i0qKNFsZlttToxrvnWRDJrs,3857
 unique_sdk/utils/sources.py,sha256=wfboE-neMKa0Wuq9QzfAEFMkNLrIrmm0v-QF33sLo6k,4952
 unique_sdk/utils/token.py,sha256=AzKuAA1AwBtnvSFxGcsHLpxXr_wWE5Mj4jYBbOz2ljA,1740
-unique_sdk-0.9.17.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
-unique_sdk-0.9.17.dist-info/METADATA,sha256=hplPuOYGU7ChHDtvKzzu3Fmh968KpFru9cvjyOLro7g,28606
-unique_sdk-0.9.17.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_sdk-0.9.17.dist-info/RECORD,,
+unique_sdk-0.9.19.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
+unique_sdk-0.9.19.dist-info/METADATA,sha256=K5Y2UoJEkFnZ8xownBWuc9zAzkGPuTaaqQSUUKKiDVU,29742
+unique_sdk-0.9.19.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_sdk-0.9.19.dist-info/RECORD,,