From f138c15859e8be0c46ec96e4c54c897fc3fa2fd0 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 15 May 2024 16:32:40 -0700 Subject: [PATCH] (ci/cd) fix test_vertex_ai_stream --- litellm/tests/test_amazing_vertex_completion.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index ad3fb3cc3..b747ca880 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -376,9 +376,8 @@ def test_vertex_ai_stream(): print("making request", model) response = completion( model=model, - messages=[ - {"role": "user", "content": "write 10 line code code for saying hi"} - ], + messages=[{"role": "user", "content": "hello tell me a short story"}], + max_tokens=15, stream=True, ) completed_str = ""