Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add output keys fix

This commit is contained in:
Henry 2023-08-07 00:14:00 +01:00
parent 3029880967
commit 5ef0324304
2 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
packages/components/nodes/agents/ConversationalRetrievalAgent/ConversationalRetrievalAgent.ts
View File

@ -76,12 +76,10 @@ class ConversationalRetrievalAgent_Agents implements INode {
async run(nodeData: INodeData, input: string, options: ICommonObject): Promise<string> { async run(nodeData: INodeData, input: string, options: ICommonObject): Promise<string> {
const executor = nodeData.instance as AgentExecutor const executor = nodeData.instance as AgentExecutor
if (options && options.chatHistory) { 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)
;(executor.memory as any).chatHistory = mapChatHistory(options)
}
} }
const loggerHandler = new ConsoleCallbackHandler(options.logger) const loggerHandler = new ConsoleCallbackHandler(options.logger)

2
packages/components/src/utils.ts
View File

@ -509,7 +509,7 @@ export const getUserHome = (): string => {
*/ */
export const mapChatHistory = (options: ICommonObject): ChatMessageHistory => { export const mapChatHistory = (options: ICommonObject): ChatMessageHistory => {
const chatHistory = [] const chatHistory = []
const histories: IMessage[] = options.chatHistory const histories: IMessage[] = options.chatHistory ?? []
for (const message of histories) { for (const message of histories) {
if (message.type === 'apiMessage') { if (message.type === 'apiMessage') {