Simplify MultiOp response handling

temporalio · Dec 18, 2024 · 3c17209 · 3c17209
1 parent 7beb8cb
commit 3c17209
Showing 1 changed file with 2 additions and 19 deletions.
diff --git a/temporalio/client.py b/temporalio/client.py
@@ -24,7 +24,6 @@
     FrozenSet,
     Generic,
     Iterable,
-    List,
     Mapping,
     Optional,
     Sequence,
@@ -6122,24 +6121,8 @@ async def _start_workflow_update_with_start(
             multiop_response = (
                 await self._client.workflow_service.execute_multi_operation(multiop_req)
             )
-            start_responses: List[
-                temporalio.api.workflowservice.v1.StartWorkflowExecutionResponse
-            ] = [
-                r.start_workflow
-                for r in multiop_response.responses
-                if r.HasField("start_workflow")
-            ]
-            update_responses: List[
-                temporalio.api.workflowservice.v1.UpdateWorkflowExecutionResponse
-            ] = [
-                r.update_workflow
-                for r in multiop_response.responses
-                if r.HasField("update_workflow")
-            ]
-            if not (len(start_responses) == 1 and len(update_responses) == 1):
-                raise RuntimeError("Invalid ExecuteMultiOperationResponse")
-            [start_response] = start_responses
-            [update_response] = update_responses
+            start_response = multiop_response.responses[0].start_workflow
+            update_response = multiop_response.responses[1].update_workflow
             on_start(start_response)
             known_outcome = (
                 update_response.outcome if update_response.HasField("outcome") else None