Class EntityMemory

Class for managing entity extraction and summarization to memory in chatbot applications. Extends the BaseChatMemory class and implements the EntityMemoryInput interface.

Example

const memory = new EntityMemory({
  llm: new ChatOpenAI({ temperature: 0 }),
  chatHistoryKey: "history",
  entitiesKey: "entities",
});
const model = new ChatOpenAI({ temperature: 0.9 });
const chain = new LLMChain({
  llm: model,
  prompt: ENTITY_MEMORY_CONVERSATION_TEMPLATE,
  memory,
});

const res1 = await chain.call({ input: "Hi! I'm Jim." });
console.log({
  res1,
  memory: await memory.loadMemoryVariables({ input: "Who is Jim?" }),
});

const res2 = await chain.call({
  input: "I work in construction. What about you?",
});
console.log({
  res2,
  memory: await memory.loadMemoryVariables({ input: "Who is Jim?" }),
});

Hierarchy (view full)

Implements

  • EntityMemoryInput

Constructors

constructor

Properties

aiPrefix? chatHistory chatHistoryKey entitiesKey entityCache entityStore humanPrefix? inputKey? k llm outputKey? returnMessages

Accessors

memoryKeys memoryVariables

Methods

clear loadMemoryVariables saveContext

Constructors

constructor

Properties

OptionalaiPrefix

aiPrefix?: string

chatHistory

chatHistory: BaseChatMessageHistory

chatHistoryKey

chatHistoryKey: string = "history"

entitiesKey

entitiesKey: string = "entities"

entityCache

entityCache: string[] = []

entityStore

entityStore: BaseEntityStore

OptionalhumanPrefix

humanPrefix?: string

OptionalinputKey

inputKey?: string

k

k: number = 3

llm

llm: BaseLanguageModelInterface<any, BaseLanguageModelCallOptions>

OptionaloutputKey

outputKey?: string

returnMessages

returnMessages: boolean = false

Accessors

memoryKeys

memoryVariables

Methods

clear

loadMemoryVariables

saveContext

Settings

Member Visibility

On This Page

Constructors
Properties
Accessors
Methods