add memory fix
This commit is contained in:
parent
a5b519d527
commit
41caf2aee7
|
|
@ -95,8 +95,12 @@ class ConversationalAgent_Agents implements INode {
|
||||||
const callbacks = await additionalCallbacks(nodeData, options)
|
const callbacks = await additionalCallbacks(nodeData, options)
|
||||||
|
|
||||||
if (options && options.chatHistory) {
|
if (options && options.chatHistory) {
|
||||||
memory.chatHistory = mapChatHistory(options)
|
const chatHistoryClassName = memory.chatHistory.constructor.name
|
||||||
executor.memory = memory
|
// Only replace when its In-Memory
|
||||||
|
if (chatHistoryClassName && chatHistoryClassName === 'ChatMessageHistory') {
|
||||||
|
memory.chatHistory = mapChatHistory(options)
|
||||||
|
executor.memory = memory
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const result = await executor.call({ input }, [...callbacks])
|
const result = await executor.call({ input }, [...callbacks])
|
||||||
|
|
|
||||||
|
|
@ -82,7 +82,11 @@ class ConversationalRetrievalAgent_Agents implements INode {
|
||||||
if (executor.memory) {
|
if (executor.memory) {
|
||||||
;(executor.memory as any).memoryKey = 'chat_history'
|
;(executor.memory as any).memoryKey = 'chat_history'
|
||||||
;(executor.memory as any).outputKey = 'output'
|
;(executor.memory as any).outputKey = 'output'
|
||||||
;(executor.memory as any).chatHistory = mapChatHistory(options)
|
const chatHistoryClassName = (executor.memory as any).chatHistory.constructor.name
|
||||||
|
// Only replace when its In-Memory
|
||||||
|
if (chatHistoryClassName && chatHistoryClassName === 'ChatMessageHistory') {
|
||||||
|
;(executor.memory as any).chatHistory = mapChatHistory(options)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const loggerHandler = new ConsoleCallbackHandler(options.logger)
|
const loggerHandler = new ConsoleCallbackHandler(options.logger)
|
||||||
|
|
|
||||||
|
|
@ -81,8 +81,12 @@ class OpenAIFunctionAgent_Agents implements INode {
|
||||||
const memory = nodeData.inputs?.memory as BaseChatMemory
|
const memory = nodeData.inputs?.memory as BaseChatMemory
|
||||||
|
|
||||||
if (options && options.chatHistory) {
|
if (options && options.chatHistory) {
|
||||||
memory.chatHistory = mapChatHistory(options)
|
const chatHistoryClassName = memory.chatHistory.constructor.name
|
||||||
executor.memory = memory
|
// Only replace when its In-Memory
|
||||||
|
if (chatHistoryClassName && chatHistoryClassName === 'ChatMessageHistory') {
|
||||||
|
memory.chatHistory = mapChatHistory(options)
|
||||||
|
executor.memory = memory
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const loggerHandler = new ConsoleCallbackHandler(options.logger)
|
const loggerHandler = new ConsoleCallbackHandler(options.logger)
|
||||||
|
|
|
||||||
|
|
@ -106,8 +106,12 @@ class ConversationChain_Chains implements INode {
|
||||||
const memory = nodeData.inputs?.memory as BufferMemory
|
const memory = nodeData.inputs?.memory as BufferMemory
|
||||||
|
|
||||||
if (options && options.chatHistory) {
|
if (options && options.chatHistory) {
|
||||||
memory.chatHistory = mapChatHistory(options)
|
const chatHistoryClassName = memory.chatHistory.constructor.name
|
||||||
chain.memory = memory
|
// Only replace when its In-Memory
|
||||||
|
if (chatHistoryClassName && chatHistoryClassName === 'ChatMessageHistory') {
|
||||||
|
memory.chatHistory = mapChatHistory(options)
|
||||||
|
chain.memory = memory
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const loggerHandler = new ConsoleCallbackHandler(options.logger)
|
const loggerHandler = new ConsoleCallbackHandler(options.logger)
|
||||||
|
|
|
||||||
|
|
@ -179,7 +179,11 @@ class ConversationalRetrievalQAChain_Chains implements INode {
|
||||||
const obj = { question: input }
|
const obj = { question: input }
|
||||||
|
|
||||||
if (options && options.chatHistory && chain.memory) {
|
if (options && options.chatHistory && chain.memory) {
|
||||||
;(chain.memory as any).chatHistory = mapChatHistory(options)
|
const chatHistoryClassName = (chain.memory as any).chatHistory.constructor.name
|
||||||
|
// Only replace when its In-Memory
|
||||||
|
if (chatHistoryClassName && chatHistoryClassName === 'ChatMessageHistory') {
|
||||||
|
;(chain.memory as any).chatHistory = mapChatHistory(options)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const loggerHandler = new ConsoleCallbackHandler(options.logger)
|
const loggerHandler = new ConsoleCallbackHandler(options.logger)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue