AWS SDK

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

pub use crate::operation::start_async_invoke::_start_async_invoke_output::StartAsyncInvokeOutputBuilder;

pub use crate::operation::start_async_invoke::_start_async_invoke_input::StartAsyncInvokeInputBuilder;

impl crate::operation::start_async_invoke::builders::StartAsyncInvokeInputBuilder {

    /// Sends a request with this input using the given client.

    pub async fn send_with(

        self,

        client: &crate::Client,

    ) -> ::std::result::Result<

        crate::operation::start_async_invoke::StartAsyncInvokeOutput,

        ::aws_smithy_runtime_api::client::result::SdkError<

            crate::operation::start_async_invoke::StartAsyncInvokeError,

            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,

        >,

    > {

        let mut fluent_builder = client.start_async_invoke();

        fluent_builder.inner = self;

        fluent_builder.send().await

    }

}

/// Fluent builder constructing a request to `StartAsyncInvoke`.

///

/// <p>Starts an asynchronous invocation.</p>

/// <p>This operation requires permission for the <code>bedrock:InvokeModel</code> action.</p><important>

/// <p>To deny all inference access to resources that you specify in the modelId field, you need to deny access to the <code>bedrock:InvokeModel</code> and <code>bedrock:InvokeModelWithResponseStream</code> actions. Doing this also denies access to the resource through the Converse API actions (<a href="https://docs.aws.amazon.com/bedrock/latest/APIReference/API_runtime_Converse.html">Converse</a> and <a href="https://docs.aws.amazon.com/bedrock/latest/APIReference/API_runtime_ConverseStream.html">ConverseStream</a>). For more information see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/security_iam_id-based-policy-examples.html#security_iam_id-based-policy-examples-deny-inference">Deny access for inference on specific models</a>.</p>

/// </important>

#[derive(::std::clone::Clone, ::std::fmt::Debug)]

pub struct StartAsyncInvokeFluentBuilder {

    handle: ::std::sync::Arc<crate::client::Handle>,

    inner: crate::operation::start_async_invoke::builders::StartAsyncInvokeInputBuilder,

    config_override: ::std::option::Option<crate::config::Builder>,

}

impl

    crate::client::customize::internal::CustomizableSend<

        crate::operation::start_async_invoke::StartAsyncInvokeOutput,

        crate::operation::start_async_invoke::StartAsyncInvokeError,

    > for StartAsyncInvokeFluentBuilder

{

    fn send(

        self,

        config_override: crate::config::Builder,

    ) -> crate::client::customize::internal::BoxFuture<

        crate::client::customize::internal::SendResult<

            crate::operation::start_async_invoke::StartAsyncInvokeOutput,

            crate::operation::start_async_invoke::StartAsyncInvokeError,

        >,

    > {

        ::std::boxed::Box::pin(async move { self.config_override(config_override).send().await })

    }

}

impl StartAsyncInvokeFluentBuilder {

    /// Creates a new `StartAsyncInvokeFluentBuilder`.

    pub(crate) fn new(handle: ::std::sync::Arc<crate::client::Handle>) -> Self {

        Self {

            handle,

            inner: ::std::default::Default::default(),

            config_override: ::std::option::Option::None,

        }

    }

    /// Access the StartAsyncInvoke as a reference.

    pub fn as_input(&self) -> &crate::operation::start_async_invoke::builders::StartAsyncInvokeInputBuilder {

        &self.inner

    }

    /// Sends the request and returns the response.

    ///

    /// If an error occurs, an `SdkError` will be returned with additional details that

    /// can be matched against.

    ///

    /// By default, any retryable failures will be retried twice. Retry behavior

    /// is configurable with the [RetryConfig](aws_smithy_types::retry::RetryConfig), which can be

    /// set when configuring the client.

    pub async fn send(

        self,

    ) -> ::std::result::Result<

        crate::operation::start_async_invoke::StartAsyncInvokeOutput,

        ::aws_smithy_runtime_api::client::result::SdkError<

            crate::operation::start_async_invoke::StartAsyncInvokeError,

            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,

        >,

    > {

        let input = self

            .inner

            .build()

            .map_err(::aws_smithy_runtime_api::client::result::SdkError::construction_failure)?;

        let runtime_plugins = crate::operation::start_async_invoke::StartAsyncInvoke::operation_runtime_plugins(

            self.handle.runtime_plugins.clone(),

            &self.handle.conf,

            self.config_override,

        );

        crate::operation::start_async_invoke::StartAsyncInvoke::orchestrate(&runtime_plugins, input).await

    }

