Respond to upstream: {get,set}_operation_definition -> {get,set}_operation

Author: dandavison

temporalio/nexus/_decorators.py

@@ -123,7 +123,7 @@ async def _start(
             return WorkflowRunOperationHandler(_start, input_type, output_type)
 
         method_name = get_callable_name(start)
-        nexusrpc.set_operation_definition(
+        nexusrpc.set_operation(
             operation_handler_factory,
             nexusrpc.Operation(
                 name=name or method_name,

temporalio/nexus/_util.py

@@ -129,15 +129,15 @@ def get_operation_factory(
 
     ``obj`` should be a decorated operation start method.
     """
-    op_defn = nexusrpc.get_operation_definition(obj)
-    if op_defn:
+    op = nexusrpc.get_operation(obj)
+    if op:
         factory = obj
     else:
         if factory := getattr(obj, "__nexus_operation_factory__", None):
-            op_defn = nexusrpc.get_operation_definition(factory)
-    if not isinstance(op_defn, nexusrpc.Operation):
+            op = nexusrpc.get_operation(factory)
+    if not isinstance(op, nexusrpc.Operation):
         return None, None
-    return factory, op_defn
+    return factory, op
 
 
 # TODO(nexus-preview) Copied from nexusrpc

tests/nexus/test_dynamic_creation_of_user_handler_classes.py

@@ -78,8 +78,8 @@ async def test_run_nexus_service_from_programmatically_created_service_handler(
     service_handler = nexusrpc.handler._core.ServiceHandler(
         service=nexusrpc.ServiceDefinition(
             name="MyService",
-            operations={
-                "increment": nexusrpc.Operation[int, int](
+            operation_definitions={
+                "increment": nexusrpc.OperationDefinition[int, int](
                     name="increment",
                     method_name="increment",
                     input_type=int,

tests/nexus/test_handler.py

@@ -773,10 +773,7 @@ async def test_start_operation_without_type_annotations(
 
 
 def test_operation_without_type_annotations_without_service_definition_raises_validation_error():
-    with pytest.raises(
-        ValueError,
-        match=r"has no input type.+has no output type",
-    ):
+    with pytest.raises(ValueError, match=r"has no input type"):
         service_handler(MyServiceHandlerWithOperationsWithoutTypeAnnotations)