import { create, insert } from '@orama/orama'
import { pluginSecureProxy } from '@orama/plugin-secure-proxy'
import { AnswerSession } from '@orama/orama'

// Step 1: Configure the Secure Proxy Plugin
const secureProxy = await pluginSecureProxy({
  apiKey: 'my-api-key',
  defaultProperty: 'embeddings',
  models: {
    // The chat model to use to generate the chat answer
    chat: 'openai/gpt-4o-mini'
  }
})

// Step 2: Create your database with the plugin
const db = await create({
  schema: {
    name: 'string',
    description: 'string'
  },
  plugins: [secureProxy]
})

// Step 3: Insert your data
await insert(db, { name: 'John Doe', description: 'Software engineer' })
await insert(db, { name: 'Jane Doe', description: 'Product manager' })

// Step 4: Create an Answer Session
const session = new AnswerSession(db, {
  systemPrompt: 'You are a helpful assistant that provides information about people.',
  events: {
    onStateChange: (state) => {
      console.log('State updated:', state)
    }
  }
})

// Step 5: Ask questions
const response = await session.ask({
  term: 'john'
})

console.log(response) // AI-generated response about John Doe