    /// Consumes this builder, creating a customizable operation that can be modified before being sent.

    pub fn customize(

        self,

    ) -> crate::client::customize::CustomizableOperation<

        crate::operation::start_async_invoke::StartAsyncInvokeOutput,

        crate::operation::start_async_invoke::StartAsyncInvokeError,

        Self,

    > {

        crate::client::customize::CustomizableOperation::new(self)

    }

    pub(crate) fn config_override(mut self, config_override: impl ::std::convert::Into<crate::config::Builder>) -> Self {

        self.set_config_override(::std::option::Option::Some(config_override.into()));

        self

    }

    pub(crate) fn set_config_override(&mut self, config_override: ::std::option::Option<crate::config::Builder>) -> &mut Self {

        self.config_override = config_override;

        self

    }

    /// <p>Specify idempotency token to ensure that requests are not duplicated.</p>

    pub fn client_request_token(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {

        self.inner = self.inner.client_request_token(input.into());

        self

    }

    /// <p>Specify idempotency token to ensure that requests are not duplicated.</p>

    pub fn set_client_request_token(mut self, input: ::std::option::Option<::std::string::String>) -> Self {

        self.inner = self.inner.set_client_request_token(input);

        self

    }

    /// <p>Specify idempotency token to ensure that requests are not duplicated.</p>

    pub fn get_client_request_token(&self) -> &::std::option::Option<::std::string::String> {

        self.inner.get_client_request_token()

    }

    /// <p>The model to invoke.</p>

    pub fn model_id(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {

        self.inner = self.inner.model_id(input.into());

        self

    }

    /// <p>The model to invoke.</p>

    pub fn set_model_id(mut self, input: ::std::option::Option<::std::string::String>) -> Self {

        self.inner = self.inner.set_model_id(input);

        self

    }

    /// <p>The model to invoke.</p>

    pub fn get_model_id(&self) -> &::std::option::Option<::std::string::String> {

        self.inner.get_model_id()

    }

    /// <p>Input to send to the model.</p>

    pub fn model_input(mut self, input: ::aws_smithy_types::Document) -> Self {

        self.inner = self.inner.model_input(input);

        self

    }

    /// <p>Input to send to the model.</p>

    pub fn set_model_input(mut self, input: ::std::option::Option<::aws_smithy_types::Document>) -> Self {

        self.inner = self.inner.set_model_input(input);

        self

    }

    /// <p>Input to send to the model.</p>

    pub fn get_model_input(&self) -> &::std::option::Option<::aws_smithy_types::Document> {

        self.inner.get_model_input()

    }

    /// <p>Where to store the output.</p>

    pub fn output_data_config(mut self, input: crate::types::AsyncInvokeOutputDataConfig) -> Self {

        self.inner = self.inner.output_data_config(input);

        self

    }

    /// <p>Where to store the output.</p>

    pub fn set_output_data_config(mut self, input: ::std::option::Option<crate::types::AsyncInvokeOutputDataConfig>) -> Self {

        self.inner = self.inner.set_output_data_config(input);

        self

    }

    /// <p>Where to store the output.</p>

    pub fn get_output_data_config(&self) -> &::std::option::Option<crate::types::AsyncInvokeOutputDataConfig> {

        self.inner.get_output_data_config()

    }

    ///

    /// Appends an item to `tags`.

    ///

    /// To override the contents of this collection use [`set_tags`](Self::set_tags).

    ///

    /// <p>Tags to apply to the invocation.</p>

    pub fn tags(mut self, input: crate::types::Tag) -> Self {

        self.inner = self.inner.tags(input);

        self

    }

    /// <p>Tags to apply to the invocation.</p>

    pub fn set_tags(mut self, input: ::std::option::Option<::std::vec::Vec<crate::types::Tag>>) -> Self {

        self.inner = self.inner.set_tags(input);

        self

    }

    /// <p>Tags to apply to the invocation.</p>

    pub fn get_tags(&self) -> &::std::option::Option<::std::vec::Vec<crate::types::Tag>> {

        self.inner.get_tags()

    }

}

pub use ::aws_smithy_types::date_time::Format as DateTimeFormat;

pub use ::aws_smithy_types::DateTime;

pub(crate) mod shape_apply_guardrail;

pub(crate) mod shape_get_async_invoke;

pub(crate) mod shape_invoke_model_with_bidirectional_stream;

pub(crate) mod shape_list_async_invokes;

pub(crate) mod shape_start_async_invoke;

pub(crate) mod shape_apply_guardrail_input;

pub(crate) mod shape_conflict_exception;

pub(crate) mod shape_invoke_model_with_bidirectional_stream_input;

pub(crate) mod shape_invoke_model_with_bidirectional_stream_output;

pub(crate) mod shape_service_unavailable_exception;

pub(crate) mod shape_start_async_invoke_input;

pub(crate) mod shape_async_invoke_output_data_config;

pub(crate) mod shape_async_invoke_summaries;

pub(crate) mod shape_bidirectional_input_payload_part;

pub(crate) mod shape_converse_trace;

pub(crate) mod shape_guardrail_assessment_list;

pub(crate) mod shape_guardrail_configuration;

pub(crate) mod shape_guardrail_content_block;

pub(crate) mod shape_guardrail_coverage;

pub(crate) mod shape_guardrail_output_content_list;

pub(crate) mod shape_guardrail_stream_configuration;

pub(crate) mod shape_guardrail_usage;

pub(crate) mod shape_performance_configuration;

pub(crate) mod shape_prompt_variable_values;

pub(crate) mod shape_tag;

pub(crate) mod shape_async_invoke_s3_output_data_config;

pub(crate) mod shape_async_invoke_summary;

pub(crate) mod shape_bidirectional_output_payload_part;

pub(crate) mod shape_cache_point_block;

pub(crate) mod shape_guardrail_assessment;

pub(crate) mod shape_guardrail_converse_content_block;

pub(crate) mod shape_guardrail_image_block;

pub(crate) mod shape_guardrail_image_coverage;

pub(crate) mod shape_guardrail_output_content;

pub(crate) mod shape_guardrail_text_block;

pub(crate) mod shape_guardrail_text_characters_coverage;

pub(crate) mod shape_guardrail_trace_assessment;

pub(crate) mod shape_prompt_router_trace;

pub(crate) mod shape_citations_content_block;

pub(crate) mod shape_document_block;

pub(crate) mod shape_guardrail_assessment_list_map;

pub(crate) mod shape_guardrail_assessment_map;

pub(crate) mod shape_guardrail_content_policy_assessment;

pub(crate) mod shape_guardrail_contextual_grounding_policy_assessment;

pub(crate) mod shape_guardrail_converse_image_block;

pub(crate) mod shape_guardrail_converse_text_block;

pub(crate) mod shape_guardrail_image_source;

pub(crate) mod shape_guardrail_invocation_metrics;

pub(crate) mod shape_guardrail_sensitive_information_policy_assessment;

pub(crate) mod shape_guardrail_topic_policy_assessment;

pub(crate) mod shape_guardrail_word_policy_assessment;

pub(crate) mod shape_model_outputs;

pub(crate) mod shape_reasoning_content_block;

pub(crate) mod shape_video_block;

pub(crate) mod shape_citation;

pub(crate) mod shape_citation_generated_content;

pub(crate) mod shape_citations_config;

pub(crate) mod shape_converse_stream_trace;

pub(crate) mod shape_document_source;

pub(crate) mod shape_guardrail_content_filter_list;

pub(crate) mod shape_guardrail_contextual_grounding_filters;

pub(crate) mod shape_guardrail_converse_image_source;

pub(crate) mod shape_guardrail_custom_word_list;

pub(crate) mod shape_guardrail_managed_word_list;

pub(crate) mod shape_guardrail_pii_entity_filter_list;

pub(crate) mod shape_guardrail_regex_filter_list;

pub(crate) mod shape_guardrail_topic_list;

pub(crate) mod shape_reasoning_text_block;

pub(crate) mod shape_video_source;

pub(crate) mod shape_citation_location;

pub(crate) mod shape_citation_source_content;

pub(crate) mod shape_citations_delta;

pub(crate) mod shape_document_content_block;

pub(crate) mod shape_guardrail_content_filter;

pub(crate) mod shape_guardrail_contextual_grounding_filter;

pub(crate) mod shape_guardrail_custom_word;

pub(crate) mod shape_guardrail_managed_word;

pub(crate) mod shape_guardrail_pii_entity_filter;

pub(crate) mod shape_guardrail_regex_filter;

pub(crate) mod shape_guardrail_topic;

pub(crate) mod shape_reasoning_content_block_delta;

pub(crate) mod shape_s3_location;

pub(crate) mod shape_citation_generated_content_list;

pub(crate) mod shape_citation_source_content_list_delta;

pub(crate) mod shape_citations;

pub(crate) mod shape_document_char_location;

pub(crate) mod shape_document_chunk_location;

pub(crate) mod shape_document_page_location;

pub(crate) mod shape_citation_source_content_delta;

pub(crate) mod shape_document_content_blocks;

pub(crate) mod shape_guardrail_converse_content_qualifier_list;

pub(crate) mod shape_citation_source_content_list;

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

#[allow(clippy::unnecessary_wraps)]

pub fn de_apply_guardrail_http_error(

    _response_status: u16,

    _response_headers: &::aws_smithy_runtime_api::http::Headers,

    _response_body: &[u8],

) -> std::result::Result<crate::operation::apply_guardrail::ApplyGuardrailOutput, crate::operation::apply_guardrail::ApplyGuardrailError> {

    #[allow(unused_mut)]

    let mut generic_builder = crate::protocol_serde::parse_http_error_metadata(_response_status, _response_headers, _response_body)

        .map_err(crate::operation::apply_guardrail::ApplyGuardrailError::unhandled)?;

    generic_builder = ::aws_types::request_id::apply_request_id(generic_builder, _response_headers);

    let generic = generic_builder.build();

    let error_code = match generic.code() {

        Some(code) => code,

        None => return Err(crate::operation::apply_guardrail::ApplyGuardrailError::unhandled(generic)),

    };

    let _error_message = generic.message().map(|msg| msg.to_owned());

    Err(match error_code {

        "AccessDeniedException" => crate::operation::apply_guardrail::ApplyGuardrailError::AccessDeniedException({

            #[allow(unused_mut)]

            let mut tmp = {

                #[allow(unused_mut)]

                let mut output = crate::types::error::builders::AccessDeniedExceptionBuilder::default();

                output = crate::protocol_serde::shape_access_denied_exception::de_access_denied_exception_json_err(_response_body, output)

                    .map_err(crate::operation::apply_guardrail::ApplyGuardrailError::unhandled)?;

                let output = output.meta(generic);

                output.build()

            };

            if tmp.message.is_none() {

                tmp.message = _error_message;

            }

            tmp

        }),

        "InternalServerException" => crate::operation::apply_guardrail::ApplyGuardrailError::InternalServerException({

            #[allow(unused_mut)]

            let mut tmp = {

                #[allow(unused_mut)]

                let mut output = crate::types::error::builders::InternalServerExceptionBuilder::default();

                output = crate::protocol_serde::shape_internal_server_exception::de_internal_server_exception_json_err(_response_body, output)

                    .map_err(crate::operation::apply_guardrail::ApplyGuardrailError::unhandled)?;

                let output = output.meta(generic);

                output.build()

            };

            if tmp.message.is_none() {

                tmp.message = _error_message;

            }

            tmp

        }),

        "ResourceNotFoundException" => crate::operation::apply_guardrail::ApplyGuardrailError::ResourceNotFoundException({

            #[allow(unused_mut)]

            let mut tmp = {

                #[allow(unused_mut)]

                let mut output = crate::types::error::builders::ResourceNotFoundExceptionBuilder::default();

                output = crate::protocol_serde::shape_resource_not_found_exception::de_resource_not_found_exception_json_err(_response_body, output)

                    .map_err(crate::operation::apply_guardrail::ApplyGuardrailError::unhandled)?;

                let output = output.meta(generic);

                output.build()

            };

            if tmp.message.is_none() {

                tmp.message = _error_message;

            }

            tmp

        }),

        "ServiceQuotaExceededException" => crate::operation::apply_guardrail::ApplyGuardrailError::ServiceQuotaExceededException({

            #[allow(unused_mut)]

            let mut tmp = {

                #[allow(unused_mut)]

                let mut output = crate::types::error::builders::ServiceQuotaExceededExceptionBuilder::default();

                output = crate::protocol_serde::shape_service_quota_exceeded_exception::de_service_quota_exceeded_exception_json_err(

                    _response_body,

                    output,

                )

                .map_err(crate::operation::apply_guardrail::ApplyGuardrailError::unhandled)?;

                let output = output.meta(generic);

                output.build()

            };

            if tmp.message.is_none() {

                tmp.message = _error_message;

            }

            tmp

        }),

        "ThrottlingException" => crate::operation::apply_guardrail::ApplyGuardrailError::ThrottlingException({

            #[allow(unused_mut)]

            let mut tmp = {

                #[allow(unused_mut)]

                let mut output = crate::types::error::builders::ThrottlingExceptionBuilder::default();

                output = crate::protocol_serde::shape_throttling_exception::de_throttling_exception_json_err(_response_body, output)

                    .map_err(crate::operation::apply_guardrail::ApplyGuardrailError::unhandled)?;

                let output = output.meta(generic);

                output.build()

            };

            if tmp.message.is_none() {

                tmp.message = _error_message;

            }

            tmp

        }),

        "ValidationException" => crate::operation::apply_guardrail::ApplyGuardrailError::ValidationException({

            #[allow(unused_mut)]

            let mut tmp = {

                #[allow(unused_mut)]

                let mut output = crate::types::error::builders::ValidationExceptionBuilder::default();

                output = crate::protocol_serde::shape_validation_exception::de_validation_exception_json_err(_response_body, output)

                    .map_err(crate::operation::apply_guardrail::ApplyGuardrailError::unhandled)?;

                let output = output.meta(generic);

                output.build()

            };

            if tmp.message.is_none() {

                tmp.message = _error_message;

            }

            tmp

        }),

        _ => crate::operation::apply_guardrail::ApplyGuardrailError::generic(generic),

    })

}

#[allow(clippy::unnecessary_wraps)]

pub fn de_apply_guardrail_http_response(

    _response_status: u16,

    _response_headers: &::aws_smithy_runtime_api::http::Headers,

    _response_body: &[u8],

) -> std::result::Result<crate::operation::apply_guardrail::ApplyGuardrailOutput, crate::operation::apply_guardrail::ApplyGuardrailError> {

    Ok({

        #[allow(unused_mut)]

        let mut output = crate::operation::apply_guardrail::builders::ApplyGuardrailOutputBuilder::default();

        output = crate::protocol_serde::shape_apply_guardrail::de_apply_guardrail(_response_body, output)

            .map_err(crate::operation::apply_guardrail::ApplyGuardrailError::unhandled)?;

        output._set_request_id(::aws_types::request_id::RequestId::request_id(_response_headers).map(str::to_string));

        crate::serde_util::apply_guardrail_output_output_correct_errors(output)

            .build()

            .map_err(crate::operation::apply_guardrail::ApplyGuardrailError::unhandled)?

    })

}

pub fn ser_apply_guardrail_input(

    input: &crate::operation::apply_guardrail::ApplyGuardrailInput,

) -> ::std::result::Result<::aws_smithy_types::body::SdkBody, ::aws_smithy_types::error::operation::SerializationError> {

    let mut out = String::new();

    let mut object = ::aws_smithy_json::serialize::JsonObjectWriter::new(&mut out);

    crate::protocol_serde::shape_apply_guardrail_input::ser_apply_guardrail_input_input(&mut object, input)?;

    object.finish();

    Ok(::aws_smithy_types::body::SdkBody::from(out))

}

pub(crate) fn de_apply_guardrail(

    value: &[u8],

    mut builder: crate::operation::apply_guardrail::builders::ApplyGuardrailOutputBuilder,

) -> ::std::result::Result<

    crate::operation::apply_guardrail::builders::ApplyGuardrailOutputBuilder,

    ::aws_smithy_json::deserialize::error::DeserializeError,

> {

    let mut tokens_owned = ::aws_smithy_json::deserialize::json_token_iter(crate::protocol_serde::or_empty_doc(value)).peekable();

    let tokens = &mut tokens_owned;

    ::aws_smithy_json::deserialize::token::expect_start_object(tokens.next())?;

    loop {

        match tokens.next().transpose()? {

            Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,

            Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {

                "action" => {

                    builder = builder.set_action(

                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                            .map(|s| s.to_unescaped().map(|u| crate::types::GuardrailAction::from(u.as_ref())))

                            .transpose()?,

                    );

                }

                "actionReason" => {

                    builder = builder.set_action_reason(

                        ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                            .map(|s| s.to_unescaped().map(|u| u.into_owned()))

                            .transpose()?,

                    );

                }

                "assessments" => {

                    builder = builder.set_assessments(crate::protocol_serde::shape_guardrail_assessment_list::de_guardrail_assessment_list(

                        tokens,

                    )?);

                }

                "guardrailCoverage" => {

                    builder = builder.set_guardrail_coverage(crate::protocol_serde::shape_guardrail_coverage::de_guardrail_coverage(tokens)?);

                }

                "outputs" => {

                    builder =

                        builder.set_outputs(crate::protocol_serde::shape_guardrail_output_content_list::de_guardrail_output_content_list(tokens)?);

                }

                "usage" => {

                    builder = builder.set_usage(crate::protocol_serde::shape_guardrail_usage::de_guardrail_usage(tokens)?);

                }

                _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,

            },

            other => {

                return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(

                    "expected object key or end object, found: {:?}",

                    other

                )))

            }

        }

    }

    if tokens.next().is_some() {

        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(

            "found more JSON tokens after completing parsing",

        ));

    }

    Ok(builder)

}

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

pub fn ser_apply_guardrail_input_input(

    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,

    input: &crate::operation::apply_guardrail::ApplyGuardrailInput,

) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {

    if let Some(var_1) = &input.content {

        let mut array_2 = object.key("content").start_array();

        for item_3 in var_1 {

            {

                #[allow(unused_mut)]

                let mut object_4 = array_2.value().start_object();

                crate::protocol_serde::shape_guardrail_content_block::ser_guardrail_content_block(&mut object_4, item_3)?;

                object_4.finish();

            }

        }

        array_2.finish();

    }

    if let Some(var_5) = &input.output_scope {

        object.key("outputScope").string(var_5.as_str());

    }

    if let Some(var_6) = &input.source {

        object.key("source").string(var_6.as_str());

    }

    Ok(())

}

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

pub(crate) fn de_async_invoke_output_data_config<'a, I>(

    tokens: &mut ::std::iter::Peekable<I>,

) -> ::std::result::Result<Option<crate::types::AsyncInvokeOutputDataConfig>, ::aws_smithy_json::deserialize::error::DeserializeError>

where

    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,

{

    let mut variant = None;

    match tokens.next().transpose()? {

        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => return Ok(None),

        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => loop {

            match tokens.next().transpose()? {

                Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,

                Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => {

                    if let ::std::option::Option::Some(::std::result::Result::Ok(::aws_smithy_json::deserialize::Token::ValueNull { .. })) =

                        tokens.peek()

                    {

                        let _ = tokens.next().expect("peek returned a token")?;

                        continue;

                    }

                    let key = key.to_unescaped()?;

                    if key == "__type" {

                        ::aws_smithy_json::deserialize::token::skip_value(tokens)?;

                        continue;

                    }

                    if variant.is_some() {

                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(

                            "encountered mixed variants in union",

                        ));

                    }

                    variant = match key.as_ref() {

                        "s3OutputDataConfig" => Some(crate::types::AsyncInvokeOutputDataConfig::S3OutputDataConfig(

                            crate::protocol_serde::shape_async_invoke_s3_output_data_config::de_async_invoke_s3_output_data_config(tokens)?

                                .ok_or_else(|| {

                                    ::aws_smithy_json::deserialize::error::DeserializeError::custom("value for 's3OutputDataConfig' cannot be null")

                                })?,

                        )),

                        _ => {

                            ::aws_smithy_json::deserialize::token::skip_value(tokens)?;

                            Some(crate::types::AsyncInvokeOutputDataConfig::Unknown)

                        }

                    };

                }

                other => {

                    return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(

                        "expected object key or end object, found: {:?}",

                        other

                    )))

                }

            }

        },

        _ => {

            return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(

                "expected start object or null",

            ))

        }

    }

    if variant.is_none() {

        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(

            "Union did not contain a valid variant.",

        ));

    }

    Ok(variant)

}

pub fn ser_async_invoke_output_data_config(

    object_5: &mut ::aws_smithy_json::serialize::JsonObjectWriter,

    input: &crate::types::AsyncInvokeOutputDataConfig,

) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {

    match input {

        crate::types::AsyncInvokeOutputDataConfig::S3OutputDataConfig(inner) => {

            #[allow(unused_mut)]

            let mut object_1 = object_5.key("s3OutputDataConfig").start_object();

            crate::protocol_serde::shape_async_invoke_s3_output_data_config::ser_async_invoke_s3_output_data_config(&mut object_1, inner)?;

            object_1.finish();

        }

        crate::types::AsyncInvokeOutputDataConfig::Unknown => {

            return Err(::aws_smithy_types::error::operation::SerializationError::unknown_variant(

                "AsyncInvokeOutputDataConfig",

            ))

        }

    }

    Ok(())

}

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

pub(crate) fn de_async_invoke_s3_output_data_config<'a, I>(

    tokens: &mut ::std::iter::Peekable<I>,

) -> ::std::result::Result<Option<crate::types::AsyncInvokeS3OutputDataConfig>, ::aws_smithy_json::deserialize::error::DeserializeError>

where

    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,

{

    match tokens.next().transpose()? {

        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),

        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {

            #[allow(unused_mut)]

            let mut builder = crate::types::builders::AsyncInvokeS3OutputDataConfigBuilder::default();

            loop {

                match tokens.next().transpose()? {

                    Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,

                    Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {

                        "s3Uri" => {

                            builder = builder.set_s3_uri(

                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))

                                    .transpose()?,

                            );

                        }

                        "kmsKeyId" => {

                            builder = builder.set_kms_key_id(

                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))

                                    .transpose()?,

                            );

                        }

                        "bucketOwner" => {

                            builder = builder.set_bucket_owner(

                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))

                                    .transpose()?,

                            );

                        }

                        _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,

                    },

                    other => {

                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(

                            "expected object key or end object, found: {:?}",

                            other

                        )))

                    }

                }

            }

            Ok(Some(

                crate::serde_util::async_invoke_s3_output_data_config_correct_errors(builder)

                    .build()

                    .map_err(|err| ::aws_smithy_json::deserialize::error::DeserializeError::custom_source("Response was invalid", err))?,

            ))

        }

        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(

            "expected start object or null",

        )),

    }

}

pub fn ser_async_invoke_s3_output_data_config(

    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,

    input: &crate::types::AsyncInvokeS3OutputDataConfig,

) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {

    {

        object.key("s3Uri").string(input.s3_uri.as_str());

    }

    if let Some(var_1) = &input.kms_key_id {

        object.key("kmsKeyId").string(var_1.as_str());

    }

    if let Some(var_2) = &input.bucket_owner {

        object.key("bucketOwner").string(var_2.as_str());

    }

    Ok(())

}

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

pub(crate) fn de_async_invoke_summaries<'a, I>(

    tokens: &mut ::std::iter::Peekable<I>,

) -> ::std::result::Result<Option<::std::vec::Vec<crate::types::AsyncInvokeSummary>>, ::aws_smithy_json::deserialize::error::DeserializeError>

where

    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,

{

    match tokens.next().transpose()? {

        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),

        Some(::aws_smithy_json::deserialize::Token::StartArray { .. }) => {

            let mut items = Vec::new();

            loop {

                match tokens.peek() {

                    Some(Ok(::aws_smithy_json::deserialize::Token::EndArray { .. })) => {

                        tokens.next().transpose().unwrap();

                        break;

                    }

                    _ => {

                        let value = crate::protocol_serde::shape_async_invoke_summary::de_async_invoke_summary(tokens)?;

                        if let Some(value) = value {

                            items.push(value);

                        }

                    }

                }

            }

            Ok(Some(items))

        }

        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(

            "expected start array or null",

        )),

    }

}

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

pub(crate) fn de_async_invoke_summary<'a, I>(

    tokens: &mut ::std::iter::Peekable<I>,

) -> ::std::result::Result<Option<crate::types::AsyncInvokeSummary>, ::aws_smithy_json::deserialize::error::DeserializeError>

where

    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,

{

    match tokens.next().transpose()? {

        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),

        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {

            #[allow(unused_mut)]

            let mut builder = crate::types::builders::AsyncInvokeSummaryBuilder::default();

            loop {

                match tokens.next().transpose()? {

                    Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,

                    Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {

                        "invocationArn" => {

                            builder = builder.set_invocation_arn(

                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))

                                    .transpose()?,

                            );

                        }

                        "modelArn" => {

                            builder = builder.set_model_arn(

                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))

                                    .transpose()?,

                            );

                        }

                        "clientRequestToken" => {

                            builder = builder.set_client_request_token(

                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))

                                    .transpose()?,

                            );

                        }

                        "status" => {

                            builder = builder.set_status(

                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                                    .map(|s| s.to_unescaped().map(|u| crate::types::AsyncInvokeStatus::from(u.as_ref())))

                                    .transpose()?,

                            );

                        }

                        "failureMessage" => {

                            builder = builder.set_failure_message(

                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                                    .map(|s| s.to_unescaped().map(|u| u.into_owned()))

                                    .transpose()?,

                            );

                        }

                        "submitTime" => {

                            builder = builder.set_submit_time(::aws_smithy_json::deserialize::token::expect_timestamp_or_null(

                                tokens.next(),

                                ::aws_smithy_types::date_time::Format::DateTimeWithOffset,

                            )?);

                        }

                        "lastModifiedTime" => {

                            builder = builder.set_last_modified_time(::aws_smithy_json::deserialize::token::expect_timestamp_or_null(

                                tokens.next(),

                                ::aws_smithy_types::date_time::Format::DateTimeWithOffset,

                            )?);

                        }

                        "endTime" => {

                            builder = builder.set_end_time(::aws_smithy_json::deserialize::token::expect_timestamp_or_null(

                                tokens.next(),

                                ::aws_smithy_types::date_time::Format::DateTimeWithOffset,

                            )?);

                        }

                        "outputDataConfig" => {

                            builder = builder.set_output_data_config(

                                crate::protocol_serde::shape_async_invoke_output_data_config::de_async_invoke_output_data_config(tokens)?,

                            );

                        }

                        _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,

                    },

                    other => {

                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(

                            "expected object key or end object, found: {:?}",

                            other

                        )))

                    }

                }

            }

            Ok(Some(crate::serde_util::async_invoke_summary_correct_errors(builder).build().map_err(

                |err| ::aws_smithy_json::deserialize::error::DeserializeError::custom_source("Response was invalid", err),

            )?))

        }

        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(

            "expected start object or null",

        )),

    }

}

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

