Merge branch 'main' into d-runner

Author: timl3136

README.md

@@ -26,7 +26,7 @@ cd cadence-python-client
    ```bash
    # macOS
    brew install protobuf@29
-   
+
    # Linux/Other
    # Install protobuf 29.x via your package manager
    ```
@@ -35,7 +35,7 @@ cd cadence-python-client
    ```bash
    # macOS
    brew install uv
-   
+
    # Linux/Other
    curl -LsSf https://astral.sh/uv/install.sh | sh
    source $HOME/.local/bin/env  # Add to your shell profile for persistence
@@ -59,6 +59,7 @@ cd cadence-python-client
 Run the generation script:
 ```bash
 # Using uv (recommended)
+uv sync --extra dev
 uv run python scripts/generate_proto.py
 
 # Or using traditional Python

cadence/_internal/decision_state_machine.py

@@ -2,7 +2,7 @@
 
 from dataclasses import dataclass, field
 from enum import Enum
-from typing import Dict, List, Optional, Callable
+from typing import Dict, List, Optional, Callable, TypedDict, Literal
 
 from cadence.api.v1 import (
     decision_pb2 as decision,
@@ -81,12 +81,18 @@ def __str__(self) -> str:
 @dataclass
 class StateTransition:
     """Represents a state transition with associated actions."""
-    next_state: DecisionState
+    next_state: Optional[DecisionState]
     action: Optional[Callable[['BaseDecisionStateMachine', history.HistoryEvent], None]] = None
     condition: Optional[Callable[['BaseDecisionStateMachine', history.HistoryEvent], bool]] = None
 
 
-decision_state_transition_map = {
+class TransitionInfo(TypedDict):
+    type: Literal["initiated", "started", "completion", "canceled", "cancel_initiated", "cancel_failed", "initiation_failed"]
+    decision_type: DecisionType
+    transition: StateTransition
+
+
+decision_state_transition_map: Dict[str, TransitionInfo] = {
     "activity_task_scheduled_event_attributes": {
         "type": "initiated",
         "decision_type": DecisionType.ACTIVITY,
@@ -247,6 +253,10 @@ class BaseDecisionStateMachine:
     Subclasses are responsible for mapping workflow history events into state
     transitions and producing the next set of decisions when queried.
     """
+    
+    # Common fields that subclasses may use
+    scheduled_event_id: Optional[int] = None
+    started_event_id: Optional[int] = None
 
     def get_id(self) -> str:
         raise NotImplementedError
@@ -890,12 +900,12 @@ def handle_history_event(self, event: history.HistoryEvent) -> None:
         if transition_info:
             event_type = transition_info["type"]
             # Route to all relevant machines using the new unified handle_event method
-            for m in list(self.activities.values()):
-                m.handle_event(event, event_type)
-            for m in list(self.timers.values()):
-                m.handle_event(event, event_type)
-            for m in list(self.children.values()):
-                m.handle_event(event, event_type)
+            for activity_machine in list(self.activities.values()):
+                activity_machine.handle_event(event, event_type)
+            for timer_machine in list(self.timers.values()):
+                timer_machine.handle_event(event, event_type)
+            for child_machine in list(self.children.values()):
+                child_machine.handle_event(event, event_type)
 
     # ----- Decision aggregation -----
 
@@ -907,11 +917,11 @@ def collect_pending_decisions(self) -> List[decision.Decision]:
             decisions.extend(machine.collect_pending_decisions())
 
         # Timers
-        for machine in list(self.timers.values()):
-            decisions.extend(machine.collect_pending_decisions())
+        for timer_machine in list(self.timers.values()):
+            decisions.extend(timer_machine.collect_pending_decisions())
 
         # Children
-        for machine in list(self.children.values()):
-            decisions.extend(machine.collect_pending_decisions())
+        for child_machine in list(self.children.values()):
+            decisions.extend(child_machine.collect_pending_decisions())
 
         return decisions

cadence/_internal/rpc/metadata.py

@@ -1,4 +1,5 @@
 import collections
+from typing import Any, Callable
 
 from grpc.aio import Metadata
 from grpc.aio import UnaryUnaryClientInterceptor, ClientCallDetails
@@ -16,7 +17,12 @@ class MetadataInterceptor(UnaryUnaryClientInterceptor):
     def __init__(self, metadata: Metadata):
         self._metadata = metadata
 
-    async def intercept_unary_unary(self, continuation, client_call_details: ClientCallDetails, request):
+    async def intercept_unary_unary(
+        self, 
+        continuation: Callable[[ClientCallDetails, Any], Any], 
+        client_call_details: ClientCallDetails, 
+        request: Any
+    ) -> Any:
         return await continuation(self._replace_details(client_call_details), request)
 
 

cadence/api/__init__.py

@@ -1,92 +1,92 @@
 # Auto-generated __init__.py file
-# Import all generated protobuf and gRPC modules
+# Import all generated protobuf modules
 from . import common_pb2
-from . import decision_pb2
-from . import domain_pb2
-from . import error_pb2
-from . import history_pb2
-from . import query_pb2
-from . import service_domain_pb2
-from . import service_meta_pb2
-from . import service_visibility_pb2
-from . import service_worker_pb2
-from . import service_workflow_pb2
-from . import tasklist_pb2
-from . import visibility_pb2
-from . import workflow_pb2
 from . import common_pb2_grpc
+from . import decision_pb2
 from . import decision_pb2_grpc
+from . import domain_pb2
 from . import domain_pb2_grpc
+from . import error_pb2
 from . import error_pb2_grpc
+from . import history_pb2
 from . import history_pb2_grpc
+from . import query_pb2
 from . import query_pb2_grpc
+from . import service_domain_pb2
 from . import service_domain_pb2_grpc
+from . import service_meta_pb2
 from . import service_meta_pb2_grpc
+from . import service_visibility_pb2
 from . import service_visibility_pb2_grpc
+from . import service_worker_pb2
 from . import service_worker_pb2_grpc
+from . import service_workflow_pb2
 from . import service_workflow_pb2_grpc
+from . import tasklist_pb2
 from . import tasklist_pb2_grpc
+from . import visibility_pb2
 from . import visibility_pb2_grpc
+from . import workflow_pb2
 from . import workflow_pb2_grpc
 
 # Create cleaner aliases for easier imports
 common = common_pb2
-decision = decision_pb2
-domain = domain_pb2
-error = error_pb2
-history = history_pb2
-query = query_pb2
-service_domain = service_domain_pb2
-service_meta = service_meta_pb2
-service_visibility = service_visibility_pb2
-service_worker = service_worker_pb2
-service_workflow = service_workflow_pb2
-tasklist = tasklist_pb2
-visibility = visibility_pb2
-workflow = workflow_pb2
 common_grpc = common_pb2_grpc
+decision = decision_pb2
 decision_grpc = decision_pb2_grpc
+domain = domain_pb2
 domain_grpc = domain_pb2_grpc
+error = error_pb2
 error_grpc = error_pb2_grpc
+history = history_pb2
 history_grpc = history_pb2_grpc
+query = query_pb2
 query_grpc = query_pb2_grpc
+service_domain = service_domain_pb2
 service_domain_grpc = service_domain_pb2_grpc
+service_meta = service_meta_pb2
 service_meta_grpc = service_meta_pb2_grpc
+service_visibility = service_visibility_pb2
 service_visibility_grpc = service_visibility_pb2_grpc
+service_worker = service_worker_pb2
 service_worker_grpc = service_worker_pb2_grpc
+service_workflow = service_workflow_pb2
 service_workflow_grpc = service_workflow_pb2_grpc
+tasklist = tasklist_pb2
 tasklist_grpc = tasklist_pb2_grpc
+visibility = visibility_pb2
 visibility_grpc = visibility_pb2_grpc
+workflow = workflow_pb2
 workflow_grpc = workflow_pb2_grpc
 
-# Only expose clean module names
+# Only expose clean module names (no _pb2)
 __all__ = [
     'common',
-    'decision',
-    'domain',
-    'error',
-    'history',
-    'query',
-    'service_domain',
-    'service_meta',
-    'service_visibility',
-    'service_worker',
-    'service_workflow',
-    'tasklist',
-    'visibility',
-    'workflow',
     'common_grpc',
+    'decision',
     'decision_grpc',
+    'domain',
     'domain_grpc',
+    'error',
     'error_grpc',
+    'history',
     'history_grpc',
+    'query',
     'query_grpc',
+    'service_domain',
     'service_domain_grpc',
+    'service_meta',
     'service_meta_grpc',
+    'service_visibility',
     'service_visibility_grpc',
+    'service_worker',
     'service_worker_grpc',
+    'service_workflow',
     'service_workflow_grpc',
+    'tasklist',
     'tasklist_grpc',
+    'visibility',
     'visibility_grpc',
+    'workflow',
     'workflow_grpc',
 ]