@things-factory/integration-label-studio 9.1.19

package/dist-server/service/label-studio-sso-service.js

@@ -0,0 +1,98 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LabelStudioSSOService = void 0;
+const tslib_1 = require("tslib");
+const axios_1 = tslib_1.__importDefault(require("axios"));
+const env_1 = require("@things-factory/env");
+/**
+ * Label Studio SSO Service
+ *
+ * Handles JWT token acquisition from Label Studio for SSO authentication.
+ * The client application requests JWT tokens from Label Studio's token API,
+ * then uses cookie-based authentication to automatically log users in.
+ */
+class LabelStudioSSOService {
+    /**
+     * Get Label Studio SSO configuration
+     */
+    static getConfig() {
+        const labelStudioConfig = env_1.config.get('labelStudio', {
+            serverUrl: 'http://localhost:8080',
+            apiToken: ''
+        });
+        return {
+            serverUrl: labelStudioConfig.serverUrl,
+            apiToken: labelStudioConfig.apiToken
+        };
+    }
+    /**
+     * Request JWT token from Label Studio for SSO authentication
+     *
+     * @param email User email address
+     * @returns JWT token and expiry time, or null if failed
+     *
+     * @example
+     * ```typescript
+     * const tokenData = await LabelStudioSSOService.getSSOToken('user@example.com')
+     * if (tokenData) {
+     *   console.log('Token:', tokenData.token)
+     *   console.log('Expires in:', tokenData.expires_in, 'seconds')
+     * }
+     * ```
+     */
+    static async getSSOToken(email) {
+        const { serverUrl, apiToken } = this.getConfig();
+        if (!apiToken) {
+            console.error('[LSS SSO] Label Studio API token not configured');
+            return null;
+        }
+        if (!email) {
+            console.error('[LSS SSO] Email is required');
+            return null;
+        }
+        try {
+            console.log(`[LSS SSO] Requesting JWT token for: ${email}`);
+            const response = await axios_1.default.post(`${serverUrl}/api/sso/token`, { email }, {
+                headers: {
+                    'Authorization': `Token ${apiToken}`,
+                    'Content-Type': 'application/json'
+                },
+                timeout: 5000
+            });
+            const { token, expires_in } = response.data;
+            console.log(`[LSS SSO] JWT token acquired for ${email} (expires in ${expires_in}s)`);
+            return { token, expires_in };
+        }
+        catch (error) {
+            if (error.response) {
+                console.error(`[LSS SSO] Label Studio API error: ${error.response.status}`, error.response.data);
+            }
+            else if (error.request) {
+                console.error('[LSS SSO] No response from Label Studio:', error.message);
+            }
+            else {
+                console.error('[LSS SSO] Request error:', error.message);
+            }
+            return null;
+        }
+    }
+    /**
+     * Verify SSO configuration is properly set up
+     *
+     * @returns True if configuration is valid
+     */
+    static verifyConfig() {
+        const { serverUrl, apiToken } = this.getConfig();
+        if (!serverUrl) {
+            console.error('[LSS SSO] Label Studio server URL not configured');
+            return false;
+        }
+        if (!apiToken) {
+            console.error('[LSS SSO] Label Studio API token not configured');
+            return false;
+        }
+        return true;
+    }
+}
+exports.LabelStudioSSOService = LabelStudioSSOService;
+//# sourceMappingURL=label-studio-sso-service.js.map

package/dist-server/service/label-studio-sso-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"label-studio-sso-service.js","sourceRoot":"","sources":["../../server/service/label-studio-sso-service.ts"],"names":[],"mappings":";;;;AAAA,0DAAyB;AACzB,6CAA4C;AAE5C;;;;;;GAMG;AACH,MAAa,qBAAqB;IAChC;;OAEG;IACK,MAAM,CAAC,SAAS;QACtB,MAAM,iBAAiB,GAAG,YAAM,CAAC,GAAG,CAAC,aAAa,EAAE;YAClD,SAAS,EAAE,uBAAuB;YAClC,QAAQ,EAAE,EAAE;SACb,CAAC,CAAA;QAEF,OAAO;YACL,SAAS,EAAE,iBAAiB,CAAC,SAAS;YACtC,QAAQ,EAAE,iBAAiB,CAAC,QAAQ;SACrC,CAAA;IACH,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,KAAa;QACpC,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,CAAA;QAEhD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,CAAC,KAAK,CAAC,iDAAiD,CAAC,CAAA;YAChE,OAAO,IAAI,CAAA;QACb,CAAC;QAED,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAA;YAC5C,OAAO,IAAI,CAAA;QACb,CAAC;QAED,IAAI,CAAC;YACH,OAAO,CAAC,GAAG,CAAC,uCAAuC,KAAK,EAAE,CAAC,CAAA;YAE3D,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,IAAI,CAC/B,GAAG,SAAS,gBAAgB,EAC5B,EAAE,KAAK,EAAE,EACT;gBACE,OAAO,EAAE;oBACP,eAAe,EAAE,SAAS,QAAQ,EAAE;oBACpC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,OAAO,EAAE,IAAI;aACd,CACF,CAAA;YAED,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAA;YAE3C,OAAO,CAAC,GAAG,CAAC,oCAAoC,KAAK,gBAAgB,UAAU,IAAI,CAAC,CAAA;YAEpF,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,CAAA;QAC9B,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACnB,OAAO,CAAC,KAAK,CAAC,qCAAqC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;YAClG,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBACzB,OAAO,CAAC,KAAK,CAAC,0CAA0C,EAAE,KAAK,CAAC,OAAO,CAAC,CAAA;YAC1E,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,OAAO,CAAC,CAAA;YAC1D,CAAC;YAED,OAAO,IAAI,CAAA;QACb,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,MAAM,CAAC,YAAY;QACjB,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,CAAA;QAEhD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAA;YACjE,OAAO,KAAK,CAAA;QACd,CAAC;QAED,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,CAAC,KAAK,CAAC,iDAAiD,CAAC,CAAA;YAChE,OAAO,KAAK,CAAA;QACd,CAAC;QAED,OAAO,IAAI,CAAA;IACb,CAAC;CACF;AAjGD,sDAiGC","sourcesContent":["import axios from 'axios'\nimport { config } from '@things-factory/env'\n\n/**\n * Label Studio SSO Service\n *\n * Handles JWT token acquisition from Label Studio for SSO authentication.\n * The client application requests JWT tokens from Label Studio's token API,\n * then uses cookie-based authentication to automatically log users in.\n */\nexport class LabelStudioSSOService {\n  /**\n   * Get Label Studio SSO configuration\n   */\n  private static getConfig() {\n    const labelStudioConfig = config.get('labelStudio', {\n      serverUrl: 'http://localhost:8080',\n      apiToken: ''\n    })\n\n    return {\n      serverUrl: labelStudioConfig.serverUrl,\n      apiToken: labelStudioConfig.apiToken\n    }\n  }\n\n  /**\n   * Request JWT token from Label Studio for SSO authentication\n   *\n   * @param email User email address\n   * @returns JWT token and expiry time, or null if failed\n   *\n   * @example\n   * ```typescript\n   * const tokenData = await LabelStudioSSOService.getSSOToken('user@example.com')\n   * if (tokenData) {\n   *   console.log('Token:', tokenData.token)\n   *   console.log('Expires in:', tokenData.expires_in, 'seconds')\n   * }\n   * ```\n   */\n  static async getSSOToken(email: string): Promise<{ token: string; expires_in: number } | null> {\n    const { serverUrl, apiToken } = this.getConfig()\n\n    if (!apiToken) {\n      console.error('[LSS SSO] Label Studio API token not configured')\n      return null\n    }\n\n    if (!email) {\n      console.error('[LSS SSO] Email is required')\n      return null\n    }\n\n    try {\n      console.log(`[LSS SSO] Requesting JWT token for: ${email}`)\n\n      const response = await axios.post(\n        `${serverUrl}/api/sso/token`,\n        { email },\n        {\n          headers: {\n            'Authorization': `Token ${apiToken}`,\n            'Content-Type': 'application/json'\n          },\n          timeout: 5000\n        }\n      )\n\n      const { token, expires_in } = response.data\n\n      console.log(`[LSS SSO] JWT token acquired for ${email} (expires in ${expires_in}s)`)\n\n      return { token, expires_in }\n    } catch (error: any) {\n      if (error.response) {\n        console.error(`[LSS SSO] Label Studio API error: ${error.response.status}`, error.response.data)\n      } else if (error.request) {\n        console.error('[LSS SSO] No response from Label Studio:', error.message)\n      } else {\n        console.error('[LSS SSO] Request error:', error.message)\n      }\n\n      return null\n    }\n  }\n\n  /**\n   * Verify SSO configuration is properly set up\n   *\n   * @returns True if configuration is valid\n   */\n  static verifyConfig(): boolean {\n    const { serverUrl, apiToken } = this.getConfig()\n\n    if (!serverUrl) {\n      console.error('[LSS SSO] Label Studio server URL not configured')\n      return false\n    }\n\n    if (!apiToken) {\n      console.error('[LSS SSO] Label Studio API token not configured')\n      return false\n    }\n\n    return true\n  }\n}\n"]}

