unique_sdk 0.9.20__tar.gz → 0.9.22__tar.gz

{unique_sdk-0.9.20 → unique_sdk-0.9.22}/CHANGELOG.md

@@ -5,6 +5,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.22] - 2025-02-25
+- update the retry_on_error to only `APIError` and `APIConnectionError` update the `resp["error"]` to be `resp.get("error")` to avoid key error
+
+## [0.9.21] - 2025-02-21
+- Add title parameter and change labels in `MessageAssessment`
+
 ## [0.9.20] - 2025-02-01
 - Add url parameter to `MessageAssessment.create_async` and `MessageAssessment.modify_async`
 

{unique_sdk-0.9.20 → unique_sdk-0.9.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.9.20
+Version: 0.9.22
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -631,8 +631,9 @@ assessment = unique_sdk.MessageAssessment.create(
     assistant_message_id="msg_...",
     status="DONE",
     explanation="This response contains incorrect information about...",
-    label="NEGATIVE",
+    label="RED",
     type="HALLUCINATION",
+    title="Hallucination detected",
     isVisible=True
 )
 ```
@@ -648,7 +649,8 @@ assessment = unique_sdk.MessageAssessment.modify(
     assistant_message_id="msg_...",
     status="DONE",
     explanation="Updated explanation...",
-    label="NEGATIVE",
+    label="RED",
+    title="update title"
     type="HALLUCINATION"
 )
 ```
@@ -938,6 +940,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.22] - 2025-02-25
+- update the retry_on_error to only `APIError` and `APIConnectionError` update the `resp["error"]` to be `resp.get("error")` to avoid key error
+
+## [0.9.21] - 2025-02-21
+- Add title parameter and change labels in `MessageAssessment`
+
 ## [0.9.20] - 2025-02-01
 - Add url parameter to `MessageAssessment.create_async` and `MessageAssessment.modify_async`
 

{unique_sdk-0.9.20 → unique_sdk-0.9.22}/README.md

@@ -616,8 +616,9 @@ assessment = unique_sdk.MessageAssessment.create(
     assistant_message_id="msg_...",
     status="DONE",
     explanation="This response contains incorrect information about...",
-    label="NEGATIVE",
+    label="RED",
     type="HALLUCINATION",
+    title="Hallucination detected",
     isVisible=True
 )
 ```
@@ -633,7 +634,8 @@ assessment = unique_sdk.MessageAssessment.modify(
     assistant_message_id="msg_...",
     status="DONE",
     explanation="Updated explanation...",
-    label="NEGATIVE",
+    label="RED",
+    title="update title"
     type="HALLUCINATION"
 )
 ```

{unique_sdk-0.9.20 → unique_sdk-0.9.22}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_sdk"
-version = "0.9.20"
+version = "0.9.22"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

{unique_sdk-0.9.20 → unique_sdk-0.9.22}/unique_sdk/_api_requestor.py

@@ -353,16 +353,7 @@ class APIRequestor(object):
         return resp
 
     def handle_error_response(self, rbody, rcode, resp, rheaders) -> NoReturn:
-        try:
-            error_data = resp["error"]
-        except (KeyError, TypeError):
-            raise _error.APIError(
-                "Invalid response object from API: %r (HTTP response code "
-                "was %d)" % (rbody, rcode),
-                rbody,
-                rcode,
-                resp,
-            )
+        error_data = resp.get("error")
 
         err = None
 
@@ -375,6 +366,7 @@ class APIRequestor(object):
         raise err
 
     def specific_api_error(self, rbody, rcode, resp, rheaders, error_data):
+        error_data = error_data or resp
         cause = error_data.get("cause", {})
         if isinstance(cause, str):
             cause = {"error": {"message": cause}, "status": rcode}

{unique_sdk-0.9.20 → unique_sdk-0.9.22}/unique_sdk/_util.py

@@ -12,7 +12,7 @@ from typing import Any, Callable, Dict, List, Optional, TypeVar, Union, cast, ov
 from typing_extensions import TYPE_CHECKING, Type
 
 import unique_sdk  # noqa: F401
-from unique_sdk._error import APIError
+from unique_sdk._error import APIConnectionError, APIError, UniqueError
 
 if TYPE_CHECKING:
     from unique_sdk._unique_object import UniqueObject
@@ -209,14 +209,20 @@ def retry_on_error(
             while attempts < max_retries:
                 try:
                     return await func(*args, **kwargs)
-                except Exception as e:
+                except (APIError, APIConnectionError) as e:
+                    e = cast(UniqueError, e)
                     logger.error(f"Retrying because of {e}")
                     should_retry = any(
                         err_msg.lower() in str(e).lower() for err_msg in error_messages
                     )
                     # Add 5xx check if `should_retry_5xx` is True
-                    if should_retry_5xx and hasattr(e, "status_code"):
-                        should_retry = should_retry or (500 <= e.status_code < 600)
+                    has_valid_status = (
+                        hasattr(e, "http_status") and e.http_status is not None
+                    )
+                    if should_retry_5xx and has_valid_status:
+                        should_retry = should_retry or (
+                            500 <= int(cast(int, e.http_status)) < 600
+                        )
 
                     if not should_retry:
                         raise e  # Raise the error if no retry condition is met
@@ -235,15 +241,20 @@ def retry_on_error(
             while attempts < max_retries:
                 try:
                     return func(*args, **kwargs)
-                except Exception as e:
+                except (APIError, APIConnectionError) as e:
+                    e = cast(UniqueError, e)
                     logger.error(f"Retrying because of {e}")
-
                     should_retry = any(
                         err_msg.lower() in str(e).lower() for err_msg in error_messages
                     )
                     # Add 5xx check if `should_retry_5xx` is True
-                    if should_retry_5xx and hasattr(e, "status_code"):
-                        should_retry = should_retry or (500 <= e.status_code < 600)
+                    has_valid_status = (
+                        hasattr(e, "http_status") and e.http_status is not None
+                    )
+                    if should_retry_5xx and has_valid_status:
+                        should_retry = should_retry or (
+                            500 <= int(cast(int, e.http_status)) < 600
+                        )
 
                     if not should_retry:
                         raise e  # Raise the error if no retry condition is met

{unique_sdk-0.9.20 → unique_sdk-0.9.22}/unique_sdk/api_resources/_chat_completion.py

@@ -37,8 +37,6 @@ class ChatCompletion(APIResource["ChatCompletion"]):
     class CreateParams(RequestOptions):
         model: NotRequired[
             Literal[
-                "AZURE_GPT_35_TURBO",
-                "AZURE_GPT_35_TURBO_16K",
                 "AZURE_GPT_4_0613",
                 "AZURE_GPT_4_32K_0613",
             ]
@@ -47,8 +45,6 @@ class ChatCompletion(APIResource["ChatCompletion"]):
         messages: List[ChatCompletionRequestMessage]
 
     model: Literal[
-        "AZURE_GPT_35_TURBO",
-        "AZURE_GPT_35_TURBO_16K",
         "AZURE_GPT_4_0613",
         "AZURE_GPT_4_32K_0613",
     ]

{unique_sdk-0.9.20 → unique_sdk-0.9.22}/unique_sdk/api_resources/_integrated.py

@@ -38,8 +38,6 @@ class Integrated(APIResource["Integrated"]):
     class CreateStream(RequestOptions):
         model: NotRequired[
             Literal[
-                "AZURE_GPT_35_TURBO",
-                "AZURE_GPT_35_TURBO_16K",
                 "AZURE_GPT_4_0613",
                 "AZURE_GPT_4_32K_0613",
             ]

{unique_sdk-0.9.20 → unique_sdk-0.9.22}/unique_sdk/api_resources/_message_assessment.py

@@ -17,15 +17,17 @@ class MessageAssessment(APIResource["MessageAssessment"]):
         status: Literal["PENDING", "DONE", "ERROR"]
         type: Literal["HALLUCINATION", "COMPLIANCE"]
         isVisible: bool
+        title: str | None
         explanation: str | None
-        label: Literal["POSITIVE", "NEGATIVE", "VERIFIED", "UNVERIFIED"] | None
+        label: Literal["RED", "YELLOW", "GREEN"] | None
 
     class ModifyParams(RequestOptions):
         messageId: str
         type: Literal["HALLUCINATION", "COMPLIANCE"]
         status: Literal["PENDING", "DONE", "ERROR"] | None
+        title: str | None
         explanation: str | None
-        label: Literal["POSITIVE", "NEGATIVE", "VERIFIED", "UNVERIFIED"] | None
+        label: Literal["RED", "YELLOW", "GREEN"] | None
 
     @classmethod
     def create(

{unique_sdk-0.9.20 → unique_sdk-0.9.22}/unique_sdk/api_resources/_search_string.py

@@ -20,9 +20,6 @@ class SearchString(APIResource["SearchString"]):
         messages: NotRequired[List[HistoryMessage]]
         languageModel: NotRequired[
             Literal[
-                "AZURE_GPT_35_TURBO",
-                "AZURE_GPT_35_TURBO_0613",
-                "AZURE_GPT_35_TURBO_16K",
                 "AZURE_GPT_4_0613",
                 "AZURE_GPT_4_32K_0613",
                 "AZURE_GPT_4_TURBO_1106",