Quickstart

import { VectoriaDB, DocumentMetadata } from 'vectoriadb';

// Define your metadata shape
interface ToolDocument extends DocumentMetadata {
  toolName: string;
  owner: string;
  tags: string[];
  risk: 'safe' | 'destructive';
}

// Create the database
const toolIndex = new VectoriaDB<ToolDocument>({
  cacheDir: './.cache/transformers',
  defaultSimilarityThreshold: 0.4,
});

// Initialize (downloads model on first run)
await toolIndex.initialize();

// Add a document
await toolIndex.add('users:list', 'List all users with pagination', {
  id: 'users:list',
  toolName: 'list',
  owner: 'users',
  tags: ['read'],
  risk: 'safe',
});

// Search
const results = await toolIndex.search('find users');
console.log(results[0].metadata.toolName); // 'list'

import { DocumentMetadata } from 'vectoriadb';

interface ToolDocument extends DocumentMetadata {
  toolName: string;
  owner: string;
  tags: string[];
  risk: 'safe' | 'destructive';
}

const db = new VectoriaDB<ToolDocument>({
  cacheDir: './.cache/transformers',
  defaultSimilarityThreshold: 0.4,
});

await db.initialize();

// Single document
await db.add('users:list', 'List all users with pagination', {
  id: 'users:list',
  toolName: 'list',
  owner: 'users',
  tags: ['read'],
  risk: 'safe',
});

// Batch add
await db.addMany([
  { id: 'users:create', text: 'Create a new user', metadata: { /* ... */ } },
  { id: 'users:delete', text: 'Delete a user', metadata: { /* ... */ } },
]);

const results = await db.search('create account', {
  topK: 5,
  threshold: 0.4,
  filter: (m) => m.risk === 'safe',
});

for (const result of results) {
  console.log(`${result.metadata.toolName}: ${result.score.toFixed(2)}`);
}