phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(utils.py): adding more logging for streaming test

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-13 16:54:16 -08:00
parent 1032b49f23
commit 39e784fb8b
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/utils.py
View file

@ -4630,8 +4630,10 @@ class CustomStreamWrapper:
else: else:
chunk = next(self.completion_stream) chunk = next(self.completion_stream)
print_verbose(f"chunk in __next__: {chunk}")
if chunk is not None: if chunk is not None:
response = self.chunk_creator(chunk=chunk) response = self.chunk_creator(chunk=chunk)
print_verbose(f"response in __next__: {response}")
if response is not None: if response is not None:
return response return response
except StopIteration: except StopIteration: