AI Knowledge Agent

import OpenAI from "openai";
import { Pinecone } from "@pinecone-database/pinecone";

const openai = new OpenAI({ apiKey: process.env.OPENAI_API_KEY });
const pinecone = new Pinecone({ apiKey: process.env.PINECONE_API_KEY });

async function answerQuestion(question: string) {
  const queryVector = await openai.embeddings.create({
    model: "text-embedding-3-small",
    input: question,
  });

  const index = pinecone.index("knowledge-base");
  const results = await index.query({
    vector: queryVector.data[0].embedding,
    topK: 5,
  });

  const context = results.matches.map((m) => m.metadata?.text).join("
");

  const response = await openai.chat.completions.create({
    model: "gpt-4o",
    messages: [
      { role: "system", content: "Answer based only on the provided context." },
      { role: "user", content: "Context:\n" + context + "\n\nQuestion: " + question },
    ],
  });

  return response.choices[0].message.content;
}