From 0a9cabdf8807654f62abe45dd26635bce4fb9866 Mon Sep 17 00:00:00 2001
From: Vansh <vanshkvanshk0@gmail.com>
Date: Fri, 2 Oct 2026 13:27:46 +0530
Subject: [PATCH] fix(litellm): keep basedpyright reportUnknownArgumentType
 budget flat for vispark

---
 litellm/__init__.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/litellm/__init__.py b/litellm/__init__.py
index b2dd33e192a..83c734e9000 100644
--- a/litellm/__init__.py
+++ b/litellm/__init__.py
@@ -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":