pub fn ser_bidirectional_input_payload_part(

    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,

    input: &crate::types::BidirectionalInputPayloadPart,

) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {

    if let Some(var_1) = &input.bytes {

        object.key("bytes").string_unchecked(&::aws_smithy_types::base64::encode(var_1));

    }

    Ok(())

}

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

pub(crate) fn de_bidirectional_output_payload_part_payload(

    input: &[u8],

) -> ::std::result::Result<crate::types::BidirectionalOutputPayloadPart, ::aws_smithy_json::deserialize::error::DeserializeError> {

    let mut tokens_owned = ::aws_smithy_json::deserialize::json_token_iter(crate::protocol_serde::or_empty_doc(input)).peekable();

    let tokens = &mut tokens_owned;

    let result = crate::protocol_serde::shape_bidirectional_output_payload_part::de_bidirectional_output_payload_part(tokens)?

        .ok_or_else(|| ::aws_smithy_json::deserialize::error::DeserializeError::custom("expected payload member value"));

    if tokens.next().is_some() {

        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(

            "found more JSON tokens after completing parsing",

        ));

    }

    result

}

pub(crate) fn de_bidirectional_output_payload_part<'a, I>(

    tokens: &mut ::std::iter::Peekable<I>,

) -> ::std::result::Result<Option<crate::types::BidirectionalOutputPayloadPart>, ::aws_smithy_json::deserialize::error::DeserializeError>

