UNPKG

@forge-ml/rag

Version:

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

11 lines (10 loc) 567 B
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;