@forge-ml/rag/dist/src/stores/docStore/minio/index.d.ts

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

import { Client } from "minio";
import { Chunk, DocStore, DocumentClass, RelevantChunk, ScoredEmbedding } from "../../../types";
declare class MinioDocStore implements DocStore {
    client: Client;
    bucketName: string;
    documentName: string;
    private static readonly DOCUMENT_FILE;
    chunksName: string;
    private static readonly CHUNKS_FILE;
    constructor(endpoint: string, port: number, useSSL: boolean, accessKey: string, secretKey: string);
    private initializeBucket;
    storeDocument(document: DocumentClass, chunks: Chunk[]): Promise<void>;
    retrieveDocument(documentId: string): Promise<DocumentClass>;
    updateDocument(document: DocumentClass, documentId: string): Promise<void>;
    deleteDocument(documentId: string): Promise<void>;
    retrieveChunks(document: DocumentClass): Promise<Chunk[]>;
    queryFromEmbeddings(embeddings: ScoredEmbedding[], document: DocumentClass): Promise<RelevantChunk[]>;
    deleteBucket(): Promise<void>;
    private streamToString;
}
export default MinioDocStore;