Skip to main content

aws_sdk_bedrockruntime/
config.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2#![allow(clippy::empty_line_after_doc_comments)]
3/// Configuration for a aws_sdk_bedrockruntime service client.
4///
5/// Service configuration allows for customization of endpoints, region, credentials providers,
6/// and retry configuration. Generally, it is constructed automatically for you from a shared
7/// configuration loaded by the `aws-config` crate. For example:
8///
9/// ```ignore
10/// // Load a shared config from the environment
11/// let shared_config = aws_config::from_env().load().await;
12/// // The client constructor automatically converts the shared config into the service config
13/// let client = Client::new(&shared_config);
14/// ```
15///
16/// The service config can also be constructed manually using its builder.
17///
18#[derive(::std::clone::Clone, ::std::fmt::Debug)]
19pub struct Config {
20    // Both `config` and `cloneable` are the same config, but the cloneable one
21    // is kept around so that it is possible to convert back into a builder. This can be
22    // optimized in the future.
23    pub(crate) config: crate::config::FrozenLayer,
24    cloneable: ::aws_smithy_types::config_bag::CloneableLayer,
25    pub(crate) runtime_components: crate::config::RuntimeComponentsBuilder,
26    pub(crate) runtime_plugins: ::std::vec::Vec<crate::config::SharedRuntimePlugin>,
27    pub(crate) behavior_version: ::std::option::Option<crate::config::BehaviorVersion>,
28}
29impl Config {
30    ///
31    /// Constructs a config builder.
32    /// <div class="warning">
33    /// Note that a config created from this builder will not have the same safe defaults as one created by
34    /// the <a href="https://crates.io/crates/aws-config" target="_blank">aws-config</a> crate.
35    /// </div>
36    ///
37    pub fn builder() -> Builder {
38        Builder::default()
39    }
40    /// Converts this config back into a builder so that it can be tweaked.
41    pub fn to_builder(&self) -> Builder {
42        Builder {
43            config: self.cloneable.clone(),
44            runtime_components: self.runtime_components.clone(),
45            runtime_plugins: self.runtime_plugins.clone(),
46            behavior_version: self.behavior_version,
47        }
48    }
49    /// Return a reference to the stalled stream protection configuration contained in this config, if any.
50    pub fn stalled_stream_protection(&self) -> ::std::option::Option<&crate::config::StalledStreamProtectionConfig> {
51        self.config.load::<crate::config::StalledStreamProtectionConfig>()
52    }
53    /// Return the [`SharedHttpClient`](crate::config::SharedHttpClient) to use when making requests, if any.
54    pub fn http_client(&self) -> Option<crate::config::SharedHttpClient> {
55        self.runtime_components.http_client()
56    }
57    /// Return the auth schemes configured on this service config
58    pub fn auth_schemes(&self) -> impl Iterator<Item = ::aws_smithy_runtime_api::client::auth::SharedAuthScheme> + '_ {
59        self.runtime_components.auth_schemes()
60    }
61
62    /// Return the auth scheme resolver configured on this service config
63    pub fn auth_scheme_resolver(&self) -> ::std::option::Option<::aws_smithy_runtime_api::client::auth::SharedAuthSchemeOptionResolver> {
64        self.runtime_components.auth_scheme_option_resolver()
65    }
66    /// Returns the configured auth scheme preference
67    pub fn auth_scheme_preference(&self) -> ::std::option::Option<&::aws_smithy_runtime_api::client::auth::AuthSchemePreference> {
68        self.config.load::<::aws_smithy_runtime_api::client::auth::AuthSchemePreference>()
69    }
70    /// Returns the endpoint resolver.
71    pub fn endpoint_resolver(&self) -> ::aws_smithy_runtime_api::client::endpoint::SharedEndpointResolver {
72        self.runtime_components.endpoint_resolver().expect("resolver defaulted if not set")
73    }
74    /// Return a reference to the retry configuration contained in this config, if any.
75    pub fn retry_config(&self) -> ::std::option::Option<&::aws_smithy_types::retry::RetryConfig> {
76        self.config.load::<::aws_smithy_types::retry::RetryConfig>()
77    }
78
79    /// Return a cloned shared async sleep implementation from this config, if any.
80    pub fn sleep_impl(&self) -> ::std::option::Option<crate::config::SharedAsyncSleep> {
81        self.runtime_components.sleep_impl()
82    }
83
84    /// Return a reference to the timeout configuration contained in this config, if any.
85    pub fn timeout_config(&self) -> ::std::option::Option<&::aws_smithy_types::timeout::TimeoutConfig> {
86        self.config.load::<::aws_smithy_types::timeout::TimeoutConfig>()
87    }
88
89    /// Returns a reference to the retry partition contained in this config, if any.
90    ///
91    /// WARNING: This method is unstable and may be removed at any time. Do not rely on this
92    /// method for anything!
93    pub fn retry_partition(&self) -> ::std::option::Option<&::aws_smithy_runtime::client::retries::RetryPartition> {
94        self.config.load::<::aws_smithy_runtime::client::retries::RetryPartition>()
95    }
96    /// Returns the configured identity cache for auth.
97    pub fn identity_cache(&self) -> ::std::option::Option<crate::config::SharedIdentityCache> {
98        self.runtime_components.identity_cache()
99    }
100    /// Returns interceptors currently registered by the user.
101    pub fn interceptors(&self) -> impl Iterator<Item = crate::config::SharedInterceptor> + '_ {
102        self.runtime_components.interceptors()
103    }
104    /// Return time source used for this service.
105    pub fn time_source(&self) -> ::std::option::Option<::aws_smithy_async::time::SharedTimeSource> {
106        self.runtime_components.time_source()
107    }
108    /// Returns retry classifiers currently registered by the user.
109    pub fn retry_classifiers(&self) -> impl Iterator<Item = ::aws_smithy_runtime_api::client::retries::classifiers::SharedRetryClassifier> + '_ {
110        self.runtime_components.retry_classifiers()
111    }
112    /// Returns the name of the app that is using the client, if it was provided.
113    ///
114    /// This _optional_ name is used to identify the application in the user agent that
115    /// gets sent along with requests.
116    pub fn app_name(&self) -> ::std::option::Option<&::aws_types::app_name::AppName> {
117        self.config.load::<::aws_types::app_name::AppName>()
118    }
119    /// Returns the framework metadata that has been configured, if any.
120    ///
121    /// This _optional_ metadata identifies software frameworks or third-party libraries
122    /// being used with the client, rendered into the user agent as `lib/{name}/{version}`.
123    /// Entries are returned in first-seen (insertion) order, matching the order they are
124    /// rendered into the user agent.
125    pub fn framework_metadata(&self) -> ::std::vec::Vec<&::aws_types::sdk_ua_metadata::FrameworkMetadata> {
126        // `StoreAppend` loads entries newest-first; reverse to first-seen order so
127        // this getter agrees with both the user agent and `SdkConfig::framework_metadata`.
128        let mut entries: ::std::vec::Vec<&::aws_types::sdk_ua_metadata::FrameworkMetadata> =
129            self.config.load::<::aws_types::sdk_ua_metadata::FrameworkMetadata>().collect();
130        entries.reverse();
131        entries
132    }
133    /// Returns the invocation ID generator if one was given in config.
134    ///
135    /// The invocation ID generator generates ID values for the `amz-sdk-invocation-id` header. By default, this will be a random UUID. Overriding it may be useful in tests that examine the HTTP request and need to be deterministic.
136    pub fn invocation_id_generator(&self) -> ::std::option::Option<::aws_runtime::invocation_id::SharedInvocationIdGenerator> {
137        self.config.load::<::aws_runtime::invocation_id::SharedInvocationIdGenerator>().cloned()
138    }
139    /// Creates a new [service config](crate::Config) from a [shared `config`](::aws_types::sdk_config::SdkConfig).
140    pub fn new(config: &::aws_types::sdk_config::SdkConfig) -> Self {
141        Builder::from(config).build()
142    }
143    /// The signature version 4 service signing name to use in the credential scope when signing requests.
144    ///
145    /// The signing service may be overridden by the `Endpoint`, or by specifying a custom
146    /// [`SigningName`](aws_types::SigningName) during operation construction
147    pub fn signing_name(&self) -> &'static str {
148        "bedrock"
149    }
150    /// Returns the AWS region, if it was provided.
151    pub fn region(&self) -> ::std::option::Option<&crate::config::Region> {
152        self.config.load::<crate::config::Region>()
153    }
154    /// This function was intended to be removed, and has been broken since release-2023-11-15 as it always returns a `None`. Do not use.
155    #[deprecated(
156        note = "This function was intended to be removed, and has been broken since release-2023-11-15 as it always returns a `None`. Do not use."
157    )]
158    pub fn credentials_provider(&self) -> Option<crate::config::SharedCredentialsProvider> {
159        ::std::option::Option::None
160    }
161}
162/// Builder for creating a `Config`.
163#[derive(::std::clone::Clone, ::std::fmt::Debug)]
164pub struct Builder {
165    pub(crate) config: ::aws_smithy_types::config_bag::CloneableLayer,
166    pub(crate) runtime_components: crate::config::RuntimeComponentsBuilder,
167    pub(crate) runtime_plugins: ::std::vec::Vec<crate::config::SharedRuntimePlugin>,
168    pub(crate) behavior_version: ::std::option::Option<crate::config::BehaviorVersion>,
169}
170impl ::std::default::Default for Builder {
171    fn default() -> Self {
172        Self {
173            config: ::std::default::Default::default(),
174            runtime_components: crate::config::RuntimeComponentsBuilder::new("service config"),
175            runtime_plugins: ::std::default::Default::default(),
176            behavior_version: ::std::default::Default::default(),
177        }
178    }
179}
180impl Builder {
181    ///
182    /// Constructs a config builder.
183    /// <div class="warning">
184    /// Note that a config created from this builder will not have the same safe defaults as one created by
185    /// the <a href="https://crates.io/crates/aws-config" target="_blank">aws-config</a> crate.
186    /// </div>
187    ///
188    pub fn new() -> Self {
189        Self::default()
190    }
191    /// Constructs a config builder from the given `config_bag`, setting only fields stored in the config bag,
192    /// but not those in runtime components.
193    #[allow(unused)]
194    pub(crate) fn from_config_bag(config_bag: &::aws_smithy_types::config_bag::ConfigBag) -> Self {
195        let mut builder = Self::new();
196        builder.set_stalled_stream_protection(config_bag.load::<crate::config::StalledStreamProtectionConfig>().cloned());
197        builder.set_auth_scheme_preference(config_bag.load::<::aws_smithy_runtime_api::client::auth::AuthSchemePreference>().cloned());
198        builder.set_retry_config(config_bag.load::<::aws_smithy_types::retry::RetryConfig>().cloned());
199        builder.set_timeout_config(config_bag.load::<::aws_smithy_types::timeout::TimeoutConfig>().cloned());
200        builder.set_retry_partition(config_bag.load::<::aws_smithy_runtime::client::retries::RetryPartition>().cloned());
201        builder.set_app_name(config_bag.load::<::aws_types::app_name::AppName>().cloned());
202        for framework_metadata in config_bag.load::<::aws_types::sdk_ua_metadata::FrameworkMetadata>() {
203            builder.push_framework_metadata(framework_metadata.clone());
204        }
205        builder.set_endpoint_url(config_bag.load::<::aws_types::endpoint_config::EndpointUrl>().map(|ty| ty.0.clone()));
206        builder.set_use_dual_stack(config_bag.load::<::aws_types::endpoint_config::UseDualStack>().map(|ty| ty.0));
207        builder.set_use_fips(config_bag.load::<::aws_types::endpoint_config::UseFips>().map(|ty| ty.0));
208        builder.set_region(config_bag.load::<crate::config::Region>().cloned());
209        builder
210    }
211    /// Names operation-input members whose values are captured *and* emitted as
212    /// attributes on the client's built-in metrics (e.g. `["Bucket"]`).
213    ///
214    /// Emitting implies capture, so an emitted member is also readable in-process
215    /// via `CapturedTelemetryAttributes` on the config bag. Names are Smithy input
216    /// member names; only string-valued, non-sensitive members are eligible, and
217    /// naming any other member has no effect. Off by default.
218    ///
219    /// Prefer bounded identifiers here: an emitted member becomes a metric label, so
220    /// high-cardinality values (like object keys) fragment the metrics and inflate
221    /// cost. Use [`Self::capture_input_attributes`] for values you want to read
222    /// in-process without emitting them on the metrics.
223    pub fn emit_input_attributes(mut self, names: impl ::std::iter::IntoIterator<Item = impl ::std::convert::Into<::std::string::String>>) -> Self {
224        let mut requested = self
225            .config
226            .load::<::aws_smithy_types::telemetry::RequestedTelemetryAttributes>()
227            .cloned()
228            .unwrap_or_default();
229        requested.emit(names.into_iter().map(|n| n.into()));
230        self.config.store_put(requested);
231        self
232    }
233
234    /// Names operation-input members whose values are captured into
235    /// `CapturedTelemetryAttributes` for in-process reads (e.g. from a custom
236    /// interceptor), but are **not** emitted on the built-in metrics.
237    ///
238    /// Use this for values you need during the operation lifecycle but do not want on
239    /// the metric label set (for example, high-cardinality identifiers). Names follow
240    /// the same eligibility rules as [`Self::emit_input_attributes`]. Off by default.
241    pub fn capture_input_attributes(
242        mut self,
243        names: impl ::std::iter::IntoIterator<Item = impl ::std::convert::Into<::std::string::String>>,
244    ) -> Self {
245        let mut requested = self
246            .config
247            .load::<::aws_smithy_types::telemetry::RequestedTelemetryAttributes>()
248            .cloned()
249            .unwrap_or_default();
250        requested.capture_only(names.into_iter().map(|n| n.into()));
251        self.config.store_put(requested);
252        self
253    }
254    /// Set the [`StalledStreamProtectionConfig`](crate::config::StalledStreamProtectionConfig)
255    /// to configure protection for stalled streams.
256    pub fn stalled_stream_protection(mut self, stalled_stream_protection_config: crate::config::StalledStreamProtectionConfig) -> Self {
257        self.set_stalled_stream_protection(::std::option::Option::Some(stalled_stream_protection_config));
258        self
259    }
260    /// Set the [`StalledStreamProtectionConfig`](crate::config::StalledStreamProtectionConfig)
261    /// to configure protection for stalled streams.
262    pub fn set_stalled_stream_protection(
263        &mut self,
264        stalled_stream_protection_config: ::std::option::Option<crate::config::StalledStreamProtectionConfig>,
265    ) -> &mut Self {
266        self.config.store_or_unset(stalled_stream_protection_config);
267        self
268    }
269    /// Sets the idempotency token provider to use for service calls that require tokens.
270    pub fn idempotency_token_provider(
271        mut self,
272        idempotency_token_provider: impl ::std::convert::Into<crate::idempotency_token::IdempotencyTokenProvider>,
273    ) -> Self {
274        self.set_idempotency_token_provider(::std::option::Option::Some(idempotency_token_provider.into()));
275        self
276    }
277    /// Sets the idempotency token provider to use for service calls that require tokens.
278    pub fn set_idempotency_token_provider(
279        &mut self,
280        idempotency_token_provider: ::std::option::Option<crate::idempotency_token::IdempotencyTokenProvider>,
281    ) -> &mut Self {
282        self.config.store_or_unset(idempotency_token_provider);
283        self
284    }
285    /// Sets the HTTP client to use when making requests.
286    ///
287    /// # Examples
288    /// ```no_run
289    /// # #[cfg(test)]
290    /// # mod tests {
291    /// # #[test]
292    /// # fn example() {
293    /// use std::time::Duration;
294    /// use aws_sdk_bedrockruntime::config::Config;
295    /// use aws_smithy_runtime::client::http::hyper_014::HyperClientBuilder;
296    ///
297    /// let https_connector = hyper_rustls::HttpsConnectorBuilder::new()
298    ///     .with_webpki_roots()
299    ///     .https_only()
300    ///     .enable_http1()
301    ///     .enable_http2()
302    ///     .build();
303    /// let hyper_client = HyperClientBuilder::new().build(https_connector);
304    ///
305    /// // This connector can then be given to a generated service Config
306    /// let config = my_service_client::Config::builder()
307    ///     .endpoint_url("https://example.com")
308    ///     .http_client(hyper_client)
309    ///     .build();
310    /// let client = my_service_client::Client::from_conf(config);
311    /// # }
312    /// # }
313    /// ```
314    pub fn http_client(mut self, http_client: impl crate::config::HttpClient + 'static) -> Self {
315        self.set_http_client(::std::option::Option::Some(crate::config::IntoShared::into_shared(http_client)));
316        self
317    }
318
319    /// Sets the HTTP client to use when making requests.
320    ///
321    /// # Examples
322    /// ```no_run
323    /// # #[cfg(test)]
324    /// # mod tests {
325    /// # #[test]
326    /// # fn example() {
327    /// use std::time::Duration;
328    /// use aws_sdk_bedrockruntime::config::{Builder, Config};
329    /// use aws_smithy_runtime::client::http::hyper_014::HyperClientBuilder;
330    ///
331    /// fn override_http_client(builder: &mut Builder) {
332    ///     let https_connector = hyper_rustls::HttpsConnectorBuilder::new()
333    ///         .with_webpki_roots()
334    ///         .https_only()
335    ///         .enable_http1()
336    ///         .enable_http2()
337    ///         .build();
338    ///     let hyper_client = HyperClientBuilder::new().build(https_connector);
339    ///     builder.set_http_client(Some(hyper_client));
340    /// }
341    ///
342    /// let mut builder = aws_sdk_bedrockruntime::Config::builder();
343    /// override_http_client(&mut builder);
344    /// let config = builder.build();
345    /// # }
346    /// # }
347    /// ```
348    pub fn set_http_client(&mut self, http_client: Option<crate::config::SharedHttpClient>) -> &mut Self {
349        self.runtime_components.set_http_client(http_client);
350        self
351    }
352    /// Sets the bearer token that will be used for HTTP bearer auth.
353    pub fn bearer_token(self, bearer_token: crate::config::Token) -> Self {
354        self.bearer_token_resolver(bearer_token)
355    }
356
357    /// Sets a bearer token provider that will be used for HTTP bearer auth.
358    pub fn bearer_token_resolver(mut self, bearer_token_resolver: impl crate::config::ResolveIdentity + 'static) -> Self {
359        self.runtime_components.set_identity_resolver(
360            ::aws_smithy_runtime_api::client::auth::http::HTTP_BEARER_AUTH_SCHEME_ID,
361            ::aws_smithy_runtime_api::shared::IntoShared::<::aws_smithy_runtime_api::client::identity::SharedIdentityResolver>::into_shared(
362                bearer_token_resolver,
363            ),
364        );
365        self
366    }
367    /// Adds an auth scheme to the builder
368    ///
369    /// If `auth_scheme` has an existing [AuthSchemeId](aws_smithy_runtime_api::client::auth::AuthSchemeId) in the runtime, the current identity
370    /// resolver and signer for that scheme will be replaced by those from `auth_scheme`.
371    ///
372    /// _Important:_ When introducing a custom auth scheme, ensure you override either
373    /// [`Self::auth_scheme_resolver`] or [`Self::set_auth_scheme_resolver`]
374    /// so that the custom auth scheme is included in the list of resolved auth scheme options.
375    /// [The default auth scheme resolver](crate::config::auth::DefaultAuthSchemeResolver) will not recognize your custom auth scheme.
376    ///
377    /// # Examples
378    /// ```no_run
379    /// # use aws_smithy_runtime_api::{
380    /// #     box_error::BoxError,
381    /// #     client::{
382    /// #         auth::{
383    /// #             AuthScheme, AuthSchemeEndpointConfig, AuthSchemeId, AuthSchemeOption,
384    /// #             AuthSchemeOptionsFuture, Sign,
385    /// #         },
386    /// #         identity::{Identity, IdentityFuture, ResolveIdentity, SharedIdentityResolver},
387    /// #         orchestrator::HttpRequest,
388    /// #         runtime_components::{GetIdentityResolver, RuntimeComponents},
389    /// #   },
390    /// #   shared::IntoShared,
391    /// # };
392    /// # use aws_smithy_types::config_bag::ConfigBag;
393    /// // Auth scheme with customer identity resolver and signer
394    /// #[derive(Debug)]
395    /// struct CustomAuthScheme {
396    ///     id: AuthSchemeId,
397    ///     identity_resolver: SharedIdentityResolver,
398    ///     signer: CustomSigner,
399    /// }
400    /// impl Default for CustomAuthScheme {
401    ///     fn default() -> Self {
402    ///         Self {
403    ///             id: AuthSchemeId::new("custom"),
404    ///             identity_resolver: CustomIdentityResolver.into_shared(),
405    ///             signer: CustomSigner,
406    ///         }
407    ///     }
408    /// }
409    /// impl AuthScheme for CustomAuthScheme {
410    ///     fn scheme_id(&self) -> AuthSchemeId {
411    ///         self.id.clone()
412    ///     }
413    ///     fn identity_resolver(
414    ///         &self,
415    ///         _identity_resolvers: &dyn GetIdentityResolver,
416    ///     ) -> Option<SharedIdentityResolver> {
417    ///         Some(self.identity_resolver.clone())
418    ///     }
419    ///     fn signer(&self) -> &dyn Sign {
420    ///         &self.signer
421    ///     }
422    /// }
423    ///
424    /// #[derive(Debug, Default)]
425    /// struct CustomSigner;
426    /// impl Sign for CustomSigner {
427    ///     fn sign_http_request(
428    ///         &self,
429    ///         _request: &mut HttpRequest,
430    ///         _identity: &Identity,
431    ///         _auth_scheme_endpoint_config: AuthSchemeEndpointConfig<'_>,
432    ///         _runtime_components: &RuntimeComponents,
433    ///         _config_bag: &ConfigBag,
434    ///     ) -> Result<(), BoxError> {
435    ///         // --snip--
436    /// #      todo!()
437    ///     }
438    /// }
439    ///
440    /// #[derive(Debug)]
441    /// struct CustomIdentityResolver;
442    /// impl ResolveIdentity for CustomIdentityResolver {
443    ///     fn resolve_identity<'a>(
444    ///         &'a self,
445    ///         _runtime_components: &'a RuntimeComponents,
446    ///         _config_bag: &'a ConfigBag,
447    ///     ) -> IdentityFuture<'a> {
448    ///         // --snip--
449    /// #      todo!()
450    ///     }
451    /// }
452    ///
453    /// // Auth scheme resolver that favors `CustomAuthScheme`
454    /// #[derive(Debug)]
455    /// struct CustomAuthSchemeResolver;
456    /// impl aws_sdk_bedrockruntime::config::auth::ResolveAuthScheme for CustomAuthSchemeResolver {
457    ///     fn resolve_auth_scheme<'a>(
458    ///         &'a self,
459    ///         _params: &'a aws_sdk_bedrockruntime::config::auth::Params,
460    ///         _cfg: &'a ConfigBag,
461    ///         _runtime_components: &'a RuntimeComponents,
462    ///     ) -> AuthSchemeOptionsFuture<'a> {
463    ///         AuthSchemeOptionsFuture::ready(Ok(vec![AuthSchemeOption::from(AuthSchemeId::new(
464    ///             "custom",
465    ///         ))]))
466    ///     }
467    /// }
468    ///
469    /// let config = aws_sdk_bedrockruntime::Config::builder()
470    ///     .push_auth_scheme(CustomAuthScheme::default())
471    ///     .auth_scheme_resolver(CustomAuthSchemeResolver)
472    ///     // other configurations
473    ///     .build();
474    /// ```
475    pub fn push_auth_scheme(mut self, auth_scheme: impl ::aws_smithy_runtime_api::client::auth::AuthScheme + 'static) -> Self {
476        self.runtime_components.push_auth_scheme(auth_scheme);
477        self
478    }
479
480    /// Set the auth scheme resolver for the builder
481    ///
482    /// # Examples
483    /// ```no_run
484    /// # use aws_smithy_runtime_api::{
485    /// #     client::{
486    /// #         auth::AuthSchemeOptionsFuture,
487    /// #         runtime_components::RuntimeComponents,
488    /// #   },
489    /// # };
490    /// # use aws_smithy_types::config_bag::ConfigBag;
491    /// #[derive(Debug)]
492    /// struct CustomAuthSchemeResolver;
493    /// impl aws_sdk_bedrockruntime::config::auth::ResolveAuthScheme for CustomAuthSchemeResolver {
494    ///     fn resolve_auth_scheme<'a>(
495    ///         &'a self,
496    ///         _params: &'a aws_sdk_bedrockruntime::config::auth::Params,
497    ///         _cfg: &'a ConfigBag,
498    ///         _runtime_components: &'a RuntimeComponents,
499    ///     ) -> AuthSchemeOptionsFuture<'a> {
500    ///         // --snip--
501    /// #      todo!()
502    ///     }
503    /// }
504    ///
505    /// let config = aws_sdk_bedrockruntime::Config::builder()
506    ///     .auth_scheme_resolver(CustomAuthSchemeResolver)
507    ///     // other configurations
508    ///     .build();
509    /// ```
510    pub fn auth_scheme_resolver(mut self, auth_scheme_resolver: impl crate::config::auth::ResolveAuthScheme + 'static) -> Self {
511        self.set_auth_scheme_resolver(auth_scheme_resolver);
512        self
513    }
514
515    /// Set the auth scheme resolver for the builder
516    ///
517    /// # Examples
518    /// See an example for [`Self::auth_scheme_resolver`].
519    pub fn set_auth_scheme_resolver(&mut self, auth_scheme_resolver: impl crate::config::auth::ResolveAuthScheme + 'static) -> &mut Self {
520        self.runtime_components
521            .set_auth_scheme_option_resolver(::std::option::Option::Some(auth_scheme_resolver.into_shared_resolver()));
522        self
523    }
524
525    /// Enable no authentication regardless of what authentication mechanisms operations support
526    ///
527    /// This adds [NoAuthScheme](aws_smithy_runtime::client::auth::no_auth::NoAuthScheme) as a fallback
528    /// and the auth scheme resolver will use it when no other auth schemes are applicable.
529    pub fn allow_no_auth(mut self) -> Self {
530        self.set_allow_no_auth();
531        self
532    }
533
534    /// Enable no authentication regardless of what authentication mechanisms operations support
535    ///
536    /// This adds [NoAuthScheme](aws_smithy_runtime::client::auth::no_auth::NoAuthScheme) as a fallback
537    /// and the auth scheme resolver will use it when no other auth schemes are applicable.
538    pub fn set_allow_no_auth(&mut self) -> &mut Self {
539        self.push_runtime_plugin(::aws_smithy_runtime::client::auth::no_auth::NoAuthRuntimePluginV2::new().into_shared());
540        self
541    }
542    /// Set the auth scheme preference for an auth scheme resolver
543    /// (typically the default auth scheme resolver).
544    ///
545    /// Each operation has a predefined order of auth schemes, as determined by the service,
546    /// for auth scheme resolution. By using the auth scheme preference, customers
547    /// can reorder the schemes resolved by the auth scheme resolver.
548    ///
549    /// The preference list is intended as a hint rather than a strict override.
550    /// Any schemes not present in the originally resolved auth schemes will be ignored.
551    ///
552    /// # Examples
553    ///
554    /// ```no_run
555    /// # use aws_smithy_runtime_api::client::auth::AuthSchemeId;
556    /// let config = aws_sdk_bedrockruntime::Config::builder()
557    ///     .auth_scheme_preference([AuthSchemeId::from("scheme1"), AuthSchemeId::from("scheme2")])
558    ///     // ...
559    ///     .build();
560    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
561    /// ```
562
563    pub fn auth_scheme_preference(
564        mut self,
565        preference: impl ::std::convert::Into<::aws_smithy_runtime_api::client::auth::AuthSchemePreference>,
566    ) -> Self {
567        self.set_auth_scheme_preference(::std::option::Option::Some(preference.into()));
568        self
569    }
570
571    /// Set the auth scheme preference for an auth scheme resolver
572    /// (typically the default auth scheme resolver).
573    ///
574    /// Each operation has a predefined order of auth schemes, as determined by the service,
575    /// for auth scheme resolution. By using the auth scheme preference, customers
576    /// can reorder the schemes resolved by the auth scheme resolver.
577    ///
578    /// The preference list is intended as a hint rather than a strict override.
579    /// Any schemes not present in the originally resolved auth schemes will be ignored.
580    ///
581    /// # Examples
582    ///
583    /// ```no_run
584    /// # use aws_smithy_runtime_api::client::auth::AuthSchemeId;
585    /// let config = aws_sdk_bedrockruntime::Config::builder()
586    ///     .auth_scheme_preference([AuthSchemeId::from("scheme1"), AuthSchemeId::from("scheme2")])
587    ///     // ...
588    ///     .build();
589    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
590    /// ```
591
592    pub fn set_auth_scheme_preference(
593        &mut self,
594        preference: ::std::option::Option<::aws_smithy_runtime_api::client::auth::AuthSchemePreference>,
595    ) -> &mut Self {
596        self.config.store_or_unset(preference);
597        self
598    }
599    /// Sets the endpoint resolver to use when making requests.
600    ///
601    ///
602    /// When unset, the client will used a generated endpoint resolver based on the endpoint resolution
603    /// rules for `aws_sdk_bedrockruntime`.
604    ///
605    ///
606    /// Note: setting an endpoint resolver will replace any endpoint URL that has been set.
607    /// This method accepts an endpoint resolver [specific to this service](crate::config::endpoint::ResolveEndpoint). If you want to
608    /// provide a shared endpoint resolver, use [`Self::set_endpoint_resolver`].
609    ///
610    /// # Examples
611    /// Create a custom endpoint resolver that resolves a different endpoing per-stage, e.g. staging vs. production.
612    /// ```no_run
613    /// use aws_sdk_bedrockruntime::config::endpoint::{ResolveEndpoint, EndpointFuture, Params, Endpoint};
614    /// #[derive(Debug)]
615    /// struct StageResolver { stage: String }
616    /// impl ResolveEndpoint for StageResolver {
617    ///     fn resolve_endpoint(&self, params: &Params) -> EndpointFuture<'_> {
618    ///         let stage = &self.stage;
619    ///         EndpointFuture::ready(Ok(Endpoint::builder().url(format!("{stage}.myservice.com")).build()))
620    ///     }
621    /// }
622    /// let resolver = StageResolver { stage: std::env::var("STAGE").unwrap() };
623    /// let config = aws_sdk_bedrockruntime::Config::builder().endpoint_resolver(resolver).build();
624    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
625    /// ```
626    pub fn endpoint_resolver(mut self, endpoint_resolver: impl crate::config::endpoint::ResolveEndpoint + 'static) -> Self {
627        self.set_endpoint_resolver(::std::option::Option::Some(endpoint_resolver.into_shared_resolver()));
628        self
629    }
630
631    /// Sets the endpoint resolver to use when making requests.
632    ///
633    ///
634    /// When unset, the client will used a generated endpoint resolver based on the endpoint resolution
635    /// rules for `aws_sdk_bedrockruntime`.
636    ///
637    pub fn set_endpoint_resolver(
638        &mut self,
639        endpoint_resolver: ::std::option::Option<::aws_smithy_runtime_api::client::endpoint::SharedEndpointResolver>,
640    ) -> &mut Self {
641        self.runtime_components.set_endpoint_resolver(endpoint_resolver);
642        self
643    }
644    /// Set the retry_config for the builder
645    ///
646    /// # Examples
647    /// ```no_run
648    /// use aws_sdk_bedrockruntime::config::Config;
649    /// use aws_sdk_bedrockruntime::config::retry::RetryConfig;
650    ///
651    /// let retry_config = RetryConfig::standard().with_max_attempts(5);
652    /// let config = Config::builder().retry_config(retry_config).build();
653    /// ```
654    ///
655    /// # Retry token bucket
656    ///
657    /// [`RetryConfig`](::aws_smithy_types::retry::RetryConfig) controls *how many* times to retry and *how long* to back
658    /// off. Retries are **also** gated by a retry token bucket (also called the retry quota) that
659    /// is shared across a [`RetryPartition`](::aws_smithy_runtime::client::retries::RetryPartition). To configure the token bucket — for
660    /// example, to set
661    /// its capacity or to give a workload its own bucket — see [`Self::retry_partition`] and
662    /// [`RetryPartition::custom`](::aws_smithy_runtime::client::retries::RetryPartition::custom).
663    pub fn retry_config(mut self, retry_config: ::aws_smithy_types::retry::RetryConfig) -> Self {
664        self.set_retry_config(Some(retry_config));
665        self
666    }
667
668    /// Set the retry_config for the builder
669    ///
670    /// # Examples
671    /// ```no_run
672    /// use aws_sdk_bedrockruntime::config::{Builder, Config};
673    /// use aws_sdk_bedrockruntime::config::retry::RetryConfig;
674    ///
675    /// fn disable_retries(builder: &mut Builder) {
676    ///     let retry_config = RetryConfig::standard().with_max_attempts(1);
677    ///     builder.set_retry_config(Some(retry_config));
678    /// }
679    ///
680    /// let mut builder = Config::builder();
681    /// disable_retries(&mut builder);
682    /// let config = builder.build();
683    /// ```
684    pub fn set_retry_config(&mut self, retry_config: ::std::option::Option<::aws_smithy_types::retry::RetryConfig>) -> &mut Self {
685        retry_config.map(|r| self.config.store_put(r));
686        self
687    }
688    /// Set the sleep_impl for the builder
689    ///
690    /// # Examples
691    ///
692    /// ```no_run
693    /// use aws_sdk_bedrockruntime::config::{AsyncSleep, Config, SharedAsyncSleep, Sleep};
694    ///
695    /// #[derive(Debug)]
696    /// pub struct ForeverSleep;
697    ///
698    /// impl AsyncSleep for ForeverSleep {
699    ///     fn sleep(&self, duration: std::time::Duration) -> Sleep {
700    ///         Sleep::new(std::future::pending())
701    ///     }
702    /// }
703    ///
704    /// let sleep_impl = SharedAsyncSleep::new(ForeverSleep);
705    /// let config = Config::builder().sleep_impl(sleep_impl).build();
706    /// ```
707    pub fn sleep_impl(mut self, sleep_impl: impl crate::config::AsyncSleep + 'static) -> Self {
708        self.set_sleep_impl(Some(::aws_smithy_runtime_api::shared::IntoShared::into_shared(sleep_impl)));
709        self
710    }
711
712    /// Set the sleep_impl for the builder
713    ///
714    /// # Examples
715    ///
716    /// ```no_run
717    /// use aws_sdk_bedrockruntime::config::{AsyncSleep, Builder, Config, SharedAsyncSleep, Sleep};
718    ///
719    /// #[derive(Debug)]
720    /// pub struct ForeverSleep;
721    ///
722    /// impl AsyncSleep for ForeverSleep {
723    ///     fn sleep(&self, duration: std::time::Duration) -> Sleep {
724    ///         Sleep::new(std::future::pending())
725    ///     }
726    /// }
727    ///
728    /// fn set_never_ending_sleep_impl(builder: &mut Builder) {
729    ///     let sleep_impl = SharedAsyncSleep::new(ForeverSleep);
730    ///     builder.set_sleep_impl(Some(sleep_impl));
731    /// }
732    ///
733    /// let mut builder = Config::builder();
734    /// set_never_ending_sleep_impl(&mut builder);
735    /// let config = builder.build();
736    /// ```
737    pub fn set_sleep_impl(&mut self, sleep_impl: ::std::option::Option<crate::config::SharedAsyncSleep>) -> &mut Self {
738        self.runtime_components.set_sleep_impl(sleep_impl);
739        self
740    }
741    /// Set the timeout_config for the builder
742    ///
743    /// # Examples
744    ///
745    /// ```no_run
746    /// # use std::time::Duration;
747    /// use aws_sdk_bedrockruntime::config::Config;
748    /// use aws_sdk_bedrockruntime::config::timeout::TimeoutConfig;
749    ///
750    /// let timeout_config = TimeoutConfig::builder()
751    ///     .operation_attempt_timeout(Duration::from_secs(1))
752    ///     .build();
753    /// let config = Config::builder().timeout_config(timeout_config).build();
754    /// ```
755    pub fn timeout_config(mut self, timeout_config: ::aws_smithy_types::timeout::TimeoutConfig) -> Self {
756        self.set_timeout_config(Some(timeout_config));
757        self
758    }
759
760    /// Set the timeout_config for the builder.
761    ///
762    /// Setting this to `None` has no effect if another source of configuration has set timeouts. If you
763    /// are attempting to disable timeouts, use [`TimeoutConfig::disabled`](::aws_smithy_types::timeout::TimeoutConfig::disabled)
764    ///
765    ///
766    /// # Examples
767    ///
768    /// ```no_run
769    /// # use std::time::Duration;
770    /// use aws_sdk_bedrockruntime::config::{Builder, Config};
771    /// use aws_sdk_bedrockruntime::config::timeout::TimeoutConfig;
772    ///
773    /// fn set_request_timeout(builder: &mut Builder) {
774    ///     let timeout_config = TimeoutConfig::builder()
775    ///         .operation_attempt_timeout(Duration::from_secs(1))
776    ///         .build();
777    ///     builder.set_timeout_config(Some(timeout_config));
778    /// }
779    ///
780    /// let mut builder = Config::builder();
781    /// set_request_timeout(&mut builder);
782    /// let config = builder.build();
783    /// ```
784    pub fn set_timeout_config(&mut self, timeout_config: ::std::option::Option<::aws_smithy_types::timeout::TimeoutConfig>) -> &mut Self {
785        // passing None has no impact.
786        let Some(mut timeout_config) = timeout_config else { return self };
787
788        if let Some(base) = self.config.load::<::aws_smithy_types::timeout::TimeoutConfig>() {
789            timeout_config.take_defaults_from(base);
790        }
791        self.config.store_put(timeout_config);
792        self
793    }
794    /// Set the partition for retry-related state. When clients share a retry partition, they will
795    /// also share components such as token buckets and client rate limiters.
796    /// See the [`RetryPartition`](::aws_smithy_runtime::client::retries::RetryPartition) documentation for more details.
797    ///
798    /// # Default Behavior
799    ///
800    /// When no retry partition is explicitly set, the SDK automatically creates a default retry partition named `bedrockruntime`
801    /// (or `bedrockruntime-<region>` if a region is configured).
802    /// All Bedrock Runtime clients without an explicit retry partition will share this default partition.
803    ///
804    /// # Notes
805    ///
806    /// - This is an advanced setting. A common reason to set it is to size or isolate the retry
807    ///   token bucket — for example, giving a high-throughput workload its own bucket. Otherwise
808    ///   most users won't need to modify it.
809    /// - A configured client rate limiter has no effect unless [`RetryConfig::adaptive`](::aws_smithy_types::retry::RetryConfig::adaptive) is used.
810    ///
811    /// # Examples
812    ///
813    /// Creating a custom retry partition with a token bucket:
814    /// ```no_run
815    /// use aws_sdk_bedrockruntime::config::Config;
816    /// use aws_sdk_bedrockruntime::config::retry::{RetryPartition, TokenBucket};
817    ///
818    /// let token_bucket = TokenBucket::new(10);
819    /// let config = Config::builder()
820    ///     .retry_partition(RetryPartition::custom("custom")
821    ///         .token_bucket(token_bucket)
822    ///         .build()
823    ///     )
824    ///     .build();
825    /// ```
826    ///
827    /// Sizing the retry token bucket (for example, for a high-throughput workload), or giving a
828    /// workload its own bucket:
829    /// ```no_run
830    /// use aws_sdk_bedrockruntime::config::Config;
831    /// use aws_sdk_bedrockruntime::config::retry::{RetryPartition, TokenBucket};
832    ///
833    /// let config = Config::builder()
834    ///     .retry_partition(
835    ///         RetryPartition::custom("high-throughput")
836    ///             .token_bucket(TokenBucket::builder().capacity(5000).build())
837    ///             .build(),
838    ///     )
839    ///     .build();
840    /// ```
841    ///
842    /// Configuring a client rate limiter with adaptive retry mode:
843    /// ```no_run
844    /// use aws_sdk_bedrockruntime::config::Config;
845    /// use aws_sdk_bedrockruntime::config::retry::{ClientRateLimiter, RetryConfig, RetryPartition};
846    ///
847    /// let client_rate_limiter = ClientRateLimiter::new(10.0);
848    /// let config = Config::builder()
849    ///     .retry_partition(RetryPartition::custom("custom")
850    ///         .client_rate_limiter(client_rate_limiter)
851    ///         .build()
852    ///     )
853    ///     .retry_config(RetryConfig::adaptive())
854    ///     .build();
855    /// ```
856    pub fn retry_partition(mut self, retry_partition: ::aws_smithy_runtime::client::retries::RetryPartition) -> Self {
857        self.set_retry_partition(Some(retry_partition));
858        self
859    }
860    /// Like [`Self::retry_partition`], but takes a mutable reference to the builder and an optional `RetryPartition`
861    pub fn set_retry_partition(
862        &mut self,
863        retry_partition: ::std::option::Option<::aws_smithy_runtime::client::retries::RetryPartition>,
864    ) -> &mut Self {
865        retry_partition.map(|r| self.config.store_put(r));
866        self
867    }
868    /// Set the identity cache for auth.
869    ///
870    /// The identity cache defaults to a lazy caching implementation that will resolve
871    /// an identity when it is requested, and place it in the cache thereafter. Subsequent
872    /// requests will take the value from the cache while it is still valid. Once it expires,
873    /// the next request will result in refreshing the identity.
874    ///
875    /// This configuration allows you to disable or change the default caching mechanism.
876    /// To use a custom caching mechanism, implement the [`ResolveCachedIdentity`](crate::config::ResolveCachedIdentity)
877    /// trait and pass that implementation into this function.
878    ///
879    /// # Examples
880    ///
881    /// Disabling identity caching:
882    /// ```no_run
883    /// use aws_sdk_bedrockruntime::config::IdentityCache;
884    ///
885    /// let config = aws_sdk_bedrockruntime::Config::builder()
886    ///     .identity_cache(IdentityCache::no_cache())
887    ///     // ...
888    ///     .build();
889    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
890    /// ```
891    ///
892    /// Customizing lazy caching:
893    /// ```no_run
894    /// use aws_sdk_bedrockruntime::config::IdentityCache;
895    /// use std::time::Duration;
896    ///
897    /// let config = aws_sdk_bedrockruntime::Config::builder()
898    ///     .identity_cache(
899    ///         IdentityCache::lazy()
900    ///             // change the load timeout to 10 seconds
901    ///             .load_timeout(Duration::from_secs(10))
902    ///             .build()
903    ///     )
904    ///     // ...
905    ///     .build();
906    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
907    /// ```
908    ///
909    pub fn identity_cache(mut self, identity_cache: impl crate::config::ResolveCachedIdentity + 'static) -> Self {
910        self.set_identity_cache(identity_cache);
911        self
912    }
913
914    /// Set the identity cache for auth.
915    ///
916    /// The identity cache defaults to a lazy caching implementation that will resolve
917    /// an identity when it is requested, and place it in the cache thereafter. Subsequent
918    /// requests will take the value from the cache while it is still valid. Once it expires,
919    /// the next request will result in refreshing the identity.
920    ///
921    /// This configuration allows you to disable or change the default caching mechanism.
922    /// To use a custom caching mechanism, implement the [`ResolveCachedIdentity`](crate::config::ResolveCachedIdentity)
923    /// trait and pass that implementation into this function.
924    ///
925    /// # Examples
926    ///
927    /// Disabling identity caching:
928    /// ```no_run
929    /// use aws_sdk_bedrockruntime::config::IdentityCache;
930    ///
931    /// let config = aws_sdk_bedrockruntime::Config::builder()
932    ///     .identity_cache(IdentityCache::no_cache())
933    ///     // ...
934    ///     .build();
935    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
936    /// ```
937    ///
938    /// Customizing lazy caching:
939    /// ```no_run
940    /// use aws_sdk_bedrockruntime::config::IdentityCache;
941    /// use std::time::Duration;
942    ///
943    /// let config = aws_sdk_bedrockruntime::Config::builder()
944    ///     .identity_cache(
945    ///         IdentityCache::lazy()
946    ///             // change the load timeout to 10 seconds
947    ///             .load_timeout(Duration::from_secs(10))
948    ///             .build()
949    ///     )
950    ///     // ...
951    ///     .build();
952    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
953    /// ```
954    ///
955    pub fn set_identity_cache(&mut self, identity_cache: impl crate::config::ResolveCachedIdentity + 'static) -> &mut Self {
956        self.runtime_components.set_identity_cache(::std::option::Option::Some(identity_cache));
957        self
958    }
959    /// Add an [interceptor](crate::config::Intercept) that runs at specific stages of the request execution pipeline.
960    ///
961    /// Interceptors targeted at a certain stage are executed according to the pre-defined priority.
962    /// The SDK provides a default set of interceptors. An interceptor configured by this method
963    /// will run after those default interceptors.
964    ///
965    /// # Examples
966    /// ```no_run
967    /// # fn example() {
968    /// use aws_smithy_runtime_api::box_error::BoxError;
969    /// use aws_smithy_runtime_api::client::interceptors::context::BeforeTransmitInterceptorContextMut;
970    /// use aws_smithy_runtime_api::client::interceptors::Intercept;
971    /// use aws_smithy_runtime_api::client::runtime_components::RuntimeComponents;
972    /// use aws_smithy_types::config_bag::ConfigBag;
973    /// use aws_sdk_bedrockruntime::config::Config;
974    /// use ::http::uri::Uri;
975    ///
976    /// fn base_url() -> String {
977    ///     // ...
978    ///     # String::new()
979    /// }
980    ///
981    /// #[derive(Debug)]
982    /// pub struct UriModifierInterceptor;
983    /// impl Intercept for UriModifierInterceptor {
984    ///     fn name(&self) -> &'static str {
985    ///         "UriModifierInterceptor"
986    ///     }
987    ///     fn modify_before_signing(
988    ///         &self,
989    ///         context: &mut BeforeTransmitInterceptorContextMut<'_>,
990    ///         _runtime_components: &RuntimeComponents,
991    ///         _cfg: &mut ConfigBag,
992    ///     ) -> Result<(), BoxError> {
993    ///         let request = context.request_mut();
994    ///         let uri = format!("{}{}", base_url(), request.uri());
995    ///         *request.uri_mut() = uri.parse::<Uri>()?.into();
996    ///
997    ///         Ok(())
998    ///     }
999    /// }
1000    ///
1001    /// let config = Config::builder()
1002    ///     .interceptor(UriModifierInterceptor)
1003    ///     .build();
1004    /// # }
1005    /// ```
1006    pub fn interceptor(mut self, interceptor: impl crate::config::Intercept + 'static) -> Self {
1007        self.push_interceptor(crate::config::SharedInterceptor::new(interceptor));
1008        self
1009    }
1010
1011    /// Like [`Self::interceptor`], but takes a [`SharedInterceptor`](crate::config::SharedInterceptor).
1012    pub fn push_interceptor(&mut self, interceptor: crate::config::SharedInterceptor) -> &mut Self {
1013        self.runtime_components.push_interceptor(interceptor);
1014        self
1015    }
1016
1017    /// Set [`SharedInterceptor`](crate::config::SharedInterceptor)s for the builder.
1018    pub fn set_interceptors(&mut self, interceptors: impl IntoIterator<Item = crate::config::SharedInterceptor>) -> &mut Self {
1019        self.runtime_components.set_interceptors(interceptors.into_iter());
1020        self
1021    }
1022    /// Sets the time source used for this service
1023    pub fn time_source(mut self, time_source: impl ::aws_smithy_async::time::TimeSource + 'static) -> Self {
1024        self.set_time_source(::std::option::Option::Some(::aws_smithy_runtime_api::shared::IntoShared::into_shared(
1025            time_source,
1026        )));
1027        self
1028    }
1029    /// Sets the time source used for this service
1030    pub fn set_time_source(&mut self, time_source: ::std::option::Option<::aws_smithy_async::time::SharedTimeSource>) -> &mut Self {
1031        self.runtime_components.set_time_source(time_source);
1032        self
1033    }
1034    /// Add type implementing [`ClassifyRetry`](::aws_smithy_runtime_api::client::retries::classifiers::ClassifyRetry) that will be used by the
1035    /// [`RetryStrategy`](::aws_smithy_runtime_api::client::retries::RetryStrategy) to determine what responses should be retried.
1036    ///
1037    /// A retry classifier configured by this method will run according to its [priority](::aws_smithy_runtime_api::client::retries::classifiers::RetryClassifierPriority).
1038    ///
1039    /// # Examples
1040    /// ```no_run
1041    /// # fn example() {
1042    /// use aws_smithy_runtime_api::client::interceptors::context::InterceptorContext;
1043    /// use aws_smithy_runtime_api::client::orchestrator::OrchestratorError;
1044    /// use aws_smithy_runtime_api::client::retries::classifiers::{
1045    ///     ClassifyRetry, RetryAction, RetryClassifierPriority,
1046    /// };
1047    /// use aws_smithy_types::error::metadata::ProvideErrorMetadata;
1048    /// use aws_smithy_types::retry::ErrorKind;
1049    /// use std::error::Error as StdError;
1050    /// use std::marker::PhantomData;
1051    /// use std::fmt;
1052    /// use aws_sdk_bedrockruntime::config::Config;
1053    /// # #[derive(Debug)]
1054    /// # struct SomeOperationError {}
1055    /// # impl StdError for SomeOperationError {}
1056    /// # impl fmt::Display for SomeOperationError {
1057    /// #    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { todo!() }
1058    /// # }
1059    /// # impl ProvideErrorMetadata for SomeOperationError {
1060    /// #    fn meta(&self) -> &aws_sdk_bedrockruntime::error::ErrorMetadata { todo!() }
1061    /// # }
1062    ///
1063    /// const RETRYABLE_ERROR_CODES: &[&str] = &[
1064    ///     // List error codes to be retried here...
1065    /// ];
1066    ///
1067    /// // When classifying at an operation's error type, classifiers require a generic parameter.
1068    /// // When classifying the HTTP response alone, no generic is needed.
1069    /// #[derive(Debug, Default)]
1070    /// pub struct ExampleErrorCodeClassifier<E> {
1071    ///     _inner: PhantomData<E>,
1072    /// }
1073    ///
1074    /// impl<E> ExampleErrorCodeClassifier<E> {
1075    ///     pub fn new() -> Self {
1076    ///         Self {
1077    ///             _inner: PhantomData,
1078    ///         }
1079    ///     }
1080    /// }
1081    ///
1082    /// impl<E> ClassifyRetry for ExampleErrorCodeClassifier<E>
1083    /// where
1084    ///     // Adding a trait bound for ProvideErrorMetadata allows us to inspect the error code.
1085    ///     E: StdError + ProvideErrorMetadata + Send + Sync + 'static,
1086    /// {
1087    ///     fn classify_retry(&self, ctx: &InterceptorContext) -> RetryAction {
1088    ///         // Check for a result
1089    ///         let output_or_error = ctx.output_or_error();
1090    ///         // Check for an error
1091    ///         let error = match output_or_error {
1092    ///             Some(Ok(_)) | None => return RetryAction::NoActionIndicated,
1093    ///               Some(Err(err)) => err,
1094    ///         };
1095    ///
1096    ///         // Downcast the generic error and extract the code
1097    ///         let error_code = OrchestratorError::as_operation_error(error)
1098    ///             .and_then(|err| err.downcast_ref::<E>())
1099    ///             .and_then(|err| err.code());
1100    ///
1101    ///         // If this error's code is in our list, return an action that tells the RetryStrategy to retry this request.
1102    ///         if let Some(error_code) = error_code {
1103    ///             if RETRYABLE_ERROR_CODES.contains(&error_code) {
1104    ///                 return RetryAction::transient_error();
1105    ///             }
1106    ///         }
1107    ///
1108    ///         // Otherwise, return that no action is indicated i.e. that this classifier doesn't require a retry.
1109    ///         // Another classifier may still classify this response as retryable.
1110    ///         RetryAction::NoActionIndicated
1111    ///     }
1112    ///
1113    ///     fn name(&self) -> &'static str { "Example Error Code Classifier" }
1114    /// }
1115    ///
1116    /// let config = Config::builder()
1117    ///     .retry_classifier(ExampleErrorCodeClassifier::<SomeOperationError>::new())
1118    ///     .build();
1119    /// # }
1120    /// ```
1121    pub fn retry_classifier(
1122        mut self,
1123        retry_classifier: impl ::aws_smithy_runtime_api::client::retries::classifiers::ClassifyRetry + 'static,
1124    ) -> Self {
1125        self.push_retry_classifier(::aws_smithy_runtime_api::client::retries::classifiers::SharedRetryClassifier::new(
1126            retry_classifier,
1127        ));
1128        self
1129    }
1130
1131    /// Like [`Self::retry_classifier`], but takes a [`SharedRetryClassifier`](::aws_smithy_runtime_api::client::retries::classifiers::SharedRetryClassifier).
1132    pub fn push_retry_classifier(
1133        &mut self,
1134        retry_classifier: ::aws_smithy_runtime_api::client::retries::classifiers::SharedRetryClassifier,
1135    ) -> &mut Self {
1136        self.runtime_components.push_retry_classifier(retry_classifier);
1137        self
1138    }
1139
1140    /// Set [`SharedRetryClassifier`](::aws_smithy_runtime_api::client::retries::classifiers::SharedRetryClassifier)s for the builder, replacing any that
1141    /// were previously set.
1142    pub fn set_retry_classifiers(
1143        &mut self,
1144        retry_classifiers: impl IntoIterator<Item = ::aws_smithy_runtime_api::client::retries::classifiers::SharedRetryClassifier>,
1145    ) -> &mut Self {
1146        self.runtime_components.set_retry_classifiers(retry_classifiers.into_iter());
1147        self
1148    }
1149    /// Sets the name of the app that is using the client.
1150    ///
1151    /// This _optional_ name is used to identify the application in the user agent that
1152    /// gets sent along with requests.
1153    pub fn app_name(mut self, app_name: ::aws_types::app_name::AppName) -> Self {
1154        self.set_app_name(Some(app_name));
1155        self
1156    }
1157    /// Sets the name of the app that is using the client.
1158    ///
1159    /// This _optional_ name is used to identify the application in the user agent that
1160    /// gets sent along with requests.
1161    pub fn set_app_name(&mut self, app_name: ::std::option::Option<::aws_types::app_name::AppName>) -> &mut Self {
1162        self.config.store_or_unset(app_name);
1163        self
1164    }
1165    /// Appends framework metadata to the user agent.
1166    ///
1167    /// This _optional_ metadata identifies a software framework or third-party library
1168    /// that is being used with the client. It is rendered into the user agent string
1169    /// (as `lib/{name}/{version}`) so that libraries built on top of the AWS SDK can
1170    /// self-identify in the requests they make. Multiple entries may be added; each call
1171    /// appends another entry rather than replacing previous ones.
1172    ///
1173    /// Entries are de-duplicated on `(name, version)`, rendered in first-seen order, and
1174    /// the total number of unique entries included in the user agent is capped (currently
1175    /// at 10); additional entries beyond the cap are dropped with a warning.
1176    pub fn framework_metadata(mut self, framework_metadata: ::aws_types::sdk_ua_metadata::FrameworkMetadata) -> Self {
1177        self.push_framework_metadata(framework_metadata);
1178        self
1179    }
1180    /// Appends framework metadata to the user agent.
1181    ///
1182    /// This _optional_ metadata identifies a software framework or third-party library
1183    /// that is being used with the client. It is rendered into the user agent string
1184    /// (as `lib/{name}/{version}`) so that libraries built on top of the AWS SDK can
1185    /// self-identify in the requests they make. Multiple entries may be added; each call
1186    /// appends another entry rather than replacing previous ones.
1187    pub fn push_framework_metadata(&mut self, framework_metadata: ::aws_types::sdk_ua_metadata::FrameworkMetadata) -> &mut Self {
1188        self.config.store_append(framework_metadata);
1189        self
1190    }
1191    /// Sets the access token provider for this service
1192    ///
1193    /// Note: the [`Self::bearer_token`] and [`Self::bearer_token_resolver`] methods are
1194    /// equivalent to this method, but take the [`Token`] and [`ResolveIdentity`] types
1195    /// respectively.
1196    ///
1197    /// [`Token`]: crate::config::Token
1198    /// [`ResolveIdentity`]: crate::config::ResolveIdentity
1199    pub fn token_provider(mut self, token_provider: impl crate::config::ProvideToken + 'static) -> Self {
1200        self.set_token_provider(::std::option::Option::Some(::aws_smithy_runtime_api::shared::IntoShared::<
1201            crate::config::SharedTokenProvider,
1202        >::into_shared(token_provider)));
1203        self
1204    }
1205
1206    /// Sets the access token provider for this service
1207    ///
1208    /// Note: the [`Self::bearer_token`] and [`Self::bearer_token_resolver`] methods are
1209    /// equivalent to this method, but take the [`Token`] and [`ResolveIdentity`] types
1210    /// respectively.
1211    ///
1212    /// [`Token`]: crate::config::Token
1213    /// [`ResolveIdentity`]: crate::config::ResolveIdentity
1214    pub fn set_token_provider(&mut self, token_provider: ::std::option::Option<crate::config::SharedTokenProvider>) -> &mut Self {
1215        if let Some(token_provider) = token_provider {
1216            self.runtime_components
1217                .set_identity_resolver(::aws_smithy_runtime_api::client::auth::http::HTTP_BEARER_AUTH_SCHEME_ID, token_provider);
1218        }
1219        self
1220    }
1221    /// Overrides the default invocation ID generator.
1222    ///
1223    /// The invocation ID generator generates ID values for the `amz-sdk-invocation-id` header. By default, this will be a random UUID. Overriding it may be useful in tests that examine the HTTP request and need to be deterministic.
1224    pub fn invocation_id_generator(mut self, gen: impl ::aws_runtime::invocation_id::InvocationIdGenerator + 'static) -> Self {
1225        self.set_invocation_id_generator(::std::option::Option::Some(
1226            ::aws_runtime::invocation_id::SharedInvocationIdGenerator::new(gen),
1227        ));
1228        self
1229    }
1230    /// Overrides the default invocation ID generator.
1231    ///
1232    /// The invocation ID generator generates ID values for the `amz-sdk-invocation-id` header. By default, this will be a random UUID. Overriding it may be useful in tests that examine the HTTP request and need to be deterministic.
1233    pub fn set_invocation_id_generator(
1234        &mut self,
1235        gen: ::std::option::Option<::aws_runtime::invocation_id::SharedInvocationIdGenerator>,
1236    ) -> &mut Self {
1237        self.config.store_or_unset(gen);
1238        self
1239    }
1240    /// Sets the endpoint URL used to communicate with this service.
1241    ///
1242    /// Note: this is used in combination with other endpoint rules, e.g. an API that applies a host-label prefix
1243    /// will be prefixed onto this URL. To fully override the endpoint resolver, use
1244    /// [`Builder::endpoint_resolver`].
1245    pub fn endpoint_url(mut self, endpoint_url: impl Into<::std::string::String>) -> Self {
1246        self.set_endpoint_url(Some(endpoint_url.into()));
1247        self
1248    }
1249    /// Sets the endpoint URL used to communicate with this service.
1250    ///
1251    /// Note: this is used in combination with other endpoint rules, e.g. an API that applies a host-label prefix
1252    /// will be prefixed onto this URL. To fully override the endpoint resolver, use
1253    /// [`Builder::endpoint_resolver`].
1254    pub fn set_endpoint_url(&mut self, endpoint_url: Option<::std::string::String>) -> &mut Self {
1255        self.config.store_or_unset(endpoint_url.map(::aws_types::endpoint_config::EndpointUrl));
1256        self
1257    }
1258    /// When true, use the dual-stack endpoint. If the configured endpoint does not support dual-stack, dispatching the request MAY return an error.
1259    pub fn use_dual_stack(mut self, use_dual_stack: impl Into<bool>) -> Self {
1260        self.set_use_dual_stack(Some(use_dual_stack.into()));
1261        self
1262    }
1263    /// When true, use the dual-stack endpoint. If the configured endpoint does not support dual-stack, dispatching the request MAY return an error.
1264    pub fn set_use_dual_stack(&mut self, use_dual_stack: Option<bool>) -> &mut Self {
1265        self.config.store_or_unset(use_dual_stack.map(::aws_types::endpoint_config::UseDualStack));
1266        self
1267    }
1268    /// When true, send this request to the FIPS-compliant regional endpoint. If the configured endpoint does not have a FIPS compliant endpoint, dispatching the request will return an error.
1269    pub fn use_fips(mut self, use_fips: impl Into<bool>) -> Self {
1270        self.set_use_fips(Some(use_fips.into()));
1271        self
1272    }
1273    /// When true, send this request to the FIPS-compliant regional endpoint. If the configured endpoint does not have a FIPS compliant endpoint, dispatching the request will return an error.
1274    pub fn set_use_fips(&mut self, use_fips: Option<bool>) -> &mut Self {
1275        self.config.store_or_unset(use_fips.map(::aws_types::endpoint_config::UseFips));
1276        self
1277    }
1278    /// Sets the AWS region to use when making requests.
1279    ///
1280    /// # Examples
1281    /// ```no_run
1282    /// use aws_types::region::Region;
1283    /// use aws_sdk_bedrockruntime::config::{Builder, Config};
1284    ///
1285    /// let config = aws_sdk_bedrockruntime::Config::builder()
1286    ///     .region(Region::new("us-east-1"))
1287    ///     .build();
1288    /// ```
1289    pub fn region(mut self, region: impl ::std::convert::Into<::std::option::Option<crate::config::Region>>) -> Self {
1290        self.set_region(region.into());
1291        self
1292    }
1293    /// Sets the AWS region to use when making requests.
1294    pub fn set_region(&mut self, region: ::std::option::Option<crate::config::Region>) -> &mut Self {
1295        self.config.store_or_unset(region);
1296        self
1297    }
1298    /// Sets the credentials provider for this service
1299    pub fn credentials_provider(mut self, credentials_provider: impl crate::config::ProvideCredentials + 'static) -> Self {
1300        self.set_credentials_provider(::std::option::Option::Some(crate::config::SharedCredentialsProvider::new(
1301            credentials_provider,
1302        )));
1303        self
1304    }
1305    /// Sets the credentials provider for this service
1306    pub fn set_credentials_provider(&mut self, credentials_provider: ::std::option::Option<crate::config::SharedCredentialsProvider>) -> &mut Self {
1307        if let Some(credentials_provider) = credentials_provider {
1308            self.runtime_components
1309                .set_identity_resolver(::aws_runtime::auth::sigv4::SCHEME_ID, credentials_provider);
1310        }
1311        self
1312    }
1313    /// Sets the [`behavior major version`](crate::config::BehaviorVersion).
1314    ///
1315    /// Over time, new best-practice behaviors are introduced. However, these behaviors might not be backwards
1316    /// compatible. For example, a change which introduces new default timeouts or a new retry-mode for
1317    /// all operations might be the ideal behavior but could break existing applications.
1318    ///
1319    /// # Examples
1320    ///
1321    /// Set the behavior major version to `latest`. This is equivalent to enabling the `behavior-version-latest` cargo feature.
1322    /// ```no_run
1323    /// use aws_sdk_bedrockruntime::config::BehaviorVersion;
1324    ///
1325    /// let config = aws_sdk_bedrockruntime::Config::builder()
1326    ///     .behavior_version(BehaviorVersion::latest())
1327    ///     // ...
1328    ///     .build();
1329    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
1330    /// ```
1331    ///
1332    /// Customizing behavior major version:
1333    /// ```no_run
1334    /// use aws_sdk_bedrockruntime::config::BehaviorVersion;
1335    ///
1336    /// let config = aws_sdk_bedrockruntime::Config::builder()
1337    ///     .behavior_version(BehaviorVersion::v2023_11_09())
1338    ///     // ...
1339    ///     .build();
1340    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
1341    /// ```
1342    ///
1343    pub fn behavior_version(mut self, behavior_version: crate::config::BehaviorVersion) -> Self {
1344        self.set_behavior_version(Some(behavior_version));
1345        self
1346    }
1347
1348    /// Sets the [`behavior major version`](crate::config::BehaviorVersion).
1349    ///
1350    /// Over time, new best-practice behaviors are introduced. However, these behaviors might not be backwards
1351    /// compatible. For example, a change which introduces new default timeouts or a new retry-mode for
1352    /// all operations might be the ideal behavior but could break existing applications.
1353    ///
1354    /// # Examples
1355    ///
1356    /// Set the behavior major version to `latest`. This is equivalent to enabling the `behavior-version-latest` cargo feature.
1357    /// ```no_run
1358    /// use aws_sdk_bedrockruntime::config::BehaviorVersion;
1359    ///
1360    /// let config = aws_sdk_bedrockruntime::Config::builder()
1361    ///     .behavior_version(BehaviorVersion::latest())
1362    ///     // ...
1363    ///     .build();
1364    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
1365    /// ```
1366    ///
1367    /// Customizing behavior major version:
1368    /// ```no_run
1369    /// use aws_sdk_bedrockruntime::config::BehaviorVersion;
1370    ///
1371    /// let config = aws_sdk_bedrockruntime::Config::builder()
1372    ///     .behavior_version(BehaviorVersion::v2023_11_09())
1373    ///     // ...
1374    ///     .build();
1375    /// let client = aws_sdk_bedrockruntime::Client::from_conf(config);
1376    /// ```
1377    ///
1378    pub fn set_behavior_version(&mut self, behavior_version: Option<crate::config::BehaviorVersion>) -> &mut Self {
1379        self.behavior_version = behavior_version;
1380        self
1381    }
1382
1383    /// Convenience method to set the latest behavior major version
1384    ///
1385    /// This is equivalent to enabling the `behavior-version-latest` Cargo feature
1386    pub fn behavior_version_latest(mut self) -> Self {
1387        self.set_behavior_version(Some(crate::config::BehaviorVersion::latest()));
1388        self
1389    }
1390    /// Adds a runtime plugin to the config.
1391    #[allow(unused)]
1392    pub(crate) fn runtime_plugin(mut self, plugin: impl crate::config::RuntimePlugin + 'static) -> Self {
1393        self.push_runtime_plugin(crate::config::SharedRuntimePlugin::new(plugin));
1394        self
1395    }
1396    /// Adds a runtime plugin to the config.
1397    #[allow(unused)]
1398    pub(crate) fn push_runtime_plugin(&mut self, plugin: crate::config::SharedRuntimePlugin) -> &mut Self {
1399        self.runtime_plugins.push(plugin);
1400        self
1401    }
1402    #[cfg(any(feature = "test-util", test))]
1403    #[allow(unused_mut)]
1404    /// Apply test defaults to the builder. NOTE: Consider migrating to use `apply_test_defaults_v2` instead.
1405    pub fn apply_test_defaults(&mut self) -> &mut Self {
1406        self.set_idempotency_token_provider(Some("00000000-0000-4000-8000-000000000000".into()));
1407        self.set_time_source(::std::option::Option::Some(::aws_smithy_async::time::SharedTimeSource::new(
1408            ::aws_smithy_async::time::StaticTimeSource::new(::std::time::UNIX_EPOCH + ::std::time::Duration::from_secs(1234567890)),
1409        )));
1410        self.config.store_put(::aws_runtime::user_agent::AwsUserAgent::for_tests());
1411        self.set_token_provider(Some(crate::config::SharedTokenProvider::new(::aws_credential_types::Token::for_tests())));
1412        self.set_credentials_provider(Some(crate::config::SharedCredentialsProvider::new(
1413            ::aws_credential_types::Credentials::for_tests(),
1414        )));
1415        self.behavior_version = ::std::option::Option::Some(crate::config::BehaviorVersion::latest());
1416        self
1417    }
1418    #[cfg(any(feature = "test-util", test))]
1419    #[allow(unused_mut)]
1420    /// Apply test defaults to the builder. NOTE: Consider migrating to use `with_test_defaults_v2` instead.
1421    pub fn with_test_defaults(mut self) -> Self {
1422        self.apply_test_defaults();
1423        self
1424    }
1425    #[cfg(any(feature = "test-util", test))]
1426    #[allow(unused_mut)]
1427    /// Apply test defaults to the builder. V2 of this function sets additional test defaults such as region configuration (if applicable).
1428    pub fn apply_test_defaults_v2(&mut self) -> &mut Self {
1429        self.apply_test_defaults();
1430        if self.config.load::<crate::config::Region>().is_none() {
1431            self.set_region(::std::option::Option::Some(crate::config::Region::new("us-east-1")));
1432        }
1433        self
1434    }
1435    #[cfg(any(feature = "test-util", test))]
1436    #[allow(unused_mut)]
1437    /// Apply test defaults to the builder. V2 of this function sets additional test defaults such as region configuration (if applicable).
1438    pub fn with_test_defaults_v2(mut self) -> Self {
1439        self.apply_test_defaults_v2();
1440        self
1441    }
1442    /// Builds a [`Config`].
1443    #[allow(unused_mut)]
1444    pub fn build(mut self) -> Config {
1445        let mut layer = self.config;
1446        if self.runtime_components.time_source().is_none() {
1447            self.runtime_components
1448                .set_time_source(::std::option::Option::Some(::std::default::Default::default()));
1449        }
1450        layer.store_put(crate::meta::API_METADATA.clone());
1451        layer.store_put(::aws_types::SigningName::from_static("bedrock"));
1452        layer
1453            .load::<::aws_types::region::Region>()
1454            .cloned()
1455            .map(|r| layer.store_put(::aws_types::region::SigningRegion::from(r)));
1456        Config {
1457            config: crate::config::Layer::from(layer.clone())
1458                .with_name("aws_sdk_bedrockruntime::config::Config")
1459                .freeze(),
1460            cloneable: layer,
1461            runtime_components: self.runtime_components,
1462            runtime_plugins: self.runtime_plugins,
1463            behavior_version: self.behavior_version,
1464        }
1465    }
1466}
1467#[derive(::std::fmt::Debug)]
1468pub(crate) struct ServiceRuntimePlugin {
1469    config: ::std::option::Option<::aws_smithy_types::config_bag::FrozenLayer>,
1470    runtime_components: ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder,
1471}
1472
1473impl ServiceRuntimePlugin {
1474    pub fn new(_service_config: crate::config::Config) -> Self {
1475        let config = {
1476            let mut cfg = ::aws_smithy_types::config_bag::Layer::new("AmazonBedrockFrontendService");
1477            cfg.store_put(crate::idempotency_token::default_provider());
1478            cfg.store_put(::aws_smithy_runtime::client::orchestrator::AuthSchemeAndEndpointOrchestrationV2);
1479            ::std::option::Option::Some(cfg.freeze())
1480        };
1481        let mut runtime_components = ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder::new("ServiceRuntimePlugin");
1482        runtime_components.push_auth_scheme(::aws_smithy_runtime_api::client::auth::SharedAuthScheme::new(
1483            ::aws_smithy_runtime::client::auth::http::BearerAuthScheme::new(),
1484        ));
1485        runtime_components.set_auth_scheme_option_resolver(::std::option::Option::Some({
1486            use crate::config::auth::ResolveAuthScheme;
1487            crate::config::auth::DefaultAuthSchemeResolver::default().into_shared_resolver()
1488        }));
1489        runtime_components.set_endpoint_resolver(::std::option::Option::Some({
1490            use crate::config::endpoint::ResolveEndpoint;
1491            crate::config::endpoint::DefaultResolver::new().into_shared_resolver()
1492        }));
1493        runtime_components.push_interceptor(::aws_smithy_runtime_api::client::interceptors::SharedInterceptor::permanent(
1494            ::aws_smithy_runtime::client::http::connection_poisoning::ConnectionPoisoningInterceptor::new(),
1495        ));
1496        runtime_components.push_retry_classifier(::aws_smithy_runtime::client::retries::classifiers::HttpStatusCodeClassifier::default());
1497        runtime_components.push_interceptor(::aws_smithy_runtime_api::client::interceptors::SharedInterceptor::permanent(
1498            crate::sdk_feature_tracker::retry_mode::RetryModeFeatureTrackerInterceptor::new(),
1499        ));
1500        runtime_components.push_interceptor(::aws_smithy_runtime_api::client::interceptors::SharedInterceptor::permanent(
1501            ::aws_runtime::service_clock_skew::ServiceClockSkewInterceptor::new(),
1502        ));
1503        runtime_components.push_interceptor(::aws_runtime::request_info::RequestInfoInterceptor::new());
1504        runtime_components.push_interceptor(::aws_runtime::user_agent::UserAgentInterceptor::new());
1505        runtime_components.push_interceptor(::aws_runtime::invocation_id::InvocationIdInterceptor::new());
1506        runtime_components.push_interceptor(::aws_smithy_runtime_api::client::interceptors::SharedInterceptor::permanent(
1507            ::aws_runtime::recursion_detection::RecursionDetectionInterceptor::new(),
1508        ));
1509        runtime_components.push_auth_scheme(::aws_smithy_runtime_api::client::auth::SharedAuthScheme::new(
1510            ::aws_runtime::auth::sigv4::SigV4AuthScheme::new(),
1511        ));
1512        runtime_components.push_interceptor(::aws_smithy_runtime_api::client::interceptors::SharedInterceptor::permanent(
1513            crate::config::endpoint::EndpointOverrideFeatureTrackerInterceptor,
1514        ));
1515        runtime_components.push_interceptor(::aws_smithy_runtime_api::client::interceptors::SharedInterceptor::permanent(
1516            crate::observability_feature::ObservabilityFeatureTrackerInterceptor,
1517        ));
1518        Self { config, runtime_components }
1519    }
1520}
1521
1522impl ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin for ServiceRuntimePlugin {
1523    fn config(&self) -> ::std::option::Option<::aws_smithy_types::config_bag::FrozenLayer> {
1524        self.config.clone()
1525    }
1526
1527    fn order(&self) -> ::aws_smithy_runtime_api::client::runtime_plugin::Order {
1528        ::aws_smithy_runtime_api::client::runtime_plugin::Order::Defaults
1529    }
1530
1531    fn runtime_components(
1532        &self,
1533        _: &::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder,
1534    ) -> ::std::borrow::Cow<'_, ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder> {
1535        ::std::borrow::Cow::Borrowed(&self.runtime_components)
1536    }
1537}
1538
1539// Cross-operation shared-state singletons
1540
1541/// A plugin that enables configuration for a single operation invocation
1542///
1543/// The `config` method will return a `FrozenLayer` by storing values from `config_override`.
1544/// In the case of default values requested, they will be obtained from `client_config`.
1545#[derive(Debug)]
1546pub(crate) struct ConfigOverrideRuntimePlugin {
1547    pub(crate) config: ::aws_smithy_types::config_bag::FrozenLayer,
1548    pub(crate) components: ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder,
1549}
1550
1551impl ConfigOverrideRuntimePlugin {
1552    #[allow(dead_code)] // unused when a service does not provide any operations
1553    pub(crate) fn new(
1554        config_override: Builder,
1555        initial_config: ::aws_smithy_types::config_bag::FrozenLayer,
1556        initial_components: &::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder,
1557    ) -> Self {
1558        let mut layer = config_override.config;
1559        let mut components = config_override.runtime_components;
1560        #[allow(unused_mut)]
1561        let mut resolver =
1562            ::aws_smithy_runtime::client::config_override::Resolver::overrid(initial_config, initial_components, &mut layer, &mut components);
1563
1564        resolver
1565            .config_mut()
1566            .load::<::aws_types::region::Region>()
1567            .cloned()
1568            .map(|r| resolver.config_mut().store_put(::aws_types::region::SigningRegion::from(r)));
1569
1570        let _ = resolver;
1571
1572        // When the config override supplies an identity resolver for any auth scheme
1573        // known to the client or the override itself, we give this operation its own
1574        // short-lived identity cache so that new partitions don't accumulate in the
1575        // shared client cache. A lazy cache (not `no_cache`) is used so that resolved
1576        // identities are served from the short-lived identity cache on retries.
1577        //
1578        // This is skipped if the override already sets its own identity cache.
1579        if components.has_identity_resolvers() && components.identity_cache().is_none() {
1580            components.set_identity_cache(::std::option::Option::Some(
1581                ::aws_smithy_runtime::client::identity::IdentityCache::lazy().max_partitions(1).build(),
1582            ));
1583        }
1584
1585        Self {
1586            config: ::aws_smithy_types::config_bag::Layer::from(layer)
1587                .with_name("aws_sdk_bedrockruntime::config::ConfigOverrideRuntimePlugin")
1588                .freeze(),
1589            components,
1590        }
1591    }
1592}
1593
1594impl ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin for ConfigOverrideRuntimePlugin {
1595    fn config(&self) -> ::std::option::Option<::aws_smithy_types::config_bag::FrozenLayer> {
1596        Some(self.config.clone())
1597    }
1598
1599    fn runtime_components(
1600        &self,
1601        _: &::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder,
1602    ) -> ::std::borrow::Cow<'_, ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder> {
1603        ::std::borrow::Cow::Borrowed(&self.components)
1604    }
1605}
1606
1607pub use ::aws_smithy_runtime::client::identity::IdentityCache;
1608pub use ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponents;
1609pub use ::aws_smithy_types::config_bag::ConfigBag;
1610
1611pub use ::aws_credential_types::Credentials;
1612
1613impl From<&::aws_types::sdk_config::SdkConfig> for Builder {
1614    fn from(input: &::aws_types::sdk_config::SdkConfig) -> Self {
1615        let mut builder = Builder::default();
1616        builder.set_credentials_provider(input.credentials_provider());
1617        builder = builder.region(input.region().cloned());
1618        builder.set_use_fips(input.use_fips());
1619        builder.set_use_dual_stack(input.use_dual_stack());
1620        if input.get_origin("endpoint_url").is_client_config() {
1621            builder.set_endpoint_url(input.endpoint_url().map(|s| s.to_string()));
1622        } else {
1623            builder.set_endpoint_url(
1624                input
1625                    .service_config()
1626                    .and_then(|conf| {
1627                        conf.load_config(service_config_key("Bedrock Runtime", "AWS_ENDPOINT_URL", "endpoint_url"))
1628                            .map(|it| it.parse().unwrap())
1629                    })
1630                    .or_else(|| input.endpoint_url().map(|s| s.to_string())),
1631            );
1632        }
1633        // resiliency
1634        builder.set_retry_config(input.retry_config().cloned());
1635        builder.set_timeout_config(input.timeout_config().cloned());
1636        builder.set_sleep_impl(input.sleep_impl());
1637
1638        builder.set_http_client(input.http_client());
1639        builder.set_time_source(input.time_source());
1640        builder.set_behavior_version(input.behavior_version());
1641        builder.set_auth_scheme_preference(input.auth_scheme_preference().cloned());
1642        // setting `None` here removes the default
1643        if let Some(config) = input.stalled_stream_protection() {
1644            builder.set_stalled_stream_protection(Some(config));
1645        }
1646
1647        if let Some(cache) = input.identity_cache() {
1648            builder.set_identity_cache(cache);
1649        }
1650        builder.set_token_provider(input.token_provider());
1651        if let ::std::option::Option::Some(val) = input.service_config().and_then(|conf| {
1652            // Passing an empty string for the last argument of `service_config_key`,
1653            // since shared config/profile for environment token provider is not supported.
1654            ::aws_types::service_config::LoadServiceConfig::load_config(conf, service_config_key("bedrock", "AWS_BEARER_TOKEN", ""))
1655                .and_then(|it| it.parse::<::std::string::String>().ok())
1656        }) {
1657            if !input.get_origin("auth_scheme_preference").is_client_config() {
1658                builder.set_auth_scheme_preference(::std::option::Option::Some(
1659                    [::aws_smithy_runtime_api::client::auth::http::HTTP_BEARER_AUTH_SCHEME_ID].into(),
1660                ));
1661            }
1662            if !input.get_origin("token_provider").is_client_config() {
1663                let mut layer = ::aws_smithy_types::config_bag::Layer::new("AwsBearerTokenBedrock");
1664                layer.store_append(::aws_credential_types::credential_feature::AwsCredentialFeature::BearerServiceEnvVars);
1665                let identity = ::aws_smithy_runtime_api::client::identity::Identity::builder()
1666                    .data(crate::config::Token::new(val, ::std::option::Option::None))
1667                    .property(layer.freeze())
1668                    .build()
1669                    .expect("required fields set");
1670                builder
1671                    .runtime_components
1672                    .set_identity_resolver(::aws_smithy_runtime_api::client::auth::http::HTTP_BEARER_AUTH_SCHEME_ID, identity);
1673            }
1674        }
1675        builder.set_app_name(input.app_name().cloned());
1676        for framework_metadata in input.framework_metadata() {
1677            builder.push_framework_metadata(framework_metadata.clone());
1678        }
1679
1680        builder
1681    }
1682}
1683
1684impl From<&::aws_types::sdk_config::SdkConfig> for Config {
1685    fn from(sdk_config: &::aws_types::sdk_config::SdkConfig) -> Self {
1686        Builder::from(sdk_config).build()
1687    }
1688}
1689
1690pub use ::aws_types::app_name::AppName;
1691pub use ::aws_types::sdk_ua_metadata::FrameworkMetadata;
1692
1693#[allow(dead_code)]
1694fn service_config_key<'a>(service_id: &'a str, env: &'a str, profile: &'a str) -> aws_types::service_config::ServiceConfigKey<'a> {
1695    ::aws_types::service_config::ServiceConfigKey::builder()
1696        .service_id(service_id)
1697        .env(env)
1698        .profile(profile)
1699        .build()
1700        .expect("all field sets explicitly, can't fail")
1701}
1702
1703pub use ::aws_smithy_async::rt::sleep::Sleep;
1704
1705pub(crate) fn base_client_runtime_plugins(mut config: crate::Config) -> ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins {
1706    let mut configured_plugins = ::std::vec::Vec::new();
1707    ::std::mem::swap(&mut config.runtime_plugins, &mut configured_plugins);
1708    #[cfg(feature = "behavior-version-latest")]
1709    {
1710        if config.behavior_version.is_none() {
1711            config.behavior_version = Some(::aws_smithy_runtime_api::client::behavior_version::BehaviorVersion::latest());
1712        }
1713    }
1714
1715    let default_retry_partition = "bedrockruntime";
1716    let default_retry_partition = match config.region() {
1717        Some(region) => ::std::borrow::Cow::from(format!("{default_retry_partition}-{region}")),
1718        None => ::std::borrow::Cow::from(default_retry_partition),
1719    };
1720
1721    let scope = "aws-sdk-bedrockruntime";
1722
1723    #[allow(deprecated)]
1724                    let mut plugins = ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins::new()
1725                        // defaults
1726                        .with_client_plugins(::aws_smithy_runtime::client::defaults::default_plugins(
1727                            ::aws_smithy_runtime::client::defaults::DefaultPluginParams::new()
1728                                .with_retry_partition_name(default_retry_partition)
1729                                .with_behavior_version(config.behavior_version.expect("Invalid client configuration: A behavior major version must be set when sending a request or constructing a client. You must set it during client construction or by enabling the `behavior-version-latest` cargo feature."))
1730                                .with_is_aws_sdk(true)
1731                        ))
1732                        // user config
1733                        .with_client_plugin(
1734                            ::aws_smithy_runtime_api::client::runtime_plugin::StaticRuntimePlugin::new()
1735                                .with_config(config.config.clone())
1736                                .with_runtime_components(config.runtime_components.clone())
1737                        )
1738                        // codegen config
1739                        .with_client_plugin(crate::config::ServiceRuntimePlugin::new(config.clone()))
1740                        .with_client_plugin(::aws_smithy_runtime::client::auth::no_auth::NoAuthRuntimePlugin::new())
1741                        .with_client_plugin(
1742                            ::aws_smithy_runtime::client::metrics::MetricsRuntimePlugin::builder()
1743                                .with_scope(scope)
1744                                .with_time_source(config.runtime_components.time_source().unwrap_or_default())
1745                                .build()
1746                                .expect("All required fields have been set")
1747                        );
1748
1749    for plugin in configured_plugins {
1750        plugins = plugins.with_client_plugin(plugin);
1751    }
1752    plugins
1753}
1754
1755pub use ::aws_smithy_types::config_bag::FrozenLayer;
1756
1757pub use ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder;
1758
1759pub use ::aws_smithy_runtime_api::client::runtime_plugin::SharedRuntimePlugin;
1760
1761pub use ::aws_smithy_runtime_api::client::behavior_version::BehaviorVersion;
1762
1763pub use ::aws_smithy_runtime_api::client::stalled_stream_protection::StalledStreamProtectionConfig;
1764
1765pub use ::aws_smithy_runtime_api::client::http::SharedHttpClient;
1766
1767pub use ::aws_smithy_async::rt::sleep::SharedAsyncSleep;
1768
1769pub use ::aws_smithy_runtime_api::client::identity::SharedIdentityCache;
1770
1771pub use ::aws_smithy_runtime_api::client::interceptors::SharedInterceptor;
1772
1773pub use ::aws_types::region::Region;
1774
1775pub use ::aws_credential_types::provider::SharedCredentialsProvider;
1776
1777pub use ::aws_smithy_runtime_api::client::http::HttpClient;
1778
1779pub use ::aws_smithy_runtime_api::shared::IntoShared;
1780
1781pub use ::aws_smithy_runtime_api::client::identity::http::Token;
1782
1783pub use ::aws_smithy_runtime_api::client::identity::ResolveIdentity;
1784
1785pub use ::aws_smithy_async::rt::sleep::AsyncSleep;
1786
1787pub use ::aws_smithy_runtime_api::client::identity::ResolveCachedIdentity;
1788
1789pub use ::aws_smithy_runtime_api::client::interceptors::Intercept;
1790
1791pub use ::aws_credential_types::provider::token::ProvideToken;
1792
1793pub use ::aws_credential_types::provider::token::SharedTokenProvider;
1794
1795pub use ::aws_credential_types::provider::ProvideCredentials;
1796
1797pub use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;
1798
1799pub use ::aws_smithy_types::config_bag::Layer;
1800
1801/// Types needed to configure endpoint resolution.
1802pub mod endpoint;
1803
1804/// HTTP request and response types.
1805pub mod http;
1806
1807/// Types needed to implement [`Intercept`](crate::config::Intercept).
1808pub mod interceptors;
1809
1810/// Retry configuration.
1811///
1812/// [`RetryConfig`](crate::config::retry::RetryConfig) sets the number of retry attempts and the backoff between them. Retries are additionally bounded by a retry token bucket (a shared retry quota): [`TokenBucket`](crate::config::retry::TokenBucket) holds the tokens and [`RetryPartition`](crate::config::retry::RetryPartition) determines which clients and operations share one. To size the token bucket or give a workload its own, use [`Builder::retry_partition`](crate::config::Builder::retry_partition).
1813pub mod retry;
1814
1815/// Timeout configuration.
1816pub mod timeout;
1817
1818/// Types needed to configure auth scheme resolution.
1819pub mod auth;