add baselanguagemodel

This commit is contained in:
Henry
2023-05-17 14:43:15 +01:00
parent 89c1de34f8
commit d44ab269e6
4 changed files with 25 additions and 25 deletions
@@ -1,8 +1,8 @@
import { INode, INodeData, INodeParams } from '../../../src/Interface'
import { initializeAgentExecutorWithOptions, AgentExecutor } from 'langchain/agents'
import { Tool } from 'langchain/tools'
import { BaseLLM } from 'langchain/llms/base'
import { getBaseClasses } from '../../../src/utils'
import { BaseLanguageModel } from 'langchain/base_language'
class MRKLAgentLLM_Agents implements INode {
label: string
@@ -30,15 +30,15 @@ class MRKLAgentLLM_Agents implements INode {
list: true
},
{
label: 'LLM Model',
label: 'Language Model',
name: 'model',
type: 'BaseLLM'
type: 'BaseLanguageModel'
}
]
}
async init(nodeData: INodeData): Promise<any> {
const model = nodeData.inputs?.model as BaseLLM
const model = nodeData.inputs?.model as BaseLanguageModel
let tools = nodeData.inputs?.tools as Tool[]
tools = tools.flat()