File tree Expand file tree Collapse file tree 2 files changed +6
-7
lines changed Expand file tree Collapse file tree 2 files changed +6
-7
lines changed Original file line number Diff line number Diff line change @@ -1208,7 +1208,6 @@ def on_start_error(
12081208 start_workflow_input = start_workflow_operation ._start_workflow_input ,
12091209 update_workflow_input = update_input ,
12101210 _on_start = on_start ,
1211- headers = {},
12121211 _on_start_error = on_start_error ,
12131212 )
12141213
@@ -5539,7 +5538,6 @@ class StartWorkflowUpdateWithStartInput:
55395538 [temporalio .api .workflowservice .v1 .StartWorkflowExecutionResponse ], None
55405539 ]
55415540 _on_start_error : Callable [[BaseException ], None ]
5542- headers : Mapping [str , temporalio .api .common .v1 .Payload ]
55435541
55445542
55455543@dataclass
@@ -6363,10 +6361,6 @@ def on_start(
63636361
63646362 err : Optional [BaseException ] = None
63656363
6366- # fan headers out to both operations
6367- input .start_workflow_input .headers = input .headers
6368- input .update_workflow_input .headers = input .headers
6369-
63706364 try :
63716365 return await self ._start_workflow_update_with_start (
63726366 input .start_workflow_input , input .update_workflow_input , on_start
Original file line number Diff line number Diff line change @@ -304,9 +304,14 @@ async def start_update_with_start_workflow(
304304 with self .root ._start_as_current_span (
305305 f"StartUpdateWithStartWorkflow:{ input .start_workflow_input .workflow } " ,
306306 attributes = attrs ,
307- input = input ,
307+ input = input . start_workflow_input ,
308308 kind = opentelemetry .trace .SpanKind .CLIENT ,
309309 ):
310+ if input .update_workflow_input :
311+ input .update_workflow_input .headers = self .root ._context_to_headers (
312+ input .update_workflow_input .headers
313+ )
314+
310315 return await super ().start_update_with_start_workflow (input )
311316
312317
You can’t perform that action at this time.
0 commit comments