package/dist-server/service/ml/ml-backend-service.d.ts

@@ -0,0 +1,23 @@
+import { MLBackend, AddMLBackendInput } from '../../types/label-studio-types.js';
+export declare class MLBackendService {
+    /**
+     * Get ML backends for a project
+     */
+    labelStudioMLBackends(projectId: number, context: ResolverContext): Promise<MLBackend[]>;
+    /**
+     * Add ML backend to project
+     */
+    addMLBackendToProject(projectId: number, input: AddMLBackendInput, context: ResolverContext): Promise<MLBackend>;
+    /**
+     * Delete ML backend
+     */
+    deleteMLBackend(mlBackendId: number, context: ResolverContext): Promise<boolean>;
+    /**
+     * Trigger predictions for tasks
+     */
+    triggerLabelStudioPredictions(projectId: number, context: ResolverContext, taskIds?: number[]): Promise<boolean>;
+    /**
+     * Train ML model
+     */
+    trainLabelStudioModel(mlBackendId: number, context: ResolverContext): Promise<boolean>;
+}

package/dist-server/service/ml/ml-backend-service.js

@@ -0,0 +1,153 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MLBackendService = void 0;
+const tslib_1 = require("tslib");
+const type_graphql_1 = require("type-graphql");
+const label_studio_api_client_js_1 = require("../../utils/label-studio-api-client.js");
+const label_studio_types_js_1 = require("../../types/label-studio-types.js");
+let MLBackendService = class MLBackendService {
+    /**
+     * Get ML backends for a project
+     */
+    async labelStudioMLBackends(projectId, context) {
+        try {
+            const backends = await label_studio_api_client_js_1.labelStudioApi.getMLBackends(projectId);
+            return backends.map(backend => ({
+                id: backend.id,
+                url: backend.url,
+                title: backend.title,
+                isInteractive: backend.is_interactive || false,
+                modelVersion: backend.model_version || 'unknown'
+            }));
+        }
+        catch (error) {
+            console.error(`Failed to fetch ML backends for project ${projectId}:`, error);
+            throw new Error(`Failed to fetch ML backends: ${error.message}`);
+        }
+    }
+    /**
+     * Add ML backend to project
+     */
+    async addMLBackendToProject(projectId, input, context) {
+        try {
+            const backend = await label_studio_api_client_js_1.labelStudioApi.addMLBackend({
+                project: projectId,
+                url: input.url,
+                title: input.title,
+                is_interactive: input.isInteractive || false
+            });
+            return {
+                id: backend.id,
+                url: backend.url,
+                title: backend.title,
+                isInteractive: backend.is_interactive || false,
+                modelVersion: backend.model_version || 'unknown'
+            };
+        }
+        catch (error) {
+            console.error(`Failed to add ML backend to project ${projectId}:`, error);
+            throw new Error(`Failed to add ML backend: ${error.message}`);
+        }
+    }
+    /**
+     * Delete ML backend
+     */
+    async deleteMLBackend(mlBackendId, context) {
+        try {
+            await label_studio_api_client_js_1.labelStudioApi.deleteMLBackend(mlBackendId);
+            return true;
+        }
+        catch (error) {
+            console.error(`Failed to delete ML backend ${mlBackendId}:`, error);
+            throw new Error(`Failed to delete ML backend: ${error.message}`);
+        }
+    }
+    /**
+     * Trigger predictions for tasks
+     */
+    async triggerLabelStudioPredictions(projectId, context, taskIds) {
+        try {
+            await label_studio_api_client_js_1.labelStudioApi.triggerPredictions(projectId, taskIds);
+            return true;
+        }
+        catch (error) {
+            console.error(`Failed to trigger predictions for project ${projectId}:`, error);
+            throw new Error(`Failed to trigger predictions: ${error.message}`);
+        }
+    }
+    /**
+     * Train ML model
+     */
+    async trainLabelStudioModel(mlBackendId, context) {
+        try {
+            await label_studio_api_client_js_1.labelStudioApi.trainModel(mlBackendId);
+            return true;
+        }
+        catch (error) {
+            console.error(`Failed to train model ${mlBackendId}:`, error);
+            throw new Error(`Failed to train model: ${error.message}`);
+        }
+    }
+};
+exports.MLBackendService = MLBackendService;
+tslib_1.__decorate([
+    (0, type_graphql_1.Query)(returns => [label_studio_types_js_1.MLBackend], {
+        description: 'Get all ML backends connected to a Label Studio project'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "query")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('projectId', type => type_graphql_1.Int)),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], MLBackendService.prototype, "labelStudioMLBackends", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => label_studio_types_js_1.MLBackend, {
+        description: 'Add ML backend to Label Studio project for predictions and training'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('projectId', type => type_graphql_1.Int)),
+    tslib_1.__param(1, (0, type_graphql_1.Arg)('input')),
+    tslib_1.__param(2, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, label_studio_types_js_1.AddMLBackendInput, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], MLBackendService.prototype, "addMLBackendToProject", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => Boolean, {
+        description: 'Remove ML backend from project'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('mlBackendId', type => type_graphql_1.Int)),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], MLBackendService.prototype, "deleteMLBackend", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => Boolean, {
+        description: 'Trigger ML predictions for tasks in a project'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('projectId', type => type_graphql_1.Int)),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__param(2, (0, type_graphql_1.Arg)('taskIds', type => [type_graphql_1.Int], { nullable: true })),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, Object, Array]),
+    tslib_1.__metadata("design:returntype", Promise)
+], MLBackendService.prototype, "triggerLabelStudioPredictions", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => Boolean, {
+        description: 'Trigger ML model training with current annotations'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('mlBackendId', type => type_graphql_1.Int)),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], MLBackendService.prototype, "trainLabelStudioModel", null);
+exports.MLBackendService = MLBackendService = tslib_1.__decorate([
+    (0, type_graphql_1.Resolver)()
+], MLBackendService);
+//# sourceMappingURL=ml-backend-service.js.map

