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

add aiplugintools

Browse Source
This commit is contained in:
Henry
2023-04-17 20:26:57 +01:00
parent 7631f651ab
commit ed81dc228f
5 changed files with 88 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts
+14 -3
View File
@@ -1,6 +1,6 @@
import { INode, INodeData, INodeParams } from '../../../src/Interface'
import { initializeAgentExecutor, AgentExecutor } from 'langchain/agents'
import { Tool } from 'langchain/tools'
import { AIPluginTool } from 'langchain/tools'
import { BaseChatModel } from 'langchain/chat_models/base'
import { getBaseClasses } from '../../../src/utils'
@@ -39,9 +39,20 @@ class MRKLAgentChat_Agents implements INode {
async init(nodeData: INodeData): Promise<any> {
const model = nodeData.inputs?.model as BaseChatModel
const tools = nodeData.inputs?.tools as Tool[]
const tools = nodeData.inputs?.tools
const executor = await initializeAgentExecutor(tools, model, 'chat-zero-shot-react-description', true)
const allowedTools = []
for (let i = 0; i < tools.length; i += 1) {
if (tools[i].pluginUrl) {
const pluginURL: string = tools[i].pluginUrl
const aiplugin = await AIPluginTool.fromPluginUrl(pluginURL)
allowedTools.push(aiplugin)
} else {
allowedTools.push(tools[i])
}
}
const executor = await initializeAgentExecutor(allowedTools, model, 'chat-zero-shot-react-description', true)
return executor
}