diff --git a/Users/remiss/PycharmProjects/aws-sdk-python/clients/aws-sdk-bedrock-runtime/src/aws_sdk_bedrock_runtime/client.py b/Users/remiss/PycharmProjects/bedrock-runtime-smithy/build/smithy/client/python-client-codegen/src/aws_sdk_bedrock_runtime/client.py index 763178c..e93f02b 100644 --- a/Users/remiss/PycharmProjects/aws-sdk-python/clients/aws-sdk-bedrock-runtime/src/aws_sdk_bedrock_runtime/client.py +++ b/Users/remiss/PycharmProjects/bedrock-runtime-smithy/build/smithy/client/python-client-codegen/src/aws_sdk_bedrock_runtime/client.py @@ -7,8 +7,7 @@ from smithy_core.aio.client import ClientCall, RequestPipeline from smithy_core.aio.eventstream import DuplexEventStream, OutputEventStream from smithy_core.exceptions import ExpectationNotMetError from smithy_core.interceptors import InterceptorChain -from smithy_core.interfaces.retries import RetryStrategy -from smithy_core.retries import RetryStrategyOptions, RetryStrategyResolver +from smithy_core.retries import RetryStrategyResolver from smithy_core.types import TypedProperties from smithy_http.plugins import user_agent_plugin @@ -110,22 +109,9 @@ class BedrockRuntimeClient: "protocol and transport MUST be set on the config to make calls." ) - # Resolve retry strategy from config - if isinstance(config.retry_strategy, RetryStrategy): - retry_strategy = config.retry_strategy - elif isinstance(config.retry_strategy, RetryStrategyOptions): - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=config.retry_strategy - ) - elif config.retry_strategy is None: - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=RetryStrategyOptions() - ) - else: - raise TypeError( - f"retry_strategy must be RetryStrategy, RetryStrategyOptions, or None, " - f"got {type(config.retry_strategy).__name__}" - ) + retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( + retry_strategy=config.retry_strategy + ) pipeline = RequestPipeline(protocol=config.protocol, transport=config.transport) call = ClientCall( @@ -201,22 +187,9 @@ class BedrockRuntimeClient: "protocol and transport MUST be set on the config to make calls." ) - # Resolve retry strategy from config - if isinstance(config.retry_strategy, RetryStrategy): - retry_strategy = config.retry_strategy - elif isinstance(config.retry_strategy, RetryStrategyOptions): - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=config.retry_strategy - ) - elif config.retry_strategy is None: - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=RetryStrategyOptions() - ) - else: - raise TypeError( - f"retry_strategy must be RetryStrategy, RetryStrategyOptions, or None, " - f"got {type(config.retry_strategy).__name__}" - ) + retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( + retry_strategy=config.retry_strategy + ) pipeline = RequestPipeline(protocol=config.protocol, transport=config.transport) call = ClientCall( @@ -300,22 +273,9 @@ class BedrockRuntimeClient: "protocol and transport MUST be set on the config to make calls." ) - # Resolve retry strategy from config - if isinstance(config.retry_strategy, RetryStrategy): - retry_strategy = config.retry_strategy - elif isinstance(config.retry_strategy, RetryStrategyOptions): - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=config.retry_strategy - ) - elif config.retry_strategy is None: - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=RetryStrategyOptions() - ) - else: - raise TypeError( - f"retry_strategy must be RetryStrategy, RetryStrategyOptions, or None, " - f"got {type(config.retry_strategy).__name__}" - ) + retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( + retry_strategy=config.retry_strategy + ) pipeline = RequestPipeline(protocol=config.protocol, transport=config.transport) call = ClientCall( @@ -384,22 +344,9 @@ class BedrockRuntimeClient: "protocol and transport MUST be set on the config to make calls." ) - # Resolve retry strategy from config - if isinstance(config.retry_strategy, RetryStrategy): - retry_strategy = config.retry_strategy - elif isinstance(config.retry_strategy, RetryStrategyOptions): - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=config.retry_strategy - ) - elif config.retry_strategy is None: - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=RetryStrategyOptions() - ) - else: - raise TypeError( - f"retry_strategy must be RetryStrategy, RetryStrategyOptions, or None, " - f"got {type(config.retry_strategy).__name__}" - ) + retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( + retry_strategy=config.retry_strategy + ) pipeline = RequestPipeline(protocol=config.protocol, transport=config.transport) call = ClientCall( @@ -438,22 +385,9 @@ class BedrockRuntimeClient: "protocol and transport MUST be set on the config to make calls." ) - # Resolve retry strategy from config - if isinstance(config.retry_strategy, RetryStrategy): - retry_strategy = config.retry_strategy - elif isinstance(config.retry_strategy, RetryStrategyOptions): - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=config.retry_strategy - ) - elif config.retry_strategy is None: - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=RetryStrategyOptions() - ) - else: - raise TypeError( - f"retry_strategy must be RetryStrategy, RetryStrategyOptions, or None, " - f"got {type(config.retry_strategy).__name__}" - ) + retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( + retry_strategy=config.retry_strategy + ) pipeline = RequestPipeline(protocol=config.protocol, transport=config.transport) call = ClientCall( @@ -510,22 +444,9 @@ class BedrockRuntimeClient: "protocol and transport MUST be set on the config to make calls." ) - # Resolve retry strategy from config - if isinstance(config.retry_strategy, RetryStrategy): - retry_strategy = config.retry_strategy - elif isinstance(config.retry_strategy, RetryStrategyOptions): - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=config.retry_strategy - ) - elif config.retry_strategy is None: - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=RetryStrategyOptions() - ) - else: - raise TypeError( - f"retry_strategy must be RetryStrategy, RetryStrategyOptions, or None, " - f"got {type(config.retry_strategy).__name__}" - ) + retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( + retry_strategy=config.retry_strategy + ) pipeline = RequestPipeline(protocol=config.protocol, transport=config.transport) call = ClientCall( @@ -578,22 +499,9 @@ class BedrockRuntimeClient: "protocol and transport MUST be set on the config to make calls." ) - # Resolve retry strategy from config - if isinstance(config.retry_strategy, RetryStrategy): - retry_strategy = config.retry_strategy - elif isinstance(config.retry_strategy, RetryStrategyOptions): - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=config.retry_strategy - ) - elif config.retry_strategy is None: - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=RetryStrategyOptions() - ) - else: - raise TypeError( - f"retry_strategy must be RetryStrategy, RetryStrategyOptions, or None, " - f"got {type(config.retry_strategy).__name__}" - ) + retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( + retry_strategy=config.retry_strategy + ) pipeline = RequestPipeline(protocol=config.protocol, transport=config.transport) call = ClientCall( @@ -665,22 +573,9 @@ class BedrockRuntimeClient: "protocol and transport MUST be set on the config to make calls." ) - # Resolve retry strategy from config - if isinstance(config.retry_strategy, RetryStrategy): - retry_strategy = config.retry_strategy - elif isinstance(config.retry_strategy, RetryStrategyOptions): - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=config.retry_strategy - ) - elif config.retry_strategy is None: - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=RetryStrategyOptions() - ) - else: - raise TypeError( - f"retry_strategy must be RetryStrategy, RetryStrategyOptions, or None, " - f"got {type(config.retry_strategy).__name__}" - ) + retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( + retry_strategy=config.retry_strategy + ) pipeline = RequestPipeline(protocol=config.protocol, transport=config.transport) call = ClientCall( @@ -721,22 +616,9 @@ class BedrockRuntimeClient: "protocol and transport MUST be set on the config to make calls." ) - # Resolve retry strategy from config - if isinstance(config.retry_strategy, RetryStrategy): - retry_strategy = config.retry_strategy - elif isinstance(config.retry_strategy, RetryStrategyOptions): - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=config.retry_strategy - ) - elif config.retry_strategy is None: - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=RetryStrategyOptions() - ) - else: - raise TypeError( - f"retry_strategy must be RetryStrategy, RetryStrategyOptions, or None, " - f"got {type(config.retry_strategy).__name__}" - ) + retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( + retry_strategy=config.retry_strategy + ) pipeline = RequestPipeline(protocol=config.protocol, transport=config.transport) call = ClientCall( @@ -784,22 +666,9 @@ class BedrockRuntimeClient: "protocol and transport MUST be set on the config to make calls." ) - # Resolve retry strategy from config - if isinstance(config.retry_strategy, RetryStrategy): - retry_strategy = config.retry_strategy - elif isinstance(config.retry_strategy, RetryStrategyOptions): - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=config.retry_strategy - ) - elif config.retry_strategy is None: - retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( - options=RetryStrategyOptions() - ) - else: - raise TypeError( - f"retry_strategy must be RetryStrategy, RetryStrategyOptions, or None, " - f"got {type(config.retry_strategy).__name__}" - ) + retry_strategy = await self._retry_strategy_resolver.resolve_retry_strategy( + retry_strategy=config.retry_strategy + ) pipeline = RequestPipeline(protocol=config.protocol, transport=config.transport) call = ClientCall(