You can use the AI SDK in a Node.js HTTP server to generate text and stream it to the client.
The examples start a simple HTTP server that listens on port 8080. You can e.g. test it using curl
:
curl -X POST http://localhost:8080
The examples use the OpenAI gpt-4o
model. Ensure that the OpenAI API key is
set in the OPENAI_API_KEY
environment variable.
Full example: github.com/vercel/ai/examples/node-http-server
You can use the pipeDataStreamToResponse
method to pipe the stream data to the server response.
import { openai } from '@ai-sdk/openai';import { streamText } from 'ai';import { createServer } from 'http';
createServer(async (req, res) => { const result = streamText({ model: openai('gpt-4o'), prompt: 'Invent a new holiday and describe its traditions.', });
result.pipeDataStreamToResponse(res);}).listen(8080);
pipeDataStreamToResponse
can be used with StreamData
to send additional data to the client.
import { openai } from '@ai-sdk/openai';import { StreamData, streamText } from 'ai';import { createServer } from 'http';
createServer(async (req, res) => { const data = new StreamData(); data.append('initialized call');
const result = streamText({ model: openai('gpt-4o'), prompt: 'Invent a new holiday and describe its traditions.', onFinish() { data.append('call completed'); data.close(); }, });
result.pipeDataStreamToResponse(res, { data });}).listen(8080);
You can send a text stream to the client using pipeTextStreamToResponse
.
import { openai } from '@ai-sdk/openai';import { streamText } from 'ai';import { createServer } from 'http';
createServer(async (req, res) => { const result = streamText({ model: openai('gpt-4o'), prompt: 'Invent a new holiday and describe its traditions.', });
result.pipeTextStreamToResponse(res);}).listen(8080);