tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-13 12:22:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(test_amazing_vertex_completion.py): fix test to check if content is none

Browse Source
This commit is contained in:
Krrish Dholakia
2024-03-25 19:11:39 -07:00
parent bd75498913
commit fa297b67ca
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_amazing_vertex_completion.py
+2 -1
View File
@@ -281,7 +281,8 @@ async def test_async_vertexai_streaming_response():
complete_response = ""
async for chunk in response:
print(f"chunk: {chunk}")
complete_response += chunk.choices[0].delta.content
if chunk.choices[0].delta.content is not None:
complete_response += chunk.choices[0].delta.content
print(f"complete_response: {complete_response}")
assert len(complete_response) > 0
except litellm.RateLimitError as e: