scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-19 00:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

wip

Browse source
This commit is contained in:
Yujong Lee 2026-08-29 12:46:18 -07:00 • committed by GitHub
parent bff6268ee2
commit 14790ef4f1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -361,20 +361,12 @@ def _provider_specs(
) -> tuple[ProviderSpec, ...]:
mistral_key: Final = os.environ.get("MISTRAL_API_KEY") or os.environ.get("LITELLM_API_KEY")
reducto_key: Final = os.environ.get("REDUCTO_API_KEY")
mistral_model: Final = os.environ.get("MISTRAL_OCR_MODEL")
reducto_model: Final = os.environ.get("REDUCTO_OCR_MODEL")
mistral_models: Final = (
(mistral_model,)
if mistral_model is not None
else (
("mistral/mistral-ocr-2512", "mistral/mistral-ocr-4-0", "mistral/mistral-ocr-4-1")
if requests_only or all_models
else ("mistral/mistral-ocr-latest",)
)
)
reducto_models: Final = (
(reducto_model,) if reducto_model is not None else ("reducto/parse-v3", "reducto/parse-legacy")
("mistral/mistral-ocr-2512", "mistral/mistral-ocr-4-0", "mistral/mistral-ocr-4-1")
if requests_only or all_models
else ("mistral/mistral-ocr-latest",)
)
reducto_models: Final = ("reducto/parse-v3", "reducto/parse-legacy")
mistral_specs: Final = (
tuple(
ProviderSpec(