@forge-ml/rag
Version:
A RAG (Retrieval-Augmented Generation) package for Forge ML
11 lines (10 loc) • 567 B
TypeScript
import { Embedder, InitializeDocumentOptions, Chunk, DocumentClass } from "../types";
import { StoresClass } from "../types";
declare const createRagger: (embedder: Embedder, stores: StoresClass) => {
embedder: Embedder;
vectorStore: import("../types").VectorStore;
docStore: import("../types").DocStore;
query: (query: string, documentIds: string[], k?: number) => Promise<import("../types").RelevantChunk[]>;
initializeDocument: (document: DocumentClass, options?: InitializeDocumentOptions) => Promise<Chunk[]>;
};
export default createRagger;