Add workflow management API - Resolves PLT-2503

Author: paulnoirel

libs/labelbox/src/labelbox/schema/workflow/__init__.py

@@ -19,7 +19,7 @@
     NodePosition,
 )
 
-# Import nodes from nodes.py (excluding LogicNode)
+# Import nodes from the nodes subdirectory
 from labelbox.schema.workflow.nodes import (
     InitialLabelingNode,
     InitialReworkNode,
@@ -28,14 +28,10 @@
     DoneNode,
     CustomReworkNode,
     UnknownWorkflowNode,
+    LogicNode,
+    AutoQANode,
 )
 
-# Import LogicNode from its own module
-from labelbox.schema.workflow.logic_node import LogicNode
-
-# Import AutoQANode from its own module
-from labelbox.schema.workflow.autoqa_node import AutoQANode
-
 from labelbox.schema.workflow.edges import (
     WorkflowEdge,
     WorkflowEdgeFactory,

libs/labelbox/src/labelbox/schema/workflow/base.py

@@ -1,8 +1,6 @@
 """Base classes and mixins for Project Workflow nodes in Labelbox."""
 
 import logging
-import random
-import string
 from typing import Dict, List, Any, Optional, Tuple
 from abc import abstractmethod
 from pydantic import BaseModel, Field, ConfigDict
@@ -12,42 +10,6 @@
 logger = logging.getLogger(__name__)
 
 
-def format_time_duration(seconds: int) -> str:
-    """Convert seconds to human-readable time format.
-
-    Args:
-        seconds: Time duration in seconds
-
-    Returns:
-        Human-readable time string (e.g., "1h 30m", "5m 30s", "45s")
-
-    Examples:
-        >>> format_time_duration(3600)
-        '1h'
-        >>> format_time_duration(90)
-        '1m 30s'
-        >>> format_time_duration(45)
-        '45s'
-    """
-    if seconds >= 3600:  # >= 1 hour
-        hours = seconds // 3600
-        remaining_seconds = seconds % 3600
-        if remaining_seconds == 0:
-            return f"{hours}h"
-        else:
-            minutes = remaining_seconds // 60
-            return f"{hours}h {minutes}m" if minutes > 0 else f"{hours}h"
-    elif seconds >= 60:  # >= 1 minute
-        minutes = seconds // 60
-        remaining_seconds = seconds % 60
-        if remaining_seconds == 0:
-            return f"{minutes}m"
-        else:
-            return f"{minutes}m {remaining_seconds}s"
-    else:
-        return f"{seconds}s"
-
-
 def format_metadata_operator(operator: str) -> Tuple[str, str]:
     """Format metadata operator for display and JSON.
 
@@ -80,15 +42,6 @@ def format_metadata_operator(operator: str) -> Tuple[str, str]:
     return operator_mappings.get(operator, (operator.upper(), operator))
 
 
-def generate_filter_id() -> str:
-    """Generate a unique filter ID for metadata filters.
-
-    Returns:
-        Random 6-character string containing lowercase letters and digits
-    """
-    return "".join(random.choices(string.ascii_lowercase + string.digits, k=6))
-
-
 class NodePosition(BaseModel):
     """Represents the position of a node in the workflow canvas.
 

libs/labelbox/src/labelbox/schema/workflow/edges.py

@@ -42,7 +42,7 @@ class WorkflowEdge(BaseModel):
     )
     targetHandle: str = Field(
         alias="targetHandle",
-        default="if",
+        default="in",
         description="Input handle on target node (typically 'in')",
     )
 

libs/labelbox/src/labelbox/schema/workflow/enums.py

@@ -6,7 +6,6 @@
 
 import logging
 from enum import Enum
-from typing import Any
 
 logger = logging.getLogger(__name__)
 
@@ -36,30 +35,7 @@ class WorkflowDefinitionId(Enum):
     Done = "done"
     CustomReworkTask = "custom_rework_task"
     AutoQA = "auto_qa"
-    Unknown = "unknown"  # Fallback for potential future types
-
-    @classmethod
-    def _missing_(cls, value: Any) -> "WorkflowDefinitionId":
-        """Handle missing enum values with appropriate fallbacks.
-
-        Args:
-            value: The value that couldn't be found in the enum
-
-        Returns:
-            A valid WorkflowDefinitionId with appropriate fallback logic
-        """
-        # If value is None, return a valid default instead of Unknown
-        if value is None:
-            logger.warning(
-                "Null WorkflowDefinitionId provided. Using default InitialLabelingTask."
-            )
-            return cls.InitialLabelingTask
-
-        # For other unrecognized values, log warning but use a valid ID
-        logger.warning(
-            f"Unknown WorkflowDefinitionId '{value}'. Using default InitialLabelingTask."
-        )
-        return cls.InitialLabelingTask
+    Unknown = "unknown"  # For unrecognized node types from API
 
 
 class NodeType(Enum):
@@ -91,19 +67,6 @@ class NodeOutput(str, Enum):
     Rejected = "else"  # Alias for review node rejected output
     Default = "out"
 
-    @classmethod
-    def _missing_(cls, value: Any) -> Any:
-        """Handle missing enum values by using the value as-is.
-
-        Args:
-            value: The value that couldn't be found in the enum
-
-        Returns:
-            The original value for flexible usage
-        """
-        logger.warning(f"Unknown NodeOutput '{value}'. Using as-is.")
-        return value
-
 
 class NodeInput(str, Enum):
     """Available input types for workflow nodes.
@@ -114,19 +77,6 @@ class NodeInput(str, Enum):
 
     Default = "in"
 
-    @classmethod
-    def _missing_(cls, value: Any) -> Any:
-        """Handle missing enum values by using the value as-is.
-
-        Args:
-            value: The value that couldn't be found in the enum
-
-        Returns:
-            The original value for flexible usage
-        """
-        logger.warning(f"Unknown NodeInput '{value}'. Using as-is.")
-        return value
-
 
 class MatchFilters(str, Enum):
     """Available match filter options for LogicNode.
@@ -137,19 +87,6 @@ class MatchFilters(str, Enum):
     Any = "any"  # Maps to filter_logic "or" - matches if any filter passes
     All = "all"  # Maps to filter_logic "and" - matches if all filters pass
 
-    @classmethod
-    def _missing_(cls, value):
-        """Handle missing enum values with All as default.
-
-        Args:
-            value: The value that couldn't be found in the enum
-
-        Returns:
-            MatchFilters.All as the safe default
-        """
-        logger.warning(f"Unknown MatchFilters '{value}'. Using All as default.")
-        return cls.All
-
 
 class Scope(str, Enum):
     """Available scope options for AutoQANode.
@@ -160,19 +97,6 @@ class Scope(str, Enum):
     Any = "any"  # Passes if any annotation meets the criteria
     All = "all"  # Passes only if all annotations meet the criteria
 
-    @classmethod
-    def _missing_(cls, value):
-        """Handle missing enum values with All as default.
-
-        Args:
-            value: The value that couldn't be found in the enum
-
-        Returns:
-            Scope.All as the safe default
-        """
-        logger.warning(f"Unknown Scope '{value}'. Using All as default.")
-        return cls.All
-
 
 class FilterField(str, Enum):
     """Available filter fields for LogicNode filters.
@@ -209,19 +133,6 @@ class FilterField(str, Enum):
     # Search filters
     NlSearch = "NlSearch"
 
-    @classmethod
-    def _missing_(cls, value: Any) -> Any:
-        """Handle missing enum values by using the value as-is.
-
-        Args:
-            value: The value that couldn't be found in the enum
-
-        Returns:
-            The original value for flexible usage
-        """
-        logger.warning(f"Unknown FilterField '{value}'. Using as-is.")
-        return value
-
 
 class FilterOperator(str, Enum):
     """Available filter operators for LogicNode filters.
@@ -250,16 +161,3 @@ class FilterOperator(str, Enum):
 
     # Range operators
     Between = "between"
-
-    @classmethod
-    def _missing_(cls, value: Any) -> Any:
-        """Handle missing enum values by using the value as-is.
-
-        Args:
-            value: The value that couldn't be found in the enum
-
-        Returns:
-            The original value for flexible usage
-        """
-        logger.warning(f"Unknown FilterOperator '{value}'. Using as-is.")
-        return value