aws-sdk-bedrockagentruntime 1.126.0

AWS SDK for Agents for Amazon Bedrock Runtime
Documentation
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
pub(crate) fn de_retrieval_result_location<'a, I>(
    tokens: &mut ::std::iter::Peekable<I>,
    _value: &'a [u8],
) -> ::std::result::Result<Option<crate::types::RetrievalResultLocation>, ::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::RetrievalResultLocationBuilder::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::RetrievalResultLocationType::from(u.as_ref())))
                                        .transpose()?,
                                );
                            }
                            "s3Location" => {
                                builder = builder.set_s3_location(
                                    crate::protocol_serde::shape_retrieval_result_s3_location::de_retrieval_result_s3_location(tokens, _value)?,
                                );
                            }
                            "webLocation" => {
                                builder = builder.set_web_location(
                                    crate::protocol_serde::shape_retrieval_result_web_location::de_retrieval_result_web_location(tokens, _value)?,
                                );
                            }
                            "confluenceLocation" => {
                                builder = builder.set_confluence_location(
                                    crate::protocol_serde::shape_retrieval_result_confluence_location::de_retrieval_result_confluence_location(
                                        tokens, _value,
                                    )?,
                                );
                            }
                            "salesforceLocation" => {
                                builder = builder.set_salesforce_location(
                                    crate::protocol_serde::shape_retrieval_result_salesforce_location::de_retrieval_result_salesforce_location(
                                        tokens, _value,
                                    )?,
                                );
                            }
                            "sharePointLocation" => {
                                builder = builder.set_share_point_location(
                                    crate::protocol_serde::shape_retrieval_result_share_point_location::de_retrieval_result_share_point_location(
                                        tokens, _value,
                                    )?,
                                );
                            }
                            "customDocumentLocation" => {
                                builder = builder.set_custom_document_location(
                                    crate::protocol_serde::shape_retrieval_result_custom_document_location::de_retrieval_result_custom_document_location(tokens, _value)?
                                );
                            }
                            "kendraDocumentLocation" => {
                                builder = builder.set_kendra_document_location(
                                    crate::protocol_serde::shape_retrieval_result_kendra_document_location::de_retrieval_result_kendra_document_location(tokens, _value)?
                                );
                            }
                            "sqlLocation" => {
                                builder = builder.set_sql_location(
                                    crate::protocol_serde::shape_retrieval_result_sql_location::de_retrieval_result_sql_location(tokens, _value)?,
                                );
                            }
                            _ => ::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::retrieval_result_location_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",
        )),
    }
}