unique_sdk 0.9.12__py3-none-any.whl → 0.9.14__py3-none-any.whl

unique_sdk/_api_resource.py

@@ -22,10 +22,12 @@ retry_dict = {
     "error_messages": [
         "problem proxying the request",
         "Upstream service reached a hard timeout",
+        "Invalid response body from API",
     ],
     "max_retries": 3,
     "backoff_factor": 2,
     "initial_delay": 1,
+    "should_retry_5xx": True,
 }
 
 

unique_sdk/_util.py

@@ -200,6 +200,7 @@ def retry_on_error(
     initial_delay: int = 1,
     backoff_factor: int = 2,
     error_class=APIError,
+    should_retry_5xx=False,
 ):
     def decorator(func: Callable) -> Callable:
         @wraps(func)
@@ -209,10 +210,15 @@ def retry_on_error(
                 try:
                     return await func(*args, **kwargs)
                 except Exception as e:
-                    if not any(
+                    should_retry = any(
                         err_msg.lower() in str(e).lower() for err_msg in error_messages
-                    ):
-                        raise e  # Raise the error if none of the messages match
+                    )
+                    # 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)
+
+                    if not should_retry:
+                        raise e  # Raise the error if no retry condition is met
 
                     attempts += 1
                     if attempts >= max_retries:
@@ -229,10 +235,15 @@ def retry_on_error(
                 try:
                     return func(*args, **kwargs)
                 except Exception as e:
-                    if not any(
+                    should_retry = any(
                         err_msg.lower() in str(e).lower() for err_msg in error_messages
-                    ):
-                        raise e  # Raise the error if none of the messages match
+                    )
+                    # 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)
+
+                    if not should_retry:
+                        raise e  # Raise the error if no retry condition is met
 
                     attempts += 1
                     if attempts >= max_retries:

unique_sdk/_version.py

@@ -1 +1 @@
-VERSION = "0.9.8"
+VERSION = "0.9.14"

unique_sdk/api_resources/_search.py

@@ -18,6 +18,7 @@ class Search(APIResource["Search"]):
         limit: NotRequired[Optional[int]]
         page: NotRequired[Optional[int]]
         metaDataFilter: NotRequired[Optional[dict[str, Any]]]
+        contentIds: NotRequired[Optional[list[str]]]
 
     id: str
     chunkId: str

{unique_sdk-0.9.12.dist-info → unique_sdk-0.9.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.9.12
+Version: 0.9.14
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -899,6 +899,13 @@ 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.14] - 2024-12-06
+- Add `contentIds` to `Search.create` and `Search.create_async`
+
+## [0.9.13] - 2024-10-23
+- Add retry for `5xx` errors, add additional error message.
+
 ## [0.9.12] - 2024-11-21
 - Include original error message in returned exceptions
 

{unique_sdk-0.9.12.dist-info → unique_sdk-0.9.14.dist-info}/RECORD

@@ -1,6 +1,6 @@
 unique_sdk/__init__.py,sha256=Cv42hZ3dlsNACMIza7LDARXiY-w7a_ijlSuY_jADJD8,3072
 unique_sdk/_api_requestor.py,sha256=O6iF_d_jc1ceTYVa3kOQHXjYKdmBp0kGW_pfMw1ClwE,14543
-unique_sdk/_api_resource.py,sha256=zKu-Sk1sJ6b9mr8ljrt9UBmi6y6lO7A29DNcBxdRYD4,6113
+unique_sdk/_api_resource.py,sha256=tE3J9sC5qwOdIMGLYEyvHjdQRYX-1unrLatO7CzgSfk,6185
 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=ddCV4IDbbf2fh1Z09X7Y2REQmXMETLrM7tKLOc33w0o,8240
-unique_sdk/_version.py,sha256=bLj15cDf-jfrKLtrOqhC3kl-ZIlt9q8ieogbrosWe_g,18
+unique_sdk/_util.py,sha256=J8L4ynMM6_zzN5uaIxriShr9lYpYa_5RMQAWKuqIcC0,8806
+unique_sdk/_version.py,sha256=Ny-C0e1cEmVxwAXD01RWMz0CqwI3WSxEXfuLbrCWv1g,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,14 +21,14 @@ 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/_search.py,sha256=UPFD-rQfvcqFAj7P5kWlp56uroK1y6pglRR3iyw02_c,1806
+unique_sdk/api_resources/_search.py,sha256=TbzNZO7OOmQqmfHkbzhDGtRIrtJw_Eohs_vo02vamzE,1859
 unique_sdk/api_resources/_search_string.py,sha256=I43XqKfYwWSwhZ4r0UHx5fs1MUGbLszWHrxqBvo6Tx8,1785
 unique_sdk/api_resources/_short_term_memory.py,sha256=FWJ-PFR8x3o15lw1L0iKICFNdSTb1C9gs8y-CBkfdag,2815
 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.12.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
-unique_sdk-0.9.12.dist-info/METADATA,sha256=qBpFA0leuy7tYkdDC1UfL0VTt2heE9SnqNVvO4YX3Wo,28057
-unique_sdk-0.9.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_sdk-0.9.12.dist-info/RECORD,,
+unique_sdk-0.9.14.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
+unique_sdk-0.9.14.dist-info/METADATA,sha256=aQkf8PdTSUR_kaitfjqDpJ_NjdN9Oak0fArbf4KbU2c,28234
+unique_sdk-0.9.14.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_sdk-0.9.14.dist-info/RECORD,,