langchain 0.0.74 → 0.0.76

package/dist/memory/prompt.js

@@ -25,3 +25,86 @@ export const SUMMARY_PROMPT = /*#__PURE__*/ new PromptTemplate({
     inputVariables: ["summary", "new_lines"],
     template: _DEFAULT_SUMMARIZER_TEMPLATE,
 });
+export const _DEFAULT_ENTITY_MEMORY_CONVERSATION_TEMPLATE = `You are an assistant to a human, powered by a large language model trained by OpenAI.
+
+You are designed to be able to assist with a wide range of tasks, from answering simple questions to providing in-depth explanations and discussions on a wide range of topics. As a language model, you are able to generate human-like text based on the input you receive, allowing you to engage in natural-sounding conversations and provide responses that are coherent and relevant to the topic at hand.
+
+You are constantly learning and improving, and your capabilities are constantly evolving. You are able to process and understand large amounts of text, and can use this knowledge to provide accurate and informative responses to a wide range of questions. You have access to some personalized information provided by the human in the Context section below. Additionally, you are able to generate your own text based on the input you receive, allowing you to engage in discussions and provide explanations and descriptions on a wide range of topics.
+
+Overall, you are a powerful tool that can help with a wide range of tasks and provide valuable insights and information on a wide range of topics. Whether the human needs help with a specific question or just wants to have a conversation about a particular topic, you are here to assist.
+
+Context:
+{entities}
+
+Current conversation:
+{history}
+Last line:
+Human: {input}
+You:`;
+export const ENTITY_MEMORY_CONVERSATION_TEMPLATE = 
+// eslint-disable-next-line spaced-comment
+/*#__PURE__*/ new PromptTemplate({
+    inputVariables: ["entities", "history", "input"],
+    template: _DEFAULT_ENTITY_MEMORY_CONVERSATION_TEMPLATE,
+});
+export const _DEFAULT_ENTITY_EXTRACTION_TEMPLATE = `You are an AI assistant reading the transcript of a conversation between an AI and a human. Extract all of the proper nouns from the last line of conversation. As a guideline, a proper noun is generally capitalized. You should definitely extract all names and places.
+
+The conversation history is provided just in case of a coreference (e.g. "What do you know about him" where "him" is defined in a previous line) -- ignore items mentioned there that are not in the last line.
+
+Return the output as a single comma-separated list, or NONE if there is nothing of note to return (e.g. the user is just issuing a greeting or having a simple conversation).
+
+EXAMPLE
+Conversation history:
+Person #1: my name is Jacob. how's it going today?
+AI: "It's going great! How about you?"
+Person #1: good! busy working on Langchain. lots to do.
+AI: "That sounds like a lot of work! What kind of things are you doing to make Langchain better?"
+Last line:
+Person #1: i'm trying to improve Langchain's interfaces, the UX, its integrations with various products the user might want ... a lot of stuff.
+Output: Jacob,Langchain
+END OF EXAMPLE
+
+EXAMPLE
+Conversation history:
+Person #1: how's it going today?
+AI: "It's going great! How about you?"
+Person #1: good! busy working on Langchain. lots to do.
+AI: "That sounds like a lot of work! What kind of things are you doing to make Langchain better?"
+Last line:
+Person #1: i'm trying to improve Langchain's interfaces, the UX, its integrations with various products the user might want ... a lot of stuff. I'm working with Person #2.
+Output: Langchain, Person #2
+END OF EXAMPLE
+
+Conversation history (for reference only):
+{history}
+Last line of conversation (for extraction):
+Human: {input}
+
+Output:`;
+// eslint-disable-next-line spaced-comment
+export const ENTITY_EXTRACTION_PROMPT = /*#__PURE__*/ new PromptTemplate({
+    inputVariables: ["history", "input"],
+    template: _DEFAULT_ENTITY_EXTRACTION_TEMPLATE,
+});
+export const _DEFAULT_ENTITY_SUMMARIZATION_TEMPLATE = `You are an AI assistant helping a human keep track of facts about relevant people, places, and concepts in their life. Update and add to the summary of the provided entity in the "Entity" section based on the last line of your conversation with the human. If you are writing the summary for the first time, return a single sentence.
+The update should only include facts that are relayed in the last line of conversation about the provided entity, and should only contain facts about the provided entity.
+
+If there is no new information about the provided entity or the information is not worth noting (not an important or relevant fact to remember long-term), output the exact string "UNCHANGED" below.
+
+Full conversation history (for context):
+{history}
+
+Entity to summarize:
+{entity}
+
+Existing summary of {entity}:
+{summary}
+
+Last line of conversation:
+Human: {input}
+Updated summary (or the exact string "UNCHANGED" if there is no new information about {entity} above):`;
+// eslint-disable-next-line spaced-comment
+export const ENTITY_SUMMARIZATION_PROMPT = /*#__PURE__*/ new PromptTemplate({
+    inputVariables: ["entity", "summary", "history", "input"],
+    template: _DEFAULT_ENTITY_SUMMARIZATION_TEMPLATE,
+});

package/dist/memory/stores/entity/in_memory.cjs

@@ -0,0 +1,32 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InMemoryEntityStore = void 0;
+const index_js_1 = require("../../../schema/index.cjs");
+class InMemoryEntityStore extends index_js_1.BaseEntityStore {
+    constructor() {
+        super();
+        Object.defineProperty(this, "store", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.store = Object.create(null);
+    }
+    async get(key, defaultValue) {
+        return key in this.store ? this.store[key] : defaultValue;
+    }
+    async set(key, value) {
+        this.store[key] = value;
+    }
+    async delete(key) {
+        delete this.store[key];
+    }
+    async exists(key) {
+        return key in this.store;
+    }
+    async clear() {
+        this.store = Object.create(null);
+    }
+}
+exports.InMemoryEntityStore = InMemoryEntityStore;

package/dist/memory/stores/entity/in_memory.d.ts

@@ -0,0 +1,10 @@
+import { BaseEntityStore } from "../../../schema/index.js";
+export declare class InMemoryEntityStore extends BaseEntityStore {
+    private store;
+    constructor();
+    get(key: string, defaultValue: string | undefined): Promise<string | undefined>;
+    set(key: string, value: string | undefined): Promise<void>;
+    delete(key: string): Promise<void>;
+    exists(key: string): Promise<boolean>;
+    clear(): Promise<void>;
+}

package/dist/memory/stores/entity/in_memory.js

@@ -0,0 +1,28 @@
+import { BaseEntityStore } from "../../../schema/index.js";
+export class InMemoryEntityStore extends BaseEntityStore {
+    constructor() {
+        super();
+        Object.defineProperty(this, "store", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.store = Object.create(null);
+    }
+    async get(key, defaultValue) {
+        return key in this.store ? this.store[key] : defaultValue;
+    }
+    async set(key, value) {
+        this.store[key] = value;
+    }
+    async delete(key) {
+        delete this.store[key];
+    }
+    async exists(key) {
+        return key in this.store;
+    }
+    async clear() {
+        this.store = Object.create(null);
+    }
+}

package/dist/prompts/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.checkValidTemplate = exports.renderTemplate = exports.parseTemplate = exports.BaseChatPromptTemplate = exports.MessagesPlaceholder = exports.ChatMessagePromptTemplate = exports.SystemMessagePromptTemplate = exports.AIMessagePromptTemplate = exports.HumanMessagePromptTemplate = exports.ChatPromptTemplate = exports.FewShotPromptTemplate = exports.SemanticSimilarityExampleSelector = exports.LengthBasedExampleSelector = exports.PromptTemplate = exports.BaseStringPromptTemplate = exports.StringPromptValue = exports.BasePromptTemplate = exports.BaseExampleSelector = void 0;
+exports.checkValidTemplate = exports.renderTemplate = exports.parseTemplate = exports.BaseChatPromptTemplate = exports.MessagesPlaceholder = exports.ChatMessagePromptTemplate = exports.SystemMessagePromptTemplate = exports.AIMessagePromptTemplate = exports.HumanMessagePromptTemplate = exports.ChatPromptTemplate = exports.FewShotPromptTemplate = exports.SemanticSimilarityExampleSelector = exports.LengthBasedExampleSelector = exports.isLLM = exports.isChatModel = exports.ConditionalPromptSelector = exports.BasePromptSelector = exports.PromptTemplate = exports.BaseStringPromptTemplate = exports.StringPromptValue = exports.BasePromptTemplate = exports.BaseExampleSelector = void 0;
 var base_js_1 = require("./base.cjs");
 Object.defineProperty(exports, "BaseExampleSelector", { enumerable: true, get: function () { return base_js_1.BaseExampleSelector; } });
 Object.defineProperty(exports, "BasePromptTemplate", { enumerable: true, get: function () { return base_js_1.BasePromptTemplate; } });
@@ -8,6 +8,11 @@ Object.defineProperty(exports, "StringPromptValue", { enumerable: true, get: fun
 Object.defineProperty(exports, "BaseStringPromptTemplate", { enumerable: true, get: function () { return base_js_1.BaseStringPromptTemplate; } });
 var prompt_js_1 = require("./prompt.cjs");
 Object.defineProperty(exports, "PromptTemplate", { enumerable: true, get: function () { return prompt_js_1.PromptTemplate; } });
+var conditional_js_1 = require("./selectors/conditional.cjs");
+Object.defineProperty(exports, "BasePromptSelector", { enumerable: true, get: function () { return conditional_js_1.BasePromptSelector; } });
+Object.defineProperty(exports, "ConditionalPromptSelector", { enumerable: true, get: function () { return conditional_js_1.ConditionalPromptSelector; } });
+Object.defineProperty(exports, "isChatModel", { enumerable: true, get: function () { return conditional_js_1.isChatModel; } });
+Object.defineProperty(exports, "isLLM", { enumerable: true, get: function () { return conditional_js_1.isLLM; } });
 var LengthBasedExampleSelector_js_1 = require("./selectors/LengthBasedExampleSelector.cjs");
 Object.defineProperty(exports, "LengthBasedExampleSelector", { enumerable: true, get: function () { return LengthBasedExampleSelector_js_1.LengthBasedExampleSelector; } });
 var SemanticSimilarityExampleSelector_js_1 = require("./selectors/SemanticSimilarityExampleSelector.cjs");

package/dist/prompts/index.d.ts

@@ -1,5 +1,6 @@
 export { BaseExampleSelector, BasePromptTemplate, BasePromptTemplateInput, StringPromptValue, BaseStringPromptTemplate, } from "./base.js";
 export { PromptTemplate, PromptTemplateInput } from "./prompt.js";
+export { BasePromptSelector, ConditionalPromptSelector, isChatModel, isLLM, } from "./selectors/conditional.js";
 export { LengthBasedExampleSelector, LengthBasedExampleSelectorInput, } from "./selectors/LengthBasedExampleSelector.js";
 export { SemanticSimilarityExampleSelector, SemanticSimilarityExampleSelectorInput, } from "./selectors/SemanticSimilarityExampleSelector.js";
 export { FewShotPromptTemplate, FewShotPromptTemplateInput, } from "./few_shot.js";

package/dist/prompts/index.js

@@ -1,5 +1,6 @@
 export { BaseExampleSelector, BasePromptTemplate, StringPromptValue, BaseStringPromptTemplate, } from "./base.js";
 export { PromptTemplate } from "./prompt.js";
+export { BasePromptSelector, ConditionalPromptSelector, isChatModel, isLLM, } from "./selectors/conditional.js";
 export { LengthBasedExampleSelector, } from "./selectors/LengthBasedExampleSelector.js";
 export { SemanticSimilarityExampleSelector, } from "./selectors/SemanticSimilarityExampleSelector.js";
 export { FewShotPromptTemplate, } from "./few_shot.js";

package/dist/{chains/prompt_selector.d.ts → prompts/selectors/conditional.d.ts}

@@ -1,7 +1,7 @@
-import { BaseChatModel } from "../chat_models/base.js";
-import { BasePromptTemplate } from "../prompts/base.js";
-import { BaseLanguageModel } from "../base_language/index.js";
-import { BaseLLM } from "../llms/base.js";
+import { BaseChatModel } from "../../chat_models/base.js";
+import { BasePromptTemplate } from "../base.js";
+import { BaseLanguageModel } from "../../base_language/index.js";
+import { BaseLLM } from "../../llms/base.js";
 export declare abstract class BasePromptSelector {
     abstract getPrompt(llm: BaseLanguageModel): BasePromptTemplate;
 }

package/dist/retrievers/document_compressors/chain_extract.cjs

@@ -59,19 +59,17 @@ class LLMChainExtractor extends index_js_2.BaseDocumentCompressor {
         this.getInput = getInput;
     }
     async compressDocuments(documents, query) {
-        const compressedDocs = [];
-        for (const doc of documents) {
+        const compressedDocs = await Promise.all(documents.map(async (doc) => {
             const input = this.getInput(query, doc);
             const output = await this.llmChain.predict(input);
-            if (output.length === 0) {
-                continue;
-            }
-            compressedDocs.push(new document_js_1.Document({
-                pageContent: output,
-                metadata: doc.metadata,
-            }));
-        }
-        return compressedDocs;
+            return output.length > 0
+                ? new document_js_1.Document({
+                    pageContent: output,
+                    metadata: doc.metadata,
+                })
+                : undefined;
+        }));
+        return compressedDocs.filter((doc) => doc !== undefined);
     }
     static fromLLM(llm, prompt, getInput) {
         const _prompt = prompt || getDefaultChainPrompt();

package/dist/retrievers/document_compressors/chain_extract.js

@@ -56,19 +56,17 @@ export class LLMChainExtractor extends BaseDocumentCompressor {
         this.getInput = getInput;
     }
     async compressDocuments(documents, query) {
-        const compressedDocs = [];
-        for (const doc of documents) {
+        const compressedDocs = await Promise.all(documents.map(async (doc) => {
             const input = this.getInput(query, doc);
             const output = await this.llmChain.predict(input);
-            if (output.length === 0) {
-                continue;
-            }
-            compressedDocs.push(new Document({
-                pageContent: output,
-                metadata: doc.metadata,
-            }));
-        }
-        return compressedDocs;
+            return output.length > 0
+                ? new Document({
+                    pageContent: output,
+                    metadata: doc.metadata,
+                })
+                : undefined;
+        }));
+        return compressedDocs.filter((doc) => doc !== undefined);
     }
     static fromLLM(llm, prompt, getInput) {
         const _prompt = prompt || getDefaultChainPrompt();

package/dist/schema/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.BaseFileStore = exports.BaseCache = exports.BaseListChatMessageHistory = exports.BaseChatMessageHistory = exports.BaseRetriever = exports.BasePromptValue = exports.ChatMessage = exports.SystemChatMessage = exports.AIChatMessage = exports.HumanChatMessage = exports.BaseChatMessage = exports.RUN_KEY = void 0;
+exports.BaseEntityStore = exports.BaseFileStore = exports.BaseCache = exports.BaseListChatMessageHistory = exports.BaseChatMessageHistory = exports.BaseRetriever = exports.BasePromptValue = exports.ChatMessage = exports.SystemChatMessage = exports.AIChatMessage = exports.HumanChatMessage = exports.BaseChatMessage = exports.RUN_KEY = void 0;
 exports.RUN_KEY = "__run";
 class BaseChatMessage {
     constructor(text) {
@@ -20,6 +20,15 @@ class BaseChatMessage {
         });
         this.text = text;
     }
+    toJSON() {
+        return {
+            type: this._getType(),
+            data: {
+                content: this.text,
+                role: "role" in this ? this.role : undefined,
+            },
+        };
+    }
 }
 exports.BaseChatMessage = BaseChatMessage;
 class HumanChatMessage extends BaseChatMessage {
@@ -86,3 +95,6 @@ exports.BaseCache = BaseCache;
 class BaseFileStore {
 }
 exports.BaseFileStore = BaseFileStore;
+class BaseEntityStore {
+}
+exports.BaseEntityStore = BaseEntityStore;

package/dist/schema/index.d.ts

@@ -34,6 +34,15 @@ export type LLMResult = {
      */
     [RUN_KEY]?: Record<string, any>;
 };
+export interface StoredMessageData {
+    content: string;
+    role: string | undefined;
+    additional_kwargs?: Record<string, any>;
+}
+export interface StoredMessage {
+    type: string;
+    data: StoredMessageData;
+}
 export type MessageType = "human" | "ai" | "generic" | "system";
 export declare abstract class BaseChatMessage {
     /** The text of the message. */
@@ -43,6 +52,7 @@ export declare abstract class BaseChatMessage {
     /** The type of the message. */
     abstract _getType(): MessageType;
     constructor(text: string);
+    toJSON(): StoredMessage;
 }
 export declare class HumanChatMessage extends BaseChatMessage {
     _getType(): MessageType;
@@ -86,6 +96,8 @@ export type AgentStep = {
     observation: string;
 };
 export type ChainValues = Record<string, any>;
+export type RunInputs = Record<string, any>;
+export type RunOutputs = Record<string, any>;
 /**
  * Base Index class. All indexes should extend this class.
  */
@@ -111,3 +123,10 @@ export declare abstract class BaseFileStore {
     abstract readFile(path: string): Promise<string>;
     abstract writeFile(path: string, contents: string): Promise<void>;
 }
+export declare abstract class BaseEntityStore {
+    abstract get(key: string, defaultValue?: string): Promise<string | undefined>;
+    abstract set(key: string, value?: string): Promise<void>;
+    abstract delete(key: string): Promise<void>;
+    abstract exists(key: string): Promise<boolean>;
+    abstract clear(): Promise<void>;
+}

package/dist/schema/index.js

@@ -17,6 +17,15 @@ export class BaseChatMessage {
         });
         this.text = text;
     }
+    toJSON() {
+        return {
+            type: this._getType(),
+            data: {
+                content: this.text,
+                role: "role" in this ? this.role : undefined,
+            },
+        };
+    }
 }
 export class HumanChatMessage extends BaseChatMessage {
     _getType() {
@@ -72,3 +81,5 @@ export class BaseCache {
 }
 export class BaseFileStore {
 }
+export class BaseEntityStore {
+}

package/dist/stores/message/dynamodb.cjs

@@ -74,10 +74,12 @@ class DynamoDBChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
         const messages = items
             .map((item) => ({
             type: item.M?.type.S,
-            role: item.M?.role?.S,
-            text: item.M?.text.S,
+            data: {
+                role: item.M?.role?.S,
+                content: item.M?.text.S,
+            },
         }))
-            .filter((x) => x.type !== undefined && x.text !== undefined);
+            .filter((x) => x.type !== undefined && x.data.content !== undefined);
         return (0, utils_js_1.mapStoredMessagesToChatMessages)(messages);
     }
     async clear() {
@@ -107,12 +109,12 @@ class DynamoDBChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
                                     S: message.type,
                                 },
                                 text: {
-                                    S: message.text,
+                                    S: message.data.content,
                                 },
                             },
                         };
-                        if (message.role) {
-                            dynamoSerializedMessage.M.role = { S: message.role };
+                        if (message.data.role) {
+                            dynamoSerializedMessage.M.role = { S: message.data.role };
                         }
                         return dynamoSerializedMessage;
                     }),

package/dist/stores/message/dynamodb.js

@@ -71,10 +71,12 @@ export class DynamoDBChatMessageHistory extends BaseListChatMessageHistory {
         const messages = items
             .map((item) => ({
             type: item.M?.type.S,
-            role: item.M?.role?.S,
-            text: item.M?.text.S,
+            data: {
+                role: item.M?.role?.S,
+                content: item.M?.text.S,
+            },
         }))
-            .filter((x) => x.type !== undefined && x.text !== undefined);
+            .filter((x) => x.type !== undefined && x.data.content !== undefined);
         return mapStoredMessagesToChatMessages(messages);
     }
     async clear() {
@@ -104,12 +106,12 @@ export class DynamoDBChatMessageHistory extends BaseListChatMessageHistory {
                                     S: message.type,
                                 },
                                 text: {
-                                    S: message.text,
+                                    S: message.data.content,
                                 },
                             },
                         };
-                        if (message.role) {
-                            dynamoSerializedMessage.M.role = { S: message.role };
+                        if (message.data.role) {
+                            dynamoSerializedMessage.M.role = { S: message.data.role };
                         }
                         return dynamoSerializedMessage;
                     }),

package/dist/stores/message/redis.cjs

@@ -0,0 +1,69 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RedisChatMessageHistory = void 0;
+const redis_1 = require("redis");
+const index_js_1 = require("../../schema/index.cjs");
+const utils_js_1 = require("./utils.cjs");
+class RedisChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
+    constructor(fields) {
+        const { sessionId, sessionTTL, config, client } = fields;
+        super();
+        Object.defineProperty(this, "client", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "sessionId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "sessionTTL", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.client = (client ?? (0, redis_1.createClient)(config ?? {}));
+        this.sessionId = sessionId;
+        this.sessionTTL = sessionTTL;
+    }
+    async ensureReadiness() {
+        if (!this.client.isReady) {
+            await this.client.connect();
+        }
+        return true;
+    }
+    async getMessages() {
+        await this.ensureReadiness();
+        const rawStoredMessages = await this.client.lRange(this.sessionId, 0, -1);
+        const orderedMessages = rawStoredMessages
+            .reverse()
+            .map((message) => JSON.parse(message));
+        const previousMessages = orderedMessages
+            .map((item) => ({
+            type: item.type,
+            data: {
+                role: item.role,
+                content: item.text,
+            },
+        }))
+            .filter((x) => x.type !== undefined && x.data.content !== undefined);
+        return (0, utils_js_1.mapStoredMessagesToChatMessages)(previousMessages);
+    }
+    async addMessage(message) {
+        await this.ensureReadiness();
+        const messageToAdd = (0, utils_js_1.mapChatMessagesToStoredMessages)([message]);
+        await this.client.lPush(this.sessionId, JSON.stringify(messageToAdd[0]));
+        if (this.sessionTTL) {
+            await this.client.expire(this.sessionId, this.sessionTTL);
+        }
+    }
+    async clear() {
+        await this.ensureReadiness();
+        await this.client.del(this.sessionId);
+    }
+}
+exports.RedisChatMessageHistory = RedisChatMessageHistory;

package/dist/stores/message/redis.d.ts

@@ -0,0 +1,18 @@
+import { RedisClientOptions, RedisClientType, RedisModules, RedisFunctions, RedisScripts } from "redis";
+import { BaseChatMessage, BaseListChatMessageHistory } from "../../schema/index.js";
+export type RedisChatMessageHistoryInput = {
+    sessionId: string;
+    sessionTTL?: number;
+    config?: RedisClientOptions;
+    client?: any;
+};
+export declare class RedisChatMessageHistory extends BaseListChatMessageHistory {
+    client: RedisClientType<RedisModules, RedisFunctions, RedisScripts>;
+    private sessionId;
+    private sessionTTL?;
+    constructor(fields: RedisChatMessageHistoryInput);
+    ensureReadiness(): Promise<boolean>;
+    getMessages(): Promise<BaseChatMessage[]>;
+    addMessage(message: BaseChatMessage): Promise<void>;
+    clear(): Promise<void>;
+}

package/dist/stores/message/redis.js

@@ -0,0 +1,65 @@
+import { createClient, } from "redis";
+import { BaseListChatMessageHistory, } from "../../schema/index.js";
+import { mapChatMessagesToStoredMessages, mapStoredMessagesToChatMessages, } from "./utils.js";
+export class RedisChatMessageHistory extends BaseListChatMessageHistory {
+    constructor(fields) {
+        const { sessionId, sessionTTL, config, client } = fields;
+        super();
+        Object.defineProperty(this, "client", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "sessionId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "sessionTTL", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.client = (client ?? createClient(config ?? {}));
+        this.sessionId = sessionId;
+        this.sessionTTL = sessionTTL;
+    }
+    async ensureReadiness() {
+        if (!this.client.isReady) {
+            await this.client.connect();
+        }
+        return true;
+    }
+    async getMessages() {
+        await this.ensureReadiness();
+        const rawStoredMessages = await this.client.lRange(this.sessionId, 0, -1);
+        const orderedMessages = rawStoredMessages
+            .reverse()
+            .map((message) => JSON.parse(message));
+        const previousMessages = orderedMessages
+            .map((item) => ({
+            type: item.type,
+            data: {
+                role: item.role,
+                content: item.text,
+            },
+        }))
+            .filter((x) => x.type !== undefined && x.data.content !== undefined);
+        return mapStoredMessagesToChatMessages(previousMessages);
+    }
+    async addMessage(message) {
+        await this.ensureReadiness();
+        const messageToAdd = mapChatMessagesToStoredMessages([message]);
+        await this.client.lPush(this.sessionId, JSON.stringify(messageToAdd[0]));
+        if (this.sessionTTL) {
+            await this.client.expire(this.sessionId, this.sessionTTL);
+        }
+    }
+    async clear() {
+        await this.ensureReadiness();
+        await this.client.del(this.sessionId);
+    }
+}

package/dist/stores/message/utils.cjs

@@ -1,31 +1,46 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.mapChatMessagesToStoredMessages = exports.mapStoredMessagesToChatMessages = void 0;
+exports.mapChatMessagesToStoredMessages = exports.mapStoredMessagesToChatMessages = exports.mapV1MessageToStoredMessage = void 0;
 const index_js_1 = require("../../schema/index.cjs");
+function mapV1MessageToStoredMessage(message) {
+    // TODO: Remove this mapper when we deprecate the old message format.
+    if (message.data !== undefined) {
+        return message;
+    }
+    else {
+        const v1Message = message;
+        return {
+            type: v1Message.type,
+            data: {
+                content: v1Message.text,
+                role: v1Message.role,
+            },
+        };
+    }
+}
+exports.mapV1MessageToStoredMessage = mapV1MessageToStoredMessage;
 function mapStoredMessagesToChatMessages(messages) {
     return messages.map((message) => {
-        switch (message.type) {
+        const storedMessage = mapV1MessageToStoredMessage(message);
+        switch (storedMessage.type) {
             case "human":
-                return new index_js_1.HumanChatMessage(message.text);
+                return new index_js_1.HumanChatMessage(storedMessage.data.content);
             case "ai":
-                return new index_js_1.AIChatMessage(message.text);
+                return new index_js_1.AIChatMessage(storedMessage.data.content);
             case "system":
-                return new index_js_1.SystemChatMessage(message.text);
-            default: {
-                if (message.role === undefined) {
-                    throw new Error("Role must be defined for generic messages");
+                return new index_js_1.SystemChatMessage(storedMessage.data.content);
+            case "chat":
+                if (storedMessage.data?.additional_kwargs?.role === undefined) {
+                    throw new Error("Role must be defined for chat messages");
                 }
-                return new index_js_1.ChatMessage(message.text, message.role);
-            }
+                return new index_js_1.ChatMessage(storedMessage.data.content, storedMessage.data.additional_kwargs.role);
+            default:
+                throw new Error(`Got unexpected type: ${storedMessage.type}`);
         }
     });
 }
 exports.mapStoredMessagesToChatMessages = mapStoredMessagesToChatMessages;
 function mapChatMessagesToStoredMessages(messages) {
-    return messages.map((message) => ({
-        type: message._getType(),
-        role: "role" in message ? message.role : undefined,
-        text: message.text,
-    }));
+    return messages.map((message) => message.toJSON());
 }
 exports.mapChatMessagesToStoredMessages = mapChatMessagesToStoredMessages;