#[allow(clippy::unnecessary_wraps)]
pub fn de_invoke_agent_http_response(
response: &mut ::aws_smithy_runtime_api::http::Response,
) -> std::result::Result<crate::operation::invoke_agent::InvokeAgentOutput, crate::operation::invoke_agent::InvokeAgentError> {
let mut _response_body = ::aws_smithy_types::body::SdkBody::taken();
std::mem::swap(&mut _response_body, response.body_mut());
let _response_body = &mut _response_body;
let _response_status = response.status().as_u16();
let _response_headers = response.headers();
Ok({
#[allow(unused_mut)]
let mut output = crate::operation::invoke_agent::builders::InvokeAgentOutputBuilder::default();
output = output.set_completion(Some(crate::protocol_serde::shape_invoke_agent_output::de_completion_payload(
_response_body,
)?));
output = output.set_content_type(
crate::protocol_serde::shape_invoke_agent_output::de_content_type_header(_response_headers).map_err(|_| {
crate::operation::invoke_agent::InvokeAgentError::unhandled(
"Failed to parse contentType from header `x-amzn-bedrock-agent-content-type",
)
})?,
);
output = output.set_memory_id(
crate::protocol_serde::shape_invoke_agent_output::de_memory_id_header(_response_headers).map_err(|_| {
crate::operation::invoke_agent::InvokeAgentError::unhandled("Failed to parse memoryId from header `x-amz-bedrock-agent-memory-id")
})?,
);
output = output.set_session_id(
crate::protocol_serde::shape_invoke_agent_output::de_session_id_header(_response_headers).map_err(|_| {
crate::operation::invoke_agent::InvokeAgentError::unhandled("Failed to parse sessionId from header `x-amz-bedrock-agent-session-id")
})?,
);
output._set_request_id(::aws_types::request_id::RequestId::request_id(_response_headers).map(str::to_string));
crate::serde_util::invoke_agent_output_output_correct_errors(output)
.build()
.map_err(crate::operation::invoke_agent::InvokeAgentError::unhandled)?
})
}
#[allow(clippy::unnecessary_wraps)]
pub fn de_invoke_agent_http_error(
_response_status: u16,
_response_headers: &::aws_smithy_runtime_api::http::Headers,
_response_body: &[u8],
) -> std::result::Result<crate::operation::invoke_agent::InvokeAgentOutput, crate::operation::invoke_agent::InvokeAgentError> {
#[allow(unused_mut)]
let mut generic_builder = crate::protocol_serde::parse_http_error_metadata(_response_status, _response_headers, _response_body)
.map_err(crate::operation::invoke_agent::InvokeAgentError::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::invoke_agent::InvokeAgentError::unhandled(generic)),
};
let _error_message = generic.message().map(|msg| msg.to_owned());
Err(match error_code {
"AccessDeniedException" => crate::operation::invoke_agent::InvokeAgentError::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::invoke_agent::InvokeAgentError::unhandled)?;
let output = output.meta(generic);
output.build()
};
if tmp.message.is_none() {
tmp.message = _error_message;
}
tmp
}),
"BadGatewayException" => crate::operation::invoke_agent::InvokeAgentError::BadGatewayException({
#[allow(unused_mut)]
let mut tmp = {
#[allow(unused_mut)]
let mut output = crate::types::error::builders::BadGatewayExceptionBuilder::default();
output = crate::protocol_serde::shape_bad_gateway_exception::de_bad_gateway_exception_json_err(_response_body, output)
.map_err(crate::operation::invoke_agent::InvokeAgentError::unhandled)?;
let output = output.meta(generic);
output.build()
};
if tmp.message.is_none() {
tmp.message = _error_message;
}
tmp
}),
"ConflictException" => crate::operation::invoke_agent::InvokeAgentError::ConflictException({
#[allow(unused_mut)]
let mut tmp = {
#[allow(unused_mut)]
let mut output = crate::types::error::builders::ConflictExceptionBuilder::default();
output = crate::protocol_serde::shape_conflict_exception::de_conflict_exception_json_err(_response_body, output)
.map_err(crate::operation::invoke_agent::InvokeAgentError::unhandled)?;
let output = output.meta(generic);
output.build()
};
if tmp.message.is_none() {
tmp.message = _error_message;
}
tmp
}),
"DependencyFailedException" => crate::operation::invoke_agent::InvokeAgentError::DependencyFailedException({
#[allow(unused_mut)]
let mut tmp = {
#[allow(unused_mut)]
let mut output = crate::types::error::builders::DependencyFailedExceptionBuilder::default();
output = crate::protocol_serde::shape_dependency_failed_exception::de_dependency_failed_exception_json_err(_response_body, output)
.map_err(crate::operation::invoke_agent::InvokeAgentError::unhandled)?;
let output = output.meta(generic);
output.build()
};
if tmp.message.is_none() {
tmp.message = _error_message;
}
tmp
}),
"InternalServerException" => crate::operation::invoke_agent::InvokeAgentError::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::invoke_agent::InvokeAgentError::unhandled)?;
let output = output.meta(generic);
output.build()
};
if tmp.message.is_none() {
tmp.message = _error_message;
}
tmp
}),
"ModelNotReadyException" => crate::operation::invoke_agent::InvokeAgentError::ModelNotReadyException({
#[allow(unused_mut)]
let mut tmp = {
#[allow(unused_mut)]
let mut output = crate::types::error::builders::ModelNotReadyExceptionBuilder::default();
output = crate::protocol_serde::shape_model_not_ready_exception::de_model_not_ready_exception_json_err(_response_body, output)
.map_err(crate::operation::invoke_agent::InvokeAgentError::unhandled)?;
let output = output.meta(generic);
output.build()
};
if tmp.message.is_none() {
tmp.message = _error_message;
}
tmp
}),
"ResourceNotFoundException" => crate::operation::invoke_agent::InvokeAgentError::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::invoke_agent::InvokeAgentError::unhandled)?;
let output = output.meta(generic);
output.build()
};
if tmp.message.is_none() {
tmp.message = _error_message;
}
tmp
}),
"ServiceQuotaExceededException" => crate::operation::invoke_agent::InvokeAgentError::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::invoke_agent::InvokeAgentError::unhandled)?;
let output = output.meta(generic);
output.build()
};
if tmp.message.is_none() {
tmp.message = _error_message;
}
tmp
}),
"ThrottlingException" => crate::operation::invoke_agent::InvokeAgentError::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::invoke_agent::InvokeAgentError::unhandled)?;
let output = output.meta(generic);
output.build()
};
if tmp.message.is_none() {
tmp.message = _error_message;
}
tmp
}),
"ValidationException" => crate::operation::invoke_agent::InvokeAgentError::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::invoke_agent::InvokeAgentError::unhandled)?;
let output = output.meta(generic);
output.build()
};
if tmp.message.is_none() {
tmp.message = _error_message;
}
tmp
}),
_ => crate::operation::invoke_agent::InvokeAgentError::generic(generic),
})
}
pub fn ser_invoke_agent_headers(
input: &crate::operation::invoke_agent::InvokeAgentInput,
mut builder: ::http_1x::request::Builder,
) -> std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
if let ::std::option::Option::Some(inner_1) = &input.source_arn {
let formatted_2 = inner_1.as_str();
let header_value = formatted_2;
let header_value: ::http_1x::HeaderValue = header_value.parse().map_err(|err| {
::aws_smithy_types::error::operation::BuildError::invalid_field(
"source_arn",
format!("`{}` cannot be used as a header value: {}", &header_value, err),
)
})?;
builder = builder.header("x-amz-source-arn", header_value);
}
Ok(builder)
}
pub fn ser_invoke_agent_input(
input: &crate::operation::invoke_agent::InvokeAgentInput,
) -> ::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_invoke_agent_input::ser_invoke_agent_input_input(&mut object, input)?;
object.finish();
Ok(::aws_smithy_types::body::SdkBody::from(out))
}