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

Ykeremy/fix router pass empty redis password (#143)

Browse source
This commit is contained in:
Kerem Yilmaz 2024-04-01 16:28:46 -07:00 • committed by GitHub
parent f175545399
commit 3cc07c9d20
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 35 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -33,8 +33,10 @@ class LLMRouterModelConfig:
@dataclass(frozen=True)
class LLMRouterConfig(LLMConfig):
model_list: list[LLMRouterModelConfig]
# All three redis parameters are required. Even if there isn't a password, it should be an empty string.
redis_host: str
redis_port: int
redis_password: str
main_model_group: str
fallback_model_group: str | None = None
routing_strategy: Literal[
@ -45,7 +47,7 @@ class LLMRouterConfig(LLMConfig):
] = "usage-based-routing"
num_retries: int = 2
retry_delay_seconds: int = 15
set_verbose: bool = True
set_verbose: bool = False
class LLMAPIHandler(Protocol):