From b5968a3ed8aca95e675cd288ff145a50b72dd05b Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 18 Nov 2023 17:17:39 -0800
Subject: [PATCH] (test) async for chunk

---
 litellm/tests/test_async_fn.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/litellm/tests/test_async_fn.py b/litellm/tests/test_async_fn.py
index 04045c1393..cd459dc1d8 100644
--- a/litellm/tests/test_async_fn.py
+++ b/litellm/tests/test_async_fn.py
@@ -111,6 +111,8 @@ def test_get_response_streaming():
             i = 0
             async for chunk in response:
                 token = chunk["choices"][0]["delta"].get("content", "")
+                if token == None:
+                    continue # openai v1.0.0 returns content=None
                 output += token
             assert output is not None, "output cannot be None."
             assert isinstance(output, str), "output needs to be of type str"