30 lines
768 B
TypeScript
30 lines
768 B
TypeScript
export interface VectorEntry {
|
|
id?: string;
|
|
vector: Float32Array | number[];
|
|
}
|
|
|
|
export interface SearchQuery {
|
|
vector: Float32Array | number[];
|
|
k: number;
|
|
efSearch?: number;
|
|
}
|
|
|
|
export interface SearchResult {
|
|
id: string;
|
|
score: number;
|
|
}
|
|
|
|
export class VectorDb {
|
|
constructor(options: { dimensions: number; storagePath?: string; distanceMetric?: string; hnswConfig?: any });
|
|
insert(entry: VectorEntry): Promise<string>;
|
|
insertBatch(entries: VectorEntry[]): Promise<string[]>;
|
|
search(query: SearchQuery): Promise<SearchResult[]>;
|
|
delete(id: string): Promise<boolean>;
|
|
get(id: string): Promise<VectorEntry | null>;
|
|
len(): Promise<number>;
|
|
isEmpty(): Promise<boolean>;
|
|
}
|
|
|
|
// Alias for backwards compatibility
|
|
export { VectorDb as VectorDB };
|