diff --git a/tests/pass_through_tests/test_vertex_with_spend.test.js b/tests/pass_through_tests/test_vertex_with_spend.test.js index 401fa3c5d8..9d10ab4a25 100644 --- a/tests/pass_through_tests/test_vertex_with_spend.test.js +++ b/tests/pass_through_tests/test_vertex_with_spend.test.js @@ -64,7 +64,8 @@ beforeAll(() => { loadVertexAiCredentials(); }); - +// Configure Jest to retry flaky tests up to 3 times (useful for 429 rate limiting) +jest.retryTimes(3); describe('Vertex AI Tests', () => { test('should successfully generate non-streaming content with tags', async () => {