mirror of
https://github.com/n8n-io/n8n.git
synced 2026-05-30 08:17:06 +02:00
113 lines
3.2 KiB
TypeScript
113 lines
3.2 KiB
TypeScript
import {
|
|
NodeOperationError,
|
|
NodeConnectionTypes,
|
|
type IExecuteFunctions,
|
|
type INodeType,
|
|
type INodeTypeDescription,
|
|
type INode,
|
|
AI_TRANSFORM_CODE_GENERATED_FOR_PROMPT,
|
|
AI_TRANSFORM_JS_CODE,
|
|
ensureError,
|
|
} from 'n8n-workflow';
|
|
|
|
import { JsTaskRunnerSandbox } from '../Code/JsTaskRunnerSandbox';
|
|
|
|
export class AiTransform implements INodeType {
|
|
description: INodeTypeDescription = {
|
|
displayName: 'AI Transform',
|
|
name: 'aiTransform',
|
|
icon: 'node:ai-transform',
|
|
iconColor: 'violet',
|
|
group: ['transform'],
|
|
version: 1,
|
|
description: 'Modify data based on instructions written in plain english',
|
|
defaults: {
|
|
name: 'AI Transform',
|
|
},
|
|
inputs: [NodeConnectionTypes.Main],
|
|
outputs: [NodeConnectionTypes.Main],
|
|
parameterPane: 'wide',
|
|
hints: [
|
|
{
|
|
message:
|
|
"This node doesn't have access to the contents of binary files. To use those contents here, use the 'Extract from File' node first.",
|
|
displayCondition: '={{ $input.all().some(i => i.binary) }}',
|
|
location: 'outputPane',
|
|
},
|
|
],
|
|
properties: [
|
|
{
|
|
displayName: 'Instructions',
|
|
name: 'instructions',
|
|
type: 'button',
|
|
default: '',
|
|
description:
|
|
"Provide instructions on how you want to transform the data, then click 'Generate code'. Use dot notation to refer to nested fields (e.g. address.street).",
|
|
placeholder:
|
|
"Example: Merge 'firstname' and 'lastname' into a field 'details.name' and sort by 'email'",
|
|
typeOptions: {
|
|
buttonConfig: {
|
|
label: 'Generate code',
|
|
hasInputField: true,
|
|
inputFieldMaxLength: 500,
|
|
action: {
|
|
type: 'askAiCodeGeneration',
|
|
target: AI_TRANSFORM_JS_CODE,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
displayName: 'Code Generated For Prompt',
|
|
name: AI_TRANSFORM_CODE_GENERATED_FOR_PROMPT,
|
|
type: 'hidden',
|
|
default: '',
|
|
},
|
|
{
|
|
displayName: 'Generated JavaScript',
|
|
name: AI_TRANSFORM_JS_CODE,
|
|
type: 'string',
|
|
typeOptions: {
|
|
editor: 'jsEditor',
|
|
editorIsReadOnly: true,
|
|
},
|
|
default: '',
|
|
hint: 'Read-only. To edit this code, adjust the instructions or copy and paste it into a Code node.',
|
|
noDataExpression: true,
|
|
},
|
|
],
|
|
};
|
|
|
|
async execute(this: IExecuteFunctions) {
|
|
const workflowMode = this.getMode();
|
|
const node = this.getNode();
|
|
const code = getValidatedCode(this, node);
|
|
|
|
const sandbox = new JsTaskRunnerSandbox(workflowMode, this);
|
|
return [await sandbox.runCodeAllItems(code)];
|
|
}
|
|
}
|
|
|
|
function getValidatedCode(executeFunctions: IExecuteFunctions, node: INode): string {
|
|
try {
|
|
const code = executeFunctions.getNodeParameter(AI_TRANSFORM_JS_CODE, 0) as string;
|
|
if (code) {
|
|
return code;
|
|
}
|
|
|
|
const instructions = executeFunctions.getNodeParameter('instructions', 0) as string;
|
|
if (!instructions) {
|
|
throw new NodeOperationError(node, 'Missing instructions to generate code', {
|
|
description: "Enter your prompt in the 'Instructions' parameter and click 'Generate code'",
|
|
});
|
|
}
|
|
throw new NodeOperationError(node, 'Missing code for data transformation', {
|
|
description: "Click the 'Generate code' button to create the code",
|
|
});
|
|
} catch (error: unknown) {
|
|
if (error instanceof NodeOperationError) throw error;
|
|
|
|
throw new NodeOperationError(node, ensureError(error));
|
|
}
|
|
}
|