@forge-ml/rag/dist/stores/vectorStore/turbopuffer/index.d.ts

A RAG (Retrieval-Augmented Generation) package for Forge ML

import { Embedding, ScoredEmbedding, VectorStore } from "../../../types";
import { TurbopufferClient } from "./client";
declare class TurbopufferVectorStore implements VectorStore {
    client: TurbopufferClient;
    constructor({ apiKey, namespace }: {
        apiKey: string;
        namespace: string;
    });
    createIndex(): Promise<void>;
    addEmbedding(embedding: Embedding): Promise<void>;
    storeEmbeddings(embeddings: Embedding[]): Promise<void>;
    queryEmbeddings(params: {
        query: number[];
        k: number;
        documentIds?: string[];
    }): Promise<ScoredEmbedding[]>;
    private createVector;
}
export default TurbopufferVectorStore;