scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-12 03:48:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 57

style(rust): format Bedrock regression tests

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Yujong Lee 2026-10-11 22:29:41 +00:00
parent d8f7217b71
commit 92789feba7
2 changed files with 6 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm-rust/crates/llms/src/bedrock/claude_platform/messages/transformation.rs
View file

@ -340,9 +340,7 @@ mod tests {
},
..Default::default()
};
let env = |name: &str| {
(name == "ANTHROPIC_AWS_WORKSPACE_ID").then(|| "workspace".into())
};
let env = |name: &str| (name == "ANTHROPIC_AWS_WORKSPACE_ID").then(|| "workspace".into());
let url = BEDROCK_CLAUDE_PLATFORM_MESSAGES_CONFIG
.get_complete_url(None, "claude-test", &params, false, &env)
.unwrap();

10
litellm-rust/crates/llms/src/bedrock_mantle/messages/transformation.rs
View file

@ -360,10 +360,7 @@ mod tests {
#[rstest]
#[case::aws_default_region("AWS_DEFAULT_REGION", "eu-west-1")]
fn aws_default_region_selects_mantle_endpoint(
#[case] env_name: &str,
#[case] region: &str,
) {
fn aws_default_region_selects_mantle_endpoint(#[case] env_name: &str, #[case] region: &str) {
let url = BEDROCK_MANTLE_MESSAGES_CONFIG
.get_complete_url(
None,
@ -516,7 +513,10 @@ mod tests {
}))
.unwrap();
let headers = BEDROCK_MANTLE_MESSAGES_CONFIG.request_headers(headers, &request);
let actual: BetaSet = header_value(&headers, "anthropic-beta").unwrap().parse().unwrap();
let actual: BetaSet = header_value(&headers, "anthropic-beta")
.unwrap()
.parse()
.unwrap();
let expected: BetaSet = expected.into_iter().collect();
assert_eq!(actual, expected);
}