Respond to upstream: operation_handler is not in the public API

Author: dandavison

tests/nexus/test_handler.py

@@ -40,10 +40,10 @@
     OperationErrorState,
     OperationHandler,
     StartOperationContext,
-    operation_handler,
     service_handler,
     sync_operation,
 )
+from nexusrpc.handler._decorators import operation_handler
 
 import temporalio.api.failure.v1
 from temporalio import nexus, workflow

tests/nexus/test_handler_async_operation.py

@@ -23,7 +23,9 @@
     OperationHandler,
     StartOperationContext,
     StartOperationResultAsync,
+    service_handler,
 )
+from nexusrpc.handler._decorators import operation_handler
 
 from temporalio.testing import WorkflowEnvironment
 from temporalio.worker import Worker
@@ -109,21 +111,21 @@ def cancel(self, ctx: CancelOperationContext, token: str) -> None:
 
 
 @dataclass
-@nexusrpc.handler.service_handler
+@service_handler
 class MyServiceHandlerWithAsyncDefs:
     executor: TaskExecutor
 
-    @nexusrpc.handler.operation_handler
+    @operation_handler
     def async_operation(self) -> OperationHandler[Input, Output]:
         return AsyncOperationWithAsyncDefs(self.executor)
 
 
 @dataclass
-@nexusrpc.handler.service_handler
+@service_handler
 class MyServiceHandlerWithNonAsyncDefs:
     executor: TaskExecutor
 
-    @nexusrpc.handler.operation_handler
+    @operation_handler
     def async_operation(self) -> OperationHandler[Input, Output]:
         return AsyncOperationWithNonAsyncDefs(self.executor)
 

tests/nexus/test_workflow_caller.py

@@ -15,10 +15,10 @@
     StartOperationContext,
     StartOperationResultAsync,
     StartOperationResultSync,
-    operation_handler,
     service_handler,
     sync_operation,
 )
+from nexusrpc.handler._decorators import operation_handler
 
 import temporalio.api
 import temporalio.api.common

tests/nexus/test_workflow_run_operation.py

@@ -8,9 +8,9 @@
     OperationHandler,
     StartOperationContext,
     StartOperationResultAsync,
-    operation_handler,
     service_handler,
 )
+from nexusrpc.handler._decorators import operation_handler
 
 from temporalio import nexus, workflow
 from temporalio.nexus._operation_handlers import WorkflowRunOperationHandler