mirror of
https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git
synced 2025-12-17 01:56:46 +00:00
feat(Think Tool Node): ToolThink, a simple tool to force the AI agent to do some thinking (#14351)
This commit is contained in:
committed by
GitHub
parent
85cbfb64c0
commit
281b70be04
@@ -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),
|
||||
};
|
||||
}
|
||||
}
|
||||
@@ -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');
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user