@zerogpu/n8n-nodes-zerogpu 0.2.0 → 0.4.0

package/dist/nodes/ZeroGpu/shared/routing.js

@@ -0,0 +1,177 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.moderationsRouting = exports.embeddingsRouting = exports.chatCompletionsRouting = void 0;
+exports.fail = fail;
+exports.getRequiredString = getRequiredString;
+exports.buildRequestBody = buildRequestBody;
+exports.parseResult = parseResult;
+const n8n_workflow_1 = require("n8n-workflow");
+const SUMMARY_PROMPT = "Summarize the user's text faithfully and concisely. Reply with the summary only.";
+function fail(context, message, description) {
+    throw new n8n_workflow_1.NodeOperationError(context.getNode(), message, {
+        itemIndex: context.getItemIndex(),
+        description,
+    });
+}
+function getRequiredString(context, name, label) {
+    var _a;
+    const value = String((_a = context.getNodeParameter(name)) !== null && _a !== void 0 ? _a : '');
+    if (value.trim() === '')
+        fail(context, `Enter the ${label}`);
+    return value;
+}
+function getLabels(context) {
+    var _a;
+    const labels = String((_a = context.getNodeParameter('labels')) !== null && _a !== void 0 ? _a : '')
+        .split(',')
+        .map((label) => label.trim())
+        .filter((label) => label !== '');
+    if (labels.length === 0)
+        fail(context, 'Enter at least one label, separated by commas');
+    return labels;
+}
+function getSchema(context, example) {
+    const shape = `Schema must be a JSON object that maps each name to a list of strings, like ${example}`;
+    const value = context.getNodeParameter('schema');
+    let schema = value;
+    if (typeof value === 'string') {
+        try {
+            schema = (0, n8n_workflow_1.jsonParse)(value);
+        }
+        catch {
+            fail(context, 'Schema is not valid JSON', shape);
+        }
+    }
+    const isValid = typeof schema === 'object' &&
+        schema !== null &&
+        !Array.isArray(schema) &&
+        Object.keys(schema).length > 0 &&
+        Object.values(schema).every((items) => Array.isArray(items) &&
+            items.length > 0 &&
+            items.every((item) => typeof item === 'string' && item.trim() !== ''));
+    if (!isValid)
+        fail(context, shape);
+    return schema;
+}
+function getThreshold(context) {
+    const { threshold } = context.getNodeParameter('options', {});
+    if (threshold === undefined)
+        return {};
+    if (typeof threshold !== 'number' || !(threshold >= 0 && threshold <= 1)) {
+        fail(context, 'Threshold must be a number from 0 to 1');
+    }
+    return { threshold };
+}
+function chatBody(model, messages, metadata) {
+    return metadata ? { model, messages, metadata } : { model, messages };
+}
+function buildBody(context, operation) {
+    const model = getRequiredString(context, 'model', 'Model');
+    const user = (content) => ({ role: 'user', content });
+    const text = () => user(getRequiredString(context, 'text', 'Text'));
+    switch (operation) {
+        case 'classifyByLabels':
+            return chatBody(model, [
+                { role: 'system', content: `[${getLabels(context).join(',')}]` },
+                text(),
+            ]);
+        case 'classifyBySchema':
+            return chatBody(model, [text()], {
+                usecase: 'classification',
+                schema: getSchema(context, '{"sentiment": ["positive", "negative"]}'),
+                ...getThreshold(context),
+            });
+        case 'classifyDomain':
+            return chatBody(model, [user(getRequiredString(context, 'domain', 'Domain'))]);
+        case 'classifyIab':
+            return chatBody(model, [text()]);
+        case 'detectPii':
+        case 'extractEntities':
+        case 'redactPii':
+            return chatBody(model, [text()], {
+                usecase: 'ner',
+                labels: getLabels(context),
+                ...getThreshold(context),
+            });
+        case 'extractJson':
+            return chatBody(model, [text()], {
+                usecase: 'json',
+                schema: getSchema(context, '{"contact": ["name::str::Full name"]}'),
+            });
+        case 'extractText': {
+            const url = getRequiredString(context, 'documentUrl', 'Document URL').trim();
+            const filename = url.split(/[?#]/)[0].split('/').pop() || 'document';
+            return chatBody(model, [
+                { role: 'user', content: [{ type: 'file', file: { filename, file_data: url } }] },
+            ]);
+        }
+        case 'embedText':
+        case 'moderate':
+            return { model, input: getRequiredString(context, 'text', 'Text') };
+        case 'summarize':
+            return chatBody(model, [{ role: 'system', content: SUMMARY_PROMPT }, text()]);
+    }
+}
+async function buildRequestBody(requestOptions) {
+    requestOptions.body = buildBody(this, this.getNodeParameter('operation'));
+    return requestOptions;
+}
+function isObject(value) {
+    return typeof value === 'object' && value !== null && !Array.isArray(value);
+}
+function redact(text, entities) {
+    return Object.entries(entities)
+        .flatMap(([label, values]) => values
+        .filter((value) => typeof value === 'string' && value !== '')
+        .map((value) => ({
+        value,
+        tag: `[${label.trim().toUpperCase().replace(/\s+/g, '_')}]`,
+    })))
+        .sort((a, b) => b.value.length - a.value.length)
+        .reduce((result, { value, tag }) => result.split(value).join(tag), text);
+}
+async function parseResult(_items, response) {
+    var _a, _b, _c;
+    const operation = this.getNodeParameter('operation');
+    const body = response.body;
+    const content = (_c = (_b = (_a = body === null || body === void 0 ? void 0 : body.choices) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.message) === null || _c === void 0 ? void 0 : _c.content;
+    if (typeof content !== 'string') {
+        fail(this, 'ZeroGPU returned no result', JSON.stringify(response.body));
+    }
+    if (operation === 'summarize')
+        return [{ json: { summary: content } }];
+    if (operation === 'extractText')
+        return [{ json: { markdown: content } }];
+    let result;
+    try {
+        result = (0, n8n_workflow_1.jsonParse)(content);
+    }
+    catch {
+        fail(this, "ZeroGPU returned a result that isn't JSON", content);
+    }
+    if (!isObject(result))
+        fail(this, "ZeroGPU returned a result that isn't a JSON object", content);
+    if (operation === 'redactPii') {
+        const { entities } = result;
+        if (!isObject(entities) || !Object.values(entities).every(Array.isArray)) {
+            fail(this, 'ZeroGPU returned entities in an unexpected shape', content);
+        }
+        const text = this.getNodeParameter('text');
+        return [{ json: { text: redact(text, entities), entities } }];
+    }
+    return [{ json: result }];
+}
+exports.chatCompletionsRouting = {
+    request: { method: 'POST', url: '/chat/completions' },
+    send: { preSend: [buildRequestBody] },
+    output: { postReceive: [parseResult] },
+};
+exports.embeddingsRouting = {
+    request: { method: 'POST', url: '/embeddings' },
+    send: { preSend: [buildRequestBody] },
+};
+exports.moderationsRouting = {
+    request: { method: 'POST', url: '/moderations' },
+    send: { preSend: [buildRequestBody] },
+};
+//# sourceMappingURL=routing.js.map

package/dist/nodes/ZeroGpu/shared/routing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"routing.js","sourceRoot":"","sources":["../../../../nodes/ZeroGpu/shared/routing.ts"],"names":[],"mappings":";;;AAgBA,oBASC;AAED,8CAQC;AA0GD,4CASC;AAqBD,kCAiCC;AApMD,+CAA6D;AAG7D,MAAM,cAAc,GACnB,kFAAkF,CAAC;AAIpF,SAAgB,IAAI,CACnB,OAAgC,EAChC,OAAe,EACf,WAAoB;IAEpB,MAAM,IAAI,iCAAkB,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE;QACxD,SAAS,EAAE,OAAO,CAAC,YAAY,EAAE;QACjC,WAAW;KACX,CAAC,CAAC;AACJ,CAAC;AAED,SAAgB,iBAAiB,CAChC,OAAgC,EAChC,IAAY,EACZ,KAAa;;IAEb,MAAM,KAAK,GAAG,MAAM,CAAC,MAAA,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,mCAAI,EAAE,CAAC,CAAC;IAC3D,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE;QAAE,IAAI,CAAC,OAAO,EAAE,aAAa,KAAK,EAAE,CAAC,CAAC;IAC7D,OAAO,KAAK,CAAC;AACd,CAAC;AAED,SAAS,SAAS,CAAC,OAAgC;;IAClD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAA,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,mCAAI,EAAE,CAAC;SAC7D,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;SAC5B,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,EAAE,CAAC,CAAC;IAClC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,IAAI,CAAC,OAAO,EAAE,+CAA+C,CAAC,CAAC;IACxF,OAAO,MAAM,CAAC;AACf,CAAC;AAED,SAAS,SAAS,CAAC,OAAgC,EAAE,OAAe;IACnE,MAAM,KAAK,GAAG,+EAA+E,OAAO,EAAE,CAAC;IACvG,MAAM,KAAK,GAAG,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAEjD,IAAI,MAAM,GAAY,KAAK,CAAC;IAC5B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC;YACJ,MAAM,GAAG,IAAA,wBAAS,EAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC;YACR,IAAI,CAAC,OAAO,EAAE,0BAA0B,EAAE,KAAK,CAAC,CAAC;QAClD,CAAC;IACF,CAAC;IAED,MAAM,OAAO,GACZ,OAAO,MAAM,KAAK,QAAQ;QAC1B,MAAM,KAAK,IAAI;QACf,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QACtB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAC1B,CAAC,KAAK,EAAE,EAAE,CACT,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YACpB,KAAK,CAAC,MAAM,GAAG,CAAC;YAChB,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CACtE,CAAC;IACH,IAAI,CAAC,OAAO;QAAE,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAEnC,OAAO,MAAqB,CAAC;AAC9B,CAAC;AAGD,SAAS,YAAY,CAAC,OAAgC;IACrD,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC,SAAS,EAAE,EAAE,CAA2B,CAAC;IACxF,IAAI,SAAS,KAAK,SAAS;QAAE,OAAO,EAAE,CAAC;IACvC,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,IAAI,SAAS,IAAI,CAAC,CAAC,EAAE,CAAC;QAC1E,IAAI,CAAC,OAAO,EAAE,wCAAwC,CAAC,CAAC;IACzD,CAAC;IACD,OAAO,EAAE,SAAS,EAAE,CAAC;AACtB,CAAC;AAED,SAAS,QAAQ,CAAC,KAAa,EAAE,QAAmB,EAAE,QAAsB;IAC3E,OAAO,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;AACvE,CAAC;AAED,SAAS,SAAS,CACjB,OAAgC,EAChC,SAA6C;IAE7C,MAAM,KAAK,GAAG,iBAAiB,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAC3D,MAAM,IAAI,GAAG,CAAC,OAAe,EAAW,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IACvE,MAAM,IAAI,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IAEpE,QAAQ,SAAS,EAAE,CAAC;QACnB,KAAK,kBAAkB;YACtB,OAAO,QAAQ,CAAC,KAAK,EAAE;gBACtB,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE;gBAChE,IAAI,EAAE;aACN,CAAC,CAAC;QACJ,KAAK,kBAAkB;YACtB,OAAO,QAAQ,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE;gBAChC,OAAO,EAAE,gBAAgB;gBACzB,MAAM,EAAE,SAAS,CAAC,OAAO,EAAE,yCAAyC,CAAC;gBACrE,GAAG,YAAY,CAAC,OAAO,CAAC;aACxB,CAAC,CAAC;QACJ,KAAK,gBAAgB;YACpB,OAAO,QAAQ,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QAChF,KAAK,aAAa;YACjB,OAAO,QAAQ,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAClC,KAAK,WAAW,CAAC;QACjB,KAAK,iBAAiB,CAAC;QACvB,KAAK,WAAW;YACf,OAAO,QAAQ,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE;gBAChC,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,SAAS,CAAC,OAAO,CAAC;gBAC1B,GAAG,YAAY,CAAC,OAAO,CAAC;aACxB,CAAC,CAAC;QACJ,KAAK,aAAa;YACjB,OAAO,QAAQ,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE;gBAChC,OAAO,EAAE,MAAM;gBACf,MAAM,EAAE,SAAS,CAAC,OAAO,EAAE,uCAAuC,CAAC;aACnE,CAAC,CAAC;QACJ,KAAK,aAAa,CAAC,CAAC,CAAC;YACpB,MAAM,GAAG,GAAG,iBAAiB,CAAC,OAAO,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC,IAAI,EAAE,CAAC;YAC7E,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,UAAU,CAAC;YACrE,OAAO,QAAQ,CAAC,KAAK,EAAE;gBACtB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,EAAE;aACjF,CAAC,CAAC;QACJ,CAAC;QACD,KAAK,WAAW,CAAC;QACjB,KAAK,UAAU;YACd,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,iBAAiB,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC;QACrE,KAAK,WAAW;YACf,OAAO,QAAQ,CAAC,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,cAAc,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAChF,CAAC;AACF,CAAC;AAEM,KAAK,UAAU,gBAAgB,CAErC,cAAmC;IAEnC,cAAc,CAAC,IAAI,GAAG,SAAS,CAC9B,IAAI,EACJ,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAuC,CACxE,CAAC;IACF,OAAO,cAAc,CAAC;AACvB,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC/B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC7E,CAAC;AAGD,SAAS,MAAM,CAAC,IAAY,EAAE,QAAqB;IAClD,OAAO,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;SAC7B,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,EAAE,CAC3B,MAAoB;SACnB,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,EAAE,CAAC;SAC7E,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAChB,KAAK;QACL,GAAG,EAAE,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG;KAC3D,CAAC,CAAC,CACJ;SACA,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;SAC/C,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;AAC3E,CAAC;AAEM,KAAK,UAAU,WAAW,CAEhC,MAA4B,EAC5B,QAA8B;;IAE9B,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAc,CAAC;IAClE,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAgE,CAAC;IACvF,MAAM,OAAO,GAAG,MAAA,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,OAAO,0CAAG,CAAC,CAAC,0CAAE,OAAO,0CAAE,OAAO,CAAC;IACrD,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QACjC,IAAI,CAAC,IAAI,EAAE,4BAA4B,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;IACzE,CAAC;IAED,IAAI,SAAS,KAAK,WAAW;QAAE,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IACvE,IAAI,SAAS,KAAK,aAAa;QAAE,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IAE1E,IAAI,MAAe,CAAC;IACpB,IAAI,CAAC;QACJ,MAAM,GAAG,IAAA,wBAAS,EAAC,OAAO,CAAC,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACR,IAAI,CAAC,IAAI,EAAE,2CAA2C,EAAE,OAAO,CAAC,CAAC;IAClE,CAAC;IACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,IAAI,CAAC,IAAI,EAAE,oDAAoD,EAAE,OAAO,CAAC,CAAC;IAEjG,IAAI,SAAS,KAAK,WAAW,EAAE,CAAC;QAC/B,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;QAC5B,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;YAC1E,IAAI,CAAC,IAAI,EAAE,kDAAkD,EAAE,OAAO,CAAC,CAAC;QACzE,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAW,CAAC;QACrD,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;AAC3B,CAAC;AAEY,QAAA,sBAAsB,GAAyB;IAC3D,OAAO,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,mBAAmB,EAAE;IACrD,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,gBAAgB,CAAC,EAAE;IACrC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC,WAAW,CAAC,EAAE;CACtC,CAAC;AAEW,QAAA,iBAAiB,GAAyB;IACtD,OAAO,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,aAAa,EAAE;IAC/C,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,gBAAgB,CAAC,EAAE;CACrC,CAAC;AAEW,QAAA,kBAAkB,GAAyB;IACvD,OAAO,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,cAAc,EAAE;IAChD,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,gBAAgB,CAAC,EAAE;CACrC,CAAC"}

package/dist/nodes/ZeroGpuChatModel/ZeroGpuChatModel.node.js

@@ -3,22 +3,21 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ZeroGpuChatModel = void 0;
 const n8n_workflow_1 = require("n8n-workflow");
 const ai_node_sdk_1 = require("@n8n/ai-node-sdk");
+const models_1 = require("../shared/models");
 const BASE_URL = 'https://api.zerogpu.ai/v1';
-const MODELS_URL = 'https://api-dashboard.zerogpu.ai/api/models';
-async function fetchModels(context) {
-    var _a;
-    const response = (await context.helpers.httpRequest({
-        method: 'GET',
-        url: MODELS_URL,
-        json: true,
-    }));
-    return (_a = response.models) !== null && _a !== void 0 ? _a : [];
+const CHAT_TASKS = ['Text Generation', 'Summarization'];
+function isChatModel({ chat, taskDisplayName, taskType, modelId }) {
+    if (chat !== undefined)
+        return chat !== null;
+    return (CHAT_TASKS.includes(taskDisplayName !== null && taskDisplayName !== void 0 ? taskDisplayName : '') &&
+        taskType !== 'moderation' &&
+        modelId !== models_1.OCR_MODEL_ID);
 }
 async function assertSettingsSupported(context, model, options, itemIndex) {
     var _a;
     let models;
     try {
-        models = await fetchModels(context);
+        models = await (0, models_1.fetchModels)(context);
     }
     catch {
         return;
@@ -128,9 +127,9 @@ class ZeroGpuChatModel {
         this.methods = {
             loadOptions: {
                 async getModels() {
-                    const models = await fetchModels(this);
+                    const models = await (0, models_1.fetchModels)(this);
                     return models
-                        .filter(({ chat }) => chat)
+                        .filter(isChatModel)
                         .map(({ modelId, modelDisplayName, pricing, chat }) => ({
                         name: (chat === null || chat === void 0 ? void 0 : chat.tools) === false ? `${modelDisplayName} (no tool calling)` : modelDisplayName,
                         value: modelId,

package/dist/nodes/ZeroGpuChatModel/ZeroGpuChatModel.node.js.map

@@ -1 +1 @@
-{"version":3,"file":"ZeroGpuChatModel.node.js","sourceRoot":"","sources":["../../../nodes/ZeroGpuChatModel/ZeroGpuChatModel.node.ts"],"names":[],"mappings":";;;AAQA,+CAAuE;AACvE,kDAA+C;AAE/C,MAAM,QAAQ,GAAG,2BAA2B,CAAC;AAC7C,MAAM,UAAU,GAAG,6CAA6C,CAAC;AAgBjE,KAAK,UAAU,WAAW,CACzB,OAAqD;;IAErD,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC;QACnD,MAAM,EAAE,KAAK;QACb,GAAG,EAAE,UAAU;QACf,IAAI,EAAE,IAAI;KACV,CAAC,CAAgC,CAAC;IAEnC,OAAO,MAAA,QAAQ,CAAC,MAAM,mCAAI,EAAE,CAAC;AAC9B,CAAC;AAGD,KAAK,UAAU,uBAAuB,CACrC,OAA6B,EAC7B,KAAa,EACb,OAAqB,EACrB,SAAiB;;IAEjB,IAAI,MAAsB,CAAC;IAC3B,IAAI,CAAC;QACJ,MAAM,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IAAC,MAAM,CAAC;QAER,OAAO;IACR,CAAC;IAED,MAAM,YAAY,GAAG,MAAA,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,CAAC,0CAAE,IAAI,CAAC;IAC3E,IAAI,CAAC,YAAY;QAAE,OAAO;IAE1B,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,IAAI,YAAY,CAAC,SAAS,KAAK,KAAK,EAAE,CAAC;QACzE,WAAW,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACpC,CAAC;IACD,IAAI,OAAO,CAAC,eAAe,KAAK,SAAS,IAAI,YAAY,CAAC,eAAe,KAAK,KAAK,EAAE,CAAC;QACrF,WAAW,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACtC,CAAC;IAED,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,iCAAkB,CAC3B,OAAO,CAAC,OAAO,EAAE,EACjB,GAAG,KAAK,oBAAoB,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,mDAAmD,EACvG,EAAE,SAAS,EAAE,CACb,CAAC;IACH,CAAC;AACF,CAAC;AAED,MAAa,gBAAgB;IAA7B;QACC,gBAAW,GAAyB;YACnC,WAAW,EAAE,oBAAoB;YACjC,IAAI,EAAE,kBAAkB;YACxB,IAAI,EAAE,EAAE,KAAK,EAAE,8BAA8B,EAAE,IAAI,EAAE,mCAAmC,EAAE;YAC1F,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,OAAO,EAAE,CAAC,CAAC,CAAC;YACZ,QAAQ,EAAE,0BAA0B;YACpC,WAAW,EAAE,4DAA4D;YACzE,QAAQ,EAAE;gBACT,IAAI,EAAE,oBAAoB;aAC1B;YACD,KAAK,EAAE;gBACN,UAAU,EAAE,CAAC,WAAW,CAAC;gBACzB,aAAa,EAAE;oBACd,EAAE,EAAE,CAAC,iBAAiB,EAAE,YAAY,CAAC;oBACrC,iBAAiB,EAAE,CAAC,2BAA2B,CAAC;iBAChD;gBACD,SAAS,EAAE;oBACV,oBAAoB,EAAE,CAAC,EAAE,GAAG,EAAE,qDAAqD,EAAE,CAAC;iBACtF;aACD;YACD,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,CAAC,kCAAmB,CAAC,eAAe,CAAC;YAC9C,WAAW,EAAE,CAAC,OAAO,CAAC;YACtB,WAAW,EAAE;gBACZ;oBACC,IAAI,EAAE,YAAY;oBAClB,QAAQ,EAAE,IAAI;iBACd;aACD;YACD,UAAU,EAAE;gBACX;oBACC,WAAW,EAAE,kBAAkB;oBAC/B,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,SAAS;oBACf,QAAQ,EAAE,IAAI;oBACd,WAAW,EAAE;wBACZ,iBAAiB,EAAE,WAAW;wBAC9B,oBAAoB,EAAE,IAAI;qBAC1B;oBACD,OAAO,EAAE,qBAAqB;oBAC9B,WAAW,EACV,+JAA+J;iBAChK;gBACD;oBACC,WAAW,EAAE,SAAS;oBACtB,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,YAAY;oBACzB,WAAW,EAAE,2BAA2B;oBACxC,IAAI,EAAE,YAAY;oBAClB,OAAO,EAAE,EAAE;oBACX,OAAO,EAAE;wBACR;4BACC,WAAW,EAAE,gBAAgB;4BAC7B,IAAI,EAAE,WAAW;4BACjB,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,IAAI;4BACb,WAAW,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE;4BAC5B,WAAW,EAAE,0DAA0D;yBACvE;wBACD;4BACC,WAAW,EAAE,kBAAkB;4BAC/B,IAAI,EAAE,iBAAiB;4BACvB,IAAI,EAAE,SAAS;4BACf,OAAO,EAAE,QAAQ;4BACjB,OAAO,EAAE;gCACR,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,OAAO,EAAE;gCACtC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE;gCAC/B,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;gCAC7B,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;gCACnC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;gCACrC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE;6BAC/B;4BACD,WAAW,EAAE,8CAA8C;yBAC3D;wBACD;4BACC,WAAW,EAAE,sBAAsB;4BACnC,IAAI,EAAE,aAAa;4BACnB,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,GAAG;4BACZ,WAAW,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE;4BAC7D,WAAW,EACV,2JAA2J;yBAC5J;qBACD;iBACD;aACD;SACD,CAAC;QAEF,YAAO,GAAG;YACT,WAAW,EAAE;gBACZ,KAAK,CAAC,SAAS;oBACd,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC;oBAEvC,OAAO,MAAM;yBACX,MAAM,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC;yBAC1B,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;wBACvD,IAAI,EACH,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,MAAK,KAAK,CAAC,CAAC,CAAC,GAAG,gBAAgB,oBAAoB,CAAC,CAAC,CAAC,gBAAgB;wBACnF,KAAK,EAAE,OAAO;wBACd,WAAW,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW;qBACjC,CAAC,CAAC;yBACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBAChD,CAAC;aACD;SACD,CAAC;IAyBH,CAAC;IAvBA,KAAK,CAAC,UAAU,CAA6B,SAAiB;QAC7D,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,SAAS,CAAW,CAAC;QAClE,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE,EAAE,CAAiB,CAAC;QAEhF,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,IAAI,OAAO,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YAC9E,MAAM,uBAAuB,CAAC,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAE5D,OAAO,IAAA,yBAAW,EAAC,IAAI,EAAE;YACxB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,QAAQ;YACjB,MAAM,EAAE,WAAW,CAAC,MAAgB;YACpC,KAAK;YACL,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,SAAS,EAAE,OAAO,CAAC,SAAS;YAE5B,gBAAgB,EAAE,OAAO,CAAC,eAAe;gBACxC,CAAC,CAAC,EAAE,gBAAgB,EAAE,OAAO,CAAC,eAAe,EAAE;gBAC/C,CAAC,CAAC,SAAS;SACZ,CAAC,CAAC;IACJ,CAAC;CACD;AAnID,4CAmIC"}
+{"version":3,"file":"ZeroGpuChatModel.node.js","sourceRoot":"","sources":["../../../nodes/ZeroGpuChatModel/ZeroGpuChatModel.node.ts"],"names":[],"mappings":";;;AAQA,+CAAuE;AACvE,kDAA+C;AAC/C,6CAAgF;AAEhF,MAAM,QAAQ,GAAG,2BAA2B,CAAC;AAE7C,MAAM,UAAU,GAAG,CAAC,iBAAiB,EAAE,eAAe,CAAC,CAAC;AAUxD,SAAS,WAAW,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,QAAQ,EAAE,OAAO,EAAgB;IAC9E,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO,IAAI,KAAK,IAAI,CAAC;IAE7C,OAAO,CACN,UAAU,CAAC,QAAQ,CAAC,eAAe,aAAf,eAAe,cAAf,eAAe,GAAI,EAAE,CAAC;QAC1C,QAAQ,KAAK,YAAY;QACzB,OAAO,KAAK,qBAAY,CACxB,CAAC;AACH,CAAC;AAGD,KAAK,UAAU,uBAAuB,CACrC,OAA6B,EAC7B,KAAa,EACb,OAAqB,EACrB,SAAiB;;IAEjB,IAAI,MAAsB,CAAC;IAC3B,IAAI,CAAC;QACJ,MAAM,GAAG,MAAM,IAAA,oBAAW,EAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IAAC,MAAM,CAAC;QAER,OAAO;IACR,CAAC;IAED,MAAM,YAAY,GAAG,MAAA,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,CAAC,0CAAE,IAAI,CAAC;IAC3E,IAAI,CAAC,YAAY;QAAE,OAAO;IAE1B,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,IAAI,YAAY,CAAC,SAAS,KAAK,KAAK,EAAE,CAAC;QACzE,WAAW,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACpC,CAAC;IACD,IAAI,OAAO,CAAC,eAAe,KAAK,SAAS,IAAI,YAAY,CAAC,eAAe,KAAK,KAAK,EAAE,CAAC;QACrF,WAAW,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACtC,CAAC;IAED,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,iCAAkB,CAC3B,OAAO,CAAC,OAAO,EAAE,EACjB,GAAG,KAAK,oBAAoB,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,mDAAmD,EACvG,EAAE,SAAS,EAAE,CACb,CAAC;IACH,CAAC;AACF,CAAC;AAED,MAAa,gBAAgB;IAA7B;QACC,gBAAW,GAAyB;YACnC,WAAW,EAAE,oBAAoB;YACjC,IAAI,EAAE,kBAAkB;YACxB,IAAI,EAAE,EAAE,KAAK,EAAE,8BAA8B,EAAE,IAAI,EAAE,mCAAmC,EAAE;YAC1F,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,OAAO,EAAE,CAAC,CAAC,CAAC;YACZ,QAAQ,EAAE,0BAA0B;YACpC,WAAW,EAAE,4DAA4D;YACzE,QAAQ,EAAE;gBACT,IAAI,EAAE,oBAAoB;aAC1B;YACD,KAAK,EAAE;gBACN,UAAU,EAAE,CAAC,WAAW,CAAC;gBACzB,aAAa,EAAE;oBACd,EAAE,EAAE,CAAC,iBAAiB,EAAE,YAAY,CAAC;oBACrC,iBAAiB,EAAE,CAAC,2BAA2B,CAAC;iBAChD;gBACD,SAAS,EAAE;oBACV,oBAAoB,EAAE,CAAC,EAAE,GAAG,EAAE,qDAAqD,EAAE,CAAC;iBACtF;aACD;YACD,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,CAAC,kCAAmB,CAAC,eAAe,CAAC;YAC9C,WAAW,EAAE,CAAC,OAAO,CAAC;YACtB,WAAW,EAAE;gBACZ;oBACC,IAAI,EAAE,YAAY;oBAClB,QAAQ,EAAE,IAAI;iBACd;aACD;YACD,UAAU,EAAE;gBACX;oBACC,WAAW,EAAE,kBAAkB;oBAC/B,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,SAAS;oBACf,QAAQ,EAAE,IAAI;oBACd,WAAW,EAAE;wBACZ,iBAAiB,EAAE,WAAW;wBAC9B,oBAAoB,EAAE,IAAI;qBAC1B;oBACD,OAAO,EAAE,qBAAqB;oBAC9B,WAAW,EACV,+JAA+J;iBAChK;gBACD;oBACC,WAAW,EAAE,SAAS;oBACtB,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,YAAY;oBACzB,WAAW,EAAE,2BAA2B;oBACxC,IAAI,EAAE,YAAY;oBAClB,OAAO,EAAE,EAAE;oBACX,OAAO,EAAE;wBACR;4BACC,WAAW,EAAE,gBAAgB;4BAC7B,IAAI,EAAE,WAAW;4BACjB,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,IAAI;4BACb,WAAW,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE;4BAC5B,WAAW,EAAE,0DAA0D;yBACvE;wBACD;4BACC,WAAW,EAAE,kBAAkB;4BAC/B,IAAI,EAAE,iBAAiB;4BACvB,IAAI,EAAE,SAAS;4BACf,OAAO,EAAE,QAAQ;4BACjB,OAAO,EAAE;gCACR,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,OAAO,EAAE;gCACtC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE;gCAC/B,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;gCAC7B,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;gCACnC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;gCACrC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE;6BAC/B;4BACD,WAAW,EAAE,8CAA8C;yBAC3D;wBACD;4BACC,WAAW,EAAE,sBAAsB;4BACnC,IAAI,EAAE,aAAa;4BACnB,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,GAAG;4BACZ,WAAW,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE;4BAC7D,WAAW,EACV,2JAA2J;yBAC5J;qBACD;iBACD;aACD;SACD,CAAC;QAEF,YAAO,GAAG;YACT,WAAW,EAAE;gBACZ,KAAK,CAAC,SAAS;oBACd,MAAM,MAAM,GAAG,MAAM,IAAA,oBAAW,EAAC,IAAI,CAAC,CAAC;oBAEvC,OAAO,MAAM;yBACX,MAAM,CAAC,WAAW,CAAC;yBACnB,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;wBACvD,IAAI,EACH,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,MAAK,KAAK,CAAC,CAAC,CAAC,GAAG,gBAAgB,oBAAoB,CAAC,CAAC,CAAC,gBAAgB;wBACnF,KAAK,EAAE,OAAO;wBACd,WAAW,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW;qBACjC,CAAC,CAAC;yBACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBAChD,CAAC;aACD;SACD,CAAC;IAyBH,CAAC;IAvBA,KAAK,CAAC,UAAU,CAA6B,SAAiB;QAC7D,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,SAAS,CAAW,CAAC;QAClE,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE,EAAE,CAAiB,CAAC;QAEhF,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,IAAI,OAAO,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YAC9E,MAAM,uBAAuB,CAAC,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAE5D,OAAO,IAAA,yBAAW,EAAC,IAAI,EAAE;YACxB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,QAAQ;YACjB,MAAM,EAAE,WAAW,CAAC,MAAgB;YACpC,KAAK;YACL,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,SAAS,EAAE,OAAO,CAAC,SAAS;YAE5B,gBAAgB,EAAE,OAAO,CAAC,eAAe;gBACxC,CAAC,CAAC,EAAE,gBAAgB,EAAE,OAAO,CAAC,eAAe,EAAE;gBAC/C,CAAC,CAAC,SAAS;SACZ,CAAC,CAAC;IACJ,CAAC;CACD;AAnID,4CAmIC"}

package/dist/nodes/ZeroGpuEmbeddings/ZeroGpuEmbeddings.node.d.ts

@@ -0,0 +1,10 @@
+import type { ILoadOptionsFunctions, INodePropertyOptions, INodeType, INodeTypeDescription, ISupplyDataFunctions, SupplyData } from 'n8n-workflow';
+export declare class ZeroGpuEmbeddings implements INodeType {
+    description: INodeTypeDescription;
+    methods: {
+        loadOptions: {
+            getModels(this: ILoadOptionsFunctions): Promise<INodePropertyOptions[]>;
+        };
+    };
+    supplyData(this: ISupplyDataFunctions, itemIndex: number): Promise<SupplyData>;
+}

package/dist/nodes/ZeroGpuEmbeddings/ZeroGpuEmbeddings.node.js

@@ -0,0 +1,154 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ZeroGpuEmbeddings = void 0;
+const n8n_workflow_1 = require("n8n-workflow");
+const models_1 = require("../shared/models");
+const EMBEDDINGS_URL = 'https://api.zerogpu.ai/v1/embeddings';
+const BATCH_SIZE = 256;
+const isVector = (value) => Array.isArray(value) && value.length > 0 && value.every((item) => typeof item === 'number');
+async function requestEmbeddings(context, model, input) {
+    let response;
+    try {
+        response = await context.helpers.httpRequestWithAuthentication.call(context, 'zeroGpuApi', {
+            method: 'POST',
+            url: EMBEDDINGS_URL,
+            body: { model, input },
+            json: true,
+        });
+    }
+    catch (error) {
+        throw new n8n_workflow_1.NodeApiError(context.getNode(), error);
+    }
+    const vectors = input.map(() => undefined);
+    const data = response === null || response === void 0 ? void 0 : response.data;
+    if (Array.isArray(data) && data.length === input.length) {
+        for (const item of data) {
+            if (typeof (item === null || item === void 0 ? void 0 : item.index) === 'number' && item.index in vectors && isVector(item.embedding)) {
+                vectors[item.index] = item.embedding;
+            }
+        }
+    }
+    if (!vectors.every(isVector)) {
+        throw new n8n_workflow_1.NodeOperationError(context.getNode(), 'ZeroGPU returned embeddings in an unexpected shape');
+    }
+    return vectors;
+}
+async function logged(context, input, run) {
+    const { index } = context.addInputData(n8n_workflow_1.NodeConnectionTypes.AiEmbedding, [[{ json: input }]]);
+    try {
+        const response = await run();
+        context.addOutputData(n8n_workflow_1.NodeConnectionTypes.AiEmbedding, index, [[{ json: { response } }]]);
+        return response;
+    }
+    catch (error) {
+        const failure = error instanceof n8n_workflow_1.NodeApiError || error instanceof n8n_workflow_1.NodeOperationError
+            ? error
+            : new n8n_workflow_1.NodeOperationError(context.getNode(), error);
+        context.addOutputData(n8n_workflow_1.NodeConnectionTypes.AiEmbedding, index, failure);
+        throw failure;
+    }
+}
+class ZeroGpuEmbeddings {
+    constructor() {
+        this.description = {
+            displayName: 'ZeroGPU Embeddings',
+            name: 'zeroGpuEmbeddings',
+            icon: { light: 'file:../../icons/zerogpu.svg', dark: 'file:../../icons/zerogpu.dark.svg' },
+            group: ['transform'],
+            version: [1],
+            subtitle: '={{$parameter["model"]}}',
+            description: 'Use ZeroGPU embedding models in vector stores and RAG workflows',
+            defaults: {
+                name: 'ZeroGPU Embeddings',
+            },
+            codex: {
+                categories: ['assistant'],
+                subcategories: {
+                    AI: ['Embeddings'],
+                },
+                resources: {
+                    primaryDocumentation: [{ url: 'https://github.com/zerogpu/n8n-nodes-zerogpu#readme' }],
+                },
+            },
+            inputs: [],
+            outputs: [n8n_workflow_1.NodeConnectionTypes.AiEmbedding],
+            outputNames: ['Embeddings'],
+            credentials: [
+                {
+                    name: 'zeroGpuApi',
+                    required: true,
+                },
+            ],
+            properties: [
+                {
+                    displayName: 'Model Name or ID',
+                    name: 'model',
+                    type: 'options',
+                    required: true,
+                    typeOptions: {
+                        loadOptionsMethod: 'getModels',
+                        allowArbitraryValues: true,
+                    },
+                    default: 'all-minilm-l6-v2',
+                    description: 'The embedding model to use. Switching to a model with a different vector size means re-indexing. Choose from the list, or specify an ID using an <a href="https://docs.n8n.io/code/expressions/">expression</a>.',
+                },
+            ],
+        };
+        this.methods = {
+            loadOptions: {
+                async getModels() {
+                    const models = await (0, models_1.fetchModels)(this);
+                    return models
+                        .filter(({ taskType }) => taskType === 'embedding')
+                        .map(({ modelId, pricing }) => ({
+                        name: modelId,
+                        value: modelId,
+                        description: pricing === null || pricing === void 0 ? void 0 : pricing.description,
+                    }))
+                        .sort((a, b) => a.name.localeCompare(b.name));
+                },
+            },
+        };
+    }
+    async supplyData(itemIndex) {
+        var _a;
+        const model = String((_a = this.getNodeParameter('model', itemIndex)) !== null && _a !== void 0 ? _a : '').trim();
+        if (model === '') {
+            throw new n8n_workflow_1.NodeOperationError(this.getNode(), 'Enter the Model', { itemIndex });
+        }
+        const embeddings = {
+            embedDocuments: async (documents) => {
+                if (!Array.isArray(documents)) {
+                    throw new n8n_workflow_1.NodeOperationError(this.getNode(), 'Documents must be an array', {
+                        description: 'Expected an array of strings to embed.',
+                    });
+                }
+                const invalid = documents.findIndex((doc) => typeof doc !== 'string' || doc === '');
+                if (invalid !== -1) {
+                    throw new n8n_workflow_1.NodeOperationError(this.getNode(), `Invalid document at index ${invalid}`, {
+                        description: `Document at index ${invalid} is empty or isn't text. All documents must be non-empty strings.`,
+                    });
+                }
+                return await logged(this, { documents }, async () => {
+                    const vectors = [];
+                    for (let start = 0; start < documents.length; start += BATCH_SIZE) {
+                        const batch = documents.slice(start, start + BATCH_SIZE);
+                        vectors.push(...(await requestEmbeddings(this, model, batch)));
+                    }
+                    return vectors;
+                });
+            },
+            embedQuery: async (query) => {
+                if (typeof query !== 'string' || query === '') {
+                    throw new n8n_workflow_1.NodeOperationError(this.getNode(), 'Cannot embed empty or undefined text', {
+                        description: 'The text to embed is empty or undefined. This can happen when an expression evaluates to nothing, or an AI agent calls a tool without the text.',
+                    });
+                }
+                return await logged(this, { query }, async () => (await requestEmbeddings(this, model, [query]))[0]);
+            },
+        };
+        return { response: embeddings };
+    }
+}
+exports.ZeroGpuEmbeddings = ZeroGpuEmbeddings;
+//# sourceMappingURL=ZeroGpuEmbeddings.node.js.map

package/dist/nodes/ZeroGpuEmbeddings/ZeroGpuEmbeddings.node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ZeroGpuEmbeddings.node.js","sourceRoot":"","sources":["../../../nodes/ZeroGpuEmbeddings/ZeroGpuEmbeddings.node.ts"],"names":[],"mappings":";;;AAUA,+CAAqF;AACrF,6CAA+C;AAE/C,MAAM,cAAc,GAAG,sCAAsC,CAAC;AAG9D,MAAM,UAAU,GAAG,GAAG,CAAC;AAEvB,MAAM,QAAQ,GAAG,CAAC,KAAc,EAAqB,EAAE,CACtD,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC;AAE7F,KAAK,UAAU,iBAAiB,CAC/B,OAA6B,EAC7B,KAAa,EACb,KAAe;IAEf,IAAI,QAA2E,CAAC;IAChF,IAAI,CAAC;QACJ,QAAQ,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,6BAA6B,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE;YAC1F,MAAM,EAAE,MAAM;YACd,GAAG,EAAE,cAAc;YACnB,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;YACtB,IAAI,EAAE,IAAI;SACV,CAAC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,IAAI,2BAAY,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,KAAmB,CAAC,CAAC;IAChE,CAAC;IAGD,MAAM,OAAO,GAAgC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACxE,MAAM,IAAI,GAAG,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,IAAI,CAAC;IAC5B,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,EAAE,CAAC;QACzD,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YACzB,IAAI,OAAO,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,CAAA,KAAK,QAAQ,IAAI,IAAI,CAAC,KAAK,IAAI,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC1F,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;YACtC,CAAC;QACF,CAAC;IACF,CAAC;IACD,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC9B,MAAM,IAAI,iCAAkB,CAC3B,OAAO,CAAC,OAAO,EAAE,EACjB,oDAAoD,CACpD,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AAChB,CAAC;AAGD,KAAK,UAAU,MAAM,CACpB,OAA6B,EAC7B,KAAkB,EAClB,GAAqB;IAErB,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC,kCAAmB,CAAC,WAAW,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;IAC7F,IAAI,CAAC;QACJ,MAAM,QAAQ,GAAG,MAAM,GAAG,EAAE,CAAC;QAC7B,OAAO,CAAC,aAAa,CAAC,kCAAmB,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1F,OAAO,QAAQ,CAAC;IACjB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,OAAO,GACZ,KAAK,YAAY,2BAAY,IAAI,KAAK,YAAY,iCAAkB;YACnE,CAAC,CAAC,KAAK;YACP,CAAC,CAAC,IAAI,iCAAkB,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,KAAc,CAAC,CAAC;QAC9D,OAAO,CAAC,aAAa,CAAC,kCAAmB,CAAC,WAAW,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACvE,MAAM,OAAO,CAAC;IACf,CAAC;AACF,CAAC;AAED,MAAa,iBAAiB;IAA9B;QACC,gBAAW,GAAyB;YACnC,WAAW,EAAE,oBAAoB;YACjC,IAAI,EAAE,mBAAmB;YACzB,IAAI,EAAE,EAAE,KAAK,EAAE,8BAA8B,EAAE,IAAI,EAAE,mCAAmC,EAAE;YAC1F,KAAK,EAAE,CAAC,WAAW,CAAC;YACpB,OAAO,EAAE,CAAC,CAAC,CAAC;YACZ,QAAQ,EAAE,0BAA0B;YACpC,WAAW,EAAE,iEAAiE;YAC9E,QAAQ,EAAE;gBACT,IAAI,EAAE,oBAAoB;aAC1B;YACD,KAAK,EAAE;gBACN,UAAU,EAAE,CAAC,WAAW,CAAC;gBACzB,aAAa,EAAE;oBACd,EAAE,EAAE,CAAC,YAAY,CAAC;iBAClB;gBACD,SAAS,EAAE;oBACV,oBAAoB,EAAE,CAAC,EAAE,GAAG,EAAE,qDAAqD,EAAE,CAAC;iBACtF;aACD;YACD,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,CAAC,kCAAmB,CAAC,WAAW,CAAC;YAC1C,WAAW,EAAE,CAAC,YAAY,CAAC;YAC3B,WAAW,EAAE;gBACZ;oBACC,IAAI,EAAE,YAAY;oBAClB,QAAQ,EAAE,IAAI;iBACd;aACD;YACD,UAAU,EAAE;gBACX;oBACC,WAAW,EAAE,kBAAkB;oBAC/B,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,SAAS;oBACf,QAAQ,EAAE,IAAI;oBACd,WAAW,EAAE;wBACZ,iBAAiB,EAAE,WAAW;wBAC9B,oBAAoB,EAAE,IAAI;qBAC1B;oBACD,OAAO,EAAE,kBAAkB;oBAC3B,WAAW,EACV,kNAAkN;iBACnN;aACD;SACD,CAAC;QAEF,YAAO,GAAG;YACT,WAAW,EAAE;gBACZ,KAAK,CAAC,SAAS;oBACd,MAAM,MAAM,GAAG,MAAM,IAAA,oBAAW,EAAC,IAAI,CAAC,CAAC;oBAEvC,OAAO,MAAM;yBACX,MAAM,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,QAAQ,KAAK,WAAW,CAAC;yBAClD,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC/B,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,OAAO;wBACd,WAAW,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW;qBACjC,CAAC,CAAC;yBACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBAChD,CAAC;aACD;SACD,CAAC;IAmDH,CAAC;IAjDA,KAAK,CAAC,UAAU,CAA6B,SAAiB;;QAC7D,MAAM,KAAK,GAAG,MAAM,CAAC,MAAA,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,SAAS,CAAC,mCAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC7E,IAAI,KAAK,KAAK,EAAE,EAAE,CAAC;YAClB,MAAM,IAAI,iCAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,iBAAiB,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;QAChF,CAAC;QAGD,MAAM,UAAU,GAAG;YAClB,cAAc,EAAE,KAAK,EAAE,SAAkB,EAAuB,EAAE;gBACjE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC/B,MAAM,IAAI,iCAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,4BAA4B,EAAE;wBAC1E,WAAW,EAAE,wCAAwC;qBACrD,CAAC,CAAC;gBACJ,CAAC;gBACD,MAAM,OAAO,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,EAAE,CAAC,CAAC;gBACpF,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;oBACpB,MAAM,IAAI,iCAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,6BAA6B,OAAO,EAAE,EAAE;wBACpF,WAAW,EAAE,qBAAqB,OAAO,mEAAmE;qBAC5G,CAAC,CAAC;gBACJ,CAAC;gBAED,OAAO,MAAM,MAAM,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,EAAE,KAAK,IAAI,EAAE;oBACnD,MAAM,OAAO,GAAe,EAAE,CAAC;oBAC/B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,SAAS,CAAC,MAAM,EAAE,KAAK,IAAI,UAAU,EAAE,CAAC;wBACnE,MAAM,KAAK,GAAI,SAAsB,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,UAAU,CAAC,CAAC;wBACvE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,iBAAiB,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;oBAChE,CAAC;oBACD,OAAO,OAAO,CAAC;gBAChB,CAAC,CAAC,CAAC;YACJ,CAAC;YAED,UAAU,EAAE,KAAK,EAAE,KAAc,EAAqB,EAAE;gBACvD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,EAAE,EAAE,CAAC;oBAC/C,MAAM,IAAI,iCAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,sCAAsC,EAAE;wBACpF,WAAW,EACV,iJAAiJ;qBAClJ,CAAC,CAAC;gBACJ,CAAC;gBAED,OAAO,MAAM,MAAM,CAClB,IAAI,EACJ,EAAE,KAAK,EAAE,EACT,KAAK,IAAI,EAAE,CAAC,CAAC,MAAM,iBAAiB,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAC9D,CAAC;YACH,CAAC;SACD,CAAC;QAEF,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;IACjC,CAAC;CACD;AAjHD,8CAiHC"}

package/dist/nodes/ZeroGpuEmbeddings/ZeroGpuEmbeddings.node.json

@@ -0,0 +1,18 @@
+{
+    "node": "@zerogpu/n8n-nodes-zerogpu",
+    "nodeVersion": "1.0",
+    "codexVersion": "1.0",
+    "categories": ["Development"],
+    "resources": {
+        "credentialDocumentation": [
+            {
+                "url": "https://docs.zerogpu.ai/docs/quickstart"
+            }
+        ],
+        "primaryDocumentation": [
+            {
+                "url": "https://github.com/zerogpu/n8n-nodes-zerogpu#readme"
+            }
+        ]
+    }
+}

package/dist/nodes/shared/models.d.ts

@@ -0,0 +1,19 @@
+import type { ILoadOptionsFunctions, ISupplyDataFunctions } from 'n8n-workflow';
+export declare const MODELS_URL = "https://api-dashboard.zerogpu.ai/api/models";
+export declare const OCR_MODEL_ID = "unlimited-ocr";
+export type ZeroGpuModel = {
+    modelId: string;
+    modelDisplayName: string;
+    taskDisplayName?: string | null;
+    taskType?: string | null;
+    modelUsecases?: Record<string, unknown> | null;
+    pricing?: {
+        description?: string;
+    } | null;
+    chat?: {
+        tools: boolean;
+        maxTokens: boolean;
+        reasoningEffort: boolean;
+    } | null;
+};
+export declare function fetchModels(context: ILoadOptionsFunctions | ISupplyDataFunctions): Promise<ZeroGpuModel[]>;

package/dist/nodes/shared/models.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OCR_MODEL_ID = exports.MODELS_URL = void 0;
+exports.fetchModels = fetchModels;
+exports.MODELS_URL = 'https://api-dashboard.zerogpu.ai/api/models';
+exports.OCR_MODEL_ID = 'unlimited-ocr';
+async function fetchModels(context) {
+    var _a;
+    const response = (await context.helpers.httpRequest({
+        method: 'GET',
+        url: exports.MODELS_URL,
+        json: true,
+    }));
+    return (_a = response.models) !== null && _a !== void 0 ? _a : [];
+}
+//# sourceMappingURL=models.js.map

package/dist/nodes/shared/models.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../../../nodes/shared/models.ts"],"names":[],"mappings":";;;AAkBA,kCAUC;AA1BY,QAAA,UAAU,GAAG,6CAA6C,CAAC;AAG3D,QAAA,YAAY,GAAG,eAAe,CAAC;AAarC,KAAK,UAAU,WAAW,CAChC,OAAqD;;IAErD,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC;QACnD,MAAM,EAAE,KAAK;QACb,GAAG,EAAE,kBAAU;QACf,IAAI,EAAE,IAAI;KACV,CAAC,CAAgC,CAAC;IAEnC,OAAO,MAAA,QAAQ,CAAC,MAAM,mCAAI,EAAE,CAAC;AAC9B,CAAC"}

package/dist/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@zerogpu/n8n-nodes-zerogpu",
-    "version": "0.2.0",
+    "version": "0.4.0",
     "description": "Official ZeroGPU nodes for n8n. ZeroGPU is a compute-efficient inference provider for apps and agents — purpose-built small and nano language models on an edge network that run the repeatable tasks frontier models shouldn’t, ~10x faster and 50%+ cheaper. Auto-scaling, with zero GPU infrastructure. Plug in and you’re live.",
     "license": "MIT",
     "homepage": "https://zerogpu.ai",
@@ -44,7 +44,9 @@
             "dist/credentials/ZeroGpuApi.credentials.js"
         ],
         "nodes": [
-            "dist/nodes/ZeroGpuChatModel/ZeroGpuChatModel.node.js"
+            "dist/nodes/ZeroGpu/ZeroGpu.node.js",
+            "dist/nodes/ZeroGpuChatModel/ZeroGpuChatModel.node.js",
+            "dist/nodes/ZeroGpuEmbeddings/ZeroGpuEmbeddings.node.js"
         ]
     },
     "devDependencies": {