@langchain/community/dist/document_loaders/web/couchbase.cjs

Third-party integrations for LangChain.js

Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
const require_runtime = require("../../_virtual/_rolldown/runtime.cjs");
let _langchain_core_documents = require("@langchain/core/documents");
let _langchain_core_document_loaders_base = require("@langchain/core/document_loaders/base");
//#region src/document_loaders/web/couchbase.ts
var couchbase_exports = /* @__PURE__ */ require_runtime.__exportAll({ CouchbaseDocumentLoader: () => CouchbaseDocumentLoader });
/**
* loader for couchbase document
*/
var CouchbaseDocumentLoader = class extends _langchain_core_document_loaders_base.BaseDocumentLoader {
	cluster;
	query;
	pageContentFields;
	metadataFields;
	/**
	* construct Couchbase document loader with a requirement for couchbase cluster client
	* @param client { Cluster } [ couchbase connected client to connect to database ]
	* @param query { string } [ query to get results from while loading the data ]
	* @param pageContentFields { Array<string> } [ filters fields of the document and shows these only ]
	* @param metadataFields { Array<string> } [ metadata fields required ]
	*/
	constructor(client, query, pageContentFields, metadataFields) {
		super();
		if (!client) throw new Error("Couchbase client cluster must be provided.");
		this.cluster = client;
		this.query = query;
		this.pageContentFields = pageContentFields;
		this.metadataFields = metadataFields;
	}
	/**
	* Function to load document based on query from couchbase
	* @returns {Promise<Document[]>} [ Returns a promise of all the documents as array ]
	*/
	async load() {
		const documents = [];
		for await (const doc of this.lazyLoad()) documents.push(doc);
		return documents;
	}
	/**
	* Function to load documents based on iterator rather than full load
	* @returns {AsyncIterable<Document>} [ Returns an iterator to fetch documents ]
	*/
	async *lazyLoad() {
		const result = await this.cluster.query(this.query);
		for await (const row of result.rows) {
			let { metadataFields, pageContentFields } = this;
			if (!pageContentFields) pageContentFields = Object.keys(row);
			if (!metadataFields) metadataFields = [];
			const metadata = metadataFields.reduce((obj, field) => ({
				...obj,
				[field]: row[field]
			}), {});
			yield new _langchain_core_documents.Document({
				pageContent: pageContentFields.map((k) => `${k}: ${JSON.stringify(row[k])}`).join("\n"),
				metadata
			});
		}
	}
};
//#endregion
exports.CouchbaseDocumentLoader = CouchbaseDocumentLoader;
Object.defineProperty(exports, "couchbase_exports", {
	enumerable: true,
	get: function() {
		return couchbase_exports;
	}
});

//# sourceMappingURL=couchbase.cjs.map