scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 181

(ci/cd) fix test_vertex_ai_stream

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-15 16:32:40 -07:00
parent f2e8b2500f
commit f138c15859
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_amazing_vertex_completion.py
View file

@ -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 = ""