@things-factory/integration-label-studio 9.1.19

package/server/types/dataset-labeling-types.ts

@@ -0,0 +1,181 @@
+import { Field, InputType, ObjectType, Int, Float } from 'type-graphql'
+
+/**
+ * Dataset Labeling Integration Types
+ *
+ * Types for integrating DataSample with Label Studio tasks and AI predictions
+ */
+
+@InputType({ description: 'Request to create Label Studio tasks from DataSamples' })
+export class CreateLabelingTasksRequest {
+  @Field(type => Int, { description: 'Label Studio project ID' })
+  projectId: number
+
+  @Field({ description: 'DataSet ID to sync' })
+  dataSetId: string
+
+  @Field({ nullable: true, description: 'Image field name in DataSample.data or DataSample.rawData' })
+  imageField?: string
+
+  @Field({ nullable: true, defaultValue: true, description: 'Auto-generate AI predictions for created tasks' })
+  autoGeneratePredictions?: boolean
+
+  @Field(type => Float, { nullable: true, description: 'Confidence threshold for AI predictions' })
+  confidenceThreshold?: number
+
+  @Field({ nullable: true, description: 'AI model ID to use for predictions' })
+  modelId?: string
+
+  @Field({ nullable: true, description: 'Only create tasks for samples after this date' })
+  sinceDate?: Date
+
+  @Field({ nullable: true, description: 'Maximum number of tasks to create' })
+  limit?: number
+}
+
+@ObjectType({ description: 'Result of task creation from dataset' })
+export class TaskCreationResult {
+  @Field(type => Int, { description: 'Total DataSamples processed' })
+  totalSamples: number
+
+  @Field(type => Int, { description: 'Number of tasks successfully created' })
+  tasksCreated: number
+
+  @Field(type => Int, { description: 'Number of tasks that failed to create' })
+  tasksFailed: number
+
+  @Field(type => Int, { description: 'Number of samples skipped (no image, already exists, etc.)' })
+  tasksSkipped: number
+
+  @Field(type => [Int], { description: 'Array of created Label Studio task IDs' })
+  taskIds: number[]
+
+  @Field(type => Int, { nullable: true, description: 'Number of AI predictions generated' })
+  predictionsCreated?: number
+
+  @Field({ nullable: true, description: 'Error message if operation partially failed' })
+  error?: string
+}
+
+@InputType({ description: 'Request to sync Label Studio annotations back to DataSamples' })
+export class SyncAnnotationsRequest {
+  @Field(type => Int, { description: 'Label Studio project ID' })
+  projectId: number
+
+  @Field({ description: 'DataSet ID to sync annotations to' })
+  dataSetId: string
+
+  @Field({ nullable: true, defaultValue: false, description: 'Only sync completed annotations' })
+  completedOnly?: boolean
+
+  @Field({ nullable: true, description: 'Only sync annotations updated after this date' })
+  sinceDate?: Date
+}
+
+@ObjectType({ description: 'Result of syncing annotations to dataset' })
+export class SyncAnnotationsResult {
+  @Field(type => Int, { description: 'Total annotations processed' })
+  totalAnnotations: number
+
+  @Field(type => Int, { description: 'Number of DataSamples successfully updated' })
+  samplesUpdated: number
+
+  @Field(type => Int, { description: 'Number of updates that failed' })
+  updatesFailed: number
+
+  @Field(type => Int, { description: 'Number of annotations skipped' })
+  skipped: number
+
+  @Field({ nullable: true, description: 'Error message if operation partially failed' })
+  error?: string
+}
+
+@ObjectType({ description: 'Labeling status for a dataset' })
+export class DatasetLabelingStatus {
+  @Field({ description: 'DataSet ID' })
+  dataSetId: string
+
+  @Field({ description: 'DataSet name' })
+  dataSetName: string
+
+  @Field(type => Int, { description: 'Total number of DataSamples in the dataset' })
+  totalSamples: number
+
+  @Field(type => Int, { description: 'Number of samples with Label Studio tasks' })
+  tasksCreated: number
+
+  @Field(type => Int, { description: 'Number of samples with AI predictions' })
+  withPredictions: number
+
+  @Field(type => Int, { description: 'Number of samples with human annotations' })
+  withAnnotations: number
+
+  @Field(type => Int, { description: 'Number of samples with completed annotations' })
+  annotationsCompleted: number
+
+  @Field(type => Int, { description: 'Number of samples not yet processed' })
+  notProcessed: number
+
+  @Field(type => Float, { description: 'Labeling completion rate (0-1)' })
+  completionRate: number
+
+  @Field({ nullable: true, description: 'Associated Label Studio project ID' })
+  projectId?: number
+
+  @Field({ nullable: true, description: 'Last sync timestamp' })
+  lastSyncedAt?: Date
+}
+
+@ObjectType({ description: 'Mapping between DataSample and Label Studio task' })
+export class DataSampleTaskMapping {
+  @Field({ description: 'DataSample ID' })
+  dataSampleId: string
+
+  @Field({ description: 'DataSample name' })
+  dataSampleName: string
+
+  @Field(type => Int, { description: 'Label Studio task ID' })
+  taskId: number
+
+  @Field({ nullable: true, description: 'Image URL used in the task' })
+  imageUrl?: string
+
+  @Field({ nullable: true, description: 'Task creation timestamp' })
+  createdAt?: Date
+
+  @Field({ nullable: true, description: 'Whether task has AI prediction' })
+  hasPrediction?: boolean
+
+  @Field({ nullable: true, description: 'Whether task has human annotation' })
+  hasAnnotation?: boolean
+
+  @Field({ nullable: true, description: 'Whether annotation is completed' })
+  isCompleted?: boolean
+}
+
+@InputType({ description: 'Request to query dataset labeling status' })
+export class DatasetLabelingStatusRequest {
+  @Field({ description: 'DataSet ID' })
+  dataSetId: string
+
+  @Field(type => Int, { nullable: true, description: 'Associated Label Studio project ID' })
+  projectId?: number
+}
+
+@InputType({ description: 'Request to generate predictions for existing tasks' })
+export class GeneratePredictionsForDatasetRequest {
+  @Field({ description: 'DataSet ID' })
+  dataSetId: string
+
+  @Field(type => Int, { description: 'Label Studio project ID' })
+  projectId: number
+
+  @Field({ nullable: true, description: 'AI model ID to use' })
+  modelId?: string
+
+  @Field(type => Float, { nullable: true, description: 'Confidence threshold' })
+  confidenceThreshold?: number
+
+  @Field({ nullable: true, defaultValue: false, description: 'Regenerate predictions even if they already exist' })
+  forceRegenerate?: boolean
+}

