vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix LiteLLMParams typing (#2285)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-05-04 01:02:01 -07:00
committed by GitHub
parent 8b834436b5
commit dbcc48833e
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
skyvern/forge/sdk/api/llm/config_registry.py
View File

@@ -747,7 +747,7 @@ if settings.ENABLE_VERTEX_AI:
add_assistant_prefix=False, add_assistant_prefix=False,
max_completion_tokens=65535, max_completion_tokens=65535,
litellm_params=LiteLLMParams( litellm_params=LiteLLMParams(
vertex_credentials=settings.VERTEX_CREDENTIALS, # type: ignore vertex_credentials=settings.VERTEX_CREDENTIALS,
), ),
), ),
) )
@@ -760,7 +760,7 @@ if settings.ENABLE_VERTEX_AI:
add_assistant_prefix=False, add_assistant_prefix=False,
max_completion_tokens=65535, max_completion_tokens=65535,
litellm_params=LiteLLMParams( litellm_params=LiteLLMParams(
vertex_credentials=settings.VERTEX_CREDENTIALS, # type: ignore vertex_credentials=settings.VERTEX_CREDENTIALS,
), ),
), ),
) )

3
skyvern/forge/sdk/api/llm/models.py
View File

@@ -9,11 +9,12 @@ from skyvern.forge.sdk.schemas.task_v2 import TaskV2, Thought
from skyvern.forge.sdk.settings_manager import SettingsManager from skyvern.forge.sdk.settings_manager import SettingsManager
class LiteLLMParams(TypedDict): class LiteLLMParams(TypedDict, total=False):
api_key: str | None api_key: str | None
api_version: str | None api_version: str | None
api_base: str | None api_base: str | None
model_info: dict[str, Any] | None model_info: dict[str, Any] | None
vertex_credentials: str | None
@dataclass(frozen=True) @dataclass(frozen=True)