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_flow_trace<'a, I>(
    tokens: &mut ::std::iter::Peekable<I>,
    _value: &'a [u8],
) -> ::std::result::Result<Option<crate::types::FlowTrace>, ::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() {
                        "nodeInputTrace" => Some(crate::types::FlowTrace::NodeInputTrace(
                            crate::protocol_serde::shape_flow_trace_node_input_event::de_flow_trace_node_input_event(tokens, _value)?.ok_or_else(
                                || ::aws_smithy_json::deserialize::error::DeserializeError::custom("value for 'nodeInputTrace' cannot be null"),
                            )?,
                        )),
                        "nodeOutputTrace" => Some(crate::types::FlowTrace::NodeOutputTrace(
                            crate::protocol_serde::shape_flow_trace_node_output_event::de_flow_trace_node_output_event(tokens, _value)?.ok_or_else(
                                || ::aws_smithy_json::deserialize::error::DeserializeError::custom("value for 'nodeOutputTrace' cannot be null"),
                            )?,
                        )),
                        "conditionNodeResultTrace" => Some(crate::types::FlowTrace::ConditionNodeResultTrace(
                            crate::protocol_serde::shape_flow_trace_condition_node_result_event::de_flow_trace_condition_node_result_event(
                                tokens, _value,
                            )?
                            .ok_or_else(|| {
                                ::aws_smithy_json::deserialize::error::DeserializeError::custom("value for 'conditionNodeResultTrace' cannot be null")
                            })?,
                        )),
                        "nodeActionTrace" => Some(crate::types::FlowTrace::NodeActionTrace(
                            crate::protocol_serde::shape_flow_trace_node_action_event::de_flow_trace_node_action_event(tokens, _value)?.ok_or_else(
                                || ::aws_smithy_json::deserialize::error::DeserializeError::custom("value for 'nodeActionTrace' cannot be null"),
                            )?,
                        )),
                        "nodeDependencyTrace" => Some(crate::types::FlowTrace::NodeDependencyTrace(
                            crate::protocol_serde::shape_flow_trace_dependency_event::de_flow_trace_dependency_event(tokens, _value)?.ok_or_else(
                                || ::aws_smithy_json::deserialize::error::DeserializeError::custom("value for 'nodeDependencyTrace' cannot be null"),
                            )?,
                        )),
                        _ => {
                            ::aws_smithy_json::deserialize::token::skip_value(tokens)?;
                            Some(crate::types::FlowTrace::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)
}