@langchain/community/dist/retrievers/zep.cjs

Third-party integrations for LangChain.js

const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
const __langchain_core_documents = require_rolldown_runtime.__toESM(require("@langchain/core/documents"));
const __getzep_zep_js = require_rolldown_runtime.__toESM(require("@getzep/zep-js"));
const __langchain_core_retrievers = require_rolldown_runtime.__toESM(require("@langchain/core/retrievers"));

//#region src/retrievers/zep.ts
var zep_exports = {};
require_rolldown_runtime.__export(zep_exports, { ZepRetriever: () => ZepRetriever });
/**
* Class for retrieving information from a Zep long-term memory store.
* Extends the BaseRetriever class.
* @example
* ```typescript
* const retriever = new ZepRetriever({
*   url: "http:
*   sessionId: "session_exampleUUID",
*   topK: 3,
* });
* const query = "Can I drive red cars in France?";
* const docs = await retriever.getRelevantDocuments(query);
* ```
*/
var ZepRetriever = class extends __langchain_core_retrievers.BaseRetriever {
	static lc_name() {
		return "ZepRetriever";
	}
	lc_namespace = [
		"langchain",
		"retrievers",
		"zep"
	];
	get lc_secrets() {
		return {
			apiKey: "ZEP_API_KEY",
			url: "ZEP_API_URL"
		};
	}
	get lc_aliases() {
		return { apiKey: "api_key" };
	}
	zepClientPromise;
	sessionId;
	topK;
	searchScope;
	searchType;
	mmrLambda;
	filter;
	constructor(config) {
		super(config);
		this.sessionId = config.sessionId;
		this.topK = config.topK;
		this.searchScope = config.searchScope;
		this.searchType = config.searchType;
		this.mmrLambda = config.mmrLambda;
		this.filter = config.filter;
		this.zepClientPromise = __getzep_zep_js.ZepClient.init(config.url, config.apiKey);
	}
	/**
	*  Converts an array of message search results to an array of Document objects.
	*  @param {MemorySearchResult[]} results - The array of search results.
	*  @returns {Document[]} An array of Document objects representing the search results.
	*/
	searchMessageResultToDoc(results) {
		return results.filter((r) => r.message).map(({ message: { content, metadata: messageMetadata } = {}, dist,...rest }) => new __langchain_core_documents.Document({
			pageContent: content ?? "",
			metadata: {
				score: dist,
				...messageMetadata,
				...rest
			}
		}));
	}
	/**
	*  Converts an array of summary search results to an array of Document objects.
	*  @param {MemorySearchResult[]} results - The array of search results.
	*  @returns {Document[]} An array of Document objects representing the search results.
	*/
	searchSummaryResultToDoc(results) {
		return results.filter((r) => r.summary).map(({ summary: { content, metadata: summaryMetadata } = {}, dist,...rest }) => new __langchain_core_documents.Document({
			pageContent: content ?? "",
			metadata: {
				score: dist,
				...summaryMetadata,
				...rest
			}
		}));
	}
	/**
	*  Retrieves the relevant documents based on the given query.
	*  @param {string} query - The query string.
	*  @returns {Promise<Document[]>} A promise that resolves to an array of relevant Document objects.
	*/
	async _getRelevantDocuments(query) {
		const payload = {
			text: query,
			metadata: this.filter,
			search_scope: this.searchScope,
			search_type: this.searchType,
			mmr_lambda: this.mmrLambda
		};
		const zepClient = await this.zepClientPromise;
		if (!zepClient) throw new Error("ZepClient is not initialized");
		try {
			const results = await zepClient.memory.searchMemory(this.sessionId, payload, this.topK);
			return this.searchScope === "summary" ? this.searchSummaryResultToDoc(results) : this.searchMessageResultToDoc(results);
		} catch (error) {
			if (error instanceof __getzep_zep_js.NotFoundError) return Promise.resolve([]);
			throw error;
		}
	}
};

//#endregion
exports.ZepRetriever = ZepRetriever;
Object.defineProperty(exports, 'zep_exports', {
  enumerable: true,
  get: function () {
    return zep_exports;
  }
});
//# sourceMappingURL=zep.cjs.map