@wiro-ai/n8n-nodes-wiroai
Version:
n8n community node for Wiro AI — 290+ AI models: video, image, audio, LLM, 3D, and more.
245 lines • 10.1 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.ControlnetSdLora = void 0;
const n8n_workflow_1 = require("n8n-workflow");
const auth_1 = require("../utils/auth");
const polling_1 = require("../utils/polling");
class ControlnetSdLora {
constructor() {
this.description = {
displayName: 'Wiro - Controlnet Sd Lora',
name: 'controlnetSdLora',
icon: { light: 'file:wiro.svg', dark: 'file:wiro.svg' },
group: ['transform'],
version: 1,
description: 'ControlNet is a neural network structure to control diffusion models by adding extra conditions',
defaults: {
name: 'Wiro - Controlnet Sd Lora',
},
inputs: [n8n_workflow_1.NodeConnectionTypes.Main],
outputs: [n8n_workflow_1.NodeConnectionTypes.Main],
usableAsTool: true,
credentials: [
{
name: 'wiroApi',
required: true,
},
],
properties: [
{
displayName: 'Select Model',
name: 'selectedModel',
type: 'options',
default: '',
description: 'Select-model-help',
options: [
{ name: '', value: '' },
],
},
{
displayName: 'Select Model Private',
name: 'selectedModelPrivate',
type: 'options',
default: '',
description: 'Select-model-private-help',
options: [],
},
{
displayName: 'Select Controlnet Model',
name: 'selectedControlnetModel',
type: 'multiOptions',
default: [],
description: 'Select-controlnet-model-help',
options: [
{ name: '', value: '' },
],
},
{
displayName: 'Lo RA Model Url',
name: 'selectedLoraModelPrivateUrl',
type: 'string',
default: '',
description: 'You can provide a direct URL to a LoRA model (.safetensors). We’ll apply it during generation. You can also use links from platforms like Google Drive, AWS S3, or any publicly accessible file host',
},
{
displayName: 'Controlnet Preprocess',
name: 'preprocess',
type: 'boolean',
default: false,
description: 'Whether to enable controlnet-preprocess-help',
},
{
displayName: 'Select Lora Model',
name: 'selectedLoraModel',
type: 'options',
default: '',
description: 'Select-lora-model-help',
options: [
{ name: '', value: '' },
],
},
{
displayName: 'Select Lora Model Private',
name: 'selectedLoraModelPrivate',
type: 'options',
default: '',
description: 'Select-lora-model-private-help',
options: [],
},
{
displayName: 'Prompt',
name: 'prompt',
type: 'string',
default: '',
required: true,
description: 'Prompt-help',
},
{
displayName: 'Negativeprompt',
name: 'negativePrompt',
type: 'string',
default: '',
description: 'Negativeprompt-help',
},
{
displayName: 'Numberofoutputs',
name: 'samples',
type: 'number',
default: 0,
description: 'Numberofoutputs-help',
},
{
displayName: 'Inferencesteps',
name: 'steps',
type: 'number',
default: 0,
description: 'Inferencesteps-help',
},
{
displayName: 'Guidancescale',
name: 'scale',
type: 'number',
default: 0,
description: 'Guidancescale-help',
},
{
displayName: 'Seed',
name: 'seed',
type: 'string',
default: '',
description: 'Seed-help',
},
{
displayName: 'Scheduler',
name: 'scheduler',
type: 'options',
default: 'DDIMScheduler',
description: 'Scheduler-help',
options: [
{ name: 'DDIMScheduler', value: 'DDIMScheduler' },
{ name: 'DDPMScheduler', value: 'DDPMScheduler' },
{ name: 'DPMSolverMultistepScheduler', value: 'DPMSolverMultistepScheduler' },
{ name: 'EulerAncestralDiscreteScheduler', value: 'EulerAncestralDiscreteScheduler' },
{ name: 'EulerDiscreteScheduler', value: 'EulerDiscreteScheduler' },
{ name: 'LMSDiscreteScheduler', value: 'LMSDiscreteScheduler' },
{ name: 'PNDMScheduler', value: 'PNDMScheduler' },
],
},
{
displayName: 'Width',
name: 'width',
type: 'number',
default: 0,
description: 'Width-help',
},
{
displayName: 'Height',
name: 'height',
type: 'number',
default: 0,
description: 'Height-help',
},
],
};
}
async execute() {
const returnData = [];
const selectedModel = this.getNodeParameter('selectedModel', 0, '');
const selectedModelPrivate = this.getNodeParameter('selectedModelPrivate', 0, '');
const selectedControlnetModel = this.getNodeParameter('selectedControlnetModel', 0, '');
const selectedLoraModelPrivateUrl = this.getNodeParameter('selectedLoraModelPrivateUrl', 0, '');
const preprocess = this.getNodeParameter('preprocess', 0, false);
const selectedLoraModel = this.getNodeParameter('selectedLoraModel', 0, '');
const selectedLoraModelPrivate = this.getNodeParameter('selectedLoraModelPrivate', 0, '');
const prompt = this.getNodeParameter('prompt', 0);
const negativePrompt = this.getNodeParameter('negativePrompt', 0, '');
const samples = this.getNodeParameter('samples', 0, 0);
const steps = this.getNodeParameter('steps', 0, 0);
const scale = this.getNodeParameter('scale', 0, 0);
const seed = this.getNodeParameter('seed', 0, '');
const scheduler = this.getNodeParameter('scheduler', 0, '');
const width = this.getNodeParameter('width', 0, 0);
const height = this.getNodeParameter('height', 0, 0);
const credentials = await this.getCredentials('wiroApi');
const apiKey = credentials.apiKey;
const apiSecret = credentials.apiSecret;
const headers = (0, auth_1.generateWiroAuthHeaders)(apiKey, apiSecret);
const response = await this.helpers.httpRequest({
method: 'POST',
url: 'https://api.wiro.ai/v1/Run/wiro/controlnet-sd-lora',
headers: {
...headers,
'Content-Type': 'application/json',
},
body: {
selectedModel,
selectedModelPrivate,
selectedControlnetModel,
selectedLoraModelPrivateUrl,
preprocess,
selectedLoraModel,
selectedLoraModelPrivate,
prompt,
negativePrompt,
samples,
steps,
scale,
seed,
scheduler,
width,
height,
},
});
if (!(response === null || response === void 0 ? void 0 : response.taskid) || !(response === null || response === void 0 ? void 0 : response.socketaccesstoken)) {
throw new n8n_workflow_1.NodeApiError(this.getNode(), {
message: 'Wiro API did not return a valid task ID or socket access token ' +
JSON.stringify(response),
});
}
const taskid = response.taskid;
const socketaccesstoken = response.socketaccesstoken;
const result = await polling_1.pollTaskUntilComplete.call(this, socketaccesstoken, headers);
const responseJSON = {
taskid: taskid,
url: '',
status: '',
};
switch (result) {
case '-1':
case '-2':
case '-3':
case '-4':
responseJSON.status = 'failed';
break;
default:
responseJSON.status = 'completed';
responseJSON.url = result !== null && result !== void 0 ? result : '';
}
returnData.push({
json: responseJSON,
});
return [returnData];
}
}
exports.ControlnetSdLora = ControlnetSdLora;
//# sourceMappingURL=ControlnetSdLora.node.js.map