embeddings-js
Version:
A NodeJS RAG framework to easily work with LLMs and custom datasets
12 lines (11 loc) • 406 B
TypeScript
import { Chunk, EmbeddedChunk } from '../global/types.js';
export interface BaseDb {
init({}: {
dimensions: number;
}): Promise<void>;
insertChunks(chunks: EmbeddedChunk[]): Promise<number>;
similaritySearch(query: number[], k: number): Promise<Chunk[]>;
getVectorCount(): Promise<number>;
deleteKeys(uniqueLoaderId: string): Promise<boolean>;
reset(): Promise<void>;
}