scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 71

fix(tokenlab): satisfy provider quality gates

Browse source
This commit is contained in:
hedging8563 2026-08-16 05:17:03 +08:00
parent ffd39f7167
commit 3a9257d3c6
2 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/__init__.py
View file

@ -647,7 +647,7 @@ dashscope_models: Set = set()
moonshot_models: Set = set()
publicai_models: Set = set()
darkbloom_models: Set = set()
tokenlab_models: Set = set()
tokenlab_models: Set = set() # mutable-ok: populated while loading the static model registry
v0_models: Set = set()
morph_models: Set = set()
lambda_ai_models: Set = set()

3
litellm/litellm_core_utils/get_llm_provider_logic.py
View file

@ -345,7 +345,8 @@ def get_llm_provider(
dynamic_api_key = get_secret_str("WANDB_API_KEY")
elif endpoint == "https://api.tokenlab.sh/v1":
custom_llm_provider = "tokenlab"
dynamic_api_key = get_secret_str("TOKENLAB_API_KEY")
tokenlab_config: Final = JSONProviderRegistry.get("tokenlab")
dynamic_api_key = get_secret_str(tokenlab_config.api_key_env) if tokenlab_config else None
elif endpoint == "https://pinstripes.io/v1":
custom_llm_provider = "pinstripes"
dynamic_api_key = get_secret_str("PINSTRIPES_API_KEY")