mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
fix(test_amazing_vertex_completion.py): fix test to check if content is none
This commit is contained in:
parent
bd75498913
commit
fa297b67ca
1 changed files with 2 additions and 1 deletions
|
@ -281,6 +281,7 @@ async def test_async_vertexai_streaming_response():
|
|||
complete_response = ""
|
||||
async for chunk in response:
|
||||
print(f"chunk: {chunk}")
|
||||
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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue