Streaming OpenAI API responses in Python without a library

stream_openai_api_no_library.py

import json
import requests


api_key = ""
url = "https://api.openai.com/v1/chat/completions"
headers = {
    "Authorization": f"Bearer {api_key}",
    "Content-Type": "application/json"
}
data = {
    "temperature": 0.0,
    "model": "gpt-4o",
    "messages": [
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "Tell me a joke."}
    ],
    "stream": True
}
response = requests.post(url, headers=headers, json=data, stream=True)
response.raise_for_status()

for line in response.iter_lines():
    line = line.decode("utf-8")

    if line.startswith("data: ") and not line.endswith("[DONE]"):
        data = json.loads(line[len("data: "):])
        chunk = data["choices"][0]["delta"].get("content", "")
        print(chunk, end="", flush=True)

print()