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

fix(tokenlab): add the model set without new type findings

Browse source 
The lint job failed on the basedpyright budget: the new provider branch
passed an untyped key to an untyped set, which counts as one more
reportUnknownArgumentType. The set is now Set[str] and the branch narrows
the key with isinstance, so the TokenLab lines add no unknown-argument,
unknown-variable or missing-type-argument findings. Model cost map keys
are always strings, so behaviour is unchanged.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
hedging8563 2026-10-03 05:47:56 +08:00
parent 9da5e2d4be
commit b5f6bffd25
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/__init__.py
View file

@ -687,7 +687,7 @@ qwen_ai_platform_models: Set = set()
moonshot_models: Set = set()
publicai_models: Set = set()
darkbloom_models: Set = set()
tokenlab_models: Set = set()
tokenlab_models: Set[str] = set()
v0_models: Set = set()
morph_models: Set = set()
lambda_ai_models: Set = set()
@ -953,7 +953,7 @@ def _populate_provider_model_sets(model_cost_map: Dict) -> None:
publicai_models.add(key)
elif value.get("litellm_provider") == "darkbloom":
darkbloom_models.add(key)
elif value.get("litellm_provider") == "tokenlab":
elif value.get("litellm_provider") == "tokenlab" and isinstance(key, str):
tokenlab_models.add(key)
elif value.get("litellm_provider") == "v0":
v0_models.add(key)