diff --git a/litellm-rust/crates/python-bridge/src/secrets/mod.rs b/litellm-rust/crates/python-bridge/src/secrets/mod.rs index 6d1677be8fd..d1553cf5105 100644 --- a/litellm-rust/crates/python-bridge/src/secrets/mod.rs +++ b/litellm-rust/crates/python-bridge/src/secrets/mod.rs @@ -23,12 +23,11 @@ use crate::python_settings::missing_module; pub(crate) fn source(py: Python<'_>) -> PyResult> { match PythonSecrets::new(py) { Ok(secrets) => Ok(Arc::new(secrets)), - Err(error) if missing_module(py, &error, "litellm")? => Ok(Arc::new( - EnvironmentSecrets::python_compatible(crate::http::host_client( - py, - litellm_http::ClientVariant::Provider, - )?), - )), + Err(error) if missing_module(py, &error, "litellm")? => { + Ok(Arc::new(EnvironmentSecrets::python_compatible( + crate::http::host_client(py, litellm_http::ClientVariant::Provider)?, + ))) + } Err(error) => Err(error), } } diff --git a/litellm-rust/crates/python-bridge/src/secrets/python.rs b/litellm-rust/crates/python-bridge/src/secrets/python.rs index 026b993fe73..eba1596878f 100644 --- a/litellm-rust/crates/python-bridge/src/secrets/python.rs +++ b/litellm-rust/crates/python-bridge/src/secrets/python.rs @@ -46,7 +46,12 @@ impl PythonSecrets { PyTuple::new(py, names) .and_then(|names| read_secrets.bind(py).call1((names,))) .and_then(|values| values.extract::>>()) - .map(|values| values.into_iter().map(|value| value.map(SecretValue::new)).collect()) + .map(|values| { + values + .into_iter() + .map(|value| value.map(SecretValue::new)) + .collect() + }) .map_err(|error| external_error(py, error)) }) .await @@ -196,7 +201,10 @@ def read_secrets(names): .await .unwrap(); - assert_eq!(resolved.get("MISTRAL_API_KEY").as_deref(), Some("vault-key")); + assert_eq!( + resolved.get("MISTRAL_API_KEY").as_deref(), + Some("vault-key") + ); assert_eq!(resolved.get("OTHER"), None); assert_eq!( resolved.get("MISTRAL_API_BASE").as_deref(),