From d77f8921b2715e0b9939918284dd5a88bcf285d5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 30 Jul 2026 21:27:39 -0700 Subject: [PATCH] chore(rust): trace Bedrock native bridge entry Co-authored-by: Cursor --- litellm-rust/crates/core/src/messages/handler.rs | 2 ++ litellm-rust/crates/core/src/messages/mod.rs | 1 + litellm/llms/custom_httpx/llm_http_handler.py | 2 +- 3 files changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm-rust/crates/core/src/messages/handler.rs b/litellm-rust/crates/core/src/messages/handler.rs index 1c895f66eba..a7fc15f8876 100644 --- a/litellm-rust/crates/core/src/messages/handler.rs +++ b/litellm-rust/crates/core/src/messages/handler.rs @@ -16,6 +16,8 @@ pub(super) async fn execute_messages_provider_call( request_builder = request_builder.timeout(duration); } + println!("IN EXECUTE MESSAGES PROVIDER CALL"); + let response = request_builder .send() .await diff --git a/litellm-rust/crates/core/src/messages/mod.rs b/litellm-rust/crates/core/src/messages/mod.rs index acb36d89daf..873a0b41e18 100644 --- a/litellm-rust/crates/core/src/messages/mod.rs +++ b/litellm-rust/crates/core/src/messages/mod.rs @@ -21,6 +21,7 @@ use prepare::prepare_messages_call; use types::{AnthropicMessagesResponse, MessagesRequest}; pub async fn messages(request: MessagesRequest<'_>) -> CoreResult { + println!("ENTERED RUST MESSAGES"); execute_messages_provider_call(prepare_messages_call(request)?).await } diff --git a/litellm/llms/custom_httpx/llm_http_handler.py b/litellm/llms/custom_httpx/llm_http_handler.py index ec1301e5923..b6b799a1089 100644 --- a/litellm/llms/custom_httpx/llm_http_handler.py +++ b/litellm/llms/custom_httpx/llm_http_handler.py @@ -2273,7 +2273,7 @@ class BaseLLMHTTPHandler: request_body: dict, timeout: float | httpx.Timeout | None, ) -> AnthropicMessagesResponse | None: - if custom_llm_provider not in ("azure_ai", "anthropic"): + if custom_llm_provider not in ("azure_ai", "anthropic", "bedrock"): return None if litellm_params.get("rust") is not True and not BaseLLMHTTPHandler._rust_env_enabled(): return None