horsies 0.1.0a4__py3-none-any.whl → 0.1.0a6__py3-none-any.whl

horsies/core/cli.py

@@ -18,7 +18,12 @@ import sys
 
 from horsies.core.app import Horsies
 from horsies.core.banner import print_banner
-from horsies.core.errors import ConfigurationError, ErrorCode, HorsiesError, ValidationReport
+from horsies.core.errors import (
+    ConfigurationError,
+    ErrorCode,
+    HorsiesError,
+    ValidationReport,
+)
 from horsies.core.logging import get_logger
 from horsies.core.scheduler import Scheduler
 from horsies.core.worker.worker import Worker, WorkerConfig
@@ -156,7 +161,7 @@ def discover_app(module_locator: str) -> tuple[Horsies, str, str, str | None]:
         if not isinstance(obj, Horsies):
             raise TypeError(
                 f"'{attr_name}' in module '{module_name}' is not a Horsies instance "
-                f"(got {type(obj).__name__})"
+                f'(got {type(obj).__name__})'
             )
         app = obj
         var_name = attr_name

horsies/core/errors.py

@@ -38,6 +38,9 @@ class ErrorCode(str, Enum):
     WORKFLOW_INVALID_JOIN = 'E013'
     WORKFLOW_UNRESOLVED_QUEUE = 'E014'
     WORKFLOW_UNRESOLVED_PRIORITY = 'E015'
+    WORKFLOW_ARGS_WITH_INJECTION = 'E016'
+    WORKFLOW_INVALID_KWARG_KEY = 'E019'
+    WORKFLOW_MISSING_REQUIRED_PARAMS = 'E020'
     WORKFLOW_INVALID_SUBWORKFLOW_RETRY_MODE = 'E017'
     WORKFLOW_SUBWORKFLOW_APP_MISSING = 'E018'
 

horsies/core/models/app.py

@@ -5,7 +5,12 @@ from horsies.core.models.queues import QueueMode, CustomQueueConfig
 from horsies.core.models.broker import PostgresConfig
 from horsies.core.models.recovery import RecoveryConfig
 from horsies.core.models.schedule import ScheduleConfig
-from horsies.core.errors import ConfigurationError, ErrorCode, ValidationReport, raise_collected
+from horsies.core.errors import (
+    ConfigurationError,
+    ErrorCode,
+    ValidationReport,
+    raise_collected,
+)
 from urllib.parse import urlparse, urlunparse
 import logging
 import os
@@ -37,91 +42,109 @@ class AppConfig(BaseModel):
 
         if self.queue_mode == QueueMode.DEFAULT:
             if self.custom_queues is not None:
-                report.add(ConfigurationError(
-                    message='custom_queues must be None in DEFAULT mode',
-                    code=ErrorCode.CONFIG_INVALID_QUEUE_MODE,
-                    notes=['queue_mode=DEFAULT but custom_queues was provided'],
-                    help_text='either remove custom_queues or set queue_mode=CUSTOM',
-                ))
+                report.add(
+                    ConfigurationError(
+                        message='custom_queues must be None in DEFAULT mode',
+                        code=ErrorCode.CONFIG_INVALID_QUEUE_MODE,
+                        notes=['queue_mode=DEFAULT but custom_queues was provided'],
+                        help_text='either remove custom_queues or set queue_mode=CUSTOM',
+                    )
+                )
         elif self.queue_mode == QueueMode.CUSTOM:
             if self.custom_queues is None or len(self.custom_queues) == 0:
-                report.add(ConfigurationError(
-                    message='custom_queues required in CUSTOM mode',
-                    code=ErrorCode.CONFIG_INVALID_QUEUE_MODE,
-                    notes=['queue_mode=CUSTOM but custom_queues is empty or None'],
-                    help_text='provide at least one CustomQueueConfig in custom_queues',
-                ))
+                report.add(
+                    ConfigurationError(
+                        message='custom_queues required in CUSTOM mode',
+                        code=ErrorCode.CONFIG_INVALID_QUEUE_MODE,
+                        notes=['queue_mode=CUSTOM but custom_queues is empty or None'],
+                        help_text='provide at least one CustomQueueConfig in custom_queues',
+                    )
+                )
             else:
                 # Validate unique queue names (only if queues exist)
                 queue_names = [q.name for q in self.custom_queues]
                 if len(queue_names) != len(set(queue_names)):
