@kernl-sdk/retrieval

import type { SearchIndex, IndexHandle } from '@kernl-sdk/retrieval';

// -- index lifecycle --
await search.createIndex({ id: 'docs', dimensions: 1536 });
const docs: IndexHandle = search.index('docs');

// query with vectors + text
const hits = await docs.query({
  query: [{ text: 'search query', tvec: [0.1, 0.2, ...] }],
  filter: { category: 'technical' },
  limit: 10,
});

// upsert documents
await docs.upsert({ id: '1', text: 'content', tvec: [...] });

import { embed, embedMany } from '@kernl-sdk/retrieval';
import { openai } from '@kernl-sdk/ai/openai';

// single text
const { embedding } = await embed({
  model: 'openai/text-embedding-3-small',
  text: 'sunny day at the beach',
});

// multiple texts
const { embeddings } = await embedMany({
  model: 'openai/text-embedding-3-small',
  texts: ['hello', 'world'],
});