unique_sdk 0.9.21__py3-none-any.whl → 0.9.22__py3-none-any.whl

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/_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/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/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/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",

{unique_sdk-0.9.21.dist-info → unique_sdk-0.9.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.9.21
+Version: 0.9.22
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -940,6 +940,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.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`
 

{unique_sdk-0.9.21.dist-info → unique_sdk-0.9.22.dist-info}/RECORD

@@ -1,5 +1,5 @@
 unique_sdk/__init__.py,sha256=_bFNx6aDyXx_P8VO3RQISQH0PNIqpY3Zev3tsVNUeps,3177
-unique_sdk/_api_requestor.py,sha256=UxFnEyLPX8orHzFcPJC4tDSVqF4sCDJpWlN6IunOYyQ,14645
+unique_sdk/_api_requestor.py,sha256=i4gCpzx8zP95sv-AhJfpQxKvWR0U-I6lclHyV55RPtg,14397
 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
@@ -10,26 +10,26 @@ 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=PlWU8ao74iK1T_On6hravH7O6MXUGSaCHrhkIAmwoag,8929
+unique_sdk/_util.py,sha256=kvJzdmwgmWpQYF2OXyzVv1449mF6ADeyuU45Ma2sIg8,9501
 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
-unique_sdk/api_resources/_chat_completion.py,sha256=Hif_kbY0OmW54nG7Ad-up_kvbhejssP5Yd7nLeDHacU,2144
+unique_sdk/api_resources/_chat_completion.py,sha256=hAPPHxoljcoHeTboxoJkcXgpqA2hNBu6a6vamdxag58,2000
 unique_sdk/api_resources/_content.py,sha256=MrAXhgZS3v2S9OwKh3ROfv5wU8Y947A6gD3wgaYe_aM,4989
 unique_sdk/api_resources/_embedding.py,sha256=C6qak7cCUBMBINfPhgH8taCJZ9n6w1MUElqDJJ8dG10,1281
 unique_sdk/api_resources/_event.py,sha256=bpWF9vstdoAWbUzr-iiGP713ceP0zPk77GJXiImf9zg,374
-unique_sdk/api_resources/_integrated.py,sha256=R5a5oSVQXkh5xyeFwor-7sfePsBdtUyh8SbTMS6AMz4,3036
+unique_sdk/api_resources/_integrated.py,sha256=l1vS8kJiSLie61mqDO3KI2MNmYwFydmCIoJpP_tPhSI,2956
 unique_sdk/api_resources/_message.py,sha256=gEDIzg3METZU2k7m69meAuf0IWmZxnYOjbBKPRMwPYE,7688
 unique_sdk/api_resources/_message_assessment.py,sha256=SSfx6eW7zb_GKe8cFJzCqW-t-_eWEXxKP5cnIb0DhIc,2276
 unique_sdk/api_resources/_search.py,sha256=m9gL-Cv347r6cM9MwGWgddORGHzPF3jK5dzL3plaeNI,1883
-unique_sdk/api_resources/_search_string.py,sha256=I43XqKfYwWSwhZ4r0UHx5fs1MUGbLszWHrxqBvo6Tx8,1785
+unique_sdk/api_resources/_search_string.py,sha256=4Idw6exgZdA8qksz9WkiA68k1hTU-7yFkgT_OLU_GkE,1662
 unique_sdk/api_resources/_short_term_memory.py,sha256=vPRN-Y0WPx74E6y-A3LocGc0TxJdzT-xGL66WzZwKRg,2820
 unique_sdk/utils/chat_history.py,sha256=5UqL9hF1O9pV7skbNOlEibF5rHdYsmG3m5-YEPUowOs,3037
 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.21.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
-unique_sdk-0.9.21.dist-info/METADATA,sha256=O5pRfpHRt3ahBTXPQ04u_9hFDwOApa6SqICRknFQPgs,30001
-unique_sdk-0.9.21.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_sdk-0.9.21.dist-info/RECORD,,
+unique_sdk-0.9.22.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
+unique_sdk-0.9.22.dist-info/METADATA,sha256=WItAhCr9G0OU078z6ihry00uK9brJDBeFtGz_QJUgfE,30171
+unique_sdk-0.9.22.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_sdk-0.9.22.dist-info/RECORD,,