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

test(ocr): assert rust path resolves key via secret manager

Browse source
This commit is contained in:
Ishaan Jaffer 2026-06-22 21:05:16 -07:00
parent a21a80751d
commit 097bab9441
No known key found for this signature in database
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/test_litellm/ocr/test_rust_bridge.py
View file

@ -119,3 +119,19 @@ def test_ocr_skips_rust_when_disabled(monkeypatch, fake_bridge):
assert called.get("hit") is True
assert fake_bridge == [] # Rust bridge never invoked
def test_ocr_resolves_key_via_secret_manager(monkeypatch, fake_bridge):
"""No explicit api_key: the Rust path must resolve it via get_secret_str so
secret-manager backends (AWS/Azure/GCP/Vault) work, matching the Python path.
Rust's own fallback only reads the process env, so Python resolves and passes it.
"""
import litellm.secret_managers.main as secret_mgr
monkeypatch.delenv("MISTRAL_API_KEY", raising=False)
monkeypatch.setattr(secret_mgr, "get_secret_str", lambda name: "sk-from-vault")
litellm.use_litellm_rust()
litellm.ocr(model=MODEL, document=DOCUMENT) # no api_key passed
assert fake_bridge[0]["api_key"] == "sk-from-vault"