package/dist-server/service/ml/ml-backend-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ml-backend-service.js","sourceRoot":"","sources":["../../../server/service/ml/ml-backend-service.ts"],"names":[],"mappings":";;;;AAAA,+CAAkF;AAClF,uFAAuE;AACvE,6EAAgF;AAGzE,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAC3B;;OAEG;IAKG,AAAN,KAAK,CAAC,qBAAqB,CACM,SAAiB,EACzC,OAAwB;QAE/B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,2CAAc,CAAC,aAAa,CAAC,SAAS,CAAC,CAAA;YAE9D,OAAO,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC9B,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,GAAG,EAAE,OAAO,CAAC,GAAG;gBAChB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,aAAa,EAAE,OAAO,CAAC,cAAc,IAAI,KAAK;gBAC9C,YAAY,EAAE,OAAO,CAAC,aAAa,IAAI,SAAS;aACjD,CAAC,CAAC,CAAA;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2CAA2C,SAAS,GAAG,EAAE,KAAK,CAAC,CAAA;YAC7E,MAAM,IAAI,KAAK,CAAC,gCAAgC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QAClE,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,qBAAqB,CACM,SAAiB,EAClC,KAAwB,EAC/B,OAAwB;QAE/B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,2CAAc,CAAC,YAAY,CAAC;gBAChD,OAAO,EAAE,SAAS;gBAClB,GAAG,EAAE,KAAK,CAAC,GAAG;gBACd,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,cAAc,EAAE,KAAK,CAAC,aAAa,IAAI,KAAK;aAC7C,CAAC,CAAA;YAEF,OAAO;gBACL,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,GAAG,EAAE,OAAO,CAAC,GAAG;gBAChB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,aAAa,EAAE,OAAO,CAAC,cAAc,IAAI,KAAK;gBAC9C,YAAY,EAAE,OAAO,CAAC,aAAa,IAAI,SAAS;aACjD,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uCAAuC,SAAS,GAAG,EAAE,KAAK,CAAC,CAAA;YACzE,MAAM,IAAI,KAAK,CAAC,6BAA6B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QAC/D,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,eAAe,CACc,WAAmB,EAC7C,OAAwB;QAE/B,IAAI,CAAC;YACH,MAAM,2CAAc,CAAC,eAAe,CAAC,WAAW,CAAC,CAAA;YACjD,OAAO,IAAI,CAAA;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,WAAW,GAAG,EAAE,KAAK,CAAC,CAAA;YACnE,MAAM,IAAI,KAAK,CAAC,gCAAgC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QAClE,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,6BAA6B,CACF,SAAiB,EACzC,OAAwB,EACoB,OAAkB;QAErE,IAAI,CAAC;YACH,MAAM,2CAAc,CAAC,kBAAkB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;YAC3D,OAAO,IAAI,CAAA;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,6CAA6C,SAAS,GAAG,EAAE,KAAK,CAAC,CAAA;YAC/E,MAAM,IAAI,KAAK,CAAC,kCAAkC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QACpE,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,qBAAqB,CACQ,WAAmB,EAC7C,OAAwB;QAE/B,IAAI,CAAC;YACH,MAAM,2CAAc,CAAC,UAAU,CAAC,WAAW,CAAC,CAAA;YAC5C,OAAO,IAAI,CAAA;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,WAAW,GAAG,EAAE,KAAK,CAAC,CAAA;YAC7D,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QAC5D,CAAC;IACH,CAAC;CACF,CAAA;AAzHY,4CAAgB;AAQrB;IAJL,IAAA,oBAAK,EAAC,OAAO,CAAC,EAAE,CAAC,CAAC,iCAAS,CAAC,EAAE;QAC7B,WAAW,EAAE,yDAAyD;KACvE,CAAC;IACD,IAAA,wBAAS,EAAC,0DAA0D,CAAC;IAEnE,mBAAA,IAAA,kBAAG,EAAC,WAAW,EAAE,IAAI,CAAC,EAAE,CAAC,kBAAG,CAAC,CAAA;IAC7B,mBAAA,IAAA,kBAAG,GAAE,CAAA;;;;6DAgBP;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,iCAAS,EAAE;QAC9B,WAAW,EAAE,qEAAqE;KACnF,CAAC;IACD,IAAA,wBAAS,EAAC,6DAA6D,CAAC;IAEtE,mBAAA,IAAA,kBAAG,EAAC,WAAW,EAAE,IAAI,CAAC,EAAE,CAAC,kBAAG,CAAC,CAAA;IAC7B,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IACZ,mBAAA,IAAA,kBAAG,GAAE,CAAA;;qDADe,yCAAiB;;6DAsBvC;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE;QAC5B,WAAW,EAAE,gCAAgC;KAC9C,CAAC;IACD,IAAA,wBAAS,EAAC,6DAA6D,CAAC;IAEtE,mBAAA,IAAA,kBAAG,EAAC,aAAa,EAAE,IAAI,CAAC,EAAE,CAAC,kBAAG,CAAC,CAAA;IAC/B,mBAAA,IAAA,kBAAG,GAAE,CAAA;;;;uDASP;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE;QAC5B,WAAW,EAAE,+CAA+C;KAC7D,CAAC;IACD,IAAA,wBAAS,EAAC,6DAA6D,CAAC;IAEtE,mBAAA,IAAA,kBAAG,EAAC,WAAW,EAAE,IAAI,CAAC,EAAE,CAAC,kBAAG,CAAC,CAAA;IAC7B,mBAAA,IAAA,kBAAG,GAAE,CAAA;IACL,mBAAA,IAAA,kBAAG,EAAC,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC,kBAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAA;;;;qEASnD;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE;QAC5B,WAAW,EAAE,oDAAoD;KAClE,CAAC;IACD,IAAA,wBAAS,EAAC,6DAA6D,CAAC;IAEtE,mBAAA,IAAA,kBAAG,EAAC,aAAa,EAAE,IAAI,CAAC,EAAE,CAAC,kBAAG,CAAC,CAAA;IAC/B,mBAAA,IAAA,kBAAG,GAAE,CAAA;;;;6DASP;2BAxHU,gBAAgB;IAD5B,IAAA,uBAAQ,GAAE;GACE,gBAAgB,CAyH5B","sourcesContent":["import { Resolver, Mutation, Query, Arg, Ctx, Int, Directive } from 'type-graphql'\nimport { labelStudioApi } from '../../utils/label-studio-api-client.js'\nimport { MLBackend, AddMLBackendInput } from '../../types/label-studio-types.js'\n\n@Resolver()\nexport class MLBackendService {\n  /**\n   * Get ML backends for a project\n   */\n  @Query(returns => [MLBackend], {\n    description: 'Get all ML backends connected to a Label Studio project'\n  })\n  @Directive('@privilege(category: \"label-studio\", privilege: \"query\")')\n  async labelStudioMLBackends(\n    @Arg('projectId', type => Int) projectId: number,\n    @Ctx() context: ResolverContext\n  ): Promise<MLBackend[]> {\n    try {\n      const backends = await labelStudioApi.getMLBackends(projectId)\n\n      return backends.map(backend => ({\n        id: backend.id,\n        url: backend.url,\n        title: backend.title,\n        isInteractive: backend.is_interactive || false,\n        modelVersion: backend.model_version || 'unknown'\n      }))\n    } catch (error) {\n      console.error(`Failed to fetch ML backends for project ${projectId}:`, error)\n      throw new Error(`Failed to fetch ML backends: ${error.message}`)\n    }\n  }\n\n  /**\n   * Add ML backend to project\n   */\n  @Mutation(returns => MLBackend, {\n    description: 'Add ML backend to Label Studio project for predictions and training'\n  })\n  @Directive('@privilege(category: \"label-studio\", privilege: \"mutation\")')\n  async addMLBackendToProject(\n    @Arg('projectId', type => Int) projectId: number,\n    @Arg('input') input: AddMLBackendInput,\n    @Ctx() context: ResolverContext\n  ): Promise<MLBackend> {\n    try {\n      const backend = await labelStudioApi.addMLBackend({\n        project: projectId,\n        url: input.url,\n        title: input.title,\n        is_interactive: input.isInteractive || false\n      })\n\n      return {\n        id: backend.id,\n        url: backend.url,\n        title: backend.title,\n        isInteractive: backend.is_interactive || false,\n        modelVersion: backend.model_version || 'unknown'\n      }\n    } catch (error) {\n      console.error(`Failed to add ML backend to project ${projectId}:`, error)\n      throw new Error(`Failed to add ML backend: ${error.message}`)\n    }\n  }\n\n  /**\n   * Delete ML backend\n   */\n  @Mutation(returns => Boolean, {\n    description: 'Remove ML backend from project'\n  })\n  @Directive('@privilege(category: \"label-studio\", privilege: \"mutation\")')\n  async deleteMLBackend(\n    @Arg('mlBackendId', type => Int) mlBackendId: number,\n    @Ctx() context: ResolverContext\n  ): Promise<boolean> {\n    try {\n      await labelStudioApi.deleteMLBackend(mlBackendId)\n      return true\n    } catch (error) {\n      console.error(`Failed to delete ML backend ${mlBackendId}:`, error)\n      throw new Error(`Failed to delete ML backend: ${error.message}`)\n    }\n  }\n\n  /**\n   * Trigger predictions for tasks\n   */\n  @Mutation(returns => Boolean, {\n    description: 'Trigger ML predictions for tasks in a project'\n  })\n  @Directive('@privilege(category: \"label-studio\", privilege: \"mutation\")')\n  async triggerLabelStudioPredictions(\n    @Arg('projectId', type => Int) projectId: number,\n    @Ctx() context: ResolverContext,\n    @Arg('taskIds', type => [Int], { nullable: true }) taskIds?: number[]\n  ): Promise<boolean> {\n    try {\n      await labelStudioApi.triggerPredictions(projectId, taskIds)\n      return true\n    } catch (error) {\n      console.error(`Failed to trigger predictions for project ${projectId}:`, error)\n      throw new Error(`Failed to trigger predictions: ${error.message}`)\n    }\n  }\n\n  /**\n   * Train ML model\n   */\n  @Mutation(returns => Boolean, {\n    description: 'Trigger ML model training with current annotations'\n  })\n  @Directive('@privilege(category: \"label-studio\", privilege: \"mutation\")')\n  async trainLabelStudioModel(\n    @Arg('mlBackendId', type => Int) mlBackendId: number,\n    @Ctx() context: ResolverContext\n  ): Promise<boolean> {\n    try {\n      await labelStudioApi.trainModel(mlBackendId)\n      return true\n    } catch (error) {\n      console.error(`Failed to train model ${mlBackendId}:`, error)\n      throw new Error(`Failed to train model: ${error.message}`)\n    }\n  }\n}\n"]}

