Skip to content

Commit 0b37fd2

Browse files
committed
Simplify MultiOp response handling
1 parent 312a548 commit 0b37fd2

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,
@@ -6123,24 +6122,8 @@ async def _start_workflow_update_with_start(
61236122
multiop_response = (
61246123
await self._client.workflow_service.execute_multi_operation(multiop_req)
61256124
)
6126-
start_responses: List[
6127-
temporalio.api.workflowservice.v1.StartWorkflowExecutionResponse
6128-
] = [
6129-
r.start_workflow
6130-
for r in multiop_response.responses
6131-
if r.HasField("start_workflow")
6132-
]
6133-
update_responses: List[
6134-
temporalio.api.workflowservice.v1.UpdateWorkflowExecutionResponse
6135-
] = [
6136-
r.update_workflow
6137-
for r in multiop_response.responses
6138-
if r.HasField("update_workflow")
6139-
]
6140-
if not (len(start_responses) == 1 and len(update_responses) == 1):
6141-
raise RuntimeError("Invalid ExecuteMultiOperationResponse")
6142-
[start_response] = start_responses
6143-
[update_response] = update_responses
6125+
start_response = multiop_response.responses[0].start_workflow
6126+
update_response = multiop_response.responses[1].update_workflow
61446127
on_start(start_response)
61456128
known_outcome = (
61466129
update_response.outcome if update_response.HasField("outcome") else None

0 commit comments

Comments
 (0)