test: handle predibase api failures

2024-07-31 19:39:58 -07:00 · 2024-07-31 19:39:58 -07:00 · 83f638100e
commit 83f638100e
parent e1696bf426
1 changed files with 2 additions and 0 deletions
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@ -273,6 +273,8 @@ async def test_completion_predibase():
        print(response)
    except litellm.Timeout as e:
        pass
+    except litellm.ServiceUnavailableError as e:
+        pass
    except Exception as e:
        pytest.fail(f"Error occurred: {e}")