@n8n/n8n-nodes-langchain
Version:
264 lines • 8.81 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.description = exports.properties = void 0;
exports.execute = execute;
const n8n_workflow_1 = require("n8n-workflow");
const transport_1 = require("../../transport");
const descriptions_1 = require("../descriptions");
exports.properties = [
{
displayName: 'Model',
name: 'modelId',
type: 'options',
options: [
{
name: 'Wan 2.6 Text-to-Video',
value: 'wan2.6-t2v',
description: 'Text-to-video generation model',
},
],
default: 'wan2.6-t2v',
description: 'The model to use for text-to-video generation',
displayOptions: {
show: { '@version': [1] },
},
},
{
...(0, descriptions_1.modelRLC)('textToVideoModelSearch'),
displayOptions: {
show: { '@version': [{ _cnd: { gte: 1.1 } }] },
},
},
{
displayName: 'Prompt',
name: 'prompt',
type: 'string',
typeOptions: {
rows: 4,
},
default: '',
required: true,
description: 'The text prompt to generate video from',
placeholder: 'A cat playing with a ball of yarn',
},
{
displayName: 'Resolution',
name: 'resolution',
type: 'options',
options: [
{
name: '720P',
value: '720P',
},
{
name: '1080P',
value: '1080P',
},
],
default: '1080P',
description: 'Resolution tier of the generated video',
},
{
displayName: 'Duration (Seconds)',
name: 'duration',
type: 'number',
typeOptions: {
minValue: 2,
maxValue: 15,
},
default: 5,
description: 'Duration of the generated video in seconds (2–15)',
},
{
displayName: 'Shot Type',
name: 'shotType',
type: 'options',
options: [
{
name: 'Single',
value: 'single',
},
{
name: 'Multi',
value: 'multi',
},
],
default: 'single',
description: 'Whether to generate a single-shot or multi-shot narrative video',
},
{
displayName: 'Download Video',
name: 'downloadVideo',
type: 'boolean',
default: true,
description: 'Whether to download the generated video as binary data. When disabled, only the video URL is returned.',
},
{
displayName: 'Simplify Output',
name: 'simplify',
type: 'boolean',
default: true,
description: 'Whether to return a simplified version of the response instead of the raw data',
},
{
displayName: 'Options',
name: 'videoOptions',
type: 'collection',
placeholder: 'Add Option',
default: {},
options: [
{
displayName: 'Prompt Extend',
name: 'promptExtend',
type: 'boolean',
default: false,
description: 'Whether to automatically extend and enhance the prompt',
},
{
displayName: 'Audio',
name: 'audio',
type: 'boolean',
default: true,
description: 'Whether to generate audio for the video',
},
{
displayName: 'Audio Input Type',
name: 'audioInputType',
type: 'options',
options: [
{
name: 'Audio URL',
value: 'url',
},
{
name: 'Binary File',
value: 'binary',
},
],
default: 'url',
},
{
displayName: 'Audio URL',
name: 'audioUrl',
type: 'string',
default: '',
placeholder: 'https://example.com/audio.mp3',
description: 'URL of the audio file to use for the video',
displayOptions: {
show: {
audioInputType: ['url'],
},
},
},
{
displayName: 'Audio Data Field Name',
name: 'audioBinaryPropertyName',
type: 'string',
default: 'audio',
placeholder: 'e.g. audio',
hint: 'The name of the input field containing the binary audio data',
typeOptions: {
binaryDataProperty: true,
},
displayOptions: {
show: {
audioInputType: ['binary'],
},
},
},
],
},
];
const displayOptions = {
show: {
resource: ['video'],
operation: ['textToVideo'],
},
};
exports.description = (0, n8n_workflow_1.updateDisplayOptions)(displayOptions, exports.properties);
async function execute(itemIndex) {
const nodeVersion = this.getNode().typeVersion;
const model = nodeVersion >= 1.1
? this.getNodeParameter('modelId', itemIndex, '', { extractValue: true })
: this.getNodeParameter('modelId', itemIndex);
const prompt = this.getNodeParameter('prompt', itemIndex);
const resolution = this.getNodeParameter('resolution', itemIndex);
const duration = this.getNodeParameter('duration', itemIndex);
const shotType = this.getNodeParameter('shotType', itemIndex);
const simplify = this.getNodeParameter('simplify', itemIndex, true);
const downloadVideo = this.getNodeParameter('downloadVideo', itemIndex, true);
const videoOptions = this.getNodeParameter('videoOptions', itemIndex, {});
const body = {
model,
input: {
prompt,
},
parameters: {
resolution,
duration,
shot_type: shotType,
},
};
if (videoOptions.promptExtend !== undefined) {
body.parameters.prompt_extend = videoOptions.promptExtend;
}
if (videoOptions.audio !== undefined) {
body.parameters.audio = videoOptions.audio;
}
const audioInputType = videoOptions.audioInputType;
if (audioInputType === 'binary') {
const binaryPropertyName = videoOptions.audioBinaryPropertyName || 'audio';
const binaryData = this.helpers.assertBinaryData(itemIndex, binaryPropertyName);
const buffer = await this.helpers.getBinaryDataBuffer(itemIndex, binaryPropertyName);
const mimeType = binaryData.mimeType || 'audio/mpeg';
body.input.audio_url = `data:${mimeType};base64,${buffer.toString('base64')}`;
}
else if (videoOptions.audioUrl) {
body.input.audio_url = videoOptions.audioUrl;
}
const createResponse = await transport_1.apiRequest.call(this, 'POST', '/api/v1/services/aigc/video-generation/video-synthesis', {
headers: {
'X-DashScope-Async': 'enable',
},
body,
});
const taskId = createResponse?.output?.task_id;
if (!taskId) {
throw new n8n_workflow_1.NodeOperationError(this.getNode(), `Failed to create video generation task: ${createResponse?.message || 'No task_id returned'}`);
}
const result = await transport_1.pollTaskResult.call(this, taskId);
const output = result.output;
const videoUrl = output?.video_url || '';
const jsonData = simplify
? { videoUrl }
: {
videoUrl,
model,
taskId,
usage: result.usage,
taskStatus: output?.task_status,
submitTime: output?.submit_time,
endTime: output?.end_time,
};
if (downloadVideo && videoUrl) {
const videoResponse = await this.helpers.httpRequest({
method: 'GET',
url: videoUrl,
encoding: 'arraybuffer',
returnFullResponse: true,
});
const contentType = videoResponse.headers?.['content-type'] || 'video/mp4';
const fileContent = Buffer.from(videoResponse.body);
const binaryData = await this.helpers.prepareBinaryData(fileContent, 'video.mp4', contentType);
return {
binary: { data: binaryData },
json: jsonData,
pairedItem: itemIndex,
};
}
return {
json: jsonData,
pairedItem: itemIndex,
};
}
//# sourceMappingURL=generate.t2v.operation.js.map