first commit

packages/@n8n/nodes-langchain/nodes/tools/ToolCalculator/ToolCalculator.node.test.ts

@@ -0,0 +1,118 @@
+import { Calculator } from '@langchain/community/tools/calculator';
+import { mock } from 'jest-mock-extended';
+import type {
+	IExecuteFunctions,
+	INode,
+	INodeExecutionData,
+	ISupplyDataFunctions,
+} from 'n8n-workflow';
+
+import { ToolCalculator } from './ToolCalculator.node';
+
+describe('ToolCalculator', () => {
+	describe('supplyData', () => {
+		beforeEach(() => {
+			jest.resetAllMocks();
+		});
+
+		it('should return Calculator tool instance', async () => {
+			const node = new ToolCalculator();
+
+			const supplyDataResult = await node.supplyData.call(
+				mock<ISupplyDataFunctions>({
+					getNode: jest.fn(() => mock<INode>({ name: 'test calculator' })),
+				}),
+			);
+
+			expect(supplyDataResult.response).toBeInstanceOf(Calculator);
+		});
+
+		it('should sanitize tool name to be LLM API compatible', async () => {
+			const node = new ToolCalculator();
+
+			const supplyDataResult = await node.supplyData.call(
+				mock<ISupplyDataFunctions>({
+					getNode: jest.fn(() => mock<INode>({ name: 'Calculator (1)' })),
+				}),
+			);
+
+			const tool = supplyDataResult.response as Calculator;
+			expect(tool.name).toBe('Calculator_1_');
+		});
+	});
+
+	describe('execute', () => {
+		beforeEach(() => {
+			jest.resetAllMocks();
+		});
+
+		it('should execute calculator and return result', async () => {
+			const node = new ToolCalculator();
+			const inputData: INodeExecutionData[] = [
+				{
+					json: { input: '2 + 2' },
+				},
+			];
+
+			const mockExecute = mock<IExecuteFunctions>({
+				getInputData: jest.fn(() => inputData),
+				getNode: jest.fn(() => mock<INode>({ name: 'test calculator' })),
+			});
+
+			const result = await node.execute.call(mockExecute);
+
+			expect(result).toEqual([
+				[
+					{
+						json: {
+							response: '4',
+						},
+						pairedItem: {
+							item: 0,
+						},
+					},
+				],
+			]);
+		});
+
+		it('should handle multiple input items', async () => {
+			const node = new ToolCalculator();
+			const inputData: INodeExecutionData[] = [
+				{
+					json: { input: '2 + 2' },
+				},
+				{
+					json: { input: '5 * 3' },
+				},
+			];
+
+			const mockExecute = mock<IExecuteFunctions>({
+				getInputData: jest.fn(() => inputData),
+				getNode: jest.fn(() => mock<INode>({ name: 'test calculator' })),
+			});
+
+			const result = await node.execute.call(mockExecute);
+
+			expect(result).toEqual([
+				[
+					{
+						json: {
+							response: '4',
+						},
+						pairedItem: {
+							item: 0,
+						},
+					},
+					{
+						json: {
+							response: '15',
+						},
+						pairedItem: {
+							item: 1,
+						},
+					},
+				],
+			]);
+		});
+	});
+});

packages/@n8n/nodes-langchain/nodes/tools/ToolCalculator/ToolCalculator.node.ts

@@ -0,0 +1,80 @@
+import { Calculator } from '@langchain/community/tools/calculator';
+import {
+	type IExecuteFunctions,
+	type INodeExecutionData,
+	NodeConnectionTypes,
+	type INodeType,
+	type INodeTypeDescription,
+	type ISupplyDataFunctions,
+	type SupplyData,
+	nodeNameToToolName,
+} from 'n8n-workflow';
+
+import { logWrapper, getConnectionHintNoticeField } from '@n8n/ai-utilities';
+
+function getTool(ctx: ISupplyDataFunctions | IExecuteFunctions): Calculator {
+	const calculator = new Calculator();
+	calculator.name = nodeNameToToolName(ctx.getNode());
+	return calculator;
+}
+
+export class ToolCalculator implements INodeType {
+	description: INodeTypeDescription = {
+		displayName: 'Calculator',
+		name: 'toolCalculator',
+		icon: 'fa:calculator',
+		iconColor: 'black',
+		group: ['transform'],
+		version: 1,
+		description: 'Make it easier for AI agents to perform arithmetic',
+		defaults: {
+			name: 'Calculator',
+		},
+		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.toolcalculator/',
+					},
+				],
+			},
+		},
+
+		inputs: [],
+
+		outputs: [NodeConnectionTypes.AiTool],
+		outputNames: ['Tool'],
+		properties: [getConnectionHintNoticeField([NodeConnectionTypes.AiAgent])],
+	};
+
+	async supplyData(this: ISupplyDataFunctions): Promise<SupplyData> {
+		return {
+			response: logWrapper(getTool(this), this),
+		};
+	}
+
+	async execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]> {
+		const calculator = getTool(this);
+		const input = this.getInputData();
+		const response: INodeExecutionData[] = [];
+		for (let i = 0; i < input.length; i++) {
+			const inputItem = input[i];
+			const result = await calculator.invoke(inputItem.json);
+			response.push({
+				json: {
+					response: result,
+				},
+				pairedItem: {
+					item: i,
+				},
+			});
+		}
+
+		return [response];
+	}
+}