add chatgpt plugin

This commit is contained in:
Henry
2023-04-19 14:48:31 +01:00
parent 2a04c8baea
commit aad473d946
3 changed files with 312 additions and 20 deletions
@@ -1,6 +1,5 @@
import { INode, INodeData, INodeParams } from '../../../src/Interface'
import { initializeAgentExecutor, AgentExecutor } from 'langchain/agents'
import { AIPluginTool } from 'langchain/tools'
import { initializeAgentExecutor, AgentExecutor, Tool } from 'langchain/agents'
import { BaseChatModel } from 'langchain/chat_models/base'
import { getBaseClasses } from '../../../src/utils'
@@ -39,20 +38,9 @@ class MRKLAgentChat_Agents implements INode {
async init(nodeData: INodeData): Promise<any> {
const model = nodeData.inputs?.model as BaseChatModel
const tools = nodeData.inputs?.tools
const tools = nodeData.inputs?.tools as Tool[]
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)
const executor = await initializeAgentExecutor(tools, model, 'chat-zero-shot-react-description', true)
return executor
}
@@ -32,11 +32,8 @@ class AIPlugin implements INode {
async init(nodeData: INodeData): Promise<any> {
const pluginUrl = nodeData.inputs?.pluginUrl as string
// Doesn't work currently
// const aiplugin = await AIPluginTool.fromPluginUrl(pluginUrl)
const aiplugin = {
pluginUrl: pluginUrl
}
const aiplugin = await AIPluginTool.fromPluginUrl(pluginUrl)
return aiplugin
}
}