Feature/prepend messages to agent graph (#2897)

prepend messages to agent graph
This commit is contained in:
Henry Heng
2024-07-29 11:09:38 +01:00
committed by GitHub
parent e39fd130d0
commit b5d969b2bb
4 changed files with 38 additions and 6 deletions
+4 -4
View File
@@ -587,10 +587,10 @@ export const mapChatMessageToBaseMessage = (chatmessages: any[] = []): BaseMessa
const chatHistory = []
for (const message of chatmessages) {
if (message.role === 'apiMessage') {
chatHistory.push(new AIMessage(message.content))
} else if (message.role === 'userMessage') {
chatHistory.push(new HumanMessage(message.content))
if (message.role === 'apiMessage' || message.type === 'apiMessage') {
chatHistory.push(new AIMessage(message.content || ''))
} else if (message.role === 'userMessage' || message.role === 'userMessage') {
chatHistory.push(new HumanMessage(message.content || ''))
}
}
return chatHistory