Fixup: eliminate references to WorkflowOperationToken

Author: dandavison

temporalio/nexus/_operation_handlers.py

@@ -42,26 +42,6 @@ class WorkflowRunOperationHandler(OperationHandler[InputT, OutputT]):
     Use this class to create an operation handler that starts a workflow by passing your
     ``start`` method to the constructor. Your ``start`` method must use
     :py:func:`temporalio.nexus.start_workflow` to start the workflow.
-
-    Example:
-
-    .. code-block:: python
-
-        @service_handler(service=MyNexusService)
-        class MyNexusServiceHandler:
-            @operation_handler
-            def my_workflow_run_operation(
-                self,
-            ) -> OperationHandler[MyInput, MyOutput]:
-                async def start(
-                    ctx: StartOperationContext, input: MyInput
-                ) -> WorkflowOperationToken[MyOutput]:
-                    return await start_workflow(
-                        WorkflowStartedByNexusOperation.run, input,
-                        id=str(uuid.uuid4()),
-                    )
-
-                return WorkflowRunOperationHandler.from_start_workflow(start)
     """
 
     def __init__(
@@ -95,13 +75,13 @@ async def start(
         if not isinstance(handle, WorkflowHandle):
             if isinstance(handle, client.WorkflowHandle):
                 raise RuntimeError(
-                    f"Expected {handle} to be a WorkflowOperationToken, but got a client.WorkflowHandle. "
-                    f"You must use temporalio.nexus.start_workflow "
+                    f"Expected {handle} to be a nexus.WorkflowHandle, but got a client.WorkflowHandle. "
+                    f"You must use WorkflowRunOperationContext.start_workflow "
                     "to start a workflow that will deliver the result of the Nexus operation, "
                     "not client.Client.start_workflow."
                 )
             raise RuntimeError(
-                f"Expected {handle} to be a WorkflowOperationToken, but got {type(handle)}. "
+                f"Expected {handle} to be a nexus.WorkflowHandle, but got {type(handle)}. "
             )
         return StartOperationResultAsync(handle.to_token())
 

temporalio/nexus/_util.py

@@ -46,7 +46,7 @@ def get_workflow_run_start_method_input_and_output_type_annotations(
         output_type = None
     elif not issubclass(origin_type, WorkflowHandle):
         warnings.warn(
-            f"Expected return type of {start.__name__} to be a subclass of WorkflowOperationToken, "
+            f"Expected return type of {start.__name__} to be a subclass of WorkflowHandle, "
             f"but is {output_type}"
         )
         output_type = None

temporalio/workflow.py

@@ -859,7 +859,7 @@ async def workflow_start_nexus_operation(
             Callable[[Any], nexusrpc.handler.OperationHandler[I, O]],
             Callable[
                 [S, nexusrpc.handler.StartOperationContext, I],
-                Awaitable[temporalio.nexus.WorkflowOperationToken[O]],
+                Awaitable[temporalio.nexus.WorkflowHandle[O]],
             ],
             str,
         ],
@@ -4429,7 +4429,7 @@ async def start_nexus_operation(
         Callable[[Any], nexusrpc.handler.OperationHandler[I, O]],
         Callable[
             [S, nexusrpc.handler.StartOperationContext, I],
-            Awaitable[temporalio.nexus.WorkflowOperationToken[O]],
+            Awaitable[temporalio.nexus.WorkflowHandle[O]],
         ],
         str,
     ],
@@ -5215,7 +5215,7 @@ async def start_operation(
             Callable[[S], nexusrpc.handler.OperationHandler[I, O]],
             Callable[
                 [S, nexusrpc.handler.StartOperationContext, I],
-                Awaitable[temporalio.nexus.WorkflowOperationToken[O]],
+                Awaitable[temporalio.nexus.WorkflowHandle[O]],
             ],
             str,
         ],
@@ -5243,7 +5243,7 @@ async def execute_operation(
             Callable[[S], nexusrpc.handler.OperationHandler[I, O]],
             Callable[
                 [S, nexusrpc.handler.StartOperationContext, I],
-                Awaitable[temporalio.nexus.WorkflowOperationToken[O]],
+                Awaitable[temporalio.nexus.WorkflowHandle[O]],
             ],
             str,
         ],