diff --git a/rig-integrations/rig-bedrock/src/types/assistant_content.rs b/rig-integrations/rig-bedrock/src/types/assistant_content.rs index d4a2e8e8cd..9521ec21b9 100644 --- a/rig-integrations/rig-bedrock/src/types/assistant_content.rs +++ b/rig-integrations/rig-bedrock/src/types/assistant_content.rs @@ -42,11 +42,16 @@ impl TryFrom for completion::CompletionResponse = + AwsConverseOutput(internal).try_into().unwrap(); + assert_eq!( + completion.usage.input_tokens, 20_100, + "cache tokens fold into input" + ); + assert_eq!(completion.usage.total_tokens, 20_150); + } + #[test] fn aws_content_block_to_assistant_content() { let content_block = aws_bedrock::ContentBlock::Text("text".into());