import { createFlowRAG } from '@flowrag/pipeline';
import { createLocalStorage } from '@flowrag/presets';
import { defineSchema } from '@flowrag/core';

const schema = defineSchema({
  entityTypes: ['SERVICE', 'DATABASE'],
  relationTypes: ['USES', 'PRODUCES'],
});

const rag = createFlowRAG({ schema, ...createLocalStorage('./data') });

// Index documents
await rag.index('./content');

// Search with dual retrieval
const results = await rag.search('how does auth work');

// Trace data flow
const flow = await rag.traceDataFlow('auth-service', 'downstream');

// Statistics
const stats = await rag.stats();