Reaching CPU time limit when streaming AI responses

export const runtime = "edge";

import { NextRequest, NextResponse } from 'next/server';

const OPENAI_API_URL = "https://api.openai.com/v1/chat/completions";

export async function POST(req: NextRequest) {
    const { messages } = await req.json();

    const response = await fetch(OPENAI_API_URL, {
        method: "POST",
        headers: {
            "Content-Type": "application/json",
            "Authorization": `Bearer ${process.env.OPENAI_API_KEY}`,
        },
        body: JSON.stringify({
            model: "gpt-4.1",
            messages,
            temperature: 0,
            stream: true,
        }),
    });

    if (!response.ok || !response.body) {
        const error = await response.text();
        return new NextResponse(error, { status: response.status });
    }

    return new NextResponse(response.body, {
        headers: {
            "Content-Type": "text/event-stream",
            "Cache-Control": "no-cache",
            "Connection": "keep-alive",
        },
    });
}

export const runtime = "edge";

import { NextRequest, NextResponse } from 'next/server';

const OPENAI_API_URL = "https://api.openai.com/v1/chat/completions";

export async function POST(req: NextRequest) {
    const { messages } = await req.json();

    const response = await fetch(OPENAI_API_URL, {
        method: "POST",
        headers: {
            "Content-Type": "application/json",
            "Authorization": `Bearer ${process.env.OPENAI_API_KEY}`,
        },
        body: JSON.stringify({
            model: "gpt-4.1",
            messages,
            temperature: 0,
            stream: true,
        }),
    });

    if (!response.ok || !response.body) {
        const error = await response.text();
        return new NextResponse(error, { status: response.status });
    }

    return new NextResponse(response.body, {
        headers: {
            "Content-Type": "text/event-stream",
            "Cache-Control": "no-cache",
            "Connection": "keep-alive",
        },
    });
}