package/dist-server/service/prediction/prediction-management.d.ts

@@ -0,0 +1,32 @@
+import { LabelStudioPrediction, PredictionInput, PredictionImportResult, BulkPredictionInput } from '../../types/label-studio-types.js';
+export declare class PredictionManagement {
+    /**
+     * Get predictions for a task
+     */
+    labelStudioTaskPredictions(taskId: number, context: ResolverContext): Promise<LabelStudioPrediction[]>;
+    /**
+     * Get predictions for a project
+     */
+    labelStudioProjectPredictions(projectId: number, context: ResolverContext): Promise<LabelStudioPrediction[]>;
+    /**
+     * Get single prediction by ID
+     */
+    labelStudioPrediction(predictionId: number, context: ResolverContext): Promise<LabelStudioPrediction | null>;
+    /**
+     * Create a prediction for a task
+     */
+    createLabelStudioPrediction(input: PredictionInput, context: ResolverContext): Promise<LabelStudioPrediction>;
+    /**
+     * Create predictions in bulk
+     */
+    createBulkPredictions(predictions: BulkPredictionInput[], context: ResolverContext): Promise<PredictionImportResult>;
+    /**
+     * Import predictions for a project
+     * This is the recommended way to bulk import AI model predictions
+     */
+    importPredictionsToProject(projectId: number, predictions: BulkPredictionInput[], context: ResolverContext): Promise<PredictionImportResult>;
+    /**
+     * Delete prediction
+     */
+    deleteLabelStudioPrediction(predictionId: number, context: ResolverContext): Promise<boolean>;
+}

