generator1.py

def mock_api_stream():
    MESSAGE_CONTENTS = [
        "mess",
        "age",
        " start",
        ":",
        " This",
        " is",
        " a",
        " nice ",
        "mess",
        "age",
        " buddy",
        "message",
        " end",
        ":",
    ]
    for x in MESSAGE_CONTENTS:
        yield x

start_message = "message start:"
end_message = "message end:"

def handle_stream(token):
    """
    Goal: do not yield the start_message or end_message
    print below:
    {'message_partial': ' This'}
    {'message_partial': ' is a'}
    {'message_partial': 'message : hello'}
    """
    stream = mock_api_stream()
    for token in stream:
        yield {"message_partial": token}
        
def main():
    for token in handle_stream():
        print({"message_partial": token})


main()