Respond to upstream: temporal_operation_context

Author: dandavison

hello_nexus/basic/handler/service_handler.py

@@ -21,9 +21,9 @@
     service_handler,
 )
 from temporalio.nexus.handler import (
-    TemporalOperationContext,
     WorkflowOperationToken,
     WorkflowRunOperationHandler,
+    temporal_operation_context,
 )
 
 from hello_nexus.basic.handler.db_client import MyDBClient
@@ -58,7 +58,7 @@ async def start(
             ctx: StartOperationContext, input: MyInput
         ) -> WorkflowOperationToken[MyOutput]:
             # You could use self.connected_db_client here.
-            tctx = TemporalOperationContext.current()
+            tctx = temporal_operation_context.get()
             return await tctx.start_workflow(
                 WorkflowStartedByNexusOperation.run,
                 input,

hello_nexus/basic/handler/service_handler_with_operation_handler_classes.py

@@ -33,8 +33,8 @@ class directly.
     service_handler,
 )
 from temporalio.nexus.handler import (
-    TemporalOperationContext,
     cancel_operation,
+    temporal_operation_context,
 )
 
 from hello_nexus.basic.handler.db_client import MyDBClient
@@ -130,7 +130,7 @@ class MyWorkflowRunOperation(OperationHandler[MyInput, MyOutput]):
     async def start(
         self, ctx: StartOperationContext, input: MyInput
     ) -> StartOperationResultAsync:
-        tctx = TemporalOperationContext.current()
+        tctx = temporal_operation_context.get()
         token = await tctx.start_workflow(
             WorkflowStartedByNexusOperation.run,
             input,

hello_nexus/without_service_definition/app.py

@@ -18,9 +18,9 @@
 from temporalio import workflow
 from temporalio.client import Client
 from temporalio.nexus.handler import (
-    TemporalOperationContext,
     WorkflowOperationToken,
     WorkflowRunOperationHandler,
+    temporal_operation_context,
 )
 from temporalio.worker import UnsandboxedWorkflowRunner, Worker
 from temporalio.workflow import NexusClient
@@ -57,7 +57,7 @@ def my_workflow_run_operation(
         async def start(
             ctx: StartOperationContext, name: str
         ) -> WorkflowOperationToken[str]:
-            tctx = TemporalOperationContext.current()
+            tctx = temporal_operation_context.get()
             return await tctx.start_workflow(
                 HandlerWorkflow.run,
                 name,