phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +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 4475f2bbdd
commit 05d720075b
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -4629,9 +4629,11 @@ class CustomStreamWrapper:
chunk = self.completion_stream chunk = self.completion_stream
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: