add override customToolFunc

This commit is contained in:
Henry
2023-08-07 01:45:40 +01:00
parent e866c55c5a
commit 68b6d4f30a
2 changed files with 5 additions and 3 deletions
@@ -36,7 +36,7 @@ class CustomTool_Tools implements INode {
//@ts-ignore
loadMethods = {
async listTools(nodeData: INodeData, options: ICommonObject): Promise<INodeOptionsValue[]> {
async listTools(_: INodeData, options: ICommonObject): Promise<INodeOptionsValue[]> {
const returnData: INodeOptionsValue[] = []
const appDataSource = options.appDataSource as DataSource
@@ -60,8 +60,9 @@ class CustomTool_Tools implements INode {
}
}
async init(nodeData: INodeData, input: string, options: ICommonObject): Promise<any> {
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
const selectedToolId = nodeData.inputs?.selectedTool as string
const customToolFunc = nodeData.inputs?.customToolFunc as string
const appDataSource = options.appDataSource as DataSource
const databaseEntities = options.databaseEntities as IDatabaseEntity
@@ -78,6 +79,7 @@ class CustomTool_Tools implements INode {
schema: z.object(convertSchemaToZod(tool.schema)),
code: tool.func
}
if (customToolFunc) obj.code = customToolFunc
return new DynamicStructuredTool(obj)
} catch (e) {
throw new Error(e)