From 3c06d44f6f5b1dcceba14300e8d8158da66883df Mon Sep 17 00:00:00 2001
From: Kerem Yilmaz <kerem@skyvern.com>
Date: Wed, 17 Apr 2024 11:51:14 -0700
Subject: [PATCH] Ykeremy/revert litellm upgrade (#204)

---
 poetry.lock                         | 6 +++---
 skyvern/forge/sdk/api/llm/models.py | 3 +--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index 1ad4be01..3a2b88f8 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -2767,13 +2767,13 @@ files = [
 
 [[package]]
 name = "litellm"
-version = "1.35.8"
+version = "1.34.41"
 description = "Library to easily interface with LLM API providers"
 optional = false
 python-versions = "!=2.7.*,!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,!=3.6.*,!=3.7.*,>=3.8"
 files = [
-    {file = "litellm-1.35.8-py3-none-any.whl", hash = "sha256:0b328b37b6c40e5ece82f701bcb92e4d1d76ffb070e03f6c3abfbc6030bf07b8"},
-    {file = "litellm-1.35.8.tar.gz", hash = "sha256:0a8c031a01c19f2cfd9d873ff89bebc120917ec88f5656c1464a64d38e7c782d"},
+    {file = "litellm-1.34.41-py3-none-any.whl", hash = "sha256:ec6b1fb7f27178ad36f3be64e2f0fb73a637c0d7c128d5aa1d845414cc21be66"},
+    {file = "litellm-1.34.41.tar.gz", hash = "sha256:e8ad1a0b9cf4abea0471394db8c95f479e94e271bc1ce3d98204e1a6ed917e93"},
 ]
 
 [package.dependencies]
diff --git a/skyvern/forge/sdk/api/llm/models.py b/skyvern/forge/sdk/api/llm/models.py
index 2605bd82..9db88873 100644
--- 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