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

chore(rust): trace Bedrock native bridge entry

Browse source 
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Ishaan Jaff 2026-07-30 21:27:39 -07:00
parent 2008816278
commit d77f8921b2
3 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm-rust/crates/core/src/messages/handler.rs
View file

@ -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

1
litellm-rust/crates/core/src/messages/mod.rs
View file

@ -21,6 +21,7 @@ use prepare::prepare_messages_call;
use types::{AnthropicMessagesResponse, MessagesRequest};
pub async fn messages(request: MessagesRequest<'_>) -> CoreResult<AnthropicMessagesResponse> {
println!("ENTERED RUST MESSAGES");
execute_messages_provider_call(prepare_messages_call(request)?).await
}

2
litellm/llms/custom_httpx/llm_http_handler.py
View file

@ -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