agent-writer/utils/stream.py

9 lines
237 B
Python

def debug_print_stream(stream):
for chunk in stream:
message = chunk['messages'][-1]
if isinstance(message, tuple):
print(chunk)
else:
message.pretty_print()
print("\n---")