edge-impulse-linux 1.4.2 → 1.4.4

package/build/sdk/studio/sdk/api/learnApi.js

@@ -30,10 +30,11 @@ var LearnApiApiKeys;
     LearnApiApiKeys[LearnApiApiKeys["JWTHttpHeaderAuthentication"] = 2] = "JWTHttpHeaderAuthentication";
 })(LearnApiApiKeys = exports.LearnApiApiKeys || (exports.LearnApiApiKeys = {}));
 class LearnApi {
-    constructor(basePathOrUsername, password, basePath) {
+    constructor(basePathOrUsername, opts, password, basePath) {
         this._basePath = defaultBasePath;
         this.defaultHeaders = {};
         this._useQuerystring = false;
+        this._opts = {};
         this.authentications = {
             'default': new models_1.VoidAuth(),
             'ApiKeyAuthentication': new models_2.ApiKeyAuth('header', 'x-api-key'),
@@ -50,6 +51,7 @@ class LearnApi {
                 this.basePath = basePathOrUsername;
             }
         }
+        this.opts = opts !== null && opts !== void 0 ? opts : {};
     }
     set useQuerystring(value) {
         this._useQuerystring = value;
@@ -60,6 +62,12 @@ class LearnApi {
     get basePath() {
         return this._basePath;
     }
+    set opts(opts) {
+        this._opts = opts;
+    }
+    get opts() {
+        return this._opts;
+    }
     setDefaultAuthentication(auth) {
         this.authentications.default = auth;
     }
@@ -74,12 +82,14 @@ class LearnApi {
      * @param featureAx1 Feature axis 1
      * @param featureAx2 Feature axis 2
      */
-    async anomalyTrainedFeatures(projectId, learnId, featureAx1, featureAx2, options = { headers: {} }) {
+    async anomalyTrainedFeatures(projectId, learnId, queryParams, options = { headers: {} }) {
         const localVarPath = this.basePath + '/api/{projectId}/training/anomaly/{learnId}/features/get-graph'
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/json'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -98,20 +108,21 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling anomalyTrainedFeatures.');
         }
         // verify required parameter 'featureAx1' is not null or undefined
-        if (featureAx1 === null || featureAx1 === undefined) {
-            throw new Error('Required parameter featureAx1 was null or undefined when calling anomalyTrainedFeatures.');
+        if (queryParams.featureAx1 === null || queryParams.featureAx1 === undefined) {
+            throw new Error('Required parameter queryParams.featureAx1 was null or undefined when calling anomalyTrainedFeatures.');
         }
         // verify required parameter 'featureAx2' is not null or undefined
-        if (featureAx2 === null || featureAx2 === undefined) {
-            throw new Error('Required parameter featureAx2 was null or undefined when calling anomalyTrainedFeatures.');
+        if (queryParams.featureAx2 === null || queryParams.featureAx2 === undefined) {
+            throw new Error('Required parameter queryParams.featureAx2 was null or undefined when calling anomalyTrainedFeatures.');
         }
-        if (featureAx1 !== undefined) {
-            localVarQueryParameters['featureAx1'] = models_1.ObjectSerializer.serialize(featureAx1, "number");
+        if (queryParams.featureAx1 !== undefined) {
+            localVarQueryParameters['featureAx1'] = models_1.ObjectSerializer.serialize(queryParams.featureAx1, "number");
         }
-        if (featureAx2 !== undefined) {
-            localVarQueryParameters['featureAx2'] = models_1.ObjectSerializer.serialize(featureAx2, "number");
+        if (queryParams.featureAx2 !== undefined) {
+            localVarQueryParameters['featureAx2'] = models_1.ObjectSerializer.serialize(queryParams.featureAx2, "number");
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -171,7 +182,9 @@ class LearnApi {
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)))
             .replace('{' + 'sampleId' + '}', encodeURIComponent(String(sampleId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/json'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -194,6 +207,7 @@ class LearnApi {
             throw new Error('Required parameter sampleId was null or undefined when calling anomalyTrainedFeaturesPerSample.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -251,7 +265,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/zip'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -270,6 +286,7 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling downloadKerasData.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -327,7 +344,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/zip'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -346,6 +365,7 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling downloadKerasExport.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -405,7 +425,9 @@ class LearnApi {
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)))
             .replace('{' + 'modelDownloadId' + '}', encodeURIComponent(String(modelDownloadId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/octet-stream'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -428,6 +450,86 @@ class LearnApi {
             throw new Error('Required parameter modelDownloadId was null or undefined when calling downloadLearnModel.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
+        let localVarUseFormData = false;
+        let localVarRequestOptions = {
+            method: 'GET',
+            qs: localVarQueryParameters,
+            headers: localVarHeaderParams,
+            uri: localVarPath,
+            useQuerystring: this._useQuerystring,
+            agentOptions: { keepAlive: false },
+            encoding: null,
+        };
+        let authenticationPromise = Promise.resolve();
+        authenticationPromise = authenticationPromise.then(() => this.authentications.ApiKeyAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTHttpHeaderAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.default.applyToRequest(localVarRequestOptions));
+        return authenticationPromise.then(() => {
+            if (Object.keys(localVarFormParams).length) {
+                if (localVarUseFormData) {
+                    localVarRequestOptions.formData = localVarFormParams;
+                }
+                else {
+                    localVarRequestOptions.form = localVarFormParams;
+                }
+            }
+            return new Promise((resolve, reject) => {
+                localVarRequest(localVarRequestOptions, (error, response, body) => {
+                    if (error) {
+                        reject(error);
+                    }
+                    else {
+                        body = models_1.ObjectSerializer.deserialize(body, "Buffer");
+                        const errString = `Failed to call "${localVarPath}", returned ${response.statusCode}: ` + response.body;
+                        if (typeof body.success === 'boolean' && !body.success) {
+                            reject(new Error(body.error || errString));
+                        }
+                        else if (response.statusCode && response.statusCode >= 200 && response.statusCode <= 299) {
+                            resolve(body);
+                        }
+                        else {
+                            reject(errString);
+                        }
+                    }
+                });
+            });
+        });
+    }
+    /**
+     * Download a pretrained model file
+     * @summary Download pretrained model
+     * @param projectId Project ID
+     * @param pretrainedModelDownloadType
+     */
+    async downloadPretrainedModel(projectId, pretrainedModelDownloadType, options = { headers: {} }) {
+        const localVarPath = this.basePath + '/api/{projectId}/pretrained-model/download/{pretrainedModelDownloadType}'
+            .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
+            .replace('{' + 'pretrainedModelDownloadType' + '}', encodeURIComponent(String(pretrainedModelDownloadType)));
+        let localVarQueryParameters = {};
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
+        const produces = ['application/octet-stream'];
+        // give precedence to 'application/json'
+        if (produces.indexOf('application/json') >= 0) {
+            localVarHeaderParams.Accept = 'application/json';
+        }
+        else {
+            localVarHeaderParams.Accept = produces.join(',');
+        }
+        let localVarFormParams = {};
+        // verify required parameter 'projectId' is not null or undefined
+        if (projectId === null || projectId === undefined) {
+            throw new Error('Required parameter projectId was null or undefined when calling downloadPretrainedModel.');
+        }
+        // verify required parameter 'pretrainedModelDownloadType' is not null or undefined
+        if (pretrainedModelDownloadType === null || pretrainedModelDownloadType === undefined) {
+            throw new Error('Required parameter pretrainedModelDownloadType was null or undefined when calling downloadPretrainedModel.');
+        }
+        Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -485,7 +587,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/json'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -504,6 +608,7 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling getAnomaly.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -561,7 +666,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/json'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -580,6 +687,7 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling getAnomalyMetadata.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -637,7 +745,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/json'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -656,6 +766,7 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling getKeras.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -713,7 +824,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/json'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -732,6 +845,7 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling getKerasDataExplorerFeatures.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -789,7 +903,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/json'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -808,6 +924,7 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling getKerasMetadata.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -865,7 +982,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/octet-stream'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -884,6 +1003,7 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling getLearnXData.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -941,7 +1061,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/octet-stream'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -960,6 +1082,7 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling getLearnYData.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'GET',
@@ -1006,6 +1129,227 @@ class LearnApi {
             });
         });
     }
+    /**
+     * Receive info back about the earlier uploaded pretrained model (via `uploadPretrainedModel`) input/output tensors. If you want to deploy a pretrained model from the API, see `startDeployPretrainedModelJob`.
+     * @summary Get pretrained model
+     * @param projectId Project ID
+     */
+    async getPretrainedModelInfo(projectId, options = { headers: {} }) {
+        const localVarPath = this.basePath + '/api/{projectId}/pretrained-model'
+            .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)));
+        let localVarQueryParameters = {};
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
+        const produces = ['application/json'];
+        // give precedence to 'application/json'
+        if (produces.indexOf('application/json') >= 0) {
+            localVarHeaderParams.Accept = 'application/json';
+        }
+        else {
+            localVarHeaderParams.Accept = produces.join(',');
+        }
+        let localVarFormParams = {};
+        // verify required parameter 'projectId' is not null or undefined
+        if (projectId === null || projectId === undefined) {
+            throw new Error('Required parameter projectId was null or undefined when calling getPretrainedModelInfo.');
+        }
+        Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
+        let localVarUseFormData = false;
+        let localVarRequestOptions = {
+            method: 'GET',
+            qs: localVarQueryParameters,
+            headers: localVarHeaderParams,
+            uri: localVarPath,
+            useQuerystring: this._useQuerystring,
+            agentOptions: { keepAlive: false },
+            json: true,
+        };
+        let authenticationPromise = Promise.resolve();
+        authenticationPromise = authenticationPromise.then(() => this.authentications.ApiKeyAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTHttpHeaderAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.default.applyToRequest(localVarRequestOptions));
+        return authenticationPromise.then(() => {
+            if (Object.keys(localVarFormParams).length) {
+                if (localVarUseFormData) {
+                    localVarRequestOptions.formData = localVarFormParams;
+                }
+                else {
+                    localVarRequestOptions.form = localVarFormParams;
+                }
+            }
+            return new Promise((resolve, reject) => {
+                localVarRequest(localVarRequestOptions, (error, response, body) => {
+                    if (error) {
+                        reject(error);
+                    }
+                    else {
+                        body = models_1.ObjectSerializer.deserialize(body, "GetPretrainedModelResponse");
+                        const errString = `Failed to call "${localVarPath}", returned ${response.statusCode}: ` + response.body;
+                        if (typeof body.success === 'boolean' && !body.success) {
+                            reject(new Error(body.error || errString));
+                        }
+                        else if (response.statusCode && response.statusCode >= 200 && response.statusCode <= 299) {
+                            resolve(body);
+                        }
+                        else {
+                            reject(errString);
+                        }
+                    }
+                });
+            });
+        });
+    }
+    /**
+     * Returns the latency, RAM and ROM used for the pretrained model - upload first via  `uploadPretrainedModel`. This is using the project\'s selected latency device. Updates are streamed over the websocket API (or can be retrieved through the /stdout endpoint). Use getProfileTfliteJobResult to get the results when the job is completed.
+     * @summary Profile pretrained model
+     * @param projectId Project ID
+     */
+    async profilePretrainedModel(projectId, options = { headers: {} }) {
+        const localVarPath = this.basePath + '/api/{projectId}/pretrained-model/profile'
+            .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)));
+        let localVarQueryParameters = {};
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
+        const produces = ['application/json'];
+        // give precedence to 'application/json'
+        if (produces.indexOf('application/json') >= 0) {
+            localVarHeaderParams.Accept = 'application/json';
+        }
+        else {
+            localVarHeaderParams.Accept = produces.join(',');
+        }
+        let localVarFormParams = {};
+        // verify required parameter 'projectId' is not null or undefined
+        if (projectId === null || projectId === undefined) {
+            throw new Error('Required parameter projectId was null or undefined when calling profilePretrainedModel.');
+        }
+        Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
+        let localVarUseFormData = false;
+        let localVarRequestOptions = {
+            method: 'POST',
+            qs: localVarQueryParameters,
+            headers: localVarHeaderParams,
+            uri: localVarPath,
+            useQuerystring: this._useQuerystring,
+            agentOptions: { keepAlive: false },
+            json: true,
+        };
+        let authenticationPromise = Promise.resolve();
+        authenticationPromise = authenticationPromise.then(() => this.authentications.ApiKeyAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTHttpHeaderAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.default.applyToRequest(localVarRequestOptions));
+        return authenticationPromise.then(() => {
+            if (Object.keys(localVarFormParams).length) {
+                if (localVarUseFormData) {
+                    localVarRequestOptions.formData = localVarFormParams;
+                }
+                else {
+                    localVarRequestOptions.form = localVarFormParams;
+                }
+            }
+            return new Promise((resolve, reject) => {
+                localVarRequest(localVarRequestOptions, (error, response, body) => {
+                    if (error) {
+                        reject(error);
+                    }
+                    else {
+                        body = models_1.ObjectSerializer.deserialize(body, "StartJobResponse");
+                        const errString = `Failed to call "${localVarPath}", returned ${response.statusCode}: ` + response.body;
+                        if (typeof body.success === 'boolean' && !body.success) {
+                            reject(new Error(body.error || errString));
+                        }
+                        else if (response.statusCode && response.statusCode >= 200 && response.statusCode <= 299) {
+                            resolve(body);
+                        }
+                        else {
+                            reject(errString);
+                        }
+                    }
+                });
+            });
+        });
+    }
+    /**
+     * Save input / model configuration for a pretrained model. This overrides the current impulse. If you want to deploy a pretrained model from the API, see `startDeployPretrainedModelJob`.
+     * @summary Save parameters for pretrained model
+     * @param projectId Project ID
+     * @param savePretrainedModelRequest
+     */
+    async savePretrainedModelParameters(projectId, savePretrainedModelRequest, options = { headers: {} }) {
+        const localVarPath = this.basePath + '/api/{projectId}/pretrained-model/save'
+            .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)));
+        let localVarQueryParameters = {};
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
+        const produces = ['application/json'];
+        // give precedence to 'application/json'
+        if (produces.indexOf('application/json') >= 0) {
+            localVarHeaderParams.Accept = 'application/json';
+        }
+        else {
+            localVarHeaderParams.Accept = produces.join(',');
+        }
+        let localVarFormParams = {};
+        // verify required parameter 'projectId' is not null or undefined
+        if (projectId === null || projectId === undefined) {
+            throw new Error('Required parameter projectId was null or undefined when calling savePretrainedModelParameters.');
+        }
+        Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
+        let localVarUseFormData = false;
+        let localVarRequestOptions = {
+            method: 'POST',
+            qs: localVarQueryParameters,
+            headers: localVarHeaderParams,
+            uri: localVarPath,
+            useQuerystring: this._useQuerystring,
+            agentOptions: { keepAlive: false },
+            json: true,
+            body: models_1.ObjectSerializer.serialize(savePretrainedModelRequest, "SavePretrainedModelRequest")
+        };
+        let authenticationPromise = Promise.resolve();
+        authenticationPromise = authenticationPromise.then(() => this.authentications.ApiKeyAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTHttpHeaderAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.default.applyToRequest(localVarRequestOptions));
+        return authenticationPromise.then(() => {
+            if (Object.keys(localVarFormParams).length) {
+                if (localVarUseFormData) {
+                    localVarRequestOptions.formData = localVarFormParams;
+                }
+                else {
+                    localVarRequestOptions.form = localVarFormParams;
+                }
+            }
+            return new Promise((resolve, reject) => {
+                localVarRequest(localVarRequestOptions, (error, response, body) => {
+                    if (error) {
+                        reject(error);
+                    }
+                    else {
+                        body = models_1.ObjectSerializer.deserialize(body, "GenericApiResponse");
+                        const errString = `Failed to call "${localVarPath}", returned ${response.statusCode}: ` + response.body;
+                        if (typeof body.success === 'boolean' && !body.success) {
+                            reject(new Error(body.error || errString));
+                        }
+                        else if (response.statusCode && response.statusCode >= 200 && response.statusCode <= 299) {
+                            resolve(body);
+                        }
+                        else {
+                            reject(errString);
+                        }
+                    }
+                });
+            });
+        });
+    }
     /**
      * Configure the anomaly block, such as its minimum confidence score. Use the impulse blocks to find the learnId.
      * @summary Anomaly settings
@@ -1018,7 +1362,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/json'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -1041,6 +1387,7 @@ class LearnApi {
             throw new Error('Required parameter setAnomalyParameterRequest was null or undefined when calling setAnomaly.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'POST',
@@ -1100,7 +1447,9 @@ class LearnApi {
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/json'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -1123,6 +1472,7 @@ class LearnApi {
             throw new Error('Required parameter setKerasParameterRequest was null or undefined when calling setKeras.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
         let localVarRequestOptions = {
             method: 'POST',
@@ -1170,6 +1520,81 @@ class LearnApi {
             });
         });
     }
+    /**
+     * Test out a pretrained model (using raw features) - upload first via  `uploadPretrainedModel`. If you want to deploy a pretrained model from the API, see `startDeployPretrainedModelJob`.
+     * @summary Test pretrained model
+     * @param projectId Project ID
+     * @param testPretrainedModelRequest
+     */
+    async testPretrainedModel(projectId, testPretrainedModelRequest, options = { headers: {} }) {
+        const localVarPath = this.basePath + '/api/{projectId}/pretrained-model/test'
+            .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)));
+        let localVarQueryParameters = {};
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
+        const produces = ['application/json'];
+        // give precedence to 'application/json'
+        if (produces.indexOf('application/json') >= 0) {
+            localVarHeaderParams.Accept = 'application/json';
+        }
+        else {
+            localVarHeaderParams.Accept = produces.join(',');
+        }
+        let localVarFormParams = {};
+        // verify required parameter 'projectId' is not null or undefined
+        if (projectId === null || projectId === undefined) {
+            throw new Error('Required parameter projectId was null or undefined when calling testPretrainedModel.');
+        }
+        Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
+        let localVarUseFormData = false;
+        let localVarRequestOptions = {
+            method: 'POST',
+            qs: localVarQueryParameters,
+            headers: localVarHeaderParams,
+            uri: localVarPath,
+            useQuerystring: this._useQuerystring,
+            agentOptions: { keepAlive: false },
+            json: true,
+            body: models_1.ObjectSerializer.serialize(testPretrainedModelRequest, "TestPretrainedModelRequest")
+        };
+        let authenticationPromise = Promise.resolve();
+        authenticationPromise = authenticationPromise.then(() => this.authentications.ApiKeyAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTHttpHeaderAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.default.applyToRequest(localVarRequestOptions));
+        return authenticationPromise.then(() => {
+            if (Object.keys(localVarFormParams).length) {
+                if (localVarUseFormData) {
+                    localVarRequestOptions.formData = localVarFormParams;
+                }
+                else {
+                    localVarRequestOptions.form = localVarFormParams;
+                }
+            }
+            return new Promise((resolve, reject) => {
+                localVarRequest(localVarRequestOptions, (error, response, body) => {
+                    if (error) {
+                        reject(error);
+                    }
+                    else {
+                        body = models_1.ObjectSerializer.deserialize(body, "TestPretrainedModelResponse");
+                        const errString = `Failed to call "${localVarPath}", returned ${response.statusCode}: ` + response.body;
+                        if (typeof body.success === 'boolean' && !body.success) {
+                            reject(new Error(body.error || errString));
+                        }
+                        else if (response.statusCode && response.statusCode >= 200 && response.statusCode <= 299) {
+                            resolve(body);
+                        }
+                        else {
+                            reject(errString);
+                        }
+                    }
+                });
+            });
+        });
+    }
     /**
      * Replace Keras block files with the contents of a zip. This is an internal API.
      * @summary Upload Keras files
@@ -1177,12 +1602,14 @@ class LearnApi {
      * @param learnId Learn Block ID, use the impulse functions to retrieve the ID
      * @param zip
      */
-    async uploadKerasFiles(projectId, learnId, zip, options = { headers: {} }) {
+    async uploadKerasFiles(projectId, learnId, params, options = { headers: {} }) {
         const localVarPath = this.basePath + '/api/{projectId}/training/keras/{learnId}/files'
             .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)))
             .replace('{' + 'learnId' + '}', encodeURIComponent(String(learnId)));
         let localVarQueryParameters = {};
-        let localVarHeaderParams = Object.assign({}, this.defaultHeaders);
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
         const produces = ['application/json'];
         // give precedence to 'application/json'
         if (produces.indexOf('application/json') >= 0) {
@@ -1201,13 +1628,14 @@ class LearnApi {
             throw new Error('Required parameter learnId was null or undefined when calling uploadKerasFiles.');
         }
         // verify required parameter 'zip' is not null or undefined
-        if (zip === null || zip === undefined) {
-            throw new Error('Required parameter zip was null or undefined when calling uploadKerasFiles.');
+        if (params.zip === null || params.zip === undefined) {
+            throw new Error('Required parameter params.zip was null or undefined when calling uploadKerasFiles.');
         }
         Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
         let localVarUseFormData = false;
-        if (zip !== undefined) {
-            localVarFormParams['zip'] = zip;
+        if (params.zip !== undefined) {
+            localVarFormParams['zip'] = params.zip;
         }
         localVarUseFormData = true;
         let localVarRequestOptions = {
@@ -1255,6 +1683,113 @@ class LearnApi {
             });
         });
     }
+    /**
+     * Upload a pretrained model and receive info back about the input/output tensors. If you want to deploy a pretrained model from the API, see `startDeployPretrainedModelJob`.
+     * @summary Upload a pretrained model
+     * @param projectId Project ID
+     * @param modelFile
+     * @param modelFileName
+     * @param modelFileType
+     * @param representativeFeatures
+     * @param device MCU used for calculating latency, query &#x60;latencyDevices&#x60; in &#x60;listProject&#x60; for a list of supported devices (and use the \\\&quot;mcu\\\&quot; property here). If this is kept empty then we\\\&#39;ll show an overview of multiple devices.
+     */
+    async uploadPretrainedModel(projectId, params, options = { headers: {} }) {
+        const localVarPath = this.basePath + '/api/{projectId}/pretrained-model/upload'
+            .replace('{' + 'projectId' + '}', encodeURIComponent(String(projectId)));
+        let localVarQueryParameters = {};
+        let localVarHeaderParams = Object.assign({
+            'User-Agent': 'edgeimpulse-api nodejs'
+        }, this.defaultHeaders);
+        const produces = ['application/json'];
+        // give precedence to 'application/json'
+        if (produces.indexOf('application/json') >= 0) {
+            localVarHeaderParams.Accept = 'application/json';
+        }
+        else {
+            localVarHeaderParams.Accept = produces.join(',');
+        }
+        let localVarFormParams = {};
+        // verify required parameter 'projectId' is not null or undefined
+        if (projectId === null || projectId === undefined) {
+            throw new Error('Required parameter projectId was null or undefined when calling uploadPretrainedModel.');
+        }
+        // verify required parameter 'modelFile' is not null or undefined
+        if (params.modelFile === null || params.modelFile === undefined) {
+            throw new Error('Required parameter params.modelFile was null or undefined when calling uploadPretrainedModel.');
+        }
+        // verify required parameter 'modelFileName' is not null or undefined
+        if (params.modelFileName === null || params.modelFileName === undefined) {
+            throw new Error('Required parameter params.modelFileName was null or undefined when calling uploadPretrainedModel.');
+        }
+        // verify required parameter 'modelFileType' is not null or undefined
+        if (params.modelFileType === null || params.modelFileType === undefined) {
+            throw new Error('Required parameter params.modelFileType was null or undefined when calling uploadPretrainedModel.');
+        }
+        Object.assign(localVarHeaderParams, options.headers);
+        Object.assign(localVarHeaderParams, this.opts.extraHeaders);
+        let localVarUseFormData = false;
+        if (params.modelFile !== undefined) {
+            localVarFormParams['modelFile'] = params.modelFile;
+        }
+        localVarUseFormData = true;
+        if (params.modelFileName !== undefined) {
+            localVarFormParams['modelFileName'] = models_1.ObjectSerializer.serialize(params.modelFileName, "string");
+        }
+        if (params.modelFileType !== undefined) {
+            localVarFormParams['modelFileType'] = models_1.ObjectSerializer.serialize(params.modelFileType, "string");
+        }
+        if (params.representativeFeatures !== undefined) {
+            localVarFormParams['representativeFeatures'] = params.representativeFeatures;
+        }
+        localVarUseFormData = true;
+        if (params.device !== undefined) {
+            localVarFormParams['device'] = models_1.ObjectSerializer.serialize(params.device, "string");
+        }
+        let localVarRequestOptions = {
+            method: 'POST',
+            qs: localVarQueryParameters,
+            headers: localVarHeaderParams,
+            uri: localVarPath,
+            useQuerystring: this._useQuerystring,
+            agentOptions: { keepAlive: false },
+            json: true,
+        };
+        let authenticationPromise = Promise.resolve();
+        authenticationPromise = authenticationPromise.then(() => this.authentications.ApiKeyAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.JWTHttpHeaderAuthentication.applyToRequest(localVarRequestOptions));
+        authenticationPromise = authenticationPromise.then(() => this.authentications.default.applyToRequest(localVarRequestOptions));
+        return authenticationPromise.then(() => {
+            if (Object.keys(localVarFormParams).length) {
+                if (localVarUseFormData) {
+                    localVarRequestOptions.formData = localVarFormParams;
+                }
+                else {
+                    localVarRequestOptions.form = localVarFormParams;
+                }
+            }
+            return new Promise((resolve, reject) => {
+                localVarRequest(localVarRequestOptions, (error, response, body) => {
+                    if (error) {
+                        reject(error);
+                    }
+                    else {
+                        body = models_1.ObjectSerializer.deserialize(body, "StartJobResponse");
+                        const errString = `Failed to call "${localVarPath}", returned ${response.statusCode}: ` + response.body;
+                        if (typeof body.success === 'boolean' && !body.success) {
+                            reject(new Error(body.error || errString));
+                        }
+                        else if (response.statusCode && response.statusCode >= 200 && response.statusCode <= 299) {
+                            resolve(body);
+                        }
+                        else {
+                            reject(errString);
+                        }
+                    }
+                });
+            });
+        });
+    }
 }
 exports.LearnApi = LearnApi;
 //# sourceMappingURL=learnApi.js.map