fix LiteLLMParams typing (#2285)

2025-05-04 01:02:01 -07:00
parent 8b834436b5
commit dbcc48833e
2 changed files with 4 additions and 3 deletions
--- a/skyvern/forge/sdk/api/llm/models.py
+++ b/skyvern/forge/sdk/api/llm/models.py
@@ -9,11 +9,12 @@ from skyvern.forge.sdk.schemas.task_v2 import TaskV2, Thought
 from skyvern.forge.sdk.settings_manager import SettingsManager


-class LiteLLMParams(TypedDict):
+class LiteLLMParams(TypedDict, total=False):
    api_key: str | None
    api_version: str | None
    api_base: str | None
    model_info: dict[str, Any] | None
+    vertex_credentials: str | None


@dataclass(frozen=True)