From 4cee518cbf93688fed26eda10c1bb4edcfffe874 Mon Sep 17 00:00:00 2001 From: vinodkiran Date: Tue, 20 Feb 2024 13:48:02 -0800 Subject: [PATCH] image uploads for mrkl agent --- .../components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/packages/components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts b/packages/components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts index 0d5b4c48..1ed831aa 100644 --- a/packages/components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts +++ b/packages/components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts @@ -79,8 +79,8 @@ class MRKLAgentChat_Agents implements INode { // Change default max token to higher when using gpt-4-vision chatModel.maxTokens = 1024 const oldTemplate = promptWithChat.template as string - let chatPromptTemplate = ChatPromptTemplate.fromMessages([SystemMessagePromptTemplate.fromTemplate(oldTemplate)]) - chatPromptTemplate.promptMessages = [new HumanMessage({ content: messageContent })] + chatPromptTemplate = ChatPromptTemplate.fromMessages([SystemMessagePromptTemplate.fromTemplate(oldTemplate)]) + chatPromptTemplate.promptMessages.push(new HumanMessage({ content: messageContent })) } else { // revert to previous values if image upload is empty chatModel.modelName = chatModel.configuredModel