RAG functions for import, export, and vector search
npm install @sf-bot/rag-functions
RAG functions for import, export, and vector search.
This package provides SQL functions for working with RAG collections:
- Import/Export: JSON format for portable collection data
- Search: Cosine similarity search using pgvector
- CSV Export: Flat format for embeddings
Export a complete collection with all documents, chunks, and embeddings as JSON.
``sql`
SELECT rag.export_collection_json('collection-uuid');
Import a collection from JSON format.
`sql`
SELECT rag.import_collection_json('{...}'::jsonb, false);
Search for similar chunks using cosine distance.
`sql`
SELECT * FROM rag.search_similar(
'collection-uuid',
'[0.1, 0.2, ...]'::vector,
'text-embedding-3-small', -- model name (optional)
10, -- limit
0.5 -- distance threshold (optional)
);
Also supports searching by collection name:
`sql`
SELECT * FROM rag.search_similar(
'my-collection',
'[0.1, 0.2, ...]'::vector
);
Export embeddings in CSV-friendly format.
`sql`
SELECT * FROM rag.export_embeddings_csv('collection-uuid', 'text-embedding-3-small');
- @sf-bot/rag-core`