Coverage for async-durable-execution/src/async_durable_execution/runner/local/processors/__init__.py: 100%

1"""Checkpoint processors module.""" 

2 

3from __future__ import annotations 

4 

5from ....models import ( 

6 OperationType, 

7) 

8from .base import ( 

9 OperationProcessor, 

10) 

11from .callback import ( 

12 CallbackProcessor, 

13) 

14from .context import ( 

15 ContextProcessor, 

16) 

17from .execution import ( 

18 ExecutionProcessor, 

19) 

20from .invoke import ( 

21 ChainedInvokeProcessor, 

22) 

23from .step import ( 

24 StepProcessor, 

25) 

26from .wait import ( 

27 WaitProcessor, 

28) 

29 

30 

31def create_default_processors() -> dict[OperationType, OperationProcessor]: 

32 """Create the default operation processors keyed by operation type.""" 

33 return { 

34 OperationType.STEP: StepProcessor(), 

35 OperationType.WAIT: WaitProcessor(), 

36 OperationType.CONTEXT: ContextProcessor(), 

37 OperationType.CALLBACK: CallbackProcessor(), 

38 OperationType.EXECUTION: ExecutionProcessor(), 

39 OperationType.CHAINED_INVOKE: ChainedInvokeProcessor(), 

40 }