capo-sagemaker-runtime 0.2.0__tar.gz → 0.3.0__tar.gz

{capo_sagemaker_runtime-0.2.0/src/capo_sagemaker_runtime.egg-info → capo_sagemaker_runtime-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: capo-sagemaker-runtime
-Version: 0.2.0
+Version: 0.3.0
 Summary: Python SDK for AWS Sagemaker Runtime.
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers

{capo_sagemaker_runtime-0.2.0 → capo_sagemaker_runtime-0.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "capo-sagemaker-runtime"
-version = "0.2.0"
+version = "0.3.0"
 description = "Python SDK for AWS Sagemaker Runtime."
 readme = "README.md"
 requires-python = ">=3.10"

{capo_sagemaker_runtime-0.2.0 → capo_sagemaker_runtime-0.3.0}/src/capo_sagemaker_runtime/_operations/amazon_sage_maker_runtime/invoke_endpoint.py

@@ -29,6 +29,7 @@ from capo_sagemaker_runtime._rule_engine._endpoint_rule_set import (
 from capo_sagemaker_runtime._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_sagemaker_runtime.errors import UnknownServiceError
 
@@ -210,7 +211,7 @@ def invoke_endpoint(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -228,7 +229,7 @@ async def async_invoke_endpoint(
     try:
         if response.status >= 300:
             await response.aread()
-            handle_error(response)
+            raise_error(response, handle_error)
         return await async_handle_response(response), response
     except BaseException:
         await response.aclose()

{capo_sagemaker_runtime-0.2.0 → capo_sagemaker_runtime-0.3.0}/src/capo_sagemaker_runtime/_operations/amazon_sage_maker_runtime/invoke_endpoint_async.py

@@ -26,6 +26,7 @@ from capo_sagemaker_runtime._rule_engine._endpoint_rule_set import (
 from capo_sagemaker_runtime._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_sagemaker_runtime.errors import UnknownServiceError
 
@@ -173,7 +174,7 @@ def invoke_endpoint_async(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -191,7 +192,7 @@ async def async_invoke_endpoint_async(
     try:
         if response.status >= 300:
             await response.aread()
-            handle_error(response)
+            raise_error(response, handle_error)
         return await async_handle_response(response), response
     except BaseException:
         await response.aclose()

{capo_sagemaker_runtime-0.2.0 → capo_sagemaker_runtime-0.3.0}/src/capo_sagemaker_runtime/_operations/amazon_sage_maker_runtime/invoke_endpoint_with_response_stream.py

@@ -35,6 +35,7 @@ from capo_sagemaker_runtime._rule_engine._endpoint_rule_set import (
 from capo_sagemaker_runtime._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_sagemaker_runtime.errors import UnknownServiceError
 
@@ -211,7 +212,7 @@ def invoke_endpoint_with_response_stream(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -229,7 +230,7 @@ async def async_invoke_endpoint_with_response_stream(
     try:
         if response.status >= 300:
             await response.aread()
-            handle_error(response)
+            raise_error(response, handle_error)
         return await async_handle_response(response), response
     except BaseException:
         await response.aclose()

{capo_sagemaker_runtime-0.2.0 → capo_sagemaker_runtime-0.3.0}/src/capo_sagemaker_runtime/_services/_pipeline.py

@@ -6,6 +6,7 @@ from collections.abc import AsyncIterator, Iterator, Sequence
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Awaitable, Callable, Generic, TypeVar, cast
 
+from typing_extensions import Never
 from zapros import (
     AsyncClient,
     Client,
@@ -135,6 +136,22 @@ async def aexecute_pipeline(
     return await make_chain(0)(request)
 
 
+def raise_error(response: Response, handle_error: Callable[[Response], Never]) -> Never:
+    try:
+        handle_error(response)
+    except ServiceError as exc:
+        # The response itself can say a retry is safe, even when the error is
+        # not marked @retryable: a 429, a transient 5xx or a Retry-After header.
+        if response.status == 429:
+            exc.is_throttling_error = True
+        if (
+            response.status in (429, 500, 502, 503, 504)
+            or "retry-after" in response.headers
+        ):
+            exc.is_retryable = True
+        raise
+
+
 def _is_retryable(exc: Exception) -> bool:
     if isinstance(exc, ServiceError):
         return exc.is_retryable

{capo_sagemaker_runtime-0.2.0 → capo_sagemaker_runtime-0.3.0/src/capo_sagemaker_runtime.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: capo-sagemaker-runtime
-Version: 0.2.0
+Version: 0.3.0
 Summary: Python SDK for AWS Sagemaker Runtime.
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers