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

fix LiteLLMParams (#958)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-10-10 23:24:27 -07:00
committed by GitHub
parent b536f1d27f
commit 60069a6256
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,5 +1,5 @@
from dataclasses import dataclass, field from dataclasses import dataclass, field
from typing import Any, Awaitable, Literal, Optional, Protocol from typing import Any, Awaitable, Literal, Optional, Protocol, TypedDict
from litellm import AllowedFailsPolicy from litellm import AllowedFailsPolicy
@@ -7,12 +7,11 @@ from skyvern.forge.sdk.models import Step
from skyvern.forge.sdk.settings_manager import SettingsManager from skyvern.forge.sdk.settings_manager import SettingsManager
@dataclass class LiteLLMParams(TypedDict):
class LiteLLMParams:
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] = field(default_factory=dict) model_info: dict[str, Any] | None
@dataclass(frozen=True) @dataclass(frozen=True)