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
+2
View File
@@ -153,6 +153,8 @@ export interface INodeCredential {
export interface IMessage {
message: string
type: MessageType
role?: MessageType
content?: string
}
export interface IUsedTool {
+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