@langchain/community
Version:
Third-party integrations for LangChain.js
71 lines (70 loc) • 2.65 kB
JavaScript
Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
const require_runtime = require("../../_virtual/_rolldown/runtime.cjs");
let _upstash_redis = require("@upstash/redis");
let _langchain_core_messages = require("@langchain/core/messages");
let _langchain_core_chat_history = require("@langchain/core/chat_history");
//#region src/stores/message/upstash_redis.ts
var upstash_redis_exports = /* @__PURE__ */ require_runtime.__exportAll({ 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() {
return (0, _langchain_core_messages.mapStoredMessagesToChatMessages)((await this.client.lrange(this.sessionId, 0, -1)).reverse().filter((x) => x.type !== void 0 && x.data.content !== void 0));
}
/**
* 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