vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-15 17:59:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

gpt4o mini pricing for massive (#957)

Browse source
This commit is contained in:
Shuchang Zheng 2024-10-10 22:57:54 -07:00 • committed by GitHub
parent c239927c72
commit b536f1d27f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
skyvern/forge/sdk/api/llm/models.py
View file

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