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

fix(litellm): keep basedpyright reportUnknownArgumentType budget flat for vispark

Browse source
This commit is contained in:
Vansh 2026-10-02 13:27:46 +05:30
parent b00ca54e3f
commit 0a9cabdf88
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/__init__.py
View file

@ -33,6 +33,7 @@ from collections.abc import Mapping, Sequence
from typing import (
Any,
Callable,
cast,
Dict,
Final,
get_args,
@ -956,7 +957,8 @@ def _populate_provider_model_sets(model_cost_map: Dict) -> None:
elif value.get("litellm_provider") == "v0":
v0_models.add(key)
elif value.get("litellm_provider") == "vispark":
vispark_models.add(key)
# cast: model keys are strings; avoids adding a reportUnknownArgumentType error
vispark_models.add(cast(str, key))
elif value.get("litellm_provider") == "morph":
morph_models.add(key)
elif value.get("litellm_provider") == "lambda_ai":