fix LiteLLMParams (#958)

This commit is contained in:
Shuchang Zheng
2024-10-10 23:24:27 -07:00
committed by GitHub
parent b536f1d27f
commit 60069a6256

View File

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