@sogni-ai/sogni-client 0.0.1

package/dist/Projects/index.d.ts

@@ -0,0 +1,38 @@
+import ApiGroup, { ApiConfig } from '../ApiGroup';
+import { EstimateRequest, ImageUrlParams, ProjectParams } from './types';
+import AvailableModels from './AvailableModels';
+import Project from './Project';
+declare class Projects extends ApiGroup {
+    readonly awailableModels: AvailableModels;
+    private projects;
+    constructor(config: ApiConfig);
+    private handleSwarmModels;
+    private handleJobState;
+    private handleJobProgress;
+    private handleJobResult;
+    private handleJobError;
+    private handleServerDisconnected;
+    /**
+     * Send new project request to the network. Returns project instance which can be used to track
+     * progress and get resulting images.
+     * @param data
+     */
+    create(data: ProjectParams): Promise<Project>;
+    /**
+     * Estimate project cost
+     * @param network - either 'fast' or 'relaxed' network
+     * @param model - model id
+     * @param imageCount - number of images to generate
+     * @param stepCount - number of steps
+     * @param previewCount - number of preview images to request
+     * @param cnEnabled - control network enabled
+     * @param denoiseStrength - denoise strength
+     */
+    estimateCost({ network, model, imageCount, stepCount, previewCount, cnEnabled, denoiseStrength }: EstimateRequest): Promise<{
+        token: string;
+        usd: string;
+    }>;
+    uploadUrl(params: ImageUrlParams): Promise<string>;
+    downloadUrl(params: ImageUrlParams): Promise<string>;
+}
+export default Projects;

package/dist/Projects/index.js

