dory-sdk 2.1.0__py3-none-any.whl → 2.1.4__py3-none-any.whl

dory/__init__.py

@@ -28,7 +28,7 @@ Quick Start (Function-based API):
             counter.value += 1
 """
 
-__version__ = "1.0.0"
+__version__ = "2.1.2"
 
 # Core API
 from dory.core.processor import BaseProcessor
@@ -50,6 +50,23 @@ from dory.utils.errors import (
     DoryConfigError,
 )
 
+# Edge node support (fencing, split-brain prevention, heartbeat)
+from dory.edge import (
+    FencingManager,
+    FencingConfig,
+    FencingToken,
+    FencingError,
+    FenceViolation,
+    StaleEpochError,
+    ProcessorRole,
+    RoleManager,
+    RoleTransition,
+    HeartbeatManager,
+    HeartbeatConfig,
+    ConnectivityStatus,
+    EdgeHealthReporter,
+)
+
 __all__ = [
     # Core API
     "BaseProcessor",
@@ -65,6 +82,20 @@ __all__ = [
     "DoryShutdownError",
     "DoryStateError",
     "DoryConfigError",
+    # Edge node support
+    "FencingManager",
+    "FencingConfig",
+    "FencingToken",
+    "FencingError",
+    "FenceViolation",
+    "StaleEpochError",
+    "ProcessorRole",
+    "RoleManager",
+    "RoleTransition",
+    "HeartbeatManager",
+    "HeartbeatConfig",
+    "ConnectivityStatus",
+    "EdgeHealthReporter",
     # Version
     "__version__",
 ]

dory/config/defaults.py

@@ -21,6 +21,12 @@ DEFAULT_CONFIG = {
     "state_s3_bucket": None,
     "state_s3_prefix": "dory-state",
 
+    # State transfer timeouts (aligned with Orchestrator's 30s default)
+    "state_capture_timeout_sec": 25,  # Must be < Orchestrator's 30s timeout
+    "state_restore_timeout_sec": 25,  # Must be < Orchestrator's 30s timeout
+    "state_max_size_bytes": 8 * 1024 * 1024,  # 8MB (Orchestrator limit is 10MB)
+    "state_size_warn_threshold": 0.75,  # Warn at 75% of max size
+
     # Recovery
     "max_restart_attempts": 3,
     "restart_backoff_sec": 10,

dory/config/schema.py

@@ -79,6 +79,32 @@ class DoryConfig(BaseModel):
         description="S3 key prefix for state objects",
     )
 
+    # State transfer timeouts (must be < Orchestrator's 30s timeout)
+    state_capture_timeout_sec: int = Field(
+        default=DEFAULT_CONFIG["state_capture_timeout_sec"],
+        ge=1,
+        le=60,
+        description="Timeout for state capture operations (must be < Orchestrator timeout)",
+    )
+    state_restore_timeout_sec: int = Field(
+        default=DEFAULT_CONFIG["state_restore_timeout_sec"],
+        ge=1,
+        le=60,
+        description="Timeout for state restore operations (must be < Orchestrator timeout)",
+    )
+    state_max_size_bytes: int = Field(
+        default=DEFAULT_CONFIG["state_max_size_bytes"],
+        ge=1024,
+        le=10 * 1024 * 1024,  # Max 10MB (Orchestrator limit)
+        description="Maximum state size in bytes (Orchestrator limit is 10MB)",
+    )
+    state_size_warn_threshold: float = Field(
+        default=DEFAULT_CONFIG["state_size_warn_threshold"],
+        ge=0.1,
+        le=1.0,
+        description="Warn when state size exceeds this fraction of max (0.0-1.0)",
+    )
+
     # Recovery
     max_restart_attempts: int = Field(
         default=DEFAULT_CONFIG["max_restart_attempts"],

dory/edge/__init__.py

@@ -0,0 +1,88 @@
+"""Edge node support for Dory SDK.
+
+Provides fencing, failover coordination, and split-brain prevention
+for edge nodes with intermittent connectivity.
+"""
+
+from dory.edge.fencing import (
+    FencingManager,
+    FencingConfig,
+    FencingToken,
+    FencingError,
+    FenceViolation,
+    StaleEpochError,
+)
+from dory.edge.role import (
+    ProcessorRole,
+    RoleManager,
+    RoleTransition,
+)
+from dory.edge.heartbeat import (
+    HeartbeatManager,
+    HeartbeatConfig,
+    HeartbeatPayload,
+    HeartbeatResponse,
+    ConnectivityStatus,
+    ConnectivityMetrics,
+    EdgeHealthReporter,
+)
+from dory.edge.detector import (
+    WorkloadDetector,
+    WorkloadContext,
+    NodeType,
+    is_edge_node,
+    is_cloud_node,
+    is_migrated_workload,
+    get_workload_context,
+    get_node_type,
+    get_recommended_env_vars,
+    get_pod_spec_yaml_snippet,
+)
+from dory.edge.adaptive import (
+    AdaptiveProcessor,
+    AdaptiveConfig,
+    EdgeConfig,
+    OperationMode,
+    create_adaptive_processor,
+    get_location_aware_settings,
+)
+
+__all__ = [
+    # Fencing
+    "FencingManager",
+    "FencingConfig",
+    "FencingToken",
+    "FencingError",
+    "FenceViolation",
+    "StaleEpochError",
+    # Role management
+    "ProcessorRole",
+    "RoleManager",
+    "RoleTransition",
+    # Heartbeat and connectivity
+    "HeartbeatManager",
+    "HeartbeatConfig",
+    "HeartbeatPayload",
+    "HeartbeatResponse",
+    "ConnectivityStatus",
+    "ConnectivityMetrics",
+    "EdgeHealthReporter",
+    # Workload detection
+    "WorkloadDetector",
+    "WorkloadContext",
+    "NodeType",
+    "is_edge_node",
+    "is_cloud_node",
+    "is_migrated_workload",
+    "get_workload_context",
+    "get_node_type",
+    "get_recommended_env_vars",
+    "get_pod_spec_yaml_snippet",
+    # Adaptive processor
+    "AdaptiveProcessor",
+    "AdaptiveConfig",
+    "EdgeConfig",
+    "OperationMode",
+    "create_adaptive_processor",
+    "get_location_aware_settings",
+]