feat(Think Tool Node): ToolThink, a simple tool to force the AI agent to do some thinking (#14351)

This commit is contained in:
Yiorgis Gozadinos
2025-04-03 12:06:23 +03:00
committed by GitHub
parent 85cbfb64c0
commit 281b70be04
3 changed files with 114 additions and 0 deletions

View File

@@ -0,0 +1,79 @@
/* eslint-disable n8n-nodes-base/node-dirname-against-convention */
import { DynamicTool } from 'langchain/tools';
import {
NodeConnectionTypes,
type INodeType,
type INodeTypeDescription,
type ISupplyDataFunctions,
type SupplyData,
} from 'n8n-workflow';
import { logWrapper } from '@utils/logWrapper';
import { getConnectionHintNoticeField } from '@utils/sharedFields';
// A thinking tool, see https://www.anthropic.com/engineering/claude-think-tool
const defaultToolDescription =
'Use the tool to think about something. It will not obtain new information or change the database, but just append the thought to the log. Use it when complex reasoning or some cache memory is needed.';
export class ToolThink implements INodeType {
description: INodeTypeDescription = {
displayName: 'Think Tool',
name: 'toolThink',
icon: 'fa:brain',
iconColor: 'black',
group: ['transform'],
version: 1,
description: 'Invite the AI agent to do some thinking',
defaults: {
name: 'Think',
},
codex: {
categories: ['AI'],
subcategories: {
AI: ['Tools'],
Tools: ['Other Tools'],
},
resources: {
primaryDocumentation: [
{
url: 'https://docs.n8n.io/integrations/builtin/cluster-nodes/sub-nodes/n8n-nodes-langchain.toolthink/',
},
],
},
},
inputs: [],
outputs: [NodeConnectionTypes.AiTool],
outputNames: ['Tool'],
properties: [
getConnectionHintNoticeField([NodeConnectionTypes.AiAgent]),
{
displayName: 'Think Tool Description',
name: 'description',
type: 'string',
default: defaultToolDescription,
placeholder: '[Describe your thinking tool here, explaining how it will help the AI think]',
description: "The thinking tool's description",
typeOptions: {
rows: 3,
},
required: true,
},
],
};
async supplyData(this: ISupplyDataFunctions, itemIndex: number): Promise<SupplyData> {
const description = this.getNodeParameter('description', itemIndex) as string;
const tool = new DynamicTool({
name: 'thinking_tool',
description,
func: async (subject: string) => {
return subject;
},
});
return {
response: logWrapper(tool, this),
};
}
}

View File

@@ -0,0 +1,34 @@
import { mock } from 'jest-mock-extended';
import { DynamicTool } from 'langchain/tools';
import type { ISupplyDataFunctions } from 'n8n-workflow';
import { ToolThink } from '../ToolThink.node';
describe('ToolThink', () => {
const thinkTool = new ToolThink();
const helpers = mock<ISupplyDataFunctions['helpers']>();
const executeFunctions = mock<ISupplyDataFunctions>({
helpers,
});
executeFunctions.addInputData.mockReturnValue({ index: 0 });
executeFunctions.getNodeParameter.mockImplementation((paramName: string) => {
switch (paramName) {
case 'description':
return 'Tool description';
default:
return undefined;
}
});
describe('Tool response', () => {
it('should return the same text as response when receiving a text input', async () => {
const { response } = (await thinkTool.supplyData.call(executeFunctions, 0)) as {
response: DynamicTool;
};
expect(response).toBeInstanceOf(DynamicTool);
expect(response.description).toEqual('Tool description');
const res = (await response.invoke('foo')) as string;
expect(res).toEqual('foo');
});
});
});

View File

@@ -104,6 +104,7 @@
"dist/nodes/tools/ToolCode/ToolCode.node.js",
"dist/nodes/tools/ToolHttpRequest/ToolHttpRequest.node.js",
"dist/nodes/tools/ToolSerpApi/ToolSerpApi.node.js",
"dist/nodes/tools/ToolThink/ToolThink.node.js",
"dist/nodes/tools/ToolVectorStore/ToolVectorStore.node.js",
"dist/nodes/tools/ToolWikipedia/ToolWikipedia.node.js",
"dist/nodes/tools/ToolWolframAlpha/ToolWolframAlpha.node.js",