RTU: ctx.start_workflow()

Author: dandavison

hello_nexus/basic/handler/service_handler.py

@@ -8,7 +8,7 @@
 
 from nexusrpc.handler import StartOperationContext, service_handler, sync_operation
 from temporalio import nexus
-from temporalio.nexus import workflow_run_operation
+from temporalio.nexus import WorkflowRunOperationContext, workflow_run_operation
 
 from hello_nexus.basic.handler.db_client import MyDBClient
 from hello_nexus.basic.handler.workflows import WorkflowStartedByNexusOperation
@@ -35,10 +35,10 @@ def __init__(self, connected_db_client: MyDBClient):
     # operation.
     @workflow_run_operation
     async def my_workflow_run_operation(
-        self, ctx: StartOperationContext, input: MyInput
+        self, ctx: WorkflowRunOperationContext, input: MyInput
     ) -> nexus.WorkflowHandle[MyOutput]:
         # You could use self.connected_db_client here.
-        return await nexus.start_workflow(
+        return await ctx.start_workflow(
             WorkflowStartedByNexusOperation.run,
             input,
             id=str(uuid.uuid4()),

hello_nexus/without_service_definition/app.py

@@ -10,12 +10,11 @@
 from typing import Optional
 
 from nexusrpc.handler import (
-    StartOperationContext,
     service_handler,
 )
 from temporalio import nexus, workflow
 from temporalio.client import Client
-from temporalio.nexus import workflow_run_operation
+from temporalio.nexus import WorkflowRunOperationContext, workflow_run_operation
 from temporalio.worker import UnsandboxedWorkflowRunner, Worker
 from temporalio.workflow import NexusClient
 
@@ -40,14 +39,14 @@ async def run(self, message: str) -> str:
 @service_handler
 class MyNexusService:
     # Here we implement a Nexus operation backed by a Temporal workflow. The start
-    # method must use nexus.start_workflow to start the workflow. (Temporal server will
-    # then take care of delivering the workflow result to the caller, using the Nexus
-    # RPC callback mechanism).
+    # method must use WorkflowRunOperationContext.start_workflow to start the workflow.
+    # (Temporal server will then take care of delivering the workflow result to the
+    # caller, using the Nexus RPC callback mechanism).
     @workflow_run_operation
     async def my_workflow_run_operation(
-        self, ctx: StartOperationContext, name: str
+        self, ctx: WorkflowRunOperationContext, name: str
     ) -> nexus.WorkflowHandle[str]:
-        return await nexus.start_workflow(
+        return await ctx.start_workflow(
             HandlerWorkflow.run,
             name,
             id=str(uuid.uuid4()),