package/server/types/global.d.ts

@@ -0,0 +1,23 @@
+/**
+ * Global type definitions for Things-Factory integration
+ */
+
+import { Domain } from '@things-factory/shell'
+import { User } from '@things-factory/auth-base'
+
+declare global {
+  /**
+   * GraphQL Resolver Context
+   * Available in all @Ctx() decorated parameters
+   */
+  interface ResolverContext {
+    state: {
+      domain: Domain
+      user: User
+      [key: string]: any
+    }
+    [key: string]: any
+  }
+}
+
+export {}

package/server/types/label-studio-types.ts

@@ -0,0 +1,346 @@
+import { Field, ObjectType, InputType, Int, Float } from 'type-graphql'
+
+// ============================================================================
+// Project Types
+// ============================================================================
+
+@ObjectType({ description: 'Label Studio project information' })
+export class LabelStudioProject {
+  @Field(type => Int, { description: 'Project ID' })
+  id: number
+
+  @Field({ description: 'Project title' })
+  title: string
+
+  @Field({ nullable: true, description: 'Project description' })
+  description?: string
+
+  @Field({ description: 'Label configuration XML' })
+  labelConfig: string
+
+  @Field({ nullable: true, description: 'Expert instruction' })
+  expertInstruction?: string
+
+  @Field(type => Int, { description: 'Number of tasks' })
+  taskCount: number
+
+  @Field(type => Int, { description: 'Number of completed tasks' })
+  completedTaskCount: number
+
+  @Field(type => Float, { description: 'Completion rate (0-1)' })
+  completionRate: number
+
+  @Field({ nullable: true, description: 'Created date' })
+  createdAt: Date
+
+  @Field({ nullable: true, description: 'Updated date' })
+  updatedAt: Date
+}
+
+@InputType({ description: 'Input for creating Label Studio project' })
+export class CreateProjectInput {
+  @Field({ description: 'Project title' })
+  title: string
+
+  @Field({ nullable: true, description: 'Project description' })
+  description?: string
+
+  @Field({ description: 'Label configuration XML' })
+  labelConfig: string
+
+  @Field({ nullable: true, description: 'Expert instruction' })
+  expertInstruction?: string
+}
+
+@InputType({ description: 'Label config specification for flexible project creation' })
+export class LabelConfigSpecInput {
+  @Field({ description: 'Data type (image, text, audio, video, timeseries, html, hypertext)' })
+  dataType: string
+
+  @Field({ nullable: true, description: 'Data name (default: data)' })
+  dataName?: string
+
+  @Field({ description: 'Control specifications as JSON array' })
+  controls: string
+}
+
+@InputType({ description: 'Input for creating project with flexible config builder' })
+export class CreateProjectWithSpecInput {
+  @Field({ description: 'Project title' })
+  title: string
+
+  @Field({ nullable: true, description: 'Project description' })
+  description?: string
+
+  @Field({ description: 'Label config specification' })
+  labelConfigSpec: LabelConfigSpecInput
+
+  @Field({ nullable: true, description: 'Expert instruction' })
+  expertInstruction?: string
+}
+
+@InputType({ description: 'Export annotations input' })
+export class ExportAnnotationsInput {
+  @Field({ description: 'Export format (json, jsonl, csv, rank-csv, coco, yolo, ner-json)' })
+  format: string
+
+  @Field({ nullable: true, description: 'Filter by task IDs' })
+  taskIds?: string
+}
+
+@ObjectType({ description: 'Annotation export result' })
+export class AnnotationExportResult {
+  @Field({ description: 'Exported data as JSON string' })
+  data: string
+
+  @Field(type => Int, { description: 'Number of annotations exported' })
+  count: number
+
+  @Field({ description: 'Export format used' })
+  format: string
+}
+
+// ============================================================================
+// Task Types
+// ============================================================================
+
+@ObjectType({ description: 'Label Studio task' })
+export class LabelStudioTask {
+  @Field(type => Int, { description: 'Task ID' })
+  id: number
+
+  @Field({ description: 'Task data (JSON)' })
+  data: string
+
+  @Field(type => Int, { description: 'Number of annotations' })
+  annotationCount: number
+
+  @Field({ description: 'Is task completed' })
+  isCompleted: boolean
+
+  @Field({ nullable: true, description: 'Task created date' })
+  createdAt?: Date
+}
+
+@InputType({ description: 'Input for creating tasks' })
+export class TaskDataInput {
+  @Field({ description: 'Task data as JSON string' })
+  data: string
+}
+
+@InputType({ description: 'Task transformation rule for flexible data import' })
+export class TaskTransformRuleInput {
+  @Field({ description: 'Field mapping as JSON (Label Studio field -> source path)' })
+  dataFields: string
+
+  @Field({ nullable: true, description: 'Prediction configuration as JSON' })
+  predictions?: string
+
+  @Field({ nullable: true, description: 'Metadata mapping as JSON' })
+  meta?: string
+}
+
+@InputType({ description: 'Input for importing tasks with transformation' })
+export class ImportTasksWithTransformInput {
+  @Field({ description: 'Source data as JSON array string' })
+  sourceData: string
+
+  @Field({ description: 'Transformation rule' })
+  transformRule: TaskTransformRuleInput
+}
+
+@ObjectType({ description: 'Result of task import operation' })
+export class TaskImportResult {
+  @Field(type => Int, { description: 'Number of tasks imported' })
+  imported: number
+
+  @Field(type => Int, { description: 'Number of tasks failed' })
+  failed: number
+
+  @Field(type => [Int], { description: 'IDs of imported tasks' })
+  taskIds: number[]
+
+  @Field(type => [String], { nullable: true, description: 'Error messages' })
+  errors?: string[]
+}
+
+// ============================================================================
+// Annotation Types
+// ============================================================================
+
+@ObjectType({ description: 'Label Studio annotation' })
+export class LabelStudioAnnotation {
+  @Field(type => Int, { description: 'Annotation ID' })
+  id: number
+
+  @Field(type => Int, { description: 'Task ID' })
+  taskId: number
+
+  @Field({ description: 'Annotation result (JSON)' })
+  result: string
+
+  @Field({ description: 'Completed by user email' })
+  completedBy: string
+
+  @Field({ description: 'Created date' })
+  createdAt: Date
+
+  @Field({ nullable: true, description: 'Lead time in seconds' })
+  leadTime?: number
+}
+
+// ============================================================================
+// Prediction Types
+// ============================================================================
+
+@ObjectType({ description: 'Label Studio prediction (AI pre-annotation)' })
+export class LabelStudioPrediction {
+  @Field(type => Int, { description: 'Prediction ID' })
+  id: number
+
+  @Field(type => Int, { description: 'Task ID' })
+  taskId: number
+
+  @Field({ description: 'Prediction result (JSON)' })
+  result: string
+
+  @Field(type => Float, { nullable: true, description: 'Prediction confidence score (0-1)' })
+  score?: number
+
+  @Field({ nullable: true, description: 'Model version that generated this prediction' })
+  modelVersion?: string
+
+  @Field({ nullable: true, description: 'Created date' })
+  createdAt?: Date
+}
+
+@InputType({ description: 'Input for creating a prediction' })
+export class PredictionInput {
+  @Field(type => Int, { description: 'Task ID to create prediction for' })
+  taskId: number
+
+  @Field({ description: 'Prediction result as JSON string' })
+  result: string
+
+  @Field(type => Float, { nullable: true, description: 'Prediction confidence score (0-1)' })
+  score?: number
+
+  @Field({ nullable: true, description: 'Model version' })
+  modelVersion?: string
+}
+
+@InputType({ description: 'Input for bulk prediction creation' })
+export class BulkPredictionInput {
+  @Field(type => Int, { description: 'Task ID' })
+  taskId: number
+
+  @Field({ description: 'Prediction result as JSON string' })
+  result: string
+
+  @Field(type => Float, { nullable: true, description: 'Prediction confidence score (0-1)' })
+  score?: number
+
+  @Field({ nullable: true, description: 'Model version' })
+  modelVersion?: string
+}
+
+@ObjectType({ description: 'Result of prediction import operation' })
+export class PredictionImportResult {
+  @Field(type => Int, { description: 'Number of predictions created' })
+  created: number
+
+  @Field(type => Int, { description: 'Number of predictions failed' })
+  failed: number
+
+  @Field(type => [String], { nullable: true, description: 'Error messages' })
+  errors?: string[]
+}
+
+@ObjectType({ description: 'Export result' })
+export class ExportResult {
+  @Field({ description: 'Export file path or URL' })
+  exportPath: string
+
+  @Field(type => Int, { description: 'Number of annotations exported' })
+  annotationCount: number
+
+  @Field({ description: 'Export format (JSON, CSV, YOLO, etc.)' })
+  format: string
+}
+
+// ============================================================================
+// Analytics Types
+// ============================================================================
+
+@ObjectType({ description: 'Project metrics and statistics' })
+export class ProjectMetrics {
+  @Field(type => Int, { description: 'Total number of tasks' })
+  totalTasks: number
+
+  @Field(type => Int, { description: 'Number of completed tasks' })
+  completedTasks: number
+
+  @Field(type => Int, { description: 'Total annotations' })
+  totalAnnotations: number
+
+  @Field(type => Float, { description: 'Average annotations per task' })
+  avgAnnotationsPerTask: number
+
+  @Field(type => Float, { description: 'Completion rate (0-1)' })
+  completionRate: number
+
+  @Field(type => Float, { nullable: true, description: 'Average time per task in seconds' })
+  avgTimePerTask?: number
+
+  @Field(type => [AnnotatorStats], { description: 'Statistics per annotator' })
+  annotatorStats: AnnotatorStats[]
+}
+
+@ObjectType({ description: 'Annotator statistics' })
+export class AnnotatorStats {
+  @Field({ description: 'User email' })
+  email: string
+
+  @Field(type => Int, { description: 'Number of annotations' })
+  annotationCount: number
+
+  @Field(type => Float, { description: 'Average time per annotation in seconds' })
+  avgTime: number
+
+  @Field({ description: 'Last annotation date' })
+  lastAnnotationDate: Date
+}
+
+// ============================================================================
+// ML Backend Types
+// ============================================================================
+
+@ObjectType({ description: 'ML Backend information' })
+export class MLBackend {
+  @Field(type => Int, { description: 'ML Backend ID' })
+  id: number
+
+  @Field({ description: 'ML Backend URL' })
+  url: string
+
+  @Field({ description: 'ML Backend title' })
+  title: string
+
+  @Field({ description: 'Is interactive preannotation enabled' })
+  isInteractive: boolean
+
+  @Field({ description: 'Model version' })
+  modelVersion: string
+}
+
+@InputType({ description: 'Input for adding ML Backend' })
+export class AddMLBackendInput {
+  @Field({ description: 'ML Backend URL' })
+  url: string
+
+  @Field({ description: 'ML Backend title' })
+  title: string
+
+  @Field({ nullable: true, description: 'Is interactive preannotation enabled' })
+  isInteractive?: boolean
+}

package/server/types/prediction-types.ts

@@ -0,0 +1,86 @@
+import { Field, ObjectType, InputType, Int, Float } from 'type-graphql'
+
+// ============================================================================
+// Label Studio Prediction Generation Types
+// ============================================================================
+
+/**
+ * Label Studio prediction generation request
+ */
+@InputType({ description: 'Request to generate Label Studio predictions' })
+export class GeneratePredictionRequest {
+  @Field(type => Int, { description: 'Label Studio task ID' })
+  taskId: number
+
+  @Field({ description: 'Image URL from task data' })
+  imageUrl: string
+
+  @Field({ nullable: true, description: 'AI model ID to use' })
+  modelId?: string
+
+  @Field(type => Float, { nullable: true, description: 'Confidence threshold (0-1)' })
+  confidenceThreshold?: number
+}
+
+/**
+ * Batch prediction generation request
+ */
+@InputType({ description: 'Request to generate predictions for multiple tasks' })
+export class BatchGeneratePredictionRequest {
+  @Field(type => Int, { description: 'Label Studio project ID' })
+  projectId: number
+
+  @Field(type => [Int], { description: 'Array of task IDs' })
+  taskIds: number[]
+
+  @Field({ nullable: true, description: 'AI model ID to use' })
+  modelId?: string
+
+  @Field(type => Float, { nullable: true, description: 'Confidence threshold (0-1)' })
+  confidenceThreshold?: number
+}
+
+/**
+ * Prediction generation result
+ */
+@ObjectType({ description: 'Prediction generation result' })
+export class PredictionGenerationResult {
+  @Field(type => Int, { description: 'Task ID' })
+  taskId: number
+
+  @Field(type => Int, { nullable: true, description: 'Created prediction ID' })
+  predictionId?: number
+
+  @Field({ description: 'Success status' })
+  success: boolean
+
+  @Field({ nullable: true, description: 'Error message if failed' })
+  error?: string
+
+  @Field(type => Int, { description: 'Number of detected objects' })
+  objectCount: number
+
+  @Field(type => Float, { nullable: true, description: 'Average confidence score' })
+  avgConfidence?: number
+}
+
+/**
+ * Batch prediction generation result
+ */
+@ObjectType({ description: 'Batch prediction generation result' })
+export class BatchPredictionResult {
+  @Field(type => Int, { description: 'Total tasks processed' })
+  total: number
+
+  @Field(type => Int, { description: 'Number of successful predictions' })
+  succeeded: number
+
+  @Field(type => Int, { description: 'Number of failed predictions' })
+  failed: number
+
+  @Field(type => [PredictionGenerationResult], { description: 'Individual results' })
+  results: PredictionGenerationResult[]
+
+  @Field({ description: 'Model version used' })
+  modelVersion: string
+}