Ykeremy/revert litellm upgrade (#204)

2024-04-17 11:51:14 -07:00
parent 2ae381a5e7
commit 3c06d44f6f
2 changed files with 4 additions and 5 deletions
--- a/skyvern/forge/sdk/api/llm/models.py
+++ b/skyvern/forge/sdk/api/llm/models.py
@@ -43,9 +43,8 @@ class LLMRouterConfig(LLMConfig):
        "simple-shuffle",
        "least-busy",
        "usage-based-routing",
-        "usage-based-routing-v2",
        "latency-based-routing",
-    ] = "usage-based-routing-v2"
+    ] = "usage-based-routing"
    num_retries: int = 2
    retry_delay_seconds: int = 15
    set_verbose: bool = False