@langchain/community/dist/stores/message/upstash_redis.cjs

Third-party integrations for LangChain.js

const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
const __upstash_redis = require_rolldown_runtime.__toESM(require("@upstash/redis"));
const __langchain_core_messages = require_rolldown_runtime.__toESM(require("@langchain/core/messages"));
const __langchain_core_chat_history = require_rolldown_runtime.__toESM(require("@langchain/core/chat_history"));

//#region src/stores/message/upstash_redis.ts
var upstash_redis_exports = {};
require_rolldown_runtime.__export(upstash_redis_exports, { UpstashRedisChatMessageHistory: () => UpstashRedisChatMessageHistory });
/**
* Class used to store chat message history in Redis. It provides methods
* to add, get, and clear messages.
*/
var UpstashRedisChatMessageHistory = class extends __langchain_core_chat_history.BaseListChatMessageHistory {
	lc_namespace = [
		"langchain",
		"stores",
		"message",
		"upstash_redis"
	];
	get lc_secrets() {
		return {
			"config.url": "UPSTASH_REDIS_REST_URL",
			"config.token": "UPSTASH_REDIS_REST_TOKEN"
		};
	}
	client;
	sessionId;
	sessionTTL;
	constructor(fields) {
		super(fields);
		const { sessionId, sessionTTL, config, client } = fields;
		if (client) this.client = client;
		else if (config) this.client = new __upstash_redis.Redis(config);
		else throw new Error(`Upstash Redis message stores require either a config object or a pre-configured client.`);
		this.sessionId = sessionId;
		this.sessionTTL = sessionTTL;
	}
	/**
	* Retrieves the chat messages from the Redis database.
	* @returns An array of BaseMessage instances representing the chat history.
	*/
	async getMessages() {
		const rawStoredMessages = await this.client.lrange(this.sessionId, 0, -1);
		const orderedMessages = rawStoredMessages.reverse();
		const previousMessages = orderedMessages.filter((x) => x.type !== void 0 && x.data.content !== void 0);
		return (0, __langchain_core_messages.mapStoredMessagesToChatMessages)(previousMessages);
	}
	/**
	* Adds a new message to the chat history in the Redis database.
	* @param message The message to be added to the chat history.
	* @returns Promise resolving to void.
	*/
	async addMessage(message) {
		const messageToAdd = (0, __langchain_core_messages.mapChatMessagesToStoredMessages)([message]);
		await this.client.lpush(this.sessionId, JSON.stringify(messageToAdd[0]));
		if (this.sessionTTL) await this.client.expire(this.sessionId, this.sessionTTL);
	}
	/**
	* Deletes all messages from the chat history in the Redis database.
	* @returns Promise resolving to void.
	*/
	async clear() {
		await this.client.del(this.sessionId);
	}
};

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