(fix-ci-cd) skip deep infra 429 errors

2024-03-13 22:05:16 -07:00 · 2024-03-13 22:05:16 -07:00 · ebfefe61ea
commit ebfefe61ea
parent e3cc0da5f1
1 changed files with 2 additions and 0 deletions
--- a/litellm/tests/test_streaming.py
+++ b/litellm/tests/test_streaming.py
@ -541,6 +541,8 @@ def test_completion_deep_infra_stream():
            raise Exception("Empty response received")
        print(f"completion_response: {complete_response}")
    except Exception as e:
+        if "Model busy, retry later" in str(e):
+            pass
        pytest.fail(f"Error occurred: {e}")