@things-factory/integration-label-studio 9.1.19

package/dist-server/route/webhook.js

@@ -0,0 +1,248 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.webhookRouter = exports.WebhookAction = void 0;
+exports.registerWebhookHandler = registerWebhookHandler;
+exports.unregisterWebhookHandler = unregisterWebhookHandler;
+exports.clearWebhookHandlers = clearWebhookHandlers;
+const tslib_1 = require("tslib");
+const koa_router_1 = tslib_1.__importDefault(require("koa-router"));
+const webhookRouter = new koa_router_1.default();
+exports.webhookRouter = webhookRouter;
+/**
+ * Label Studio Webhook Event Types
+ */
+var WebhookAction;
+(function (WebhookAction) {
+    WebhookAction["ANNOTATION_CREATED"] = "ANNOTATION_CREATED";
+    WebhookAction["ANNOTATION_UPDATED"] = "ANNOTATION_UPDATED";
+    WebhookAction["ANNOTATION_DELETED"] = "ANNOTATION_DELETED";
+    WebhookAction["TASK_CREATED"] = "TASK_CREATED";
+    WebhookAction["TASK_UPDATED"] = "TASK_UPDATED";
+    WebhookAction["TASK_DELETED"] = "TASK_DELETED";
+    WebhookAction["PROJECT_UPDATED"] = "PROJECT_UPDATED";
+})(WebhookAction || (exports.WebhookAction = WebhookAction = {}));
+/**
+ * Registry for custom webhook handlers
+ * Multiple handlers can be registered per action
+ */
+const customHandlers = new Map();
+/**
+ * Register a custom webhook handler
+ * Handlers are executed in registration order
+ *
+ * @param action - Webhook action to handle
+ * @param handler - Handler function
+ *
+ * @example
+ * registerWebhookHandler(WebhookAction.ANNOTATION_CREATED, async (payload, ctx) => {
+ *   console.log('Custom handler:', payload.annotation?.id)
+ *   // Store annotation in database
+ *   // Trigger ML training
+ *   // Send notifications
+ * })
+ */
+function registerWebhookHandler(action, handler) {
+    if (!customHandlers.has(action)) {
+        customHandlers.set(action, []);
+    }
+    customHandlers.get(action).push(handler);
+}
+/**
+ * Unregister a webhook handler
+ */
+function unregisterWebhookHandler(action, handler) {
+    const handlers = customHandlers.get(action);
+    if (handlers) {
+        const index = handlers.indexOf(handler);
+        if (index > -1) {
+            handlers.splice(index, 1);
+        }
+    }
+}
+/**
+ * Clear all custom handlers for an action
+ */
+function clearWebhookHandlers(action) {
+    if (action) {
+        customHandlers.delete(action);
+    }
+    else {
+        customHandlers.clear();
+    }
+}
+/**
+ * Execute all registered handlers for an action
+ */
+async function executeCustomHandlers(action, payload, ctx) {
+    const handlers = customHandlers.get(action);
+    if (!handlers || handlers.length === 0) {
+        return;
+    }
+    // Execute all handlers in sequence
+    for (const handler of handlers) {
+        try {
+            await handler(payload, ctx);
+        }
+        catch (error) {
+            console.error(`[Webhook] Custom handler error for ${action}:`, error);
+            // Continue executing other handlers even if one fails
+        }
+    }
+}
+/**
+ * Handle annotation created event
+ */
+async function handleAnnotationCreated(payload) {
+    console.log(`[Webhook] Annotation created:`, {
+        projectId: payload.project.id,
+        taskId: payload.task?.id,
+        annotationId: payload.annotation?.id,
+        completedBy: payload.annotation?.completed_by?.email
+    });
+    // TODO: Implement business logic
+    // 1. Store annotation in Things-Factory database
+    // 2. Trigger downstream processes (e.g., model training)
+    // 3. Send notifications to stakeholders
+    // 4. Update task status in external systems
+}
+/**
+ * Handle annotation updated event
+ */
+async function handleAnnotationUpdated(payload) {
+    console.log(`[Webhook] Annotation updated:`, {
+        projectId: payload.project.id,
+        annotationId: payload.annotation?.id
+    });
+    // TODO: Update annotation in database
+}
+/**
+ * Handle annotation deleted event
+ */
+async function handleAnnotationDeleted(payload) {
+    console.log(`[Webhook] Annotation deleted:`, {
+        projectId: payload.project.id,
+        annotationId: payload.annotation?.id
+    });
+    // TODO: Remove annotation from database
+}
+/**
+ * Handle task created event
+ */
+async function handleTaskCreated(payload) {
+    console.log(`[Webhook] Task created:`, {
+        projectId: payload.project.id,
+        taskId: payload.task?.id
+    });
+    // TODO: Store task reference
+}
+/**
+ * Handle project updated event
+ */
+async function handleProjectUpdated(payload) {
+    console.log(`[Webhook] Project updated:`, {
+        projectId: payload.project.id,
+        projectTitle: payload.project.title
+    });
+    // TODO: Update project metadata
+}
+/**
+ * Main webhook endpoint
+ * Label Studio will POST events here
+ */
+webhookRouter.post('/label-studio/webhook', async (ctx) => {
+    try {
+        const payload = ctx.request.body;
+        console.log(`[Webhook] Received event: ${payload.action}`);
+        // Execute default handlers first
+        switch (payload.action) {
+            case WebhookAction.ANNOTATION_CREATED:
+                await handleAnnotationCreated(payload);
+                break;
+            case WebhookAction.ANNOTATION_UPDATED:
+                await handleAnnotationUpdated(payload);
+                break;
+            case WebhookAction.ANNOTATION_DELETED:
+                await handleAnnotationDeleted(payload);
+                break;
+            case WebhookAction.TASK_CREATED:
+                await handleTaskCreated(payload);
+                break;
+            case WebhookAction.TASK_UPDATED:
+                // Optional: handle task updates
+                break;
+            case WebhookAction.TASK_DELETED:
+                // Optional: handle task deletions
+                break;
+            case WebhookAction.PROJECT_UPDATED:
+                await handleProjectUpdated(payload);
+                break;
+            default:
+                console.warn(`[Webhook] Unknown action: ${payload.action}`);
+        }
+        // Execute custom handlers
+        await executeCustomHandlers(payload.action, payload, ctx);
+        ctx.status = 200;
+        ctx.body = { success: true };
+    }
+    catch (error) {
+        console.error('[Webhook] Error processing webhook:', error);
+        ctx.status = 500;
+        ctx.body = { error: error.message };
+    }
+});
+/**
+ * Webhook registration helper
+ * Call this to register webhook with Label Studio
+ */
+webhookRouter.post('/label-studio/webhook/register', async (ctx) => {
+    try {
+        const { projectId } = ctx.request.body;
+        // Get Things-Factory server URL
+        const serverUrl = process.env.SERVER_URL || 'http://localhost:3000';
+        const webhookUrl = `${serverUrl}/label-studio/webhook`;
+        // Register webhook using Label Studio API
+        const { labelStudioApi } = await Promise.resolve().then(() => tslib_1.__importStar(require('../utils/label-studio-api-client.js')));
+        const webhook = await labelStudioApi.createWebhook({
+            project: projectId,
+            url: webhookUrl,
+            send_payload: true,
+            send_for_all_actions: true,
+            headers: {
+                'Content-Type': 'application/json'
+            }
+        });
+        ctx.status = 200;
+        ctx.body = {
+            success: true,
+            webhook
+        };
+    }
+    catch (error) {
+        console.error('[Webhook] Error registering webhook:', error);
+        ctx.status = 500;
+        ctx.body = { error: error.message };
+    }
+});
+/**
+ * Get webhooks for a project
+ */
+webhookRouter.get('/label-studio/webhook/:projectId', async (ctx) => {
+    try {
+        const projectId = parseInt(ctx.params.projectId);
+        const { labelStudioApi } = await Promise.resolve().then(() => tslib_1.__importStar(require('../utils/label-studio-api-client.js')));
+        const webhooks = await labelStudioApi.getWebhooks(projectId);
+        ctx.status = 200;
+        ctx.body = webhooks;
+    }
+    catch (error) {
+        console.error('[Webhook] Error fetching webhooks:', error);
+        ctx.status = 500;
+        ctx.body = { error: error.message };
+    }
+});
+// Register routes with Things-Factory
+process.on('bootstrap-module-domain-private-route', (_app, router) => {
+    router.use(webhookRouter.routes());
+    router.use(webhookRouter.allowedMethods());
+});
+//# sourceMappingURL=webhook.js.map

package/dist-server/route/webhook.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"webhook.js","sourceRoot":"","sources":["../../server/route/webhook.ts"],"names":[],"mappings":";;;AAoEA,wDAKC;AAKD,4DAQC;AAKD,oDAMC;;AAhGD,oEAA+B;AAG/B,MAAM,aAAa,GAAG,IAAI,oBAAM,EAAE,CAAA;AA2SzB,sCAAa;AAzStB;;GAEG;AACH,IAAY,aAQX;AARD,WAAY,aAAa;IACvB,0DAAyC,CAAA;IACzC,0DAAyC,CAAA;IACzC,0DAAyC,CAAA;IACzC,8CAA6B,CAAA;IAC7B,8CAA6B,CAAA;IAC7B,8CAA6B,CAAA;IAC7B,oDAAmC,CAAA;AACrC,CAAC,EARW,aAAa,6BAAb,aAAa,QAQxB;AA8BD;;;GAGG;AACH,MAAM,cAAc,GAAyC,IAAI,GAAG,EAAE,CAAA;AAEtE;;;;;;;;;;;;;;GAcG;AACH,SAAgB,sBAAsB,CAAC,MAAqB,EAAE,OAAuB;IACnF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;QAChC,cAAc,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;IAChC,CAAC;IACD,cAAc,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;AAC3C,CAAC;AAED;;GAEG;AACH,SAAgB,wBAAwB,CAAC,MAAqB,EAAE,OAAuB;IACrF,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;IAC3C,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QACvC,IAAI,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC;YACf,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;QAC3B,CAAC;IACH,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,oBAAoB,CAAC,MAAsB;IACzD,IAAI,MAAM,EAAE,CAAC;QACX,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;IAC/B,CAAC;SAAM,CAAC;QACN,cAAc,CAAC,KAAK,EAAE,CAAA;IACxB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,qBAAqB,CAAC,MAAqB,EAAE,OAAuB,EAAE,GAAgB;IACnG,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;IAC3C,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvC,OAAM;IACR,CAAC;IAED,mCAAmC;IACnC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAA;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,sCAAsC,MAAM,GAAG,EAAE,KAAK,CAAC,CAAA;YACrE,sDAAsD;QACxD,CAAC;IACH,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,uBAAuB,CAAC,OAAuB;IAC5D,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE;QAC3C,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;QAC7B,MAAM,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE;QACxB,YAAY,EAAE,OAAO,CAAC,UAAU,EAAE,EAAE;QACpC,WAAW,EAAE,OAAO,CAAC,UAAU,EAAE,YAAY,EAAE,KAAK;KACrD,CAAC,CAAA;IAEF,iCAAiC;IACjC,iDAAiD;IACjD,yDAAyD;IACzD,wCAAwC;IACxC,4CAA4C;AAC9C,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,uBAAuB,CAAC,OAAuB;IAC5D,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE;QAC3C,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;QAC7B,YAAY,EAAE,OAAO,CAAC,UAAU,EAAE,EAAE;KACrC,CAAC,CAAA;IAEF,sCAAsC;AACxC,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,uBAAuB,CAAC,OAAuB;IAC5D,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE;QAC3C,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;QAC7B,YAAY,EAAE,OAAO,CAAC,UAAU,EAAE,EAAE;KACrC,CAAC,CAAA;IAEF,wCAAwC;AAC1C,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,iBAAiB,CAAC,OAAuB;IACtD,OAAO,CAAC,GAAG,CAAC,yBAAyB,EAAE;QACrC,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;QAC7B,MAAM,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE;KACzB,CAAC,CAAA;IAEF,6BAA6B;AAC/B,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,oBAAoB,CAAC,OAAuB;IACzD,OAAO,CAAC,GAAG,CAAC,4BAA4B,EAAE;QACxC,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;QAC7B,YAAY,EAAE,OAAO,CAAC,OAAO,CAAC,KAAK;KACpC,CAAC,CAAA;IAEF,gCAAgC;AAClC,CAAC;AAED;;;GAGG;AACH,aAAa,CAAC,IAAI,CAAC,uBAAuB,EAAE,KAAK,EAAE,GAAgB,EAAE,EAAE;IACrE,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC,IAAsB,CAAA;QAElD,OAAO,CAAC,GAAG,CAAC,6BAA6B,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;QAE1D,iCAAiC;QACjC,QAAQ,OAAO,CAAC,MAAM,EAAE,CAAC;YACvB,KAAK,aAAa,CAAC,kBAAkB;gBACnC,MAAM,uBAAuB,CAAC,OAAO,CAAC,CAAA;gBACtC,MAAK;YAEP,KAAK,aAAa,CAAC,kBAAkB;gBACnC,MAAM,uBAAuB,CAAC,OAAO,CAAC,CAAA;gBACtC,MAAK;YAEP,KAAK,aAAa,CAAC,kBAAkB;gBACnC,MAAM,uBAAuB,CAAC,OAAO,CAAC,CAAA;gBACtC,MAAK;YAEP,KAAK,aAAa,CAAC,YAAY;gBAC7B,MAAM,iBAAiB,CAAC,OAAO,CAAC,CAAA;gBAChC,MAAK;YAEP,KAAK,aAAa,CAAC,YAAY;gBAC7B,gCAAgC;gBAChC,MAAK;YAEP,KAAK,aAAa,CAAC,YAAY;gBAC7B,kCAAkC;gBAClC,MAAK;YAEP,KAAK,aAAa,CAAC,eAAe;gBAChC,MAAM,oBAAoB,CAAC,OAAO,CAAC,CAAA;gBACnC,MAAK;YAEP;gBACE,OAAO,CAAC,IAAI,CAAC,6BAA6B,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;QAC/D,CAAC;QAED,0BAA0B;QAC1B,MAAM,qBAAqB,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,CAAA;QAEzD,GAAG,CAAC,MAAM,GAAG,GAAG,CAAA;QAChB,GAAG,CAAC,IAAI,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAA;IAC9B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAA;QAC3D,GAAG,CAAC,MAAM,GAAG,GAAG,CAAA;QAChB,GAAG,CAAC,IAAI,GAAG,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAA;IACrC,CAAC;AACH,CAAC,CAAC,CAAA;AAEF;;;GAGG;AACH,aAAa,CAAC,IAAI,CAAC,gCAAgC,EAAE,KAAK,EAAE,GAAgB,EAAE,EAAE;IAC9E,IAAI,CAAC;QACH,MAAM,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC,OAAO,CAAC,IAA6B,CAAA;QAE/D,gCAAgC;QAChC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,uBAAuB,CAAA;QACnE,MAAM,UAAU,GAAG,GAAG,SAAS,uBAAuB,CAAA;QAEtD,0CAA0C;QAC1C,MAAM,EAAE,cAAc,EAAE,GAAG,gEAAa,qCAAqC,GAAC,CAAA;QAE9E,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,aAAa,CAAC;YACjD,OAAO,EAAE,SAAS;YAClB,GAAG,EAAE,UAAU;YACf,YAAY,EAAE,IAAI;YAClB,oBAAoB,EAAE,IAAI;YAC1B,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;SACF,CAAC,CAAA;QAEF,GAAG,CAAC,MAAM,GAAG,GAAG,CAAA;QAChB,GAAG,CAAC,IAAI,GAAG;YACT,OAAO,EAAE,IAAI;YACb,OAAO;SACR,CAAA;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE,KAAK,CAAC,CAAA;QAC5D,GAAG,CAAC,MAAM,GAAG,GAAG,CAAA;QAChB,GAAG,CAAC,IAAI,GAAG,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAA;IACrC,CAAC;AACH,CAAC,CAAC,CAAA;AAEF;;GAEG;AACH,aAAa,CAAC,GAAG,CAAC,kCAAkC,EAAE,KAAK,EAAE,GAAgB,EAAE,EAAE;IAC/E,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;QAEhD,MAAM,EAAE,cAAc,EAAE,GAAG,gEAAa,qCAAqC,GAAC,CAAA;QAC9E,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,WAAW,CAAC,SAAS,CAAC,CAAA;QAE5D,GAAG,CAAC,MAAM,GAAG,GAAG,CAAA;QAChB,GAAG,CAAC,IAAI,GAAG,QAAQ,CAAA;IACrB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAA;QAC1D,GAAG,CAAC,MAAM,GAAG,GAAG,CAAA;QAChB,GAAG,CAAC,IAAI,GAAG,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAA;IACrC,CAAC;AACH,CAAC,CAAC,CAAA;AAEF,sCAAsC;AACtC,OAAO,CAAC,EAAE,CAAC,uCAAuC,EAAE,CAAC,IAAS,EAAE,MAAc,EAAE,EAAE;IAChF,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAA;IAClC,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,cAAc,EAAE,CAAC,CAAA;AAC5C,CAAC,CAAC,CAAA","sourcesContent":["import Koa from 'koa'\nimport Router from 'koa-router'\nimport { getRepository } from 'typeorm'\n\nconst webhookRouter = new Router()\n\n/**\n * Label Studio Webhook Event Types\n */\nexport enum WebhookAction {\n  ANNOTATION_CREATED = 'ANNOTATION_CREATED',\n  ANNOTATION_UPDATED = 'ANNOTATION_UPDATED',\n  ANNOTATION_DELETED = 'ANNOTATION_DELETED',\n  TASK_CREATED = 'TASK_CREATED',\n  TASK_UPDATED = 'TASK_UPDATED',\n  TASK_DELETED = 'TASK_DELETED',\n  PROJECT_UPDATED = 'PROJECT_UPDATED'\n}\n\nexport interface WebhookPayload {\n  action: WebhookAction\n  project: {\n    id: number\n    title: string\n  }\n  task?: {\n    id: number\n    data: any\n    annotations: any[]\n  }\n  annotation?: {\n    id: number\n    result: any[]\n    completed_by: {\n      id: number\n      email: string\n    }\n    lead_time: number\n  }\n}\n\n/**\n * Webhook handler function type\n * Applications can register custom handlers for any webhook action\n */\nexport type WebhookHandler = (payload: WebhookPayload, context: Koa.Context) => Promise<void>\n\n/**\n * Registry for custom webhook handlers\n * Multiple handlers can be registered per action\n */\nconst customHandlers: Map<WebhookAction, WebhookHandler[]> = new Map()\n\n/**\n * Register a custom webhook handler\n * Handlers are executed in registration order\n *\n * @param action - Webhook action to handle\n * @param handler - Handler function\n *\n * @example\n * registerWebhookHandler(WebhookAction.ANNOTATION_CREATED, async (payload, ctx) => {\n *   console.log('Custom handler:', payload.annotation?.id)\n *   // Store annotation in database\n *   // Trigger ML training\n *   // Send notifications\n * })\n */\nexport function registerWebhookHandler(action: WebhookAction, handler: WebhookHandler): void {\n  if (!customHandlers.has(action)) {\n    customHandlers.set(action, [])\n  }\n  customHandlers.get(action)!.push(handler)\n}\n\n/**\n * Unregister a webhook handler\n */\nexport function unregisterWebhookHandler(action: WebhookAction, handler: WebhookHandler): void {\n  const handlers = customHandlers.get(action)\n  if (handlers) {\n    const index = handlers.indexOf(handler)\n    if (index > -1) {\n      handlers.splice(index, 1)\n    }\n  }\n}\n\n/**\n * Clear all custom handlers for an action\n */\nexport function clearWebhookHandlers(action?: WebhookAction): void {\n  if (action) {\n    customHandlers.delete(action)\n  } else {\n    customHandlers.clear()\n  }\n}\n\n/**\n * Execute all registered handlers for an action\n */\nasync function executeCustomHandlers(action: WebhookAction, payload: WebhookPayload, ctx: Koa.Context): Promise<void> {\n  const handlers = customHandlers.get(action)\n  if (!handlers || handlers.length === 0) {\n    return\n  }\n\n  // Execute all handlers in sequence\n  for (const handler of handlers) {\n    try {\n      await handler(payload, ctx)\n    } catch (error) {\n      console.error(`[Webhook] Custom handler error for ${action}:`, error)\n      // Continue executing other handlers even if one fails\n    }\n  }\n}\n\n/**\n * Handle annotation created event\n */\nasync function handleAnnotationCreated(payload: WebhookPayload) {\n  console.log(`[Webhook] Annotation created:`, {\n    projectId: payload.project.id,\n    taskId: payload.task?.id,\n    annotationId: payload.annotation?.id,\n    completedBy: payload.annotation?.completed_by?.email\n  })\n\n  // TODO: Implement business logic\n  // 1. Store annotation in Things-Factory database\n  // 2. Trigger downstream processes (e.g., model training)\n  // 3. Send notifications to stakeholders\n  // 4. Update task status in external systems\n}\n\n/**\n * Handle annotation updated event\n */\nasync function handleAnnotationUpdated(payload: WebhookPayload) {\n  console.log(`[Webhook] Annotation updated:`, {\n    projectId: payload.project.id,\n    annotationId: payload.annotation?.id\n  })\n\n  // TODO: Update annotation in database\n}\n\n/**\n * Handle annotation deleted event\n */\nasync function handleAnnotationDeleted(payload: WebhookPayload) {\n  console.log(`[Webhook] Annotation deleted:`, {\n    projectId: payload.project.id,\n    annotationId: payload.annotation?.id\n  })\n\n  // TODO: Remove annotation from database\n}\n\n/**\n * Handle task created event\n */\nasync function handleTaskCreated(payload: WebhookPayload) {\n  console.log(`[Webhook] Task created:`, {\n    projectId: payload.project.id,\n    taskId: payload.task?.id\n  })\n\n  // TODO: Store task reference\n}\n\n/**\n * Handle project updated event\n */\nasync function handleProjectUpdated(payload: WebhookPayload) {\n  console.log(`[Webhook] Project updated:`, {\n    projectId: payload.project.id,\n    projectTitle: payload.project.title\n  })\n\n  // TODO: Update project metadata\n}\n\n/**\n * Main webhook endpoint\n * Label Studio will POST events here\n */\nwebhookRouter.post('/label-studio/webhook', async (ctx: Koa.Context) => {\n  try {\n    const payload = ctx.request.body as WebhookPayload\n\n    console.log(`[Webhook] Received event: ${payload.action}`)\n\n    // Execute default handlers first\n    switch (payload.action) {\n      case WebhookAction.ANNOTATION_CREATED:\n        await handleAnnotationCreated(payload)\n        break\n\n      case WebhookAction.ANNOTATION_UPDATED:\n        await handleAnnotationUpdated(payload)\n        break\n\n      case WebhookAction.ANNOTATION_DELETED:\n        await handleAnnotationDeleted(payload)\n        break\n\n      case WebhookAction.TASK_CREATED:\n        await handleTaskCreated(payload)\n        break\n\n      case WebhookAction.TASK_UPDATED:\n        // Optional: handle task updates\n        break\n\n      case WebhookAction.TASK_DELETED:\n        // Optional: handle task deletions\n        break\n\n      case WebhookAction.PROJECT_UPDATED:\n        await handleProjectUpdated(payload)\n        break\n\n      default:\n        console.warn(`[Webhook] Unknown action: ${payload.action}`)\n    }\n\n    // Execute custom handlers\n    await executeCustomHandlers(payload.action, payload, ctx)\n\n    ctx.status = 200\n    ctx.body = { success: true }\n  } catch (error) {\n    console.error('[Webhook] Error processing webhook:', error)\n    ctx.status = 500\n    ctx.body = { error: error.message }\n  }\n})\n\n/**\n * Webhook registration helper\n * Call this to register webhook with Label Studio\n */\nwebhookRouter.post('/label-studio/webhook/register', async (ctx: Koa.Context) => {\n  try {\n    const { projectId } = ctx.request.body as { projectId: number }\n\n    // Get Things-Factory server URL\n    const serverUrl = process.env.SERVER_URL || 'http://localhost:3000'\n    const webhookUrl = `${serverUrl}/label-studio/webhook`\n\n    // Register webhook using Label Studio API\n    const { labelStudioApi } = await import('../utils/label-studio-api-client.js')\n\n    const webhook = await labelStudioApi.createWebhook({\n      project: projectId,\n      url: webhookUrl,\n      send_payload: true,\n      send_for_all_actions: true,\n      headers: {\n        'Content-Type': 'application/json'\n      }\n    })\n\n    ctx.status = 200\n    ctx.body = {\n      success: true,\n      webhook\n    }\n  } catch (error) {\n    console.error('[Webhook] Error registering webhook:', error)\n    ctx.status = 500\n    ctx.body = { error: error.message }\n  }\n})\n\n/**\n * Get webhooks for a project\n */\nwebhookRouter.get('/label-studio/webhook/:projectId', async (ctx: Koa.Context) => {\n  try {\n    const projectId = parseInt(ctx.params.projectId)\n\n    const { labelStudioApi } = await import('../utils/label-studio-api-client.js')\n    const webhooks = await labelStudioApi.getWebhooks(projectId)\n\n    ctx.status = 200\n    ctx.body = webhooks\n  } catch (error) {\n    console.error('[Webhook] Error fetching webhooks:', error)\n    ctx.status = 500\n    ctx.body = { error: error.message }\n  }\n})\n\n// Register routes with Things-Factory\nprocess.on('bootstrap-module-domain-private-route', (_app: Koa, router: Router) => {\n  router.use(webhookRouter.routes())\n  router.use(webhookRouter.allowedMethods())\n})\n\nexport { webhookRouter }\n"]}

package/dist-server/route.d.ts

@@ -0,0 +1 @@
+export {};

package/dist-server/route.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const webhook_1 = require("./route/webhook");
+const label_studio_sso_1 = require("./route/label-studio-sso");
+// Public routes - No authentication required
+process.on('bootstrap-module-global-public-route', (_app, routes) => {
+    /*
+     * Register webhook routes (Label Studio → Things Factory)
+     * These must be public so Label Studio can send events
+     */
+    routes.use(webhook_1.webhookRouter.routes(), webhook_1.webhookRouter.allowedMethods());
+});
+// Private routes - Authentication required
+process.on('bootstrap-module-domain-private-route', (_app, routes) => {
+    /*
+     * Register Label Studio SSO routes
+     * This requires authentication - ctx.state.user will be available
+     */
+    routes.use(label_studio_sso_1.ssoRouter.routes(), label_studio_sso_1.ssoRouter.allowedMethods());
+});
+//# sourceMappingURL=route.js.map

package/dist-server/route.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"route.js","sourceRoot":"","sources":["../server/route.ts"],"names":[],"mappings":";;AAeA,6CAA+C;AAC/C,+DAAoD;AAEpD,6CAA6C;AAC7C,OAAO,CAAC,EAAE,CAAC,sCAA6C,EAAE,CAAC,IAAS,EAAE,MAAc,EAAE,EAAE;IACtF;;;OAGG;IACH,MAAM,CAAC,GAAG,CAAC,uBAAa,CAAC,MAAM,EAAE,EAAE,uBAAa,CAAC,cAAc,EAAE,CAAC,CAAA;AACpE,CAAC,CAAC,CAAA;AAEF,2CAA2C;AAC3C,OAAO,CAAC,EAAE,CAAC,uCAA8C,EAAE,CAAC,IAAS,EAAE,MAAc,EAAE,EAAE;IACvF;;;OAGG;IACH,MAAM,CAAC,GAAG,CAAC,4BAAS,CAAC,MAAM,EAAE,EAAE,4BAAS,CAAC,cAAc,EAAE,CAAC,CAAA;AAC5D,CAAC,CAAC,CAAA","sourcesContent":["/**\n * Label Studio Integration Routes\n *\n * This module registers all Label Studio-related routes:\n * - Webhook handler for Label Studio events (PUBLIC)\n * - SSO authentication endpoint (PRIVATE - requires authentication)\n *\n * ## Subdomain Cookie Sharing Architecture:\n * This integration uses subdomain-based cookie sharing instead of proxying.\n * - Backend gets JWT token from Label Studio and sets shared domain cookie\n * - Frontend loads Label Studio directly (not through proxy)\n * - Cookie is automatically sent with all subdomain requests\n */\nimport Koa from 'koa'\nimport Router from 'koa-router'\nimport { webhookRouter } from './route/webhook'\nimport { ssoRouter } from './route/label-studio-sso'\n\n// Public routes - No authentication required\nprocess.on('bootstrap-module-global-public-route' as any, (_app: Koa, routes: Router) => {\n  /*\n   * Register webhook routes (Label Studio → Things Factory)\n   * These must be public so Label Studio can send events\n   */\n  routes.use(webhookRouter.routes(), webhookRouter.allowedMethods())\n})\n\n// Private routes - Authentication required\nprocess.on('bootstrap-module-domain-private-route' as any, (_app: Koa, routes: Router) => {\n  /*\n   * Register Label Studio SSO routes\n   * This requires authentication - ctx.state.user will be available\n   */\n  routes.use(ssoRouter.routes(), ssoRouter.allowedMethods())\n})\n"]}

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

