first commit

packages/nodes-base/nodes/OpenAi/ImageDescription.ts

@@ -0,0 +1,399 @@
+import type { INodeExecutionData, INodeProperties } from 'n8n-workflow';
+
+import { sendErrorPostReceive } from './GenericFunctions';
+
+export const imageOperations: INodeProperties[] = [
+	{
+		displayName: 'Operation',
+		name: 'operation',
+		type: 'options',
+		noDataExpression: true,
+		displayOptions: {
+			show: {
+				resource: ['image'],
+			},
+		},
+		options: [
+			{
+				name: 'Create',
+				value: 'create',
+				action: 'Create an Image',
+				description: 'Create an image for a given text',
+				routing: {
+					request: {
+						method: 'POST',
+						url: '/v1/images/generations',
+					},
+					output: { postReceive: [sendErrorPostReceive] },
+				},
+			},
+		],
+		routing: {
+			output: {
+				postReceive: [
+					{
+						type: 'rootProperty',
+						properties: {
+							property: 'data',
+						},
+					},
+				],
+			},
+		},
+		default: 'create',
+	},
+];
+
+const createOperations: INodeProperties[] = [
+	{
+		displayName: 'Prompt',
+		name: 'prompt',
+		type: 'string',
+		placeholder: 'e.g. A cute cat eating a dinosaur',
+		description:
+			'A text description of the desired image(s). The maximum length is 1000 characters.',
+		displayOptions: {
+			show: {
+				resource: ['image'],
+				operation: ['create'],
+			},
+		},
+		default: '',
+		routing: {
+			send: {
+				type: 'body',
+				property: 'prompt',
+			},
+		},
+	},
+	{
+		displayName: 'Model',
+		name: 'model',
+		type: 'options',
+		default: 'dall-e-2',
+		description: 'The model to use for image generation',
+		typeOptions: {
+			loadOptions: {
+				routing: {
+					request: {
+						method: 'GET',
+						url: '/v1/models',
+					},
+					output: {
+						postReceive: [
+							{
+								type: 'rootProperty',
+								properties: {
+									property: 'data',
+								},
+							},
+							{
+								type: 'filter',
+								properties: {
+									pass: "={{ $responseItem.id.startsWith('dall-') }}",
+								},
+							},
+							{
+								type: 'setKeyValue',
+								properties: {
+									name: '={{$responseItem.id}}',
+									value: '={{$responseItem.id}}',
+								},
+							},
+							{
+								type: 'sort',
+								properties: {
+									key: 'name',
+								},
+							},
+						],
+					},
+				},
+			},
+		},
+		displayOptions: {
+			show: {
+				resource: ['image'],
+				operation: ['create'],
+				'@version': [1],
+			},
+		},
+		routing: {
+			send: {
+				type: 'body',
+				property: 'model',
+			},
+		},
+	},
+	{
+		displayName: 'Model',
+		name: 'imageModel',
+		type: 'options',
+		default: 'dall-e-2',
+		description: 'The model to use for image generation',
+		typeOptions: {
+			loadOptions: {
+				routing: {
+					request: {
+						method: 'GET',
+						url: '/v1/models',
+					},
+					output: {
+						postReceive: [
+							{
+								type: 'rootProperty',
+								properties: {
+									property: 'data',
+								},
+							},
+							{
+								type: 'filter',
+								properties: {
+									pass: "={{ $responseItem.id.startsWith('dall-') }}",
+								},
+							},
+							{
+								type: 'setKeyValue',
+								properties: {
+									name: '={{$responseItem.id}}',
+									value: '={{$responseItem.id}}',
+								},
+							},
+							{
+								type: 'sort',
+								properties: {
+									key: 'name',
+								},
+							},
+						],
+					},
+				},
+			},
+		},
+		displayOptions: {
+			show: {
+				resource: ['image'],
+				operation: ['create'],
+			},
+			hide: {
+				'@version': [1],
+			},
+		},
+		routing: {
+			send: {
+				type: 'body',
+				property: 'model',
+			},
+		},
+	},
+	{
+		displayName: 'Response Format',
+		name: 'responseFormat',
+		type: 'options',
+		default: 'binaryData',
+		description: 'The format in which to return the image(s)',
+		displayOptions: {
+			show: {
+				resource: ['image'],
+				operation: ['create'],
+			},
+		},
+		options: [
+			{
+				name: 'Binary File',
+				value: 'binaryData',
+			},
+			{
+				name: 'Image Url',
+				value: 'imageUrl',
+			},
+		],
+		routing: {
+			send: {
+				type: 'body',
+				property: 'response_format',
+				value: '={{ $value === "imageUrl" ? "url" : "b64_json" }}',
+			},
+			output: {
+				postReceive: [
+					async function (items: INodeExecutionData[]): Promise<INodeExecutionData[]> {
+						if (this.getNode().parameters.responseFormat === 'imageUrl') {
+							return items;
+						}
+
+						const result: INodeExecutionData[] = [];
+						for (let i = 0; i < items.length; i++) {
+							result.push({
+								json: {},
+								binary: {
+									data: await this.helpers.prepareBinaryData(
+										Buffer.from(items[i].json.b64_json as string, 'base64'),
+										'data',
+									),
+								},
+							} as INodeExecutionData);
+						}
+
+						return result;
+					},
+				],
+			},
+		},
+	},
+	{
+		displayName: 'Options',
+		name: 'options',
+		placeholder: 'Add option',
+		description: 'Additional options to add',
+		type: 'collection',
+		default: {},
+		displayOptions: {
+			show: {
+				resource: ['image'],
+				operation: ['create'],
+			},
+		},
+		options: [
+			{
+				displayName: 'Number of Images',
+				name: 'n',
+				default: 1,
+				description: 'Number of images to generate',
+				type: 'number',
+				typeOptions: {
+					minValue: 1,
+					maxValue: 10,
+				},
+				routing: {
+					send: {
+						type: 'body',
+						property: 'n',
+					},
+				},
+			},
+			{
+				displayName: 'Quality',
+				name: 'quality',
+				type: 'options',
+				options: [
+					{
+						name: 'HD',
+						value: 'hd',
+					},
+					{
+						name: 'Standard',
+						value: 'standard',
+					},
+				],
+				displayOptions: {
+					show: {
+						'/model': ['dall-e-3'],
+					},
+				},
+				routing: {
+					send: {
+						type: 'body',
+						property: 'quality',
+					},
+				},
+				default: 'standard',
+			},
+			{
+				displayName: 'Resolution',
+				name: 'size',
+				type: 'options',
+				options: [
+					{
+						name: '256x256',
+						value: '256x256',
+					},
+					{
+						name: '512x512',
+						value: '512x512',
+					},
+					{
+						name: '1024x1024',
+						value: '1024x1024',
+					},
+				],
+				displayOptions: {
+					show: {
+						'/model': ['dall-e-2'],
+					},
+				},
+				routing: {
+					send: {
+						type: 'body',
+						property: 'size',
+					},
+				},
+				default: '1024x1024',
+			},
+			{
+				displayName: 'Resolution',
+				name: 'size',
+				type: 'options',
+				options: [
+					{
+						name: '1024x1024',
+						value: '1024x1024',
+					},
+					{
+						name: '1792x1024',
+						value: '1792x1024',
+					},
+					{
+						name: '1024x1792',
+						value: '1024x1792',
+					},
+				],
+				displayOptions: {
+					show: {
+						'/model': ['dall-e-3'],
+					},
+				},
+				routing: {
+					send: {
+						type: 'body',
+						property: 'size',
+					},
+				},
+				default: '1024x1024',
+			},
+			{
+				displayName: 'Style',
+				name: 'style',
+				type: 'options',
+				options: [
+					{
+						name: 'Natural',
+						value: 'natural',
+					},
+					{
+						name: 'Vivid',
+						value: 'vivid',
+					},
+				],
+				displayOptions: {
+					show: {
+						'/model': ['dall-e-3'],
+					},
+				},
+				routing: {
+					send: {
+						type: 'body',
+						property: 'style',
+					},
+				},
+				default: 'vivid',
+			},
+		],
+	},
+];
+
+export const imageFields: INodeProperties[] = [
+	/* -------------------------------------------------------------------------- */
+	/*                                image:create                                */
+	/* -------------------------------------------------------------------------- */
+	...createOperations,
+];