OpenAI is a language model provider. Check out OpenAI API for more information about their models and pricing.

Install RAG Chat SDK

Initialize the project and install the required packages:

npm init es6
npm install dotenv
npm install @upstash/rag-chat

Setup Upstash Redis

Create a Redis database using Upstash Console or Upstash CLI and copy the UPSTASH_REDIS_REST_URL and UPSTASH_REDIS_REST_TOKEN into your .env file.

.env
UPSTASH_REDIS_REST_URL=<YOUR_URL>
UPSTASH_REDIS_REST_TOKEN=<YOUR_TOKEN>

Setup Upstash Vector

Create a Vector index using Upstash Console or Upstash CLI and copy the UPSTASH_VECTOR_REST_URL and UPSTASH_VECTOR_REST_TOKEN into your .env file.

.env
UPSTASH_VECTOR_REST_URL=<YOUR_URL>
UPSTASH_VECTOR_REST_TOKEN=<YOUR_TOKEN>

Setup OpenAI

Create an OpenAI account and get an API key from OpenAI Platform -> Dashboard -> API keys. Set your OpenAI API key as an environment variable:

.env
OPENAI_API_KEY=<YOUR_API_KEY>

Setup the Project

Initialize RAGChat with the OpenAI model:

index.ts
import { RAGChat, openai } from "@upstash/rag-chat";
import "dotenv/config";

export const ragChat = new RAGChat({
  model: openai("gpt-4-turbo"),
});

Add context to the RAG Chat:

index.ts
await ragChat.context.add("The speed of light is approximately 299,792,458 meters per second.");

Chat with the RAG Chat:

index.ts
const response = await ragChat.chat("What is the speed of light?");
console.log(response);

Run

Run the project:

npx tsx index.ts