where

    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,

{

    match tokens.next().transpose()? {

        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),

        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {

            #[allow(unused_mut)]

            let mut builder = crate::types::builders::BidirectionalOutputPayloadPartBuilder::default();

            loop {

                match tokens.next().transpose()? {

                    Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,

                    Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {

                        "bytes" => {

                            builder = builder.set_bytes(::aws_smithy_json::deserialize::token::expect_blob_or_null(tokens.next())?);

                        }

                        _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,

                    },

                    other => {

                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(

                            "expected object key or end object, found: {:?}",

                            other

                        )))

                    }

                }

            }

            Ok(Some(builder.build()))

        }

        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(

            "expected start object or null",

        )),

    }

}

// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

pub fn ser_cache_point_block(

    object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,

    input: &crate::types::CachePointBlock,

) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {

    {

        object.key("type").string(input.r#type.as_str());

    }

    Ok(())

}

pub(crate) fn de_cache_point_block<'a, I>(

    tokens: &mut ::std::iter::Peekable<I>,

) -> ::std::result::Result<Option<crate::types::CachePointBlock>, ::aws_smithy_json::deserialize::error::DeserializeError>

where

    I: Iterator<Item = Result<::aws_smithy_json::deserialize::Token<'a>, ::aws_smithy_json::deserialize::error::DeserializeError>>,

{

    match tokens.next().transpose()? {

        Some(::aws_smithy_json::deserialize::Token::ValueNull { .. }) => Ok(None),

        Some(::aws_smithy_json::deserialize::Token::StartObject { .. }) => {

            #[allow(unused_mut)]

            let mut builder = crate::types::builders::CachePointBlockBuilder::default();

            loop {

                match tokens.next().transpose()? {

                    Some(::aws_smithy_json::deserialize::Token::EndObject { .. }) => break,

                    Some(::aws_smithy_json::deserialize::Token::ObjectKey { key, .. }) => match key.to_unescaped()?.as_ref() {

                        "type" => {

                            builder = builder.set_type(

                                ::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?

                                    .map(|s| s.to_unescaped().map(|u| crate::types::CachePointType::from(u.as_ref())))

                                    .transpose()?,

                            );

                        }

                        _ => ::aws_smithy_json::deserialize::token::skip_value(tokens)?,

                    },

                    other => {

                        return Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(format!(

                            "expected object key or end object, found: {:?}",

                            other

                        )))

                    }

                }

            }

            Ok(Some(crate::serde_util::cache_point_block_correct_errors(builder).build().map_err(

                |err| ::aws_smithy_json::deserialize::error::DeserializeError::custom_source("Response was invalid", err),

            )?))

        }

        _ => Err(::aws_smithy_json::deserialize::error::DeserializeError::custom(

            "expected start object or null",

        )),

    }

}