test(test_amazing_vertex_completion.py): fix test to handle rate limit errors

2026-08-06 10:21:32 +00:00 · 2024-02-26 22:07:33 -08:00
parent 6b14c6287b
commit 647e867cf8
1 changed files with 2 additions and 0 deletions
@@ -222,6 +222,8 @@ async def test_async_vertexai_response():
                model=model, messages=messages, temperature=0.7, timeout=5
            )
            print(f"response: {response}")
+        except litellm.RateLimitError as e:
+            pass
        except litellm.Timeout as e:
            pass
        except Exception as e: