UNPKG

embeddings-js

Version:

A NodeJS RAG framework to easily work with LLMs and custom datasets

18 lines (17 loc) 600 B
import { BaseDb } from '../interfaces/base-db.js'; import { Chunk, EmbeddedChunk } from '../global/types.js'; export declare class HNSWDb implements BaseDb { private readonly debug; private index; private docCount; private docMap; constructor(); init({ dimensions }: { 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>; }