diff --git a/litellm-rust/crates/inference-messages/tests/messages/request.rs b/litellm-rust/crates/inference-messages/tests/messages/request.rs index df914fe27e1..93bbbb071ba 100644 --- a/litellm-rust/crates/inference-messages/tests/messages/request.rs +++ b/litellm-rust/crates/inference-messages/tests/messages/request.rs @@ -385,7 +385,7 @@ async fn safeguards_add_the_dangerous_tool_use_beta_only_where_the_beta_map_has_ #[tokio::test] async fn null_safeguards_are_dropped_instead_of_sent_without_their_beta( call: MessagesCall, - #[values("anthropic", "azure_ai", "deepseek", "minimax")] provider: &str, + #[values("anthropic", "azure_ai", "bedrock_mantle", "deepseek", "minimax")] provider: &str, ) { let upstream = upstream([message_response()]).await; diff --git a/tests/integration/providers/_anthropic_safeguards_beta_support.py b/tests/integration/providers/_anthropic_safeguards_beta_support.py index 4211a80240a..0307ea11724 100644 --- a/tests/integration/providers/_anthropic_safeguards_beta_support.py +++ b/tests/integration/providers/_anthropic_safeguards_beta_support.py @@ -44,7 +44,7 @@ DEEPSEEK_TARGET: Final = "/anthropic/v1/messages" SCRIPTED_TARGETS: Final = frozenset({ANTHROPIC_TARGET, DEEPSEEK_TARGET, VERTEX_TARGET, VERTEX_STREAM_TARGET}) REPLY_TEXT: Final = "safeguards beta control" EXTRA_INPUTS_MESSAGE: Final = "safeguards: Extra inputs are not permitted" -NOT_A_LIST_MESSAGE: Final = "safeguards: Input should be a valid list" +NOT_A_LIST_MESSAGE: Final = "safeguards: Input should be a valid array" OVERLOADED_MESSAGE: Final = "Overloaded" SAFEGUARDS: Final[tuple[JsonValue, ...]] = ( {