vertex_ai/codestral@2405 is very unstable - handle their instability …

…in our tests
haringsrob · Nov 18, 2024 · e1ca956 · e1ca956
1 parent 585b54e
commit e1ca956
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py
@@ -995,6 +995,8 @@ async def test_partner_models_httpx(model, sync_mode):
         pass
     except litellm.APIConnectionError as e:
         pass
+    except litellm.ServiceUnavailableError as e:
+        pass
     except Exception as e:
         if "429 Quota exceeded" in str(e):
             pass

diff --git a/tests/local_testing/test_text_completion.py b/tests/local_testing/test_text_completion.py
@@ -4193,6 +4193,9 @@ async def test_completion_codestral_fim_api_stream(model):
     except litellm.APIConnectionError as e:
         print(e)
         pass
+    except litellm.ServiceUnavailableError as e:
+        print(e)
+        pass
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")