From 39e784fb8b81d7ca43fbcb142c37ec62e58835cd Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 13 Nov 2023 16:54:16 -0800
Subject: [PATCH] test(utils.py): adding more logging for streaming test

---
 litellm/utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/litellm/utils.py b/litellm/utils.py
index 630cb3b9f5..b4c9ad4914 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -4629,9 +4629,11 @@ class CustomStreamWrapper:
                     chunk = self.completion_stream
                 else:
                     chunk = next(self.completion_stream)
-
+                
+                print_verbose(f"chunk in __next__: {chunk}")
                 if chunk is not None:
                     response = self.chunk_creator(chunk=chunk)
+                    print_verbose(f"response in __next__: {response}")
                     if response is not None:
                         return response
         except StopIteration: