unique_sdk 0.9.18__py3-none-any.whl → 0.9.20__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_requestor.py

@@ -376,6 +376,8 @@ class APIRequestor(object):
 
     def specific_api_error(self, rbody, rcode, resp, rheaders, error_data):
         cause = error_data.get("cause", {})
+        if isinstance(cause, str):
+            cause = {"error": {"message": cause}, "status": rcode}
         status = cause.get("status", rcode)
 
         _util.log_info(

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):
+        messageId: str
+        status: Literal["PENDING", "DONE", "ERROR"]
+        type: Literal["HALLUCINATION", "COMPLIANCE"]
+        isVisible: bool
+        explanation: str | None
+        label: Literal["POSITIVE", "NEGATIVE", "VERIFIED", "UNVERIFIED"] | None
+
+    class ModifyParams(RequestOptions):
+        messageId: str
+        type: Literal["HALLUCINATION", "COMPLIANCE"]
+        status: Literal["PENDING", "DONE", "ERROR"] | None
+        explanation: str | None
+        label: Literal["POSITIVE", "NEGATIVE", "VERIFIED", "UNVERIFIED"] | None
+
+    @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 await 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":
+        url = f"{cls.RESOURCE_URL}/{params['messageId']}"
+        return cls._static_request("patch", url, user_id, company_id, params=params)
+
+    @classmethod
+    async def modify_async(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["MessageAssessment.ModifyParams"],
+    ) -> "MessageAssessment":
+        url = f"{cls.RESOURCE_URL}/{params['messageId']}"
+        return await cls._static_request_async(
+            "patch", url, user_id, company_id, params=params
+        )

{unique_sdk-0.9.18.dist-info → unique_sdk-0.9.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.9.18
+Version: 0.9.20
 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.20] - 2025-02-01
+- Add url parameter to `MessageAssessment.create_async` and `MessageAssessment.modify_async`
+
+## [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.
 

{unique_sdk-0.9.18.dist-info → unique_sdk-0.9.20.dist-info}/RECORD

@@ -1,5 +1,5 @@
-unique_sdk/__init__.py,sha256=Cv42hZ3dlsNACMIza7LDARXiY-w7a_ijlSuY_jADJD8,3072
-unique_sdk/_api_requestor.py,sha256=O6iF_d_jc1ceTYVa3kOQHXjYKdmBp0kGW_pfMw1ClwE,14543
+unique_sdk/__init__.py,sha256=_bFNx6aDyXx_P8VO3RQISQH0PNIqpY3Zev3tsVNUeps,3177
+unique_sdk/_api_requestor.py,sha256=UxFnEyLPX8orHzFcPJC4tDSVqF4sCDJpWlN6IunOYyQ,14645
 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
@@ -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=EEf-cThCEd2N6hceENG3HnJQPoe_2Oiq5MBy5hfAxmM,2272
 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.18.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
-unique_sdk-0.9.18.dist-info/METADATA,sha256=8dO6nMtffbsxzKrcO63phs6g5XLl-W5YDjdy0-xz43M,28727
-unique_sdk-0.9.18.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_sdk-0.9.18.dist-info/RECORD,,
+unique_sdk-0.9.20.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
+unique_sdk-0.9.20.dist-info/METADATA,sha256=8uMy1QbABEhlg1xysGcmfp0QOApqaqkSZyL0g1bdu9M,29861
+unique_sdk-0.9.20.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_sdk-0.9.20.dist-info/RECORD,,