From 1e539895a2d07005bc8528663b0b8ff2c12e7e2b Mon Sep 17 00:00:00 2001 From: mateo-berri <277851410+mateo-berri@users.noreply.github.com> Date: Mon, 12 Oct 2026 16:49:35 -0700 Subject: [PATCH] test(rust): check the dangerous-tool-use beta reaches the wire with safeguards --- .../crates/inference-messages/tests/messages/request.rs | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm-rust/crates/inference-messages/tests/messages/request.rs b/litellm-rust/crates/inference-messages/tests/messages/request.rs index 7c02d2a2e5a..543cdf1d8a1 100644 --- a/litellm-rust/crates/inference-messages/tests/messages/request.rs +++ b/litellm-rust/crates/inference-messages/tests/messages/request.rs @@ -301,6 +301,10 @@ fn sent_betas(request: &wiremock::Request) -> BetaSet { json!({"tools": [{"type": "advisor_20260301", "name": "advisor", "model": MODEL}]}), &[AnthropicBeta::AdvisorTool20260301] )] +#[case::safeguards( + json!({"safeguards": [{"type": "dangerous_tool_use", "classifier_context": {"v": 1}}]}), + &[AnthropicBeta::DangerousToolUse20260903] +)] #[case::several_features_at_once( json!({"speed": "fast", "output_format": {"type": "json_schema"}}), &[AnthropicBeta::StructuredOutputs20251113, AnthropicBeta::FastMode20260201]