Don't expose underscore-prefixed context variables

Author: dandavison

temporalio/nexus/__init__.py

@@ -4,12 +4,6 @@
 from ._operation_context import (
     WorkflowRunOperationContext as WorkflowRunOperationContext,
 )
-from ._operation_context import (
-    _TemporalCancelOperationContext as _TemporalCancelOperationContext,
-)
-from ._operation_context import (
-    _TemporalStartOperationContext as _TemporalStartOperationContext,
-)
 from ._operation_context import client as client
 from ._operation_context import in_operation as in_operation
 from ._operation_context import info as info

temporalio/worker/_nexus.py

@@ -172,7 +172,7 @@ async def _handle_cancel_operation_task(
             operation=request.operation,
             headers=headers,
         )
-        temporalio.nexus._TemporalCancelOperationContext(
+        temporalio.nexus._operation_context._TemporalCancelOperationContext(
             info=lambda: Info(task_queue=self._task_queue),
             nexus_context=ctx,
             client=self._client,
@@ -274,7 +274,7 @@ async def _start_operation(
             ],
             callback_headers=dict(start_request.callback_header),
         )
-        temporalio.nexus._TemporalStartOperationContext(
+        temporalio.nexus._operation_context._TemporalStartOperationContext(
             nexus_context=ctx,
             client=self._client,
             info=lambda: Info(task_queue=self._task_queue),