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;