package/dist-server/service/prediction/prediction-management.js

@@ -0,0 +1,299 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PredictionManagement = void 0;
+const tslib_1 = require("tslib");
+const type_graphql_1 = require("type-graphql");
+const label_studio_api_client_js_1 = require("../../utils/label-studio-api-client.js");
+const label_studio_types_js_1 = require("../../types/label-studio-types.js");
+let PredictionManagement = class PredictionManagement {
+    /**
+     * Get predictions for a task
+     */
+    async labelStudioTaskPredictions(taskId, context) {
+        try {
+            const predictions = await label_studio_api_client_js_1.labelStudioApi.getPredictions(taskId);
+            return predictions.map((prediction) => ({
+                id: prediction.id,
+                taskId: prediction.task || taskId,
+                result: JSON.stringify(prediction.result),
+                score: prediction.score || undefined,
+                modelVersion: prediction.model_version || undefined,
+                createdAt: prediction.created_at ? new Date(prediction.created_at) : undefined
+            }));
+        }
+        catch (error) {
+            console.error(`Failed to fetch predictions for task ${taskId}:`, error);
+            throw new Error(`Failed to fetch predictions: ${error.message}`);
+        }
+    }
+    /**
+     * Get predictions for a project
+     */
+    async labelStudioProjectPredictions(projectId, context) {
+        try {
+            const predictions = await label_studio_api_client_js_1.labelStudioApi.getProjectPredictions(projectId);
+            return predictions.map((prediction) => ({
+                id: prediction.id,
+                taskId: prediction.task,
+                result: JSON.stringify(prediction.result),
+                score: prediction.score || undefined,
+                modelVersion: prediction.model_version || undefined,
+                createdAt: prediction.created_at ? new Date(prediction.created_at) : undefined
+            }));
+        }
+        catch (error) {
+            console.error(`Failed to fetch predictions for project ${projectId}:`, error);
+            throw new Error(`Failed to fetch predictions: ${error.message}`);
+        }
+    }
+    /**
+     * Get single prediction by ID
+     */
+    async labelStudioPrediction(predictionId, context) {
+        try {
+            const prediction = await label_studio_api_client_js_1.labelStudioApi.getPrediction(predictionId);
+            if (!prediction) {
+                return null;
+            }
+            return {
+                id: prediction.id,
+                taskId: prediction.task,
+                result: JSON.stringify(prediction.result),
+                score: prediction.score || undefined,
+                modelVersion: prediction.model_version || undefined,
+                createdAt: prediction.created_at ? new Date(prediction.created_at) : undefined
+            };
+        }
+        catch (error) {
+            console.error(`Failed to fetch prediction ${predictionId}:`, error);
+            return null;
+        }
+    }
+    /**
+     * Create a prediction for a task
+     */
+    async createLabelStudioPrediction(input, context) {
+        try {
+            // Parse result JSON
+            const result = JSON.parse(input.result);
+            // Create prediction
+            const prediction = await label_studio_api_client_js_1.labelStudioApi.createPrediction({
+                task: input.taskId,
+                result,
+                score: input.score,
+                model_version: input.modelVersion
+            });
+            return {
+                id: prediction.id,
+                taskId: prediction.task,
+                result: JSON.stringify(prediction.result),
+                score: prediction.score || undefined,
+                modelVersion: prediction.model_version || undefined,
+                createdAt: prediction.created_at ? new Date(prediction.created_at) : undefined
+            };
+        }
+        catch (error) {
+            console.error(`Failed to create prediction for task ${input.taskId}:`, error);
+            throw new Error(`Failed to create prediction: ${error.message}`);
+        }
+    }
+    /**
+     * Create predictions in bulk
+     */
+    async createBulkPredictions(predictions, context) {
+        const errors = [];
+        let created = 0;
+        let failed = 0;
+        try {
+            // Parse and validate predictions
+            const parsedPredictions = predictions.map((pred, index) => {
+                try {
+                    return {
+                        task: pred.taskId,
+                        result: JSON.parse(pred.result),
+                        score: pred.score,
+                        model_version: pred.modelVersion
+                    };
+                }
+                catch (e) {
+                    failed++;
+                    errors.push(`Invalid JSON in prediction ${index}: ${pred.result}`);
+                    return null;
+                }
+            });
+            const validPredictions = parsedPredictions.filter(p => p !== null);
+            if (validPredictions.length === 0) {
+                return {
+                    created: 0,
+                    failed: predictions.length,
+                    errors
+                };
+            }
+            // Create predictions
+            const response = await label_studio_api_client_js_1.labelStudioApi.createPredictions(validPredictions);
+            created = response.created || validPredictions.length;
+            return {
+                created,
+                failed,
+                errors: errors.length > 0 ? errors : undefined
+            };
+        }
+        catch (error) {
+            console.error('Failed to create bulk predictions:', error);
+            throw new Error(`Failed to create predictions: ${error.message}`);
+        }
+    }
+    /**
+     * Import predictions for a project
+     * This is the recommended way to bulk import AI model predictions
+     */
+    async importPredictionsToProject(projectId, predictions, context) {
+        const errors = [];
+        let created = 0;
+        let failed = 0;
+        try {
+            // Parse and validate predictions
+            const parsedPredictions = predictions.map((pred, index) => {
+                try {
+                    return {
+                        task: pred.taskId,
+                        result: JSON.parse(pred.result),
+                        score: pred.score,
+                        model_version: pred.modelVersion
+                    };
+                }
+                catch (e) {
+                    failed++;
+                    errors.push(`Invalid JSON in prediction ${index}: ${pred.result}`);
+                    return null;
+                }
+            });
+            const validPredictions = parsedPredictions.filter(p => p !== null);
+            if (validPredictions.length === 0) {
+                return {
+                    created: 0,
+                    failed: predictions.length,
+                    errors
+                };
+            }
+            // Import predictions using project endpoint
+            const response = await label_studio_api_client_js_1.labelStudioApi.importPredictions(projectId, validPredictions);
+            // Parse response
+            if (response.created !== undefined) {
+                created = response.created;
+            }
+            else if (response.task_count !== undefined) {
+                created = response.task_count;
+            }
+            else if (Array.isArray(response)) {
+                created = response.length;
+            }
+            return {
+                created,
+                failed,
+                errors: errors.length > 0 ? errors : undefined
+            };
+        }
+        catch (error) {
+            console.error(`Failed to import predictions to project ${projectId}:`, error);
+            throw new Error(`Failed to import predictions: ${error.message}`);
+        }
+    }
+    /**
+     * Delete prediction
+     */
+    async deleteLabelStudioPrediction(predictionId, context) {
+        try {
+            await label_studio_api_client_js_1.labelStudioApi.deletePrediction(predictionId);
+            return true;
+        }
+        catch (error) {
+            console.error(`Failed to delete prediction ${predictionId}:`, error);
+            throw new Error(`Failed to delete prediction: ${error.message}`);
+        }
+    }
+};
+exports.PredictionManagement = PredictionManagement;
+tslib_1.__decorate([
+    (0, type_graphql_1.Query)(returns => [label_studio_types_js_1.LabelStudioPrediction], {
+        description: 'Get all predictions for a Label Studio task'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "query")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('taskId', type => type_graphql_1.Int)),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], PredictionManagement.prototype, "labelStudioTaskPredictions", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Query)(returns => [label_studio_types_js_1.LabelStudioPrediction], {
+        description: 'Get all predictions for a Label Studio project'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "query")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('projectId', type => type_graphql_1.Int)),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], PredictionManagement.prototype, "labelStudioProjectPredictions", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Query)(returns => label_studio_types_js_1.LabelStudioPrediction, {
+        description: 'Get a single prediction by ID',
+        nullable: true
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "query")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('predictionId', type => type_graphql_1.Int)),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], PredictionManagement.prototype, "labelStudioPrediction", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => label_studio_types_js_1.LabelStudioPrediction, {
+        description: 'Create a prediction for a Label Studio task'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('input')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [label_studio_types_js_1.PredictionInput, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], PredictionManagement.prototype, "createLabelStudioPrediction", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => label_studio_types_js_1.PredictionImportResult, {
+        description: 'Create multiple predictions at once'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('predictions', type => [label_studio_types_js_1.BulkPredictionInput])),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Array, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], PredictionManagement.prototype, "createBulkPredictions", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => label_studio_types_js_1.PredictionImportResult, {
+        description: 'Import predictions for a Label Studio project in bulk'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('projectId', type => type_graphql_1.Int)),
+    tslib_1.__param(1, (0, type_graphql_1.Arg)('predictions', type => [label_studio_types_js_1.BulkPredictionInput])),
+    tslib_1.__param(2, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, Array, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], PredictionManagement.prototype, "importPredictionsToProject", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => Boolean, {
+        description: 'Delete a prediction from Label Studio'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "label-studio", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('predictionId', type => type_graphql_1.Int)),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], PredictionManagement.prototype, "deleteLabelStudioPrediction", null);
+exports.PredictionManagement = PredictionManagement = tslib_1.__decorate([
+    (0, type_graphql_1.Resolver)()
+], PredictionManagement);
+//# sourceMappingURL=prediction-management.js.map