@@ -0,0 +1,27 @@
+import { GeneratePredictionRequest, BatchGeneratePredictionRequest, PredictionGenerationResult, BatchPredictionResult } from '../types/prediction-types.js';
+/**
+ * Label Studio AI Prediction Service
+ *
+ * Integrates AI inference with Label Studio prediction system
+ * Uses ai-inference module for pure AI operations
+ */
+export declare class LabelStudioAIPredictionService {
+    /**
+     * Generate Label Studio prediction for a task
+     * Runs AI model and creates prediction in Label Studio
+     */
+    generatePrediction(input: GeneratePredictionRequest, context: ResolverContext): Promise<PredictionGenerationResult>;
+    /**
+     * Generate predictions for multiple tasks in batch
+     */
+    generateBatchPredictions(input: BatchGeneratePredictionRequest, context: ResolverContext): Promise<BatchPredictionResult>;
+    /**
+     * Auto-generate predictions for all unlabeled tasks in a project
+     */
+    autoGeneratePredictions(projectId: number, modelId: string, confidenceThreshold: number, context: ResolverContext): Promise<BatchPredictionResult>;
+    /**
+     * Convert AI detection results to Label Studio format
+     * This handles the Label Studio-specific data structure
+     */
+    private convertToLabelStudioFormat;
+}

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

