farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-28 23:01:09 +03:00
Code Issues Packages Projects Releases Wiki Activity

update mrkl agents

Browse Source
This commit is contained in:
Henry
2024-02-20 18:23:39 +08:00
parent d1fdd8b3bd
commit 15afb8a2dd
4 changed files with 345 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts
+32 -8
View File
@@ -1,12 +1,13 @@
import { flatten } from 'lodash'
import { AgentExecutor, createReactAgent } from 'langchain/agents'
import { AgentExecutor } from 'langchain/agents'
import { pull } from 'langchain/hub'
import { Tool } from '@langchain/core/tools'
import type { PromptTemplate } from '@langchain/core/prompts'
import { BaseChatModel } from '@langchain/core/language_models/chat_models'
import { additionalCallbacks } from '../../../src/handler'
import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface'
import { FlowiseMemory, ICommonObject, IMessage, INode, INodeData, INodeParams } from '../../../src/Interface'
import { getBaseClasses } from '../../../src/utils'
import { createReactAgent } from '../../../src/agents'
class MRKLAgentChat_Agents implements INode {
label: string
@@ -18,11 +19,12 @@ class MRKLAgentChat_Agents implements INode {
category: string
baseClasses: string[]
inputs: INodeParams[]
sessionId?: string
constructor() {
constructor(fields?: { sessionId?: string }) {
this.label = 'ReAct Agent for Chat Models'
this.name = 'mrklAgentChat'
this.version = 2.0
this.version = 3.0
this.type = 'AgentExecutor'
this.category = 'Agents'
this.icon = 'agent.svg'
@@ -39,8 +41,14 @@ class MRKLAgentChat_Agents implements INode {
label: 'Chat Model',
name: 'model',
type: 'BaseChatModel'
},
{
label: 'Memory',
name: 'memory',
type: 'BaseChatMemory'
}
]
this.sessionId = fields?.sessionId
}
async init(): Promise<any> {
@@ -48,6 +56,7 @@ class MRKLAgentChat_Agents implements INode {
}
async run(nodeData: INodeData, input: string, options: ICommonObject): Promise<string> {
const memory = nodeData.inputs?.memory as FlowiseMemory
const model = nodeData.inputs?.model as BaseChatModel
let tools = nodeData.inputs?.tools as Tool[]
tools = flatten(tools)
@@ -68,10 +77,25 @@ class MRKLAgentChat_Agents implements INode {
const callbacks = await additionalCallbacks(nodeData, options)
const result = await executor.invoke({
input,
callbacks
})
const prevChatHistory = options.chatHistory
const chatHistory = ((await memory.getChatMessages(this.sessionId, false, prevChatHistory)) as IMessage[]) ?? []
const chatHistoryString = chatHistory.map((hist) => hist.message).join('\\n')
const result = await executor.invoke({ input, chat_history: chatHistoryString }, { callbacks })
await memory.addChatMessages(
[
{
text: input,
type: 'userMessage'
},
{
text: result?.output,
type: 'apiMessage'
}
],
this.sessionId
)
return result?.output
}
packages/components/nodes/agents/MRKLAgentLLM/MRKLAgentLLM.ts
+3 -5
View File
@@ -1,5 +1,5 @@
import { flatten } from 'lodash'
import { AgentExecutor, createReactAgent } from 'langchain/agents'
import { AgentExecutor } from 'langchain/agents'
import { pull } from 'langchain/hub'
import { Tool } from '@langchain/core/tools'
import type { PromptTemplate } from '@langchain/core/prompts'
@@ -7,6 +7,7 @@ import { BaseLanguageModel } from 'langchain/base_language'
import { additionalCallbacks } from '../../../src/handler'
import { getBaseClasses } from '../../../src/utils'
import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface'
import { createReactAgent } from '../../../src/agents'
class MRKLAgentLLM_Agents implements INode {
label: string
@@ -68,10 +69,7 @@ class MRKLAgentLLM_Agents implements INode {
const callbacks = await additionalCallbacks(nodeData, options)
const result = await executor.invoke({
input,
callbacks
})
const result = await executor.invoke({ input }, { callbacks })
return result?.output
}