Skip to content

Commit 3c17209

Browse files
committed
Simplify MultiOp response handling
1 parent 7beb8cb commit 3c17209

File tree

1 file changed

+2
-19
lines changed

1 file changed

+2
-19
lines changed

temporalio/client.py

Lines changed: 2 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@
2424
FrozenSet,
2525
Generic,
2626
Iterable,
27-
List,
2827
Mapping,
2928
Optional,
3029
Sequence,
@@ -6122,24 +6121,8 @@ async def _start_workflow_update_with_start(
61226121
multiop_response = (
61236122
await self._client.workflow_service.execute_multi_operation(multiop_req)
61246123
)
6125-
start_responses: List[
6126-
temporalio.api.workflowservice.v1.StartWorkflowExecutionResponse
6127-
] = [
6128-
r.start_workflow
6129-
for r in multiop_response.responses
6130-
if r.HasField("start_workflow")
6131-
]
6132-
update_responses: List[
6133-
temporalio.api.workflowservice.v1.UpdateWorkflowExecutionResponse
6134-
] = [
6135-
r.update_workflow
6136-
for r in multiop_response.responses
6137-
if r.HasField("update_workflow")
6138-
]
6139-
if not (len(start_responses) == 1 and len(update_responses) == 1):
6140-
raise RuntimeError("Invalid ExecuteMultiOperationResponse")
6141-
[start_response] = start_responses
6142-
[update_response] = update_responses
6124+
start_response = multiop_response.responses[0].start_workflow
6125+
update_response = multiop_response.responses[1].update_workflow
61436126
on_start(start_response)
61446127
known_outcome = (
61456128
update_response.outcome if update_response.HasField("outcome") else None

0 commit comments

Comments
 (0)