@@ -0,0 +1,222 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LabelStudioAIPredictionService = void 0;
+const tslib_1 = require("tslib");
+const type_graphql_1 = require("type-graphql");
+const ai_inference_1 = require("@things-factory/ai-inference");
+const label_studio_api_client_js_1 = require("../utils/label-studio-api-client.js");
+const prediction_types_js_1 = require("../types/prediction-types.js");
+/**
+ * Label Studio AI Prediction Service
+ *
+ * Integrates AI inference with Label Studio prediction system
+ * Uses ai-inference module for pure AI operations
+ */
+let LabelStudioAIPredictionService = class LabelStudioAIPredictionService {
+    /**
+     * Generate Label Studio prediction for a task
+     * Runs AI model and creates prediction in Label Studio
+     */
+    async generatePrediction(input, context) {
+        try {
+            // 1. Run AI model using ai-inference module
+            const modelClient = input.modelId
+                ? ai_inference_1.AIModelClientFactory.getClient(input.modelId)
+                : ai_inference_1.AIModelClientFactory.getDefaultClient();
+            const objects = await modelClient.detectObjects(input.imageUrl, {
+                confidenceThreshold: input.confidenceThreshold
+            });
+            if (objects.length === 0) {
+                return {
+                    taskId: input.taskId,
+                    success: true,
+                    objectCount: 0,
+                    error: 'No objects detected'
+                };
+            }
+            // 2. Convert AI results to Label Studio format
+            const labelStudioResult = this.convertToLabelStudioFormat(objects);
+            // 3. Calculate average confidence
+            const avgConfidence = objects.reduce((sum, obj) => sum + obj.confidence, 0) / objects.length;
+            // 4. Create prediction in Label Studio
+            const prediction = await label_studio_api_client_js_1.labelStudioApi.createPrediction({
+                task: input.taskId,
+                result: labelStudioResult,
+                score: avgConfidence,
+                model_version: input.modelId || 'default-model-v1.0'
+            });
+            console.log(`[LS AI Prediction] Created prediction ${prediction.id} for task ${input.taskId} with ${objects.length} objects`);
+            return {
+                taskId: input.taskId,
+                predictionId: prediction.id,
+                success: true,
+                objectCount: objects.length,
+                avgConfidence
+            };
+        }
+        catch (error) {
+            console.error(`[LS AI Prediction] Failed for task ${input.taskId}:`, error);
+            return {
+                taskId: input.taskId,
+                success: false,
+                objectCount: 0,
+                error: error.message
+            };
+        }
+    }
+    /**
+     * Generate predictions for multiple tasks in batch
+     */
+    async generateBatchPredictions(input, context) {
+        const results = [];
+        let succeeded = 0;
+        let failed = 0;
+        const modelId = input.modelId || 'default-model-v1.0';
+        try {
+            // Process tasks in parallel (limit concurrency to avoid overload)
+            const BATCH_SIZE = 5;
+            for (let i = 0; i < input.taskIds.length; i += BATCH_SIZE) {
+                const batch = input.taskIds.slice(i, i + BATCH_SIZE);
+                const batchPromises = batch.map(async (taskId) => {
+                    try {
+                        // Get task data from Label Studio
+                        const task = await label_studio_api_client_js_1.labelStudioApi.getTask(taskId);
+                        if (!task || !task.data || !task.data.image) {
+                            failed++;
+                            return {
+                                taskId,
+                                success: false,
+                                objectCount: 0,
+                                error: 'Task has no image data'
+                            };
+                        }
+                        // Generate prediction
+                        const result = await this.generatePrediction({
+                            taskId,
+                            imageUrl: task.data.image,
+                            modelId: input.modelId,
+                            confidenceThreshold: input.confidenceThreshold
+                        }, context);
+                        if (result.success) {
+                            succeeded++;
+                        }
+                        else {
+                            failed++;
+                        }
+                        return result;
+                    }
+                    catch (error) {
+                        failed++;
+                        return {
+                            taskId,
+                            success: false,
+                            objectCount: 0,
+                            error: error.message
+                        };
+                    }
+                });
+                const batchResults = await Promise.all(batchPromises);
+                results.push(...batchResults);
+            }
+            console.log(`[LS AI Prediction] Batch completed: ${succeeded} succeeded, ${failed} failed out of ${input.taskIds.length} tasks`);
+            return {
+                total: input.taskIds.length,
+                succeeded,
+                failed,
+                results,
+                modelVersion: modelId
+            };
+        }
+        catch (error) {
+            console.error('[LS AI Prediction] Batch processing failed:', error);
+            throw new Error(`Batch prediction generation failed: ${error.message}`);
+        }
+    }
+    /**
+     * Auto-generate predictions for all unlabeled tasks in a project
+     */
+    async autoGeneratePredictions(projectId, modelId, confidenceThreshold, context) {
+        try {
+            // Get all tasks from project
+            const tasksResponse = await label_studio_api_client_js_1.labelStudioApi.getTasks(projectId, {
+                page_size: 1000 // Adjust as needed
+            });
+            const tasks = tasksResponse.tasks || tasksResponse.results || [];
+            // Filter unlabeled tasks (no annotations)
+            const unlabeledTasks = tasks.filter((task) => {
+                const annotationCount = task.annotations?.length || task.total_annotations || 0;
+                return annotationCount === 0;
+            });
+            console.log(`[LS AI Prediction] Auto-generating predictions for ${unlabeledTasks.length} unlabeled tasks in project ${projectId}`);
+            // Generate predictions in batch
+            return await this.generateBatchPredictions({
+                projectId,
+                taskIds: unlabeledTasks.map((t) => t.id),
+                modelId,
+                confidenceThreshold
+            }, context);
+        }
+        catch (error) {
+            console.error(`[LS AI Prediction] Auto-generation failed for project ${projectId}:`, error);
+            throw new Error(`Auto prediction generation failed: ${error.message}`);
+        }
+    }
+    /**
+     * Convert AI detection results to Label Studio format
+     * This handles the Label Studio-specific data structure
+     */
+    convertToLabelStudioFormat(objects) {
+        return objects.map(obj => ({
+            from_name: 'label',
+            to_name: 'image',
+            type: 'rectanglelabels',
+            value: {
+                x: obj.bbox.x,
+                y: obj.bbox.y,
+                width: obj.bbox.width,
+                height: obj.bbox.height,
+                rectanglelabels: [obj.className]
+            }
+        }));
+    }
+};
+exports.LabelStudioAIPredictionService = LabelStudioAIPredictionService;
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => prediction_types_js_1.PredictionGenerationResult, {
+        description: 'Generate Label Studio prediction for a task using AI model'
+    }),
+    (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", [prediction_types_js_1.GeneratePredictionRequest, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], LabelStudioAIPredictionService.prototype, "generatePrediction", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => prediction_types_js_1.BatchPredictionResult, {
+        description: 'Generate Label Studio predictions for multiple tasks in batch'
+    }),
+    (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", [prediction_types_js_1.BatchGeneratePredictionRequest, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], LabelStudioAIPredictionService.prototype, "generateBatchPredictions", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => prediction_types_js_1.BatchPredictionResult, {
+        description: 'Auto-generate predictions for all unlabeled tasks in a Label Studio 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.Arg)('modelId', { nullable: true })),
+    tslib_1.__param(2, (0, type_graphql_1.Arg)('confidenceThreshold', type => type_graphql_1.Float, { nullable: true })),
+    tslib_1.__param(3, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Number, String, Number, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], LabelStudioAIPredictionService.prototype, "autoGeneratePredictions", null);
+exports.LabelStudioAIPredictionService = LabelStudioAIPredictionService = tslib_1.__decorate([
+    (0, type_graphql_1.Resolver)()
+], LabelStudioAIPredictionService);
+//# sourceMappingURL=ai-prediction-service.js.map

