Vercel AI SDK

The Vercel AI SDK is a library for building edge-ready AI-powered streaming text and chat UIs.

Features

SWR-powered React and Svelte helpers for streaming text responses and building chat and completion UIs
First-class support for LangChain and OpenAI, Anthropic, and HuggingFace
Edge Runtime compatibility
Callbacks for saving completed streaming responses to a database (in the same request)

Installation

pnpm install ai

Example: An AI Chatbot with Next.js and OpenAI

With the Vercel AI SDK, you can build a ChatGPT-like app in just a few lines of code:

// ./app/api/chat/route.js
import { Configuration, OpenAIApi } from 'openai-edge'
import { OpenAIStream, StreamingTextResponse } from 'ai'

const config = new Configuration({
  apiKey: process.env.OPENAI_API_KEY
})
const openai = new OpenAIApi(config)

export const runtime = 'edge'

export async function POST(req) {
  const { messages } = await req.json()
  const response = await openai.createChatCompletion({
    model: 'gpt-4',
    stream: true,
    messages
  })
  const stream = OpenAIStream(response)
  return new StreamingTextResponse(stream)
}

// ./app/page.js
'use client'

import { useChat } from 'ai/react'

export default function Chat() {
  const { messages, input, handleInputChange, handleSubmit } = useChat()

  return (
    <div className="mx-auto w-full max-w-md py-24 flex flex-col stretch">
      {messages.length > 0
        ? messages.map(m => (
            <div key={m.id}>
              {m.role === 'user' ? 'User: ' : 'AI: '}
              {m.content}
            </div>
          ))
        : null}

      <form onSubmit={handleSubmit}>
        <input
          className="fixed w-full max-w-md bottom-0 border border-gray-300 rounded mb-8 shadow-xl p-2"
          value={input}
          placeholder="Say something..."
          onChange={handleInputChange}
        />
      </form>
    </div>
  )
}

View the full documentation and examples on play.vercel.ai/docs

Authors

This library is created by Vercel and Next.js team members, with contributions from:

Jared Palmer (@jaredpalmer) - Vercel
Shu Ding (@shuding_) - Vercel
Max Leiter (@max_leiter) - Vercel
Malte Ubl (@cramforce) - Vercel
Justin Ridgewell (@jridgewell) - Vercel

Contributors

License

theluk/ai

ErrorLooks like something went wrong!

About

Resources

License

Uh oh!

Stars

Watchers

Forks

Releases

Packages 0

Languages

Error
Looks like something went wrong!

Packages