capo-codeguruprofiler 0.4.0__tar.gz → 0.5.0__tar.gz

{capo_codeguruprofiler-0.4.0/src/capo_codeguruprofiler.egg-info → capo_codeguruprofiler-0.5.0}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: capo-codeguruprofiler
-Version: 0.4.0
-Summary: Python SDK for Codeguruprofiler.
+Version: 0.5.0
+Summary: Python SDK for AWS Codeguruprofiler.
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License

{capo_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "capo-codeguruprofiler"
-version = "0.4.0"
-description = "Python SDK for Codeguruprofiler."
+version = "0.5.0"
+description = "Python SDK for AWS Codeguruprofiler."
 readme = "README.md"
 requires-python = ">=3.10"
 classifiers = [

{capo_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/add_notification_channels.py

@@ -30,6 +30,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -160,7 +161,7 @@ def add_notification_channels(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -178,7 +179,7 @@ async def async_add_notification_channels(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/batch_get_frame_metric_data.py

@@ -31,6 +31,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -177,7 +178,7 @@ def batch_get_frame_metric_data(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -195,7 +196,7 @@ async def async_batch_get_frame_metric_data(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/configure_agent.py

@@ -28,6 +28,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -152,7 +153,7 @@ def configure_agent(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -170,7 +171,7 @@ async def async_configure_agent(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/create_profiling_group.py

@@ -29,6 +29,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -155,7 +156,7 @@ def create_profiling_group(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -173,7 +174,7 @@ async def async_create_profiling_group(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/delete_profiling_group.py

@@ -27,6 +27,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -140,7 +141,7 @@ def delete_profiling_group(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -158,7 +159,7 @@ async def async_delete_profiling_group(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/describe_profiling_group.py

@@ -27,6 +27,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -144,7 +145,7 @@ def describe_profiling_group(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -162,7 +163,7 @@ async def async_describe_profiling_group(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/get_findings_report_account_summary.py

@@ -25,6 +25,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -139,7 +140,7 @@ def get_findings_report_account_summary(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -157,7 +158,7 @@ async def async_get_findings_report_account_summary(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/get_notification_configuration.py

@@ -27,6 +27,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -143,7 +144,7 @@ def get_notification_configuration(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -161,7 +162,7 @@ async def async_get_notification_configuration(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/get_policy.py

@@ -25,6 +25,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -137,7 +138,7 @@ def get_policy(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -154,7 +155,7 @@ async def async_get_policy(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/get_profile.py

@@ -28,6 +28,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -172,7 +173,7 @@ def get_profile(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -189,7 +190,7 @@ async def async_get_profile(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/get_recommendations.py

@@ -29,6 +29,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -171,7 +172,7 @@ def get_recommendations(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -189,7 +190,7 @@ async def async_get_recommendations(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/list_findings_reports.py

@@ -28,6 +28,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -172,7 +173,7 @@ def list_findings_reports(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -190,7 +191,7 @@ async def async_list_findings_reports(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/list_profile_times.py

@@ -28,6 +28,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -171,7 +172,7 @@ def list_profile_times(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -189,7 +190,7 @@ async def async_list_profile_times(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/list_profiling_groups.py

@@ -25,6 +25,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -135,7 +136,7 @@ def list_profiling_groups(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -153,7 +154,7 @@ async def async_list_profiling_groups(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/list_tags_for_resource.py

@@ -26,6 +26,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -133,7 +134,7 @@ def list_tags_for_resource(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -151,7 +152,7 @@ async def async_list_tags_for_resource(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/post_agent_profile.py

@@ -27,6 +27,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -143,7 +144,7 @@ def post_agent_profile(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -161,7 +162,7 @@ async def async_post_agent_profile(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/put_permission.py

@@ -28,6 +28,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -157,7 +158,7 @@ def put_permission(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -175,7 +176,7 @@ async def async_put_permission(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/remove_notification_channel.py

@@ -27,6 +27,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -144,7 +145,7 @@ def remove_notification_channel(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -162,7 +163,7 @@ async def async_remove_notification_channel(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/remove_permission.py

@@ -27,6 +27,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -150,7 +151,7 @@ def remove_permission(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -168,7 +169,7 @@ async def async_remove_permission(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/submit_feedback.py

@@ -26,6 +26,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -145,7 +146,7 @@ def submit_feedback(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -163,7 +164,7 @@ async def async_submit_feedback(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/tag_resource.py

@@ -26,6 +26,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -133,7 +134,7 @@ def tag_resource(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -151,7 +152,7 @@ async def async_tag_resource(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/untag_resource.py

@@ -26,6 +26,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -131,7 +132,7 @@ def untag_resource(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -149,7 +150,7 @@ async def async_untag_resource(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_operations/code_guru_profiler/update_profiling_group.py

@@ -29,6 +29,7 @@ from capo_codeguruprofiler._rule_engine._endpoint_rule_set import (
 from capo_codeguruprofiler._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_codeguruprofiler.errors import UnknownServiceError
 
@@ -156,7 +157,7 @@ def update_profiling_group(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -174,7 +175,7 @@ async def async_update_profiling_group(
     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_codeguruprofiler-0.4.0 → capo_codeguruprofiler-0.5.0}/src/capo_codeguruprofiler/_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