-                    report.add(ConfigurationError(
-                        message='duplicate queue names in custom_queues',
-                        code=ErrorCode.CONFIG_INVALID_QUEUE_MODE,
-                        notes=[f'queue names: {queue_names}'],
-                        help_text='each queue name must be unique',
-                    ))
+                    report.add(
+                        ConfigurationError(
+                            message='duplicate queue names in custom_queues',
+                            code=ErrorCode.CONFIG_INVALID_QUEUE_MODE,
+                            notes=[f'queue names: {queue_names}'],
+                            help_text='each queue name must be unique',
+                        )
+                    )
 
         # Validate cluster_wide_cap if provided
         if self.cluster_wide_cap is not None and self.cluster_wide_cap <= 0:
-            report.add(ConfigurationError(
-                message='cluster_wide_cap must be positive',
-                code=ErrorCode.CONFIG_INVALID_CLUSTER_CAP,
-                notes=[f'got cluster_wide_cap={self.cluster_wide_cap}'],
-                help_text='use a positive integer or None for unlimited',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='cluster_wide_cap must be positive',
+                    code=ErrorCode.CONFIG_INVALID_CLUSTER_CAP,
+                    notes=[f'got cluster_wide_cap={self.cluster_wide_cap}'],
+                    help_text='use a positive integer or None for unlimited',
+                )
+            )
 
         # Validate prefetch_buffer
         if self.prefetch_buffer < 0:
-            report.add(ConfigurationError(
-                message='prefetch_buffer must be non-negative',
-                code=ErrorCode.CONFIG_INVALID_PREFETCH,
-                notes=[f'got prefetch_buffer={self.prefetch_buffer}'],
-                help_text='use 0 for hard cap mode or positive integer for soft cap',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='prefetch_buffer must be non-negative',
+                    code=ErrorCode.CONFIG_INVALID_PREFETCH,
+                    notes=[f'got prefetch_buffer={self.prefetch_buffer}'],
+                    help_text='use 0 for hard cap mode or positive integer for soft cap',
+                )
+            )
 
         # Validate claim_lease_ms
         if self.prefetch_buffer > 0 and self.claim_lease_ms is None:
-            report.add(ConfigurationError(
-                message='claim_lease_ms required when prefetch_buffer > 0',
-                code=ErrorCode.CONFIG_INVALID_PREFETCH,
-                notes=[
-                    f'prefetch_buffer={self.prefetch_buffer} but claim_lease_ms is None',
-                ],
-                help_text='set claim_lease_ms (e.g., 30000 for 30 seconds)',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='claim_lease_ms required when prefetch_buffer > 0',
+                    code=ErrorCode.CONFIG_INVALID_PREFETCH,
+                    notes=[
+                        f'prefetch_buffer={self.prefetch_buffer} but claim_lease_ms is None',
+                    ],
+                    help_text='set claim_lease_ms (e.g., 30000 for 30 seconds)',
+                )
+            )
         if self.claim_lease_ms is not None and self.claim_lease_ms <= 0:
-            report.add(ConfigurationError(
-                message='claim_lease_ms must be positive',
-                code=ErrorCode.CONFIG_INVALID_PREFETCH,
-                notes=[f'got claim_lease_ms={self.claim_lease_ms}'],
-                help_text='use a positive integer in milliseconds',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='claim_lease_ms must be positive',
+                    code=ErrorCode.CONFIG_INVALID_PREFETCH,
+                    notes=[f'got claim_lease_ms={self.claim_lease_ms}'],
+                    help_text='use a positive integer in milliseconds',
+                )
+            )
 
         # Forbid claim_lease_ms in hard cap mode
         if self.prefetch_buffer == 0 and self.claim_lease_ms is not None:
-            report.add(ConfigurationError(
-                message='claim_lease_ms incompatible with hard cap mode',
-                code=ErrorCode.CONFIG_INVALID_PREFETCH,
-                notes=[
-                    'prefetch_buffer=0 (hard cap mode)',
-                    f'but claim_lease_ms={self.claim_lease_ms} was set',
-                ],
-                help_text='remove claim_lease_ms or set prefetch_buffer > 0',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='claim_lease_ms incompatible with hard cap mode',
+                    code=ErrorCode.CONFIG_INVALID_PREFETCH,
+                    notes=[
+                        'prefetch_buffer=0 (hard cap mode)',
+                        f'but claim_lease_ms={self.claim_lease_ms} was set',
+                    ],
+                    help_text='remove claim_lease_ms or set prefetch_buffer > 0',
+                )
+            )
 
         # Validate prefetch_buffer vs cluster_wide_cap conflict
         if self.cluster_wide_cap is not None and self.prefetch_buffer > 0:
-            report.add(ConfigurationError(
-                message='cluster_wide_cap incompatible with prefetch mode',
-                code=ErrorCode.CONFIG_INVALID_CLUSTER_CAP,
-                notes=[
-                    f'cluster_wide_cap={self.cluster_wide_cap}',
-                    f'prefetch_buffer={self.prefetch_buffer}',
-                    'cluster_wide_cap requires hard cap mode (prefetch_buffer=0)',
-                ],
-                help_text='set prefetch_buffer=0 when using cluster_wide_cap',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='cluster_wide_cap incompatible with prefetch mode',
+                    code=ErrorCode.CONFIG_INVALID_CLUSTER_CAP,
+                    notes=[
+                        f'cluster_wide_cap={self.cluster_wide_cap}',
+                        f'prefetch_buffer={self.prefetch_buffer}',
+                        'cluster_wide_cap requires hard cap mode (prefetch_buffer=0)',
+                    ],
+                    help_text='set prefetch_buffer=0 when using cluster_wide_cap',
+                )
+            )
 
         raise_collected(report)
         return self

horsies/core/models/recovery.py

@@ -2,7 +2,12 @@
 from __future__ import annotations
 from typing import Annotated, Self
 from pydantic import BaseModel, Field, model_validator
-from horsies.core.errors import ConfigurationError, ErrorCode, ValidationReport, raise_collected
+from horsies.core.errors import (
+    ConfigurationError,
+    ErrorCode,
+    ValidationReport,
+    raise_collected,
+)
 
 
 class RecoveryConfig(BaseModel):
@@ -73,29 +78,33 @@ class RecoveryConfig(BaseModel):
 
         # Validate runner heartbeat vs running stale threshold
         if self.running_stale_threshold_ms < min_running:
-            report.add(ConfigurationError(
-                message='running_stale_threshold_ms too low',
-                code=ErrorCode.CONFIG_INVALID_RECOVERY,
-                notes=[
-                    f'running_stale_threshold_ms={self.running_stale_threshold_ms}ms ({self.running_stale_threshold_ms/1000:.1f}s)',
-                    f'runner_heartbeat_interval_ms={self.runner_heartbeat_interval_ms}ms ({self.runner_heartbeat_interval_ms/1000:.1f}s)',
-                    'threshold must be at least 2x heartbeat interval',
-                ],
-                help_text=f'set running_stale_threshold_ms >= {min_running}ms ({min_running/1000:.1f}s)',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='running_stale_threshold_ms too low',
+                    code=ErrorCode.CONFIG_INVALID_RECOVERY,
+                    notes=[
+                        f'running_stale_threshold_ms={self.running_stale_threshold_ms}ms ({self.running_stale_threshold_ms/1000:.1f}s)',
+                        f'runner_heartbeat_interval_ms={self.runner_heartbeat_interval_ms}ms ({self.runner_heartbeat_interval_ms/1000:.1f}s)',
+                        'threshold must be at least 2x heartbeat interval',
+                    ],
+                    help_text=f'set running_stale_threshold_ms >= {min_running}ms ({min_running/1000:.1f}s)',
+                )
+            )
 
         # Validate claimer heartbeat vs claimed stale threshold
         if self.claimed_stale_threshold_ms < min_claimed:
-            report.add(ConfigurationError(
-                message='claimed_stale_threshold_ms too low',
-                code=ErrorCode.CONFIG_INVALID_RECOVERY,
-                notes=[
-                    f'claimed_stale_threshold_ms={self.claimed_stale_threshold_ms}ms ({self.claimed_stale_threshold_ms/1000:.1f}s)',
-                    f'claimer_heartbeat_interval_ms={self.claimer_heartbeat_interval_ms}ms ({self.claimer_heartbeat_interval_ms/1000:.1f}s)',
-                    'threshold must be at least 2x heartbeat interval',
-                ],
-                help_text=f'set claimed_stale_threshold_ms >= {min_claimed}ms ({min_claimed/1000:.1f}s)',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='claimed_stale_threshold_ms too low',
+                    code=ErrorCode.CONFIG_INVALID_RECOVERY,
+                    notes=[
+                        f'claimed_stale_threshold_ms={self.claimed_stale_threshold_ms}ms ({self.claimed_stale_threshold_ms/1000:.1f}s)',
+                        f'claimer_heartbeat_interval_ms={self.claimer_heartbeat_interval_ms}ms ({self.claimer_heartbeat_interval_ms/1000:.1f}s)',
+                        'threshold must be at least 2x heartbeat interval',
+                    ],
+                    help_text=f'set claimed_stale_threshold_ms >= {min_claimed}ms ({min_claimed/1000:.1f}s)',
+                )
+            )
 
         raise_collected(report)
         return self

horsies/core/models/schedule.py

@@ -5,7 +5,12 @@ from typing import Literal, Union, Optional, Any
 from pydantic import BaseModel, Field, model_validator
 from typing_extensions import Self
 from enum import Enum
-from horsies.core.errors import ConfigurationError, ErrorCode, ValidationReport, raise_collected
+from horsies.core.errors import (
+    ConfigurationError,
+    ErrorCode,
+    ValidationReport,
+    raise_collected,
+)
 
 
 class Weekday(str, Enum):
@@ -54,12 +59,14 @@ class IntervalSchedule(BaseModel):
         """Ensure at least one time unit is specified."""
         report = ValidationReport('schedule')
         if not any([self.seconds, self.minutes, self.hours, self.days]):
-            report.add(ConfigurationError(
-                message='IntervalSchedule requires at least one time unit',
-                code=ErrorCode.CONFIG_INVALID_SCHEDULE,
-                notes=['all time units (seconds, minutes, hours, days) are None'],
-                help_text='specify at least one: seconds, minutes, hours, or days',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='IntervalSchedule requires at least one time unit',
+                    code=ErrorCode.CONFIG_INVALID_SCHEDULE,
+                    notes=['all time units (seconds, minutes, hours, days) are None'],
+                    help_text='specify at least one: seconds, minutes, hours, or days',
+                )
+            )
         raise_collected(report)
         return self
 
@@ -126,12 +133,14 @@ class WeeklySchedule(BaseModel):
         """Ensure no duplicate days."""
         report = ValidationReport('schedule')
         if len(self.days) != len(set(self.days)):
-            report.add(ConfigurationError(
-                message='WeeklySchedule has duplicate days',
-                code=ErrorCode.CONFIG_INVALID_SCHEDULE,
-                notes=[f'days: {[d.value for d in self.days]}'],
-                help_text='each day should appear only once in the list',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='WeeklySchedule has duplicate days',
+                    code=ErrorCode.CONFIG_INVALID_SCHEDULE,
+                    notes=[f'days: {[d.value for d in self.days]}'],
+                    help_text='each day should appear only once in the list',
+                )
+            )
         raise_collected(report)
         return self
 
@@ -219,11 +228,13 @@ class ScheduleConfig(BaseModel):
         report = ValidationReport('schedule')
         names: list[str] = [s.name for s in self.schedules]
         if len(names) != len(set(names)):
-            report.add(ConfigurationError(
-                message='duplicate schedule names',
-                code=ErrorCode.CONFIG_INVALID_SCHEDULE,
-                notes=[f'schedule names: {names}'],
-                help_text='each schedule must have a unique name',
-            ))
+            report.add(
+                ConfigurationError(
+                    message='duplicate schedule names',
+                    code=ErrorCode.CONFIG_INVALID_SCHEDULE,
+                    notes=[f'schedule names: {names}'],
+                    help_text='each schedule must have a unique name',
+                )
+            )
         raise_collected(report)
         return self

horsies/core/models/tasks.py

@@ -22,6 +22,7 @@ if TYPE_CHECKING:
     from horsies.core.models.workflow import SubWorkflowSummary
 
 from horsies.core.types.status import TaskStatus
+
 T = TypeVar('T')  # success payload
 E = TypeVar('E')  # error payload (TaskError )