@@ -537,6 +537,7 @@ class ReplayDecider:
537537 decision_events : DecisionEvents = None
538538 decisions : OrderedDict [DecisionId , DecisionStateMachine ] = field (default_factory = OrderedDict )
539539 decision_context : DecisionContext = None
540+ workflow_id : str = None
540541
541542 activity_id_to_scheduled_event_id : Dict [str , int ] = field (default_factory = dict )
542543
@@ -910,14 +911,16 @@ def poll(self) -> Optional[PollForDecisionTaskResponse]:
910911
911912 def process_task (self , decision_task : PollForDecisionTaskResponse ) -> List [Decision ]:
912913 execution_id = str (decision_task .workflow_execution )
913- decider = ReplayDecider (execution_id , decision_task .workflow_type , self .worker )
914+ decider = ReplayDecider (execution_id , decision_task .workflow_type , self .worker ,
915+ workflow_id = decision_task .workflow_execution .workflow_id )
914916 decisions : List [Decision ] = decider .decide (decision_task .history .events )
915917 decider .destroy ()
916918 return decisions
917919
918920 def process_query (self , decision_task : PollForDecisionTaskResponse ) -> bytes :
919921 execution_id = str (decision_task .workflow_execution )
920- decider = ReplayDecider (execution_id , decision_task .workflow_type , self .worker )
922+ decider = ReplayDecider (execution_id , decision_task .workflow_type , self .worker ,
923+ workflow_id = decision_task .workflow_execution .workflow_id )
921924 decider .decide (decision_task .history .events )
922925 try :
923926 result = decider .query (decision_task , decision_task .query )
0 commit comments