mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-10 03:28:53 +00:00
fix(litellm): satisfy type-discipline budget for vispark cast
This commit is contained in:
parent
0a9cabdf88
commit
130138b771
1 changed files with 2 additions and 2 deletions
|
|
@ -957,8 +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":
|
||||
# cast: model keys are strings; avoids adding a reportUnknownArgumentType error
|
||||
vispark_models.add(cast(str, key))
|
||||
# cast keeps the basedpyright reportUnknownArgumentType budget flat (keys are model-name strings)
|
||||
vispark_models.add(cast(str, key)) # cast-ok: model keys are strings
|
||||
elif value.get("litellm_provider") == "morph":
|
||||
morph_models.add(key)
|
||||
elif value.get("litellm_provider") == "lambda_ai":
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue