image uploads for mrkl agent

feature/Vision
vinodkiran 2024-02-20 13:48:02 -08:00
parent 0bc85593d5
commit 4cee518cbf
1 changed files with 2 additions and 2 deletions

View File

@ -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