pub fn ser_knowledge_base_flow_node_configuration(
object: &mut ::aws_smithy_json::serialize::JsonObjectWriter,
input: &crate::types::KnowledgeBaseFlowNodeConfiguration,
) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::SerializationError> {
{
object.key("knowledgeBaseId").string(input.knowledge_base_id.as_str());
}
if let Some(var_1) = &input.model_id {
object.key("modelId").string(var_1.as_str());
}
if let Some(var_2) = &input.guardrail_configuration {
#[allow(unused_mut)]
let mut object_3 = object.key("guardrailConfiguration").start_object();
crate::protocol_serde::shape_guardrail_configuration::ser_guardrail_configuration(&mut object_3, var_2)?;
object_3.finish();
}
if let Some(var_4) = &input.number_of_results {
object.key("numberOfResults").number(
#[allow(clippy::useless_conversion)]
::aws_smithy_types::Number::NegInt((*var_4).into()),
);
}
if let Some(var_5) = &input.prompt_template {
#[allow(unused_mut)]
let mut object_6 = object.key("promptTemplate").start_object();
crate::protocol_serde::shape_knowledge_base_prompt_template::ser_knowledge_base_prompt_template(&mut object_6, var_5)?;
object_6.finish();
}
if let Some(var_7) = &input.inference_configuration {
#[allow(unused_mut)]
let mut object_8 = object.key("inferenceConfiguration").start_object();
crate::protocol_serde::shape_prompt_inference_configuration::ser_prompt_inference_configuration(&mut object_8, var_7)?;
object_8.finish();
}
if let Some(var_9) = &input.reranking_configuration {
#[allow(unused_mut)]
let mut object_10 = object.key("rerankingConfiguration").start_object();
crate::protocol_serde::shape_vector_search_reranking_configuration::ser_vector_search_reranking_configuration(&mut object_10, var_9)?;
object_10.finish();
}
if let Some(var_11) = &input.orchestration_configuration {
#[allow(unused_mut)]
let mut object_12 = object.key("orchestrationConfiguration").start_object();
crate::protocol_serde::shape_knowledge_base_orchestration_configuration::ser_knowledge_base_orchestration_configuration(
&mut object_12,
var_11,
)?;
object_12.finish();
}
Ok(())
}
pub(crate) fn de_knowledge_base_flow_node_configuration<'a, I>(
tokens: &mut ::std::iter::Peekable<I>,
) -> ::std::result::Result<Option<crate::types::KnowledgeBaseFlowNodeConfiguration>, ::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::KnowledgeBaseFlowNodeConfigurationBuilder::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() {
"knowledgeBaseId" => {
builder = builder.set_knowledge_base_id(
::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
.map(|s| s.to_unescaped().map(|u| u.into_owned()))
.transpose()?,
);
}
"modelId" => {
builder = builder.set_model_id(
::aws_smithy_json::deserialize::token::expect_string_or_null(tokens.next())?
.map(|s| s.to_unescaped().map(|u| u.into_owned()))
.transpose()?,
);
}
"guardrailConfiguration" => {
builder = builder.set_guardrail_configuration(
crate::protocol_serde::shape_guardrail_configuration::de_guardrail_configuration(tokens)?,
);
}
"numberOfResults" => {
builder = builder.set_number_of_results(
::aws_smithy_json::deserialize::token::expect_number_or_null(tokens.next())?
.map(i32::try_from)
.transpose()?,
);
}
"promptTemplate" => {
builder = builder.set_prompt_template(
crate::protocol_serde::shape_knowledge_base_prompt_template::de_knowledge_base_prompt_template(tokens)?,
);
}
"inferenceConfiguration" => {
builder = builder.set_inference_configuration(
crate::protocol_serde::shape_prompt_inference_configuration::de_prompt_inference_configuration(tokens)?,
);
}
"rerankingConfiguration" => {
builder = builder.set_reranking_configuration(
crate::protocol_serde::shape_vector_search_reranking_configuration::de_vector_search_reranking_configuration(tokens)?,
);
}
"orchestrationConfiguration" => {
builder = builder.set_orchestration_configuration(
crate::protocol_serde::shape_knowledge_base_orchestration_configuration::de_knowledge_base_orchestration_configuration(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::knowledge_base_flow_node_configuration_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",
)),
}
}