Using the Inkeep RAG API with the Vercel AI SDK

import { generateObject} from 'ai'
import { z } from 'zod';
import dotenv from 'dotenv';
import { createOpenAI } from '@ai-sdk/openai';
dotenv.config();
 
if (!process.env.INKEEP_API_KEY) {
  throw new Error('INKEEP_API_KEY is required');
}
 
export const provider = createOpenAI({
  baseURL: 'https://api.inkeep.com/v1/',
  apiKey: process.env.INKEEP_API_KEY,
});
 
 
/* Inkeep RAG Document Schema */
 
export const InkeepSourceContentSchema = z.object({
  type: z.union([z.literal('text'), z.string()]),
  media_type: z.union([z.literal('text/plain'), z.string()]).optional(),
  text: z.string().optional(),
  data: z.string().optional(),
}).passthrough();
 
export const InkeepSourceSchema = z.object({
  content: z.array(InkeepSourceContentSchema).optional(),
  type: z.union([z.literal('content'), z.string()]).optional(),
  media_type: z.string().optional(),
  data: z.string().optional(),
}).passthrough();
 
const InkeepRecordTypes = z.enum([
    'documentation',
    'site',
    'discourse_post',
    'github_issue',
    'github_discussion',
    'stackoverflow_question',
    'discord_forum_post',
    'discord_message',
    'custom_question_answer',
]);
 
export const InkeepRAGDocumentSchema = z.object({
  type: z.string(),
  source: InkeepSourceSchema,
  title: z.string().optional(),
  context: z.string().optional(),
  record_type: z.union([z.string(), InkeepRecordTypes]).optional(),
  url: z.string().optional(),
}).passthrough();
 
export const InkeepRAGResponseSchema = z.object({
  content: z.array(InkeepRAGDocumentSchema),
}).passthrough();
 
export type InkeepSourceContent = z.infer<typeof InkeepSourceContentSchema>;
export type InkeepSource = z.infer<typeof InkeepSourceSchema>;
export type InkeepRAGDocument = z.infer<typeof InkeepRAGDocumentSchema>;
export type InkeepRAGResponse = z.infer<typeof InkeepRAGResponseSchema>;

import { createOpenAI } from '@ai-sdk/openai';
import { z } from 'zod';
import dotenv from 'dotenv';
import fetch from 'node-fetch';
dotenv.config();
 
if (!process.env.INKEEP_API_KEY) {
  throw new Error('INKEEP_API_KEY is required');
}
 
// Initialize OpenAI client with Inkeep's base URL
const provider = createOpenAI({
  apiKey: process.env.INKEEP_API_KEY,
  baseURL: 'https://api.inkeep.com/v1'
});
 
// Define the RAG document schema
const InkeepSourceContentSchema = z.object({
  type: z.union([z.literal('text'), z.string()]),
  media_type: z.union([z.literal('text/plain'), z.string()]).optional(),
  text: z.string().optional(),
  data: z.string().optional(),
}).passthrough();
 
const InkeepSourceSchema = z.object({
  content: z.array(InkeepSourceContentSchema).optional(),
  type: z.union([z.literal('content'), z.string()]).optional(),
  media_type: z.string().optional(),
  data: z.string().optional(),
}).passthrough();
 
const InkeepRAGDocumentSchema = z.object({
  type: z.string(),
  source: InkeepSourceSchema,
  title: z.string().optional(),
  context: z.string().optional(),
  record_type: z.string().optional(),
  url: z.string().optional(),
}).passthrough();
 
const InkeepRAGResponseSchema = z.object({
  content: z.array(InkeepRAGDocumentSchema),
}).passthrough();
 
const {object} = await generateObject({
    model: provider.languageModel('inkeep-rag'),
    messages: [{ role: 'user', content: 'How do I get started?' }],
    schema: InkeepRAGResponseSchema,
    mode: 'json',
  }
)
// Process the RAG documents
object.content.forEach((doc, index) => {
  console.log(`Document ${index + 1}:`);
  console.log(`Title: ${doc.title || 'N/A'}`);
  console.log(`URL: ${doc.url || 'N/A'}`);
  console.log(`Context: ${doc.context || 'N/A'}`);
  console.log(`Record Type: ${doc.record_type || 'N/A'}`);
  console.log('---');
  });