mirror of
https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git
synced 2025-12-20 03:12:15 +00:00
fix(Think Tool Node): Use dynamic tool name based on node name (#17364)
This commit is contained in:
@@ -1,27 +1,34 @@
|
||||
import { mock } from 'jest-mock-extended';
|
||||
import { DynamicTool } from 'langchain/tools';
|
||||
import type { ISupplyDataFunctions } from 'n8n-workflow';
|
||||
import type { ISupplyDataFunctions, INode } 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;
|
||||
}
|
||||
});
|
||||
|
||||
const createExecuteFunctions = (node: Partial<INode> = { typeVersion: 1 }) => {
|
||||
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;
|
||||
}
|
||||
});
|
||||
executeFunctions.getNode.mockReturnValue(mock<INode>(node));
|
||||
return executeFunctions;
|
||||
};
|
||||
|
||||
describe('Tool response', () => {
|
||||
it('should return the same text as response when receiving a text input', async () => {
|
||||
const executeFunctions = createExecuteFunctions();
|
||||
|
||||
const { response } = (await thinkTool.supplyData.call(executeFunctions, 0)) as {
|
||||
response: DynamicTool;
|
||||
};
|
||||
@@ -30,5 +37,26 @@ describe('ToolThink', () => {
|
||||
const res = (await response.invoke('foo')) as string;
|
||||
expect(res).toEqual('foo');
|
||||
});
|
||||
|
||||
it('should use hardcoded name for version 1', async () => {
|
||||
const executeFunctions = createExecuteFunctions({ typeVersion: 1 });
|
||||
|
||||
const { response } = (await thinkTool.supplyData.call(executeFunctions, 0)) as {
|
||||
response: DynamicTool;
|
||||
};
|
||||
expect(response.name).toEqual('thinking_tool');
|
||||
});
|
||||
|
||||
it('should use dynamic name from node for version 1.1', async () => {
|
||||
const executeFunctions = createExecuteFunctions({
|
||||
typeVersion: 1.1,
|
||||
name: 'My Thinking Tool',
|
||||
});
|
||||
|
||||
const { response } = (await thinkTool.supplyData.call(executeFunctions, 0)) as {
|
||||
response: DynamicTool;
|
||||
};
|
||||
expect(response.name).toEqual('My_Thinking_Tool');
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user