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