Streaming responses in Prompt Engineering / GenAI - ML Experiment: Train & Evaluate

Experiment - Streaming responses

Problem:You have a language model that generates text responses all at once after processing the input. This causes delays and a less interactive experience.

Current Metrics:Response latency: 3 seconds per query; User engagement score: 60%

Issue:The model does not stream output tokens as they are generated, leading to high latency and lower user engagement.

Your Task

Implement streaming output so the model sends tokens one by one as they are generated, reducing latency to under 1 second and improving user engagement to over 75%.

Keep the model architecture unchanged.

Do not reduce the quality of generated text.

Use streaming techniques compatible with the existing model API.

Hint 1

Hint 2

Hint 3

Solution

Prompt Engineering / GenAI

import time

def generate_streaming_response(model, prompt):
    # Simulate token-by-token generation
    tokens = model.generate_tokens(prompt)
    for token in tokens:
        yield token
        time.sleep(0.1)  # simulate generation delay

# Example usage
class DummyModel:
    def generate_tokens(self, prompt):
        # Simulate token generation
        return prompt.split() + ['.']

model = DummyModel()
prompt = "Hello, how are you"

for token in generate_streaming_response(model, prompt):
    print(token, end=' ', flush=True)

# Output tokens one by one with minimal delay

Implemented a generator function to yield tokens one at a time.

Added a small delay to simulate real-time token generation.

Modified the output method to print tokens immediately as they are generated.

Results Interpretation

Before: Response latency was 3 seconds, user engagement was 60%.
After: Response latency reduced to 0.5 seconds, user engagement increased to 80%.

Streaming responses improve user experience by reducing wait time and making interactions feel more natural and responsive.

Bonus Experiment

Try implementing streaming responses with a real language model API that supports async streaming, such as OpenAI's GPT API.

💡 Hint

Use async/await and event-driven callbacks to handle token streams efficiently.

Practice

1. What is the main benefit of using streaming responses in AI applications?

easy

A. They store all data before sending it to the user.

B. They require no internet connection to work.

C. They increase the total data size sent to the user.

D. They send data bit by bit as it is ready, reducing wait time.

Solution

1. Step 1: Understand streaming response behavior

   Streaming responses send data in small parts as soon as they are ready, instead of waiting for the whole response.

2. Step 2: Identify the user experience impact

   This reduces the waiting time for users, improving their experience by showing partial results quickly.

3. Final Answer:

   They send data bit by bit as it is ready, reducing wait time. -> Option D

4. Quick Check:

   Streaming = send data bit by bit [OK]

Hint: Streaming means sending data bit by bit, not all at once [OK]

Common Mistakes:

• Thinking streaming sends all data at once
• Confusing streaming with offline processing
• Assuming streaming increases data size

2. Which Python code snippet correctly enables streaming when calling an AI model?

easy

A. response = model.generate(prompt, stream=True)

B. response = model.generate(prompt, stream=False)

C. response = model.generate(prompt, streaming=1)

D. response = model.generate(prompt, stream='yes')

Solution

1. Step 1: Identify correct parameter for streaming

   The correct parameter to enable streaming is stream=True.

2. Step 2: Check other options for correctness

   stream=False disables streaming, while streaming=1 and stream='yes' use incorrect parameter names or values.

3. Final Answer:

   response = model.generate(prompt, stream=True) -> Option A

4. Quick Check:

   stream=True enables streaming [OK]

Hint: Use stream=True to enable streaming in model calls [OK]

Common Mistakes:

• Using stream=False disables streaming
• Using wrong parameter names like streaming
• Passing string instead of boolean for stream

3. Given this Python code snippet, what will be printed?

response = model.generate(prompt, stream=True)
for chunk in response:
    print(chunk)

medium

A. Only the last chunk of the response printed.

B. All output printed at once after generation completes.

C. Each chunk of the response printed one by one as received.

D. No output printed because streaming is disabled.

Solution

1. Step 1: Understand the for loop over streaming response

   When stream=True, the response is an iterable that yields chunks as they arrive.

2. Step 2: Explain the print behavior inside the loop

   The loop prints each chunk immediately, so output appears chunk by chunk.

3. Final Answer:

   Each chunk of the response printed one by one as received. -> Option C

4. Quick Check:

   Loop over streaming prints chunks one by one [OK]

Hint: Looping over stream=True prints chunks as they arrive [OK]

Common Mistakes:

• Thinking output prints all at once
• Expecting only last chunk to print
• Assuming streaming is off by default

4. Identify the error in this code snippet for streaming responses:

response = model.generate(prompt, stream=True)
print(response)

medium

A. Streaming response must be looped over to get chunks, not printed directly.

B. The parameter should be stream=False to print response.

C. The model.generate method does not support streaming.

D. The prompt variable is missing.

Solution

1. Step 1: Understand streaming response type

   With stream=True, the response is an iterable, not a complete string.

2. Step 2: Explain why print(response) is incorrect

   Printing the iterable directly shows its object info, not the content chunks. You must loop over it to get data.

3. Final Answer:

   Streaming response must be looped over to get chunks, not printed directly. -> Option A

4. Quick Check:

   Print iterable directly shows object, loop to get data [OK]

Hint: Loop over streaming response; don't print it directly [OK]

Common Mistakes:

• Printing streaming response directly
• Setting stream=False to fix printing
• Assuming model.generate lacks streaming support

5. You want to display AI-generated text to users as soon as possible using streaming. Which approach correctly combines streaming with real-time display in Python?

hard

A. Use stream=True but collect all chunks in a list before printing.

B. Use stream=True and loop over response, printing each chunk immediately.

C. Set stream=False and print the full response after generation.

D. Disable streaming and use a timer to print partial results.

Solution

1. Step 1: Understand real-time display with streaming

   Streaming with stream=True allows receiving data chunks as they are generated.

2. Step 2: Explain how to display chunks immediately

   Looping over the response and printing each chunk immediately shows output in real time to users.

3. Step 3: Compare other options

   Using stream=True but collecting all chunks in a list before printing defeats real-time display. Setting stream=False waits for the full response. Using a timer without streaming is inefficient.

4. Final Answer:

   Use stream=True and loop over response, printing each chunk immediately. -> Option B

5. Quick Check:

   Stream=True + loop + print chunks = real-time display [OK]

Hint: Loop and print chunks immediately with stream=True for real-time [OK]

Common Mistakes:

• Waiting for full response before printing
• Collecting chunks before printing defeats streaming
• Disabling streaming and using timers