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

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

import { Vector, DistanceMetric, Schema, QueryResults, Filters } from "@turbopuffer/turbopuffer";
export declare class TurbopufferClient {
    private apiKey;
    private namespaceId;
    constructor({ apiKey, namespaceId }: {
        apiKey: string;
        namespaceId: string;
    });
    private makeRequest;
    upsert(params: {
        vectors: Vector[];
        distance_metric: DistanceMetric;
        schema?: Schema;
    }): Promise<void>;
    private formatAttributes;
    query(params: {
        vector?: number[];
        distance_metric?: DistanceMetric;
        top_k?: number;
        include_vectors?: boolean;
        include_attributes?: boolean | string[];
        filters?: Filters;
        rank_by?: string[];
    }): Promise<QueryResults>;
    deleteNamespace(): Promise<{
        status: string;
    }>;
}