langchain 0.0.75 → 0.0.76

package/dist/memory/base.js

@@ -42,3 +42,10 @@ export function getBufferString(messages, humanPrefix = "Human", aiPrefix = "AI"
     }
     return string_messages.join("\n");
 }
+export function getPromptInputKey(inputs, memoryVariables) {
+    const promptInputKeys = Object.keys(inputs).filter((key) => !memoryVariables.includes(key) && key !== "stop");
+    if (promptInputKeys.length !== 1) {
+        throw new Error(`One input key expected, but got ${promptInputKeys.length}`);
+    }
+    return promptInputKeys[0];
+}

package/dist/memory/entity_memory.cjs

@@ -0,0 +1,151 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EntityMemory = void 0;
+const chat_memory_js_1 = require("./chat_memory.cjs");
+const prompt_js_1 = require("./prompt.cjs");
+const base_js_1 = require("./base.cjs");
+const llm_chain_js_1 = require("../chains/llm_chain.cjs");
+const in_memory_js_1 = require("./stores/entity/in_memory.cjs");
+// Entity extractor & summarizer to memory.
+class EntityMemory extends chat_memory_js_1.BaseChatMemory {
+    constructor(fields) {
+        super({
+            chatHistory: fields.chatHistory,
+            returnMessages: fields.returnMessages ?? false,
+            inputKey: fields.inputKey,
+            outputKey: fields.outputKey,
+        });
+        Object.defineProperty(this, "entityExtractionChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "entitySummarizationChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "entityStore", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "entityCache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
+        Object.defineProperty(this, "k", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 3
+        });
+        Object.defineProperty(this, "chatHistoryKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "history"
+        });
+        Object.defineProperty(this, "llm", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "entitiesKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "entities"
+        });
+        Object.defineProperty(this, "humanPrefix", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "aiPrefix", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.llm = fields.llm;
+        this.humanPrefix = fields.humanPrefix;
+        this.aiPrefix = fields.aiPrefix;
+        this.chatHistoryKey = fields.chatHistoryKey ?? this.chatHistoryKey;
+        this.entitiesKey = fields.entitiesKey ?? this.entitiesKey;
+        this.entityExtractionChain = new llm_chain_js_1.LLMChain({
+            llm: this.llm,
+            prompt: fields.entityExtractionPrompt ?? prompt_js_1.ENTITY_EXTRACTION_PROMPT,
+        });
+        this.entitySummarizationChain = new llm_chain_js_1.LLMChain({
+            llm: this.llm,
+            prompt: fields.entitySummarizationPrompt ?? prompt_js_1.ENTITY_SUMMARIZATION_PROMPT,
+        });
+        this.entityStore = fields.entityStore ?? new in_memory_js_1.InMemoryEntityStore();
+        this.entityCache = fields.entityCache ?? this.entityCache;
+        this.k = fields.k ?? this.k;
+    }
+    get memoryKeys() {
+        return [this.chatHistoryKey];
+    }
+    // Will always return list of memory variables.
+    get memoryVariables() {
+        return [this.entitiesKey, this.chatHistoryKey];
+    }
+    // Return history buffer.
+    async loadMemoryVariables(inputs) {
+        const promptInputKey = this.inputKey ?? (0, base_js_1.getPromptInputKey)(inputs, this.memoryVariables);
+        const messages = await this.chatHistory.getMessages();
+        const serializedMessages = (0, base_js_1.getBufferString)(messages.slice(-this.k * 2), this.humanPrefix, this.aiPrefix);
+        const output = await this.entityExtractionChain.predict({
+            history: serializedMessages,
+            input: inputs[promptInputKey],
+        });
+        const entities = output.trim() === "NONE" ? [] : output.split(",").map((w) => w.trim());
+        const entitySummaries = {};
+        for (const entity of entities) {
+            entitySummaries[entity] = await this.entityStore.get(entity, "No current information known.");
+        }
+        this.entityCache = [...entities];
+        const buffer = this.returnMessages
+            ? messages.slice(-this.k * 2)
+            : serializedMessages;
+        return {
+            [this.chatHistoryKey]: buffer,
+            [this.entitiesKey]: entitySummaries,
+        };
+    }
+    // Save context from this conversation to buffer.
+    async saveContext(inputs, outputs) {
+        await super.saveContext(inputs, outputs);
+        const promptInputKey = this.inputKey ?? (0, base_js_1.getPromptInputKey)(inputs, this.memoryVariables);
+        const messages = await this.chatHistory.getMessages();
+        const serializedMessages = (0, base_js_1.getBufferString)(messages.slice(-this.k * 2), this.humanPrefix, this.aiPrefix);
+        const inputData = inputs[promptInputKey];
+        for (const entity of this.entityCache) {
+            const existingSummary = await this.entityStore.get(entity, "No current information known.");
+            const output = await this.entitySummarizationChain.predict({
+                summary: existingSummary,
+                entity,
+                history: serializedMessages,
+                input: inputData,
+            });
+            if (output.trim() !== "UNCHANGED") {
+                await this.entityStore.set(entity, output.trim());
+            }
+        }
+    }
+    // Clear memory contents.
+    async clear() {
+        await super.clear();
+        await this.entityStore.clear();
+    }
+}
+exports.EntityMemory = EntityMemory;

package/dist/memory/entity_memory.d.ts

@@ -0,0 +1,35 @@
+import { BaseLanguageModel } from "../base_language/index.js";
+import { BaseEntityStore } from "../schema/index.js";
+import { BaseChatMemory, BaseChatMemoryInput } from "./chat_memory.js";
+import { InputValues, MemoryVariables, OutputValues } from "./base.js";
+import { PromptTemplate } from "../prompts/prompt.js";
+export interface EntityMemoryInput extends BaseChatMemoryInput {
+    llm: BaseLanguageModel;
+    humanPrefix?: string;
+    aiPrefix?: string;
+    entityExtractionPrompt?: PromptTemplate;
+    entitySummarizationPrompt?: PromptTemplate;
+    entityCache?: string[];
+    k?: number;
+    chatHistoryKey?: string;
+    entitiesKey?: string;
+    entityStore?: BaseEntityStore;
+}
+export declare class EntityMemory extends BaseChatMemory implements EntityMemoryInput {
+    private entityExtractionChain;
+    private entitySummarizationChain;
+    entityStore: BaseEntityStore;
+    entityCache: string[];
+    k: number;
+    chatHistoryKey: string;
+    llm: BaseLanguageModel;
+    entitiesKey: string;
+    humanPrefix?: string;
+    aiPrefix?: string;
+    constructor(fields: EntityMemoryInput);
+    get memoryKeys(): string[];
+    get memoryVariables(): string[];
+    loadMemoryVariables(inputs: InputValues): Promise<MemoryVariables>;
+    saveContext(inputs: InputValues, outputs: OutputValues): Promise<void>;
+    clear(): Promise<void>;
+}

package/dist/memory/entity_memory.js

@@ -0,0 +1,147 @@
+import { BaseChatMemory } from "./chat_memory.js";
+import { ENTITY_EXTRACTION_PROMPT, ENTITY_SUMMARIZATION_PROMPT, } from "./prompt.js";
+import { getBufferString, getPromptInputKey, } from "./base.js";
+import { LLMChain } from "../chains/llm_chain.js";
+import { InMemoryEntityStore } from "./stores/entity/in_memory.js";
+// Entity extractor & summarizer to memory.
+export class EntityMemory extends BaseChatMemory {
+    constructor(fields) {
+        super({
+            chatHistory: fields.chatHistory,
+            returnMessages: fields.returnMessages ?? false,
+            inputKey: fields.inputKey,
+            outputKey: fields.outputKey,
+        });
+        Object.defineProperty(this, "entityExtractionChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "entitySummarizationChain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "entityStore", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "entityCache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
+        Object.defineProperty(this, "k", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 3
+        });
+        Object.defineProperty(this, "chatHistoryKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "history"
+        });
+        Object.defineProperty(this, "llm", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "entitiesKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "entities"
+        });
+        Object.defineProperty(this, "humanPrefix", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "aiPrefix", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.llm = fields.llm;
+        this.humanPrefix = fields.humanPrefix;
+        this.aiPrefix = fields.aiPrefix;
+        this.chatHistoryKey = fields.chatHistoryKey ?? this.chatHistoryKey;
+        this.entitiesKey = fields.entitiesKey ?? this.entitiesKey;
+        this.entityExtractionChain = new LLMChain({
+            llm: this.llm,
+            prompt: fields.entityExtractionPrompt ?? ENTITY_EXTRACTION_PROMPT,
+        });
+        this.entitySummarizationChain = new LLMChain({
+            llm: this.llm,
+            prompt: fields.entitySummarizationPrompt ?? ENTITY_SUMMARIZATION_PROMPT,
+        });
+        this.entityStore = fields.entityStore ?? new InMemoryEntityStore();
+        this.entityCache = fields.entityCache ?? this.entityCache;
+        this.k = fields.k ?? this.k;
+    }
+    get memoryKeys() {
+        return [this.chatHistoryKey];
+    }
+    // Will always return list of memory variables.
+    get memoryVariables() {
+        return [this.entitiesKey, this.chatHistoryKey];
+    }
+    // Return history buffer.
+    async loadMemoryVariables(inputs) {
+        const promptInputKey = this.inputKey ?? getPromptInputKey(inputs, this.memoryVariables);
+        const messages = await this.chatHistory.getMessages();
+        const serializedMessages = getBufferString(messages.slice(-this.k * 2), this.humanPrefix, this.aiPrefix);
+        const output = await this.entityExtractionChain.predict({
+            history: serializedMessages,
+            input: inputs[promptInputKey],
+        });
+        const entities = output.trim() === "NONE" ? [] : output.split(",").map((w) => w.trim());
+        const entitySummaries = {};
+        for (const entity of entities) {
+            entitySummaries[entity] = await this.entityStore.get(entity, "No current information known.");
+        }
+        this.entityCache = [...entities];
+        const buffer = this.returnMessages
+            ? messages.slice(-this.k * 2)
+            : serializedMessages;
+        return {
+            [this.chatHistoryKey]: buffer,
+            [this.entitiesKey]: entitySummaries,
+        };
+    }
+    // Save context from this conversation to buffer.
+    async saveContext(inputs, outputs) {
+        await super.saveContext(inputs, outputs);
+        const promptInputKey = this.inputKey ?? getPromptInputKey(inputs, this.memoryVariables);
+        const messages = await this.chatHistory.getMessages();
+        const serializedMessages = getBufferString(messages.slice(-this.k * 2), this.humanPrefix, this.aiPrefix);
+        const inputData = inputs[promptInputKey];
+        for (const entity of this.entityCache) {
+            const existingSummary = await this.entityStore.get(entity, "No current information known.");
+            const output = await this.entitySummarizationChain.predict({
+                summary: existingSummary,
+                entity,
+                history: serializedMessages,
+                input: inputData,
+            });
+            if (output.trim() !== "UNCHANGED") {
+                await this.entityStore.set(entity, output.trim());
+            }
+        }
+    }
+    // Clear memory contents.
+    async clear() {
+        await super.clear();
+        await this.entityStore.clear();
+    }
+}