package/dist-server/service/ai-prediction-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-prediction-service.js","sourceRoot":"","sources":["../../server/service/ai-prediction-service.ts"],"names":[],"mappings":";;;;AAAA,+CAAkF;AAClF,+DAAmF;AACnF,oFAAoE;AACpE,sEAKqC;AAErC;;;;;GAKG;AAEI,IAAM,8BAA8B,GAApC,MAAM,8BAA8B;IACzC;;;OAGG;IAKG,AAAN,KAAK,CAAC,kBAAkB,CACR,KAAgC,EACvC,OAAwB;QAE/B,IAAI,CAAC;YACH,4CAA4C;YAC5C,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO;gBAC/B,CAAC,CAAC,mCAAoB,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC;gBAC/C,CAAC,CAAC,mCAAoB,CAAC,gBAAgB,EAAE,CAAA;YAE3C,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,aAAa,CAAC,KAAK,CAAC,QAAQ,EAAE;gBAC9D,mBAAmB,EAAE,KAAK,CAAC,mBAAmB;aAC/C,CAAC,CAAA;YAEF,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO;oBACL,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,OAAO,EAAE,IAAI;oBACb,WAAW,EAAE,CAAC;oBACd,KAAK,EAAE,qBAAqB;iBAC7B,CAAA;YACH,CAAC;YAED,+CAA+C;YAC/C,MAAM,iBAAiB,GAAG,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC,CAAA;YAElE,kCAAkC;YAClC,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM,CAAA;YAE5F,uCAAuC;YACvC,MAAM,UAAU,GAAG,MAAM,2CAAc,CAAC,gBAAgB,CAAC;gBACvD,IAAI,EAAE,KAAK,CAAC,MAAM;gBAClB,MAAM,EAAE,iBAAiB;gBACzB,KAAK,EAAE,aAAa;gBACpB,aAAa,EAAE,KAAK,CAAC,OAAO,IAAI,oBAAoB;aACrD,CAAC,CAAA;YAEF,OAAO,CAAC,GAAG,CACT,yCAAyC,UAAU,CAAC,EAAE,aAAa,KAAK,CAAC,MAAM,SAAS,OAAO,CAAC,MAAM,UAAU,CACjH,CAAA;YAED,OAAO;gBACL,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,YAAY,EAAE,UAAU,CAAC,EAAE;gBAC3B,OAAO,EAAE,IAAI;gBACb,WAAW,EAAE,OAAO,CAAC,MAAM;gBAC3B,aAAa;aACd,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,sCAAsC,KAAK,CAAC,MAAM,GAAG,EAAE,KAAK,CAAC,CAAA;YAC3E,OAAO;gBACL,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,OAAO,EAAE,KAAK;gBACd,WAAW,EAAE,CAAC;gBACd,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAA;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,wBAAwB,CACd,KAAqC,EAC5C,OAAwB;QAE/B,MAAM,OAAO,GAAiC,EAAE,CAAA;QAChD,IAAI,SAAS,GAAG,CAAC,CAAA;QACjB,IAAI,MAAM,GAAG,CAAC,CAAA;QAEd,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,oBAAoB,CAAA;QAErD,IAAI,CAAC;YACH,kEAAkE;YAClE,MAAM,UAAU,GAAG,CAAC,CAAA;YACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;gBAC1D,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,CAAA;gBAEpD,MAAM,aAAa,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,EAAC,MAAM,EAAC,EAAE;oBAC7C,IAAI,CAAC;wBACH,kCAAkC;wBAClC,MAAM,IAAI,GAAG,MAAM,2CAAc,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;wBAEjD,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;4BAC5C,MAAM,EAAE,CAAA;4BACR,OAAO;gCACL,MAAM;gCACN,OAAO,EAAE,KAAK;gCACd,WAAW,EAAE,CAAC;gCACd,KAAK,EAAE,wBAAwB;6BAChC,CAAA;wBACH,CAAC;wBAED,sBAAsB;wBACtB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAC1C;4BACE,MAAM;4BACN,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK;4BACzB,OAAO,EAAE,KAAK,CAAC,OAAO;4BACtB,mBAAmB,EAAE,KAAK,CAAC,mBAAmB;yBAC/C,EACD,OAAO,CACR,CAAA;wBAED,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BACnB,SAAS,EAAE,CAAA;wBACb,CAAC;6BAAM,CAAC;4BACN,MAAM,EAAE,CAAA;wBACV,CAAC;wBAED,OAAO,MAAM,CAAA;oBACf,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,EAAE,CAAA;wBACR,OAAO;4BACL,MAAM;4BACN,OAAO,EAAE,KAAK;4BACd,WAAW,EAAE,CAAC;4BACd,KAAK,EAAE,KAAK,CAAC,OAAO;yBACrB,CAAA;oBACH,CAAC;gBACH,CAAC,CAAC,CAAA;gBAEF,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAA;gBACrD,OAAO,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAA;YAC/B,CAAC;YAED,OAAO,CAAC,GAAG,CACT,uCAAuC,SAAS,eAAe,MAAM,kBAAkB,KAAK,CAAC,OAAO,CAAC,MAAM,QAAQ,CACpH,CAAA;YAED,OAAO;gBACL,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM;gBAC3B,SAAS;gBACT,MAAM;gBACN,OAAO;gBACP,YAAY,EAAE,OAAO;aACtB,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,6CAA6C,EAAE,KAAK,CAAC,CAAA;YACnE,MAAM,IAAI,KAAK,CAAC,uCAAuC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QACzE,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,uBAAuB,CACI,SAAiB,EACZ,OAAe,EACY,mBAA2B,EACnF,OAAwB;QAE/B,IAAI,CAAC;YACH,6BAA6B;YAC7B,MAAM,aAAa,GAAG,MAAM,2CAAc,CAAC,QAAQ,CAAC,SAAS,EAAE;gBAC7D,SAAS,EAAE,IAAI,CAAC,mBAAmB;aACpC,CAAC,CAAA;YAEF,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,IAAI,aAAa,CAAC,OAAO,IAAI,EAAE,CAAA;YAEhE,0CAA0C;YAC1C,MAAM,cAAc,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAS,EAAE,EAAE;gBAChD,MAAM,eAAe,GAAG,IAAI,CAAC,WAAW,EAAE,MAAM,IAAI,IAAI,CAAC,iBAAiB,IAAI,CAAC,CAAA;gBAC/E,OAAO,eAAe,KAAK,CAAC,CAAA;YAC9B,CAAC,CAAC,CAAA;YAEF,OAAO,CAAC,GAAG,CACT,sDAAsD,cAAc,CAAC,MAAM,+BAA+B,SAAS,EAAE,CACtH,CAAA;YAED,gCAAgC;YAChC,OAAO,MAAM,IAAI,CAAC,wBAAwB,CACxC;gBACE,SAAS;gBACT,OAAO,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7C,OAAO;gBACP,mBAAmB;aACpB,EACD,OAAO,CACR,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,yDAAyD,SAAS,GAAG,EAAE,KAAK,CAAC,CAAA;YAC3F,MAAM,IAAI,KAAK,CAAC,sCAAsC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QACxE,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,0BAA0B,CAAC,OAAyB;QAC1D,OAAO,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACzB,SAAS,EAAE,OAAO;YAClB,OAAO,EAAE,OAAO;YAChB,IAAI,EAAE,iBAAiB;YACvB,KAAK,EAAE;gBACL,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;gBACb,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;gBACb,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,KAAK;gBACrB,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM;gBACvB,eAAe,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC;aACjC;SACF,CAAC,CAAC,CAAA;IACL,CAAC;CACF,CAAA;AA7NY,wEAA8B;AASnC;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,gDAA0B,EAAE;QAC/C,WAAW,EAAE,4DAA4D;KAC1E,CAAC;IACD,IAAA,wBAAS,EAAC,6DAA6D,CAAC;IAEtE,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IACZ,mBAAA,IAAA,kBAAG,GAAE,CAAA;;6CADe,+CAAyB;;wEAwD/C;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,2CAAqB,EAAE;QAC1C,WAAW,EAAE,+DAA+D;KAC7E,CAAC;IACD,IAAA,wBAAS,EAAC,6DAA6D,CAAC;IAEtE,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IACZ,mBAAA,IAAA,kBAAG,GAAE,CAAA;;6CADe,oDAA8B;;8EA8EpD;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,2CAAqB,EAAE;QAC1C,WAAW,EAAE,6EAA6E;KAC3F,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,SAAS,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAA;IAClC,mBAAA,IAAA,kBAAG,EAAC,qBAAqB,EAAE,IAAI,CAAC,EAAE,CAAC,oBAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAA;IAC7D,mBAAA,IAAA,kBAAG,GAAE,CAAA;;;;6EAkCP;yCAzMU,8BAA8B;IAD1C,IAAA,uBAAQ,GAAE;GACE,8BAA8B,CA6N1C","sourcesContent":["import { Resolver, Mutation, Arg, Ctx, Int, Float, Directive } from 'type-graphql'\nimport { AIModelClientFactory, DetectedObject } from '@things-factory/ai-inference'\nimport { labelStudioApi } from '../utils/label-studio-api-client.js'\nimport {\n  GeneratePredictionRequest,\n  BatchGeneratePredictionRequest,\n  PredictionGenerationResult,\n  BatchPredictionResult\n} from '../types/prediction-types.js'\n\n/**\n * Label Studio AI Prediction Service\n *\n * Integrates AI inference with Label Studio prediction system\n * Uses ai-inference module for pure AI operations\n */\n@Resolver()\nexport class LabelStudioAIPredictionService {\n  /**\n   * Generate Label Studio prediction for a task\n   * Runs AI model and creates prediction in Label Studio\n   */\n  @Mutation(returns => PredictionGenerationResult, {\n    description: 'Generate Label Studio prediction for a task using AI model'\n  })\n  @Directive('@privilege(category: \"label-studio\", privilege: \"mutation\")')\n  async generatePrediction(\n    @Arg('input') input: GeneratePredictionRequest,\n    @Ctx() context: ResolverContext\n  ): Promise<PredictionGenerationResult> {\n    try {\n      // 1. Run AI model using ai-inference module\n      const modelClient = input.modelId\n        ? AIModelClientFactory.getClient(input.modelId)\n        : AIModelClientFactory.getDefaultClient()\n\n      const objects = await modelClient.detectObjects(input.imageUrl, {\n        confidenceThreshold: input.confidenceThreshold\n      })\n\n      if (objects.length === 0) {\n        return {\n          taskId: input.taskId,\n          success: true,\n          objectCount: 0,\n          error: 'No objects detected'\n        }\n      }\n\n      // 2. Convert AI results to Label Studio format\n      const labelStudioResult = this.convertToLabelStudioFormat(objects)\n\n      // 3. Calculate average confidence\n      const avgConfidence = objects.reduce((sum, obj) => sum + obj.confidence, 0) / objects.length\n\n      // 4. Create prediction in Label Studio\n      const prediction = await labelStudioApi.createPrediction({\n        task: input.taskId,\n        result: labelStudioResult,\n        score: avgConfidence,\n        model_version: input.modelId || 'default-model-v1.0'\n      })\n\n      console.log(\n        `[LS AI Prediction] Created prediction ${prediction.id} for task ${input.taskId} with ${objects.length} objects`\n      )\n\n      return {\n        taskId: input.taskId,\n        predictionId: prediction.id,\n        success: true,\n        objectCount: objects.length,\n        avgConfidence\n      }\n    } catch (error) {\n      console.error(`[LS AI Prediction] Failed for task ${input.taskId}:`, error)\n      return {\n        taskId: input.taskId,\n        success: false,\n        objectCount: 0,\n        error: error.message\n      }\n    }\n  }\n\n  /**\n   * Generate predictions for multiple tasks in batch\n   */\n  @Mutation(returns => BatchPredictionResult, {\n    description: 'Generate Label Studio predictions for multiple tasks in batch'\n  })\n  @Directive('@privilege(category: \"label-studio\", privilege: \"mutation\")')\n  async generateBatchPredictions(\n    @Arg('input') input: BatchGeneratePredictionRequest,\n    @Ctx() context: ResolverContext\n  ): Promise<BatchPredictionResult> {\n    const results: PredictionGenerationResult[] = []\n    let succeeded = 0\n    let failed = 0\n\n    const modelId = input.modelId || 'default-model-v1.0'\n\n    try {\n      // Process tasks in parallel (limit concurrency to avoid overload)\n      const BATCH_SIZE = 5\n      for (let i = 0; i < input.taskIds.length; i += BATCH_SIZE) {\n        const batch = input.taskIds.slice(i, i + BATCH_SIZE)\n\n        const batchPromises = batch.map(async taskId => {\n          try {\n            // Get task data from Label Studio\n            const task = await labelStudioApi.getTask(taskId)\n\n            if (!task || !task.data || !task.data.image) {\n              failed++\n              return {\n                taskId,\n                success: false,\n                objectCount: 0,\n                error: 'Task has no image data'\n              }\n            }\n\n            // Generate prediction\n            const result = await this.generatePrediction(\n              {\n                taskId,\n                imageUrl: task.data.image,\n                modelId: input.modelId,\n                confidenceThreshold: input.confidenceThreshold\n              },\n              context\n            )\n\n            if (result.success) {\n              succeeded++\n            } else {\n              failed++\n            }\n\n            return result\n          } catch (error) {\n            failed++\n            return {\n              taskId,\n              success: false,\n              objectCount: 0,\n              error: error.message\n            }\n          }\n        })\n\n        const batchResults = await Promise.all(batchPromises)\n        results.push(...batchResults)\n      }\n\n      console.log(\n        `[LS AI Prediction] Batch completed: ${succeeded} succeeded, ${failed} failed out of ${input.taskIds.length} tasks`\n      )\n\n      return {\n        total: input.taskIds.length,\n        succeeded,\n        failed,\n        results,\n        modelVersion: modelId\n      }\n    } catch (error) {\n      console.error('[LS AI Prediction] Batch processing failed:', error)\n      throw new Error(`Batch prediction generation failed: ${error.message}`)\n    }\n  }\n\n  /**\n   * Auto-generate predictions for all unlabeled tasks in a project\n   */\n  @Mutation(returns => BatchPredictionResult, {\n    description: 'Auto-generate predictions for all unlabeled tasks in a Label Studio project'\n  })\n  @Directive('@privilege(category: \"label-studio\", privilege: \"mutation\")')\n  async autoGeneratePredictions(\n    @Arg('projectId', type => Int) projectId: number,\n    @Arg('modelId', { nullable: true }) modelId: string,\n    @Arg('confidenceThreshold', type => Float, { nullable: true }) confidenceThreshold: number,\n    @Ctx() context: ResolverContext\n  ): Promise<BatchPredictionResult> {\n    try {\n      // Get all tasks from project\n      const tasksResponse = await labelStudioApi.getTasks(projectId, {\n        page_size: 1000 // Adjust as needed\n      })\n\n      const tasks = tasksResponse.tasks || tasksResponse.results || []\n\n      // Filter unlabeled tasks (no annotations)\n      const unlabeledTasks = tasks.filter((task: any) => {\n        const annotationCount = task.annotations?.length || task.total_annotations || 0\n        return annotationCount === 0\n      })\n\n      console.log(\n        `[LS AI Prediction] Auto-generating predictions for ${unlabeledTasks.length} unlabeled tasks in project ${projectId}`\n      )\n\n      // Generate predictions in batch\n      return await this.generateBatchPredictions(\n        {\n          projectId,\n          taskIds: unlabeledTasks.map((t: any) => t.id),\n          modelId,\n          confidenceThreshold\n        },\n        context\n      )\n    } catch (error) {\n      console.error(`[LS AI Prediction] Auto-generation failed for project ${projectId}:`, error)\n      throw new Error(`Auto prediction generation failed: ${error.message}`)\n    }\n  }\n\n  /**\n   * Convert AI detection results to Label Studio format\n   * This handles the Label Studio-specific data structure\n   */\n  private convertToLabelStudioFormat(objects: DetectedObject[]) {\n    return objects.map(obj => ({\n      from_name: 'label',\n      to_name: 'image',\n      type: 'rectanglelabels',\n      value: {\n        x: obj.bbox.x,\n        y: obj.bbox.y,\n        width: obj.bbox.width,\n        height: obj.bbox.height,\n        rectanglelabels: [obj.className]\n      }\n    }))\n  }\n}\n"]}

package/dist-server/service/dataset-labeling-integration.d.ts

@@ -0,0 +1,44 @@
+import { CreateLabelingTasksRequest, TaskCreationResult, SyncAnnotationsRequest, SyncAnnotationsResult, DatasetLabelingStatus, DatasetLabelingStatusRequest, GeneratePredictionsForDatasetRequest } from '../types/dataset-labeling-types.js';
+import { BatchPredictionResult } from '../types/prediction-types.js';
+/**
+ * Dataset Labeling Integration Service
+ *
+ * Integrates Things-Factory Dataset module with Label Studio for AI-assisted labeling
+ *
+ * Features:
+ * - Create Label Studio tasks from DataSamples
+ * - Auto-generate AI predictions for labeling tasks
+ * - Sync human annotations back to DataSamples
+ * - Track labeling progress and status
+ */
+export declare class DatasetLabelingIntegration {
+    /**
+     * Create Label Studio labeling tasks from DataSamples in a DataSet
+     * Optionally generates AI predictions automatically
+     */
+    createLabelingTasksFromDataset(input: CreateLabelingTasksRequest, context: ResolverContext): Promise<TaskCreationResult>;
+    /**
+     * Sync completed annotations from Label Studio back to DataSamples
+     */
+    syncAnnotationsToDataset(input: SyncAnnotationsRequest, context: ResolverContext): Promise<SyncAnnotationsResult>;
+    /**
+     * Generate AI predictions for existing DataSet samples
+     */
+    generatePredictionsForDataset(input: GeneratePredictionsForDatasetRequest, context: ResolverContext): Promise<BatchPredictionResult>;
+    /**
+     * Query labeling status for a DataSet
+     */
+    datasetLabelingStatus(input: DatasetLabelingStatusRequest, context: ResolverContext): Promise<DatasetLabelingStatus>;
+    /**
+     * Helper: Extract image URL from DataSample
+     */
+    private extractImageUrl;
+    /**
+     * Helper: Convert AI detection results to Label Studio format
+     */
+    private convertToLabelStudioFormat;
+    /**
+     * Helper: Convert Label Studio annotation to DataSample judgment format
+     */
+    private convertAnnotationToJudgment;
+}