@forge-ml/rag
Version:
A RAG (Retrieval-Augmented Generation) package for Forge ML
25 lines (24 loc) • 663 B
JavaScript
import { generateUUID } from "../utils/generateID";
class Document {
forgeMetadata; // forge metadata: documentId, page?
metadata; //user metadata
text;
//@QUESTION: should the user be given the option to set their own doc id
constructor(text, metadata, forgeMetadata) {
this.forgeMetadata = forgeMetadata || {
documentId: generateUUID(),
};
this.metadata = metadata || {};
this.text = text;
}
getForgeMetadata() {
return this.forgeMetadata;
}
getText() {
return this.text;
}
getUserMetadata() {
return this.metadata;
}
}
export default Document;