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

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

import { Embedding, VectorStore } from "../../../types";
import { Pool } from 'pg';
import { VECTOR_MODEL_DIM } from "../types";
declare class PostgresVectorStore implements VectorStore {
    pool: Pool;
    constructor(connectionString: string);
    createIndex(opts?: {
        dim: VECTOR_MODEL_DIM;
    }): Promise<void>;
    addEmbedding(embedding: Embedding): Promise<void>;
    storeEmbeddings(embeddings: Embedding[]): Promise<void>;
    queryEmbeddings(params: {
        query: number[];
        k: number;
        documentIds?: string[];
    }): Promise<{
        chunkId: any;
        documentId: any;
        score: any;
    }[]>;
    close(): Promise<void>;
}
export default PostgresVectorStore;