@0xplaygrounds/rig-wasm
Version:
A TS and WebAssembly-based port of the Rust agentic AI framework Rig.
15 lines (14 loc) • 590 B
TypeScript
import { Embedding } from "../generated/rig_wasm";
import { CanEmbed, DocumentId, Metadata, TopNIdsResult, TopNResult, VectorSearchOpts, VectorStore } from "../types";
/**
* A basic in memory vector store.
* Usable for small datasets.
*/
export declare class InMemoryVectorStore implements VectorStore {
private store;
private model;
constructor(model: CanEmbed);
addDocument(id: DocumentId, embedding: Embedding, metadata?: Metadata): Promise<void>;
topN(req: VectorSearchOpts): Promise<TopNResult[]>;
topNIds(req: VectorSearchOpts): Promise<TopNIdsResult[]>;
}