creae exclusive aio package for all the async code, also the same for tests

Signed-off-by: Filinto Duran <[email protected]>

Author: filintod

.pre-commit-config.yaml

@@ -0,0 +1,38 @@
+# Pre-commit configuration for durabletask-python
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.4.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-yaml
+      - id: check-added-large-files
+      - id: check-merge-conflict
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.2.2
+    hooks:
+      - id: ruff
+        args: [--fix, --exit-non-zero-on-fix]
+      - id: ruff-format
+
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.8.0
+    hooks:
+      - id: mypy
+        additional_dependencies: [types-protobuf]
+        args: [--config-file=mypy.ini]
+        files: ^durabletask/
+        exclude: ^durabletask/internal/.*_pb2\.py$
+
+  - repo: local
+    hooks:
+      - id: pytest-asyncio
+        name: Run asyncio tests
+        entry: python -m pytest tests/aio/ -q
+        language: system
+        pass_filenames: false
+        stages: [pre-push]  # Only run on git push, not every commit
+
+
+

durabletask/__init__.py

@@ -6,8 +6,8 @@
 
 PACKAGE_NAME = "durabletask"
 
-# Public async compatibility exports
-from durabletask.asyncio_compat import (  # noqa: F401
-    AsyncWorkflowContext,
-    CoroutineOrchestratorRunner,
+# Public async exports (import directly from durabletask.aio)
+from durabletask.aio import (  # noqa: F401
+                             AsyncWorkflowContext,
+                             CoroutineOrchestratorRunner,
 )

durabletask/aio/__init__.py

@@ -0,0 +1,102 @@
+"""Async workflow primitives (aio namespace).
+
+This package contains the async implementation previously under
+`durabletask.asyncio`, now moved to `durabletask.aio` for naming
+consistency.
+"""
+
+# Deterministic utilities
+from durabletask.deterministic import (
+                                       DeterminismSeed,
+                                       DeterministicContextMixin,
+                                       derive_seed,
+                                       deterministic_random,
+                                       deterministic_uuid4,
+)
+
+# Awaitable classes
+from .awaitables import (
+                                       ActivityAwaitable,
+                                       AwaitableBase,
+                                       ExternalEventAwaitable,
+                                       SleepAwaitable,
+                                       SubOrchestratorAwaitable,
+                                       SwallowExceptionAwaitable,
+                                       TimeoutAwaitable,
+                                       WhenAllAwaitable,
+                                       WhenAnyAwaitable,
+                                       WhenAnyResultAwaitable,
+                                       gather,
+)
+
+# Compatibility protocol (core functionality only)
+from .compatibility import OrchestrationContextProtocol, ensure_compatibility
+
+# Core context and driver
+from .context import AsyncWorkflowContext
+from .driver import CoroutineOrchestratorRunner, WorkflowFunction
+
+# Sandbox and error handling
+from .errors import (
+                                       AsyncWorkflowError,
+                                       NonDeterminismWarning,
+                                       SandboxViolationError,
+                                       WorkflowTimeoutError,
+                                       WorkflowValidationError,
+)
+from .sandbox import (
+                                       SandboxMode,
+                                       _NonDeterminismDetector,
+                                       sandbox_best_effort,
+                                       sandbox_off,
+                                       sandbox_scope,
+                                       sandbox_strict,
+)
+
+__all__ = [
+    # Core classes
+    "AsyncWorkflowContext",
+    "CoroutineOrchestratorRunner",
+    "WorkflowFunction",
+
+    # Deterministic utilities
+    "DeterministicContextMixin",
+    "DeterminismSeed",
+    "derive_seed",
+    "deterministic_random",
+    "deterministic_uuid4",
+
+    # Awaitable classes
+    "AwaitableBase",
+    "ActivityAwaitable",
+    "SubOrchestratorAwaitable",
+    "SleepAwaitable",
+    "ExternalEventAwaitable",
+    "WhenAllAwaitable",
+    "WhenAnyAwaitable",
+    "WhenAnyResultAwaitable",
+    "TimeoutAwaitable",
+    "SwallowExceptionAwaitable",
+    "gather",
+
+    # Sandbox and utilities
+    "sandbox_scope",
+    "SandboxMode",
+    "sandbox_off",
+    "sandbox_best_effort",
+    "sandbox_strict",
+    "_NonDeterminismDetector",
+
+    # Compatibility protocol
+    "OrchestrationContextProtocol",
+    "ensure_compatibility",
+
+    # Exceptions
+    "AsyncWorkflowError",
+    "NonDeterminismWarning",
+    "WorkflowTimeoutError",
+    "WorkflowValidationError",
+    "SandboxViolationError",
+]
+
+