package/dist/memory/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.VectorStoreRetrieverMemory = exports.MotorheadMemory = exports.ChatMessageHistory = exports.BaseChatMemory = exports.BufferWindowMemory = exports.ConversationSummaryMemory = exports.getBufferString = exports.getInputValue = exports.BaseMemory = exports.BufferMemory = void 0;
+exports.ENTITY_MEMORY_CONVERSATION_TEMPLATE = exports.EntityMemory = exports.VectorStoreRetrieverMemory = exports.MotorheadMemory = exports.ChatMessageHistory = exports.BaseChatMemory = exports.BufferWindowMemory = exports.ConversationSummaryMemory = exports.getBufferString = exports.getInputValue = exports.BaseMemory = exports.BufferMemory = void 0;
 var buffer_memory_js_1 = require("./buffer_memory.cjs");
 Object.defineProperty(exports, "BufferMemory", { enumerable: true, get: function () { return buffer_memory_js_1.BufferMemory; } });
 var base_js_1 = require("./base.cjs");
@@ -19,3 +19,7 @@ var motorhead_memory_js_1 = require("./motorhead_memory.cjs");
 Object.defineProperty(exports, "MotorheadMemory", { enumerable: true, get: function () { return motorhead_memory_js_1.MotorheadMemory; } });
 var vector_store_js_1 = require("./vector_store.cjs");
 Object.defineProperty(exports, "VectorStoreRetrieverMemory", { enumerable: true, get: function () { return vector_store_js_1.VectorStoreRetrieverMemory; } });
+var entity_memory_js_1 = require("./entity_memory.cjs");
+Object.defineProperty(exports, "EntityMemory", { enumerable: true, get: function () { return entity_memory_js_1.EntityMemory; } });
+var prompt_js_1 = require("./prompt.cjs");
+Object.defineProperty(exports, "ENTITY_MEMORY_CONVERSATION_TEMPLATE", { enumerable: true, get: function () { return prompt_js_1.ENTITY_MEMORY_CONVERSATION_TEMPLATE; } });

package/dist/memory/index.d.ts

@@ -6,3 +6,5 @@ export { BaseChatMemory, BaseChatMemoryInput } from "./chat_memory.js";
 export { ChatMessageHistory } from "../stores/message/in_memory.js";
 export { MotorheadMemory, MotorheadMemoryInput } from "./motorhead_memory.js";
 export { VectorStoreRetrieverMemory, VectorStoreRetrieverMemoryParams, } from "./vector_store.js";
+export { EntityMemory } from "./entity_memory.js";
+export { ENTITY_MEMORY_CONVERSATION_TEMPLATE } from "./prompt.js";

package/dist/memory/index.js

@@ -6,3 +6,5 @@ export { BaseChatMemory } from "./chat_memory.js";
 export { ChatMessageHistory } from "../stores/message/in_memory.js";
 export { MotorheadMemory } from "./motorhead_memory.js";
 export { VectorStoreRetrieverMemory, } from "./vector_store.js";
+export { EntityMemory } from "./entity_memory.js";
+export { ENTITY_MEMORY_CONVERSATION_TEMPLATE } from "./prompt.js";

package/dist/memory/prompt.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SUMMARY_PROMPT = void 0;
+exports.ENTITY_SUMMARIZATION_PROMPT = exports._DEFAULT_ENTITY_SUMMARIZATION_TEMPLATE = exports.ENTITY_EXTRACTION_PROMPT = exports._DEFAULT_ENTITY_EXTRACTION_TEMPLATE = exports.ENTITY_MEMORY_CONVERSATION_TEMPLATE = exports._DEFAULT_ENTITY_MEMORY_CONVERSATION_TEMPLATE = exports.SUMMARY_PROMPT = void 0;
 const prompt_js_1 = require("../prompts/prompt.cjs");
 const _DEFAULT_SUMMARIZER_TEMPLATE = `Progressively summarize the lines of conversation provided, adding onto the previous summary returning a new summary.
 
@@ -28,3 +28,86 @@ exports.SUMMARY_PROMPT = new prompt_js_1.PromptTemplate({
     inputVariables: ["summary", "new_lines"],
     template: _DEFAULT_SUMMARIZER_TEMPLATE,
 });
+exports._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:`;
+exports.ENTITY_MEMORY_CONVERSATION_TEMPLATE = 
+// eslint-disable-next-line spaced-comment
+/*#__PURE__*/ new prompt_js_1.PromptTemplate({
+    inputVariables: ["entities", "history", "input"],
+    template: exports._DEFAULT_ENTITY_MEMORY_CONVERSATION_TEMPLATE,
+});
+exports._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
+exports.ENTITY_EXTRACTION_PROMPT = new prompt_js_1.PromptTemplate({
+    inputVariables: ["history", "input"],
+    template: exports._DEFAULT_ENTITY_EXTRACTION_TEMPLATE,
+});
+exports._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
+exports.ENTITY_SUMMARIZATION_PROMPT = new prompt_js_1.PromptTemplate({
+    inputVariables: ["entity", "summary", "history", "input"],
+    template: exports._DEFAULT_ENTITY_SUMMARIZATION_TEMPLATE,
+});

package/dist/memory/prompt.d.ts

@@ -1,2 +1,8 @@
 import { PromptTemplate } from "../prompts/prompt.js";
 export declare const SUMMARY_PROMPT: PromptTemplate;
+export declare const _DEFAULT_ENTITY_MEMORY_CONVERSATION_TEMPLATE = "You are an assistant to a human, powered by a large language model trained by OpenAI.\n\nYou 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.\n\nYou 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.\n\nOverall, 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.\n\nContext:\n{entities}\n\nCurrent conversation:\n{history}\nLast line:\nHuman: {input}\nYou:";
+export declare const ENTITY_MEMORY_CONVERSATION_TEMPLATE: PromptTemplate;
+export declare 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.\n\nThe 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.\n\nReturn 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).\n\nEXAMPLE\nConversation history:\nPerson #1: my name is Jacob. how's it going today?\nAI: \"It's going great! How about you?\"\nPerson #1: good! busy working on Langchain. lots to do.\nAI: \"That sounds like a lot of work! What kind of things are you doing to make Langchain better?\"\nLast line:\nPerson #1: i'm trying to improve Langchain's interfaces, the UX, its integrations with various products the user might want ... a lot of stuff.\nOutput: Jacob,Langchain\nEND OF EXAMPLE\n\nEXAMPLE\nConversation history:\nPerson #1: how's it going today?\nAI: \"It's going great! How about you?\"\nPerson #1: good! busy working on Langchain. lots to do.\nAI: \"That sounds like a lot of work! What kind of things are you doing to make Langchain better?\"\nLast line:\nPerson #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.\nOutput: Langchain, Person #2\nEND OF EXAMPLE\n\nConversation history (for reference only):\n{history}\nLast line of conversation (for extraction):\nHuman: {input}\n\nOutput:";
+export declare const ENTITY_EXTRACTION_PROMPT: PromptTemplate;
+export declare 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.\nThe 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.\n\nIf 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.\n\nFull conversation history (for context):\n{history}\n\nEntity to summarize:\n{entity}\n\nExisting summary of {entity}:\n{summary}\n\nLast line of conversation:\nHuman: {input}\nUpdated summary (or the exact string \"UNCHANGED\" if there is no new information about {entity} above):";
+export declare const ENTITY_SUMMARIZATION_PROMPT: PromptTemplate;

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);
+    }
+}