@@ -0,0 +1,224 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const ApiGroup_1 = __importDefault(require("../ApiGroup"));
+const models_json_1 = __importDefault(require("./models.json"));
+const AvailableModels_1 = __importDefault(require("./AvailableModels"));
+const Project_1 = __importDefault(require("./Project"));
+const createJobRequestMessage_1 = __importDefault(require("./createJobRequestMessage"));
+class Projects extends ApiGroup_1.default {
+    constructor(config) {
+        super(config);
+        this.awailableModels = new AvailableModels_1.default({ list: [], index: {} });
+        this.projects = [];
+        this.client.socket.on('swarmModels', this.handleSwarmModels.bind(this));
+        this.client.socket.on('jobState', this.handleJobState.bind(this));
+        this.client.socket.on('jobProgress', this.handleJobProgress.bind(this));
+        this.client.socket.on('jobError', this.handleJobError.bind(this));
+        this.client.socket.on('jobResult', this.handleJobResult.bind(this));
+        this.client.on('disconnected', this.handleServerDisconnected.bind(this));
+    }
+    handleSwarmModels(data) {
+        const modelsData = models_json_1.default.reduce((acc, model) => {
+            if (data[model.modelId]) {
+                const item = { id: model.modelId, name: model.modelShortName };
+                acc.index[item.id] = item;
+                acc.list.push(item);
+            }
+            return acc;
+        }, { list: [], index: {} });
+        this.awailableModels._update(modelsData);
+    }
+    handleJobState(data) {
+        const project = this.projects.find((p) => p.id === data.jobID);
+        if (!project) {
+            return;
+        }
+        switch (data.type) {
+            case 'queued':
+                project._update({
+                    status: 'queued',
+                    queuePosition: data.queuePosition
+                });
+                break;
+            case 'jobCompleted':
+                project._update({ status: 'completed' });
+                break;
+            case 'initiatingModel':
+            case 'jobStarted': {
+                const status = data.type === 'initiatingModel' ? 'initiating' : 'started';
+                const job = project.job(data.imgID);
+                if (job) {
+                    job._update({ status });
+                }
+                else {
+                    project._addJob({
+                        id: data.imgID,
+                        status,
+                        step: 0,
+                        stepCount: project.params.steps
+                    });
+                }
+                break;
+            }
+        }
+    }
+    handleJobProgress(data) {
+        const project = this.projects.find((p) => p.id === data.jobID);
+        if (!project) {
+            return;
+        }
+        let job = project.job(data.imgID);
+        if (job) {
+            job._update({
+                status: 'creating',
+                step: data.step,
+                stepCount: data.stepCount
+            });
+        }
+        else {
+            job = project._addJob({
+                id: data.imgID,
+                status: 'creating',
+                step: data.step,
+                stepCount: data.stepCount
+            });
+        }
+        if (data.hasImage) {
+            this.downloadUrl({
+                jobId: project.id,
+                imageId: job.id,
+                type: 'preview'
+            }).then((url) => {
+                job._update({ previewUrl: url });
+            });
+        }
+    }
+    handleJobResult(data) {
+        const project = this.projects.find((p) => p.id === data.jobID);
+        if (!project) {
+            return;
+        }
+        let job = project.job(data.imgID);
+        if (job) {
+            job._update({
+                status: 'completed',
+                step: data.performedStepCount
+            });
+        }
+        else {
+            job = project._addJob({
+                id: data.imgID,
+                status: 'completed',
+                step: data.performedStepCount,
+                stepCount: project.params.steps
+            });
+        }
+        this.downloadUrl({
+            jobId: project.id,
+            imageId: job.id,
+            type: 'complete'
+        }).then((url) => {
+            job === null || job === void 0 ? void 0 : job._update({ resultUrl: url });
+        });
+    }
+    handleJobError(data) {
+        const project = this.projects.find((p) => p.id === data.jobID);
+        if (!project) {
+            return;
+        }
+        if (!data.imgID) {
+            project._update({
+                status: 'failed',
+                error: {
+                    code: data.error,
+                    message: data.error_message
+                }
+            });
+            return;
+        }
+        const job = project.job(data.imgID);
+        if (!job) {
+            project._addJob({
+                id: data.imgID,
+                status: 'failed',
+                step: 0,
+                stepCount: project.params.steps,
+                error: {
+                    code: data.error,
+                    message: data.error_message
+                }
+            });
+        }
+        else {
+            job._update({
+                status: 'failed',
+                error: {
+                    code: data.error,
+                    message: data.error_message
+                }
+            });
+        }
+    }
+    handleServerDisconnected() {
+        this.awailableModels._update({ list: [], index: {} });
+    }
+    /**
+     * Send new project request to the network. Returns project instance which can be used to track
+     * progress and get resulting images.
+     * @param data
+     */
+    create(data) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const project = new Project_1.default(Object.assign({}, data));
+            const request = (0, createJobRequestMessage_1.default)(project.id, data);
+            yield this.client.socket.send('jobRequest', request);
+            this.projects.push(project);
+            return project;
+        });
+    }
+    /**
+     * Estimate project cost
+     * @param network - either 'fast' or 'relaxed' network
+     * @param model - model id
+     * @param imageCount - number of images to generate
+     * @param stepCount - number of steps
+     * @param previewCount - number of preview images to request
+     * @param cnEnabled - control network enabled
+     * @param denoiseStrength - denoise strength
+     */
+    estimateCost(_a) {
+        return __awaiter(this, arguments, void 0, function* ({ network, model, imageCount, stepCount, previewCount, cnEnabled, denoiseStrength }) {
+            const r = yield this.client.socket.get(`/api/v1/job/estimate/${network}/${model}/${imageCount}/${stepCount}/${previewCount}/${cnEnabled ? 1 : 0}/${denoiseStrength || 0}`);
+            return {
+                token: r.quote.project.costInToken,
+                usd: r.quote.project.costInUSD
+            };
+        });
+    }
+    uploadUrl(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const r = yield this.client.rest.get(`/v1/image/uploadUrl`, params);
+            return r.data.uploadUrl;
+        });
+    }
+    downloadUrl(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const r = yield this.client.rest.get(`/v1/image/downloadUrl`, params);
+            return r.data.downloadUrl;
+        });
+    }
+}
+exports.default = Projects;
+//# sourceMappingURL=index.js.map

package/dist/Projects/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/Projects/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAAA,2DAAkD;AAClD,gEAAmC;AASnC,wEAAyE;AACzE,wDAAgC;AAChC,wFAAgE;AAIhE,MAAM,QAAS,SAAQ,kBAAQ;IAI7B,YAAY,MAAiB;QAC3B,KAAK,CAAC,MAAM,CAAC,CAAC;QAJP,oBAAe,GAAG,IAAI,yBAAe,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QAChE,aAAQ,GAAc,EAAE,CAAC;QAI/B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,aAAa,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACxE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,aAAa,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACxE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACpE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,cAAc,EAAE,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3E,CAAC;IAEO,iBAAiB,CAAC,IAAmC;QAC3D,MAAM,UAAU,GAAG,qBAAM,CAAC,MAAM,CAC9B,CAAC,GAAwB,EAAE,KAAK,EAAE,EAAE;YAClC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;gBACxB,MAAM,IAAI,GAAG,EAAE,EAAE,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC;gBAC/D,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;gBAC1B,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CACxB,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC;IAEO,cAAc,CAAC,IAAkB;QACvC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QACD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,QAAQ;gBACX,OAAO,CAAC,OAAO,CAAC;oBACd,MAAM,EAAE,QAAQ;oBAChB,aAAa,EAAE,IAAI,CAAC,aAAa;iBAClC,CAAC,CAAC;gBACH,MAAM;YACR,KAAK,cAAc;gBACjB,OAAO,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;gBACzC,MAAM;YACR,KAAK,iBAAiB,CAAC;YACvB,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC1E,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACpC,IAAI,GAAG,EAAE,CAAC;oBACR,GAAG,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC1B,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,OAAO,CAAC;wBACd,EAAE,EAAE,IAAI,CAAC,KAAK;wBACd,MAAM;wBACN,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;qBAChC,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,IAAqB;QAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QACD,IAAI,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,IAAI,GAAG,EAAE,CAAC;YACR,GAAG,CAAC,OAAO,CAAC;gBACV,MAAM,EAAE,UAAU;gBAClB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,SAAS,EAAE,IAAI,CAAC,SAAS;aAC1B,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC;gBACpB,EAAE,EAAE,IAAI,CAAC,KAAK;gBACd,MAAM,EAAE,UAAU;gBAClB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,SAAS,EAAE,IAAI,CAAC,SAAS;aAC1B,CAAC,CAAC;QACL,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,WAAW,CAAC;gBACf,KAAK,EAAE,OAAO,CAAC,EAAE;gBACjB,OAAO,EAAE,GAAG,CAAC,EAAE;gBACf,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE;gBACd,GAAG,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;YACnC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,eAAe,CAAC,IAAmB;QACzC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QACD,IAAI,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,IAAI,GAAG,EAAE,CAAC;YACR,GAAG,CAAC,OAAO,CAAC;gBACV,MAAM,EAAE,WAAW;gBACnB,IAAI,EAAE,IAAI,CAAC,kBAAkB;aAC9B,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC;gBACpB,EAAE,EAAE,IAAI,CAAC,KAAK;gBACd,MAAM,EAAE,WAAW;gBACnB,IAAI,EAAE,IAAI,CAAC,kBAAkB;gBAC7B,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;aAChC,CAAC,CAAC;QACL,CAAC;QACD,IAAI,CAAC,WAAW,CAAC;YACf,KAAK,EAAE,OAAO,CAAC,EAAE;YACjB,OAAO,EAAE,GAAG,CAAC,EAAE;YACf,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE;YACd,GAAG,aAAH,GAAG,uBAAH,GAAG,CAAE,OAAO,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,IAAkB;QACvC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,OAAO,CAAC;gBACd,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE;oBACL,IAAI,EAAE,IAAI,CAAC,KAAK;oBAChB,OAAO,EAAE,IAAI,CAAC,aAAa;iBAC5B;aACF,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QACD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,OAAO,CAAC,OAAO,CAAC;gBACd,EAAE,EAAE,IAAI,CAAC,KAAK;gBACd,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,CAAC;gBACP,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;gBAC/B,KAAK,EAAE;oBACL,IAAI,EAAE,IAAI,CAAC,KAAK;oBAChB,OAAO,EAAE,IAAI,CAAC,aAAa;iBAC5B;aACF,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,GAAG,CAAC,OAAO,CAAC;gBACV,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE;oBACL,IAAI,EAAE,IAAI,CAAC,KAAK;oBAChB,OAAO,EAAE,IAAI,CAAC,aAAa;iBAC5B;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,wBAAwB;QAC9B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;IACxD,CAAC;IAED;;;;OAIG;IACG,MAAM,CAAC,IAAmB;;YAC9B,MAAM,OAAO,GAAG,IAAI,iBAAO,mBAAM,IAAI,EAAG,CAAC;YACzC,MAAM,OAAO,GAAG,IAAA,iCAAuB,EAAC,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YAC1D,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACrD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5B,OAAO,OAAO,CAAC;QACjB,CAAC;KAAA;IAED;;;;;;;;;OASG;IACG,YAAY;6DAAC,EACjB,OAAO,EACP,KAAK,EACL,UAAU,EACV,SAAS,EACT,YAAY,EACZ,SAAS,EACT,eAAe,EACC;YAChB,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CACpC,wBAAwB,OAAO,IAAI,KAAK,IAAI,UAAU,IAAI,SAAS,IAAI,YAAY,IAAI,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,eAAe,IAAI,CAAC,EAAE,CACnI,CAAC;YACF,OAAO;gBACL,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW;gBAClC,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS;aAC/B,CAAC;QACJ,CAAC;KAAA;IAEK,SAAS,CAAC,MAAsB;;YACpC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAClC,qBAAqB,EACrB,MAAM,CACP,CAAC;YACF,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;QAC1B,CAAC;KAAA;IAEK,WAAW,CAAC,MAAsB;;YACtC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAClC,uBAAuB,EACvB,MAAM,CACP,CAAC;YACF,OAAO,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;QAC5B,CAAC;KAAA;CACF;AAED,kBAAe,QAAQ,CAAC"}