Feat/async

.github/actions/test/action.yml

@@ -16,3 +16,13 @@ runs:
     - name: run tests
       shell: bash
       run: poetry run just test
+    - name: generate report
+      if: failure()
+      shell: bash
+      run: poetry run coverage html
+    - name: upload report
+      if: failure()
+      uses: actions/upload-artifact@v4
+      with:
+        name: py-${{ inputs.python-version }}-${{ github.sha }}-coverage-report
+        path: htmlcov/

flake.nix

@@ -12,7 +12,7 @@
         {
           devShells.default = pkgs.mkShell {
             packages = [
-              pkgs.python312
+              pkgs.python313
               pkgs.just
               pkgs.poetry
             ];

justfile

@@ -1,5 +1,6 @@
 test:
-    coverage run --source=src -m pytest tests
+    coverage run -m pytest tests
+    coverage combine
     coverage report --fail-under=100
 
 lint:

pyproject.toml

@@ -18,7 +18,7 @@ ipython = "^8.17.2"
 pytest = "^7.4.3"
 pyright = "^1.1.336"
 pre-commit = "^3.5.0"
-ruff = "^0.1.6"
+ruff = "^0.14.2"
 coverage = "^7.3.2"
 toml = "^0.10.2"
 
@@ -34,13 +34,28 @@ warn_return_any = true
 strict_equality = true
 disallow_any_generics = true
 
-[tool.ruff]
+[[tool.mypy.overrides]]
+module = [
+    "cloudpickle",
+]
+ignore_missing_imports = true
+
+[tool.ruff.lint]
 select = ["I", "F", "N", "RUF", "D"]
 ignore = ["D107", "D213", "D203", "D202", "D212"]
 
-[tool.ruff.per-file-ignores]
+[tool.ruff.lint.per-file-ignores]
 "tests/**/*" = ["D100", "D101", "D102", "D103", "D104", "D105", "D107"]
 
+[tool.coverage.report]
+exclude_also = [
+    "@overload"
+]
+
+[tool.coverage.run]
+concurrency = ["multiprocessing", "thread"]
+source = ["src/"]
+
 
 [build-system]
 requires = ["poetry-core"]

src/stateless/__init__.py

@@ -2,20 +2,22 @@
 
 # ruff: noqa: F401
 
-from stateless.abilities import Abilities
+from stateless.ability import Ability
+from stateless.async_ import Async, Task, fork, wait
 from stateless.effect import (
     Depend,
     Effect,
     Success,
     Try,
     catch,
-    depend,
     memoize,
     run,
+    run_async,
     success,
     throw,
     throws,
 )
-from stateless.functions import repeat, retry
-from stateless.parallel import parallel, process
+from stateless.functions import as_type, repeat, retry
+from stateless.handler import Handler, handle
+from stateless.need import Need, need, supply
 from stateless.schedule import Schedule

src/stateless/ability.py

@@ -0,0 +1,21 @@
+"""Module containing the base ability type."""
+
+from typing import Generator, Generic, TypeVar
+
+from typing_extensions import Self
+
+from stateless.errors import MissingAbilityError
+
+T = TypeVar("T", covariant=True)
+
+
+class Ability(Generic[T]):
+    """The base ability type."""
+
+    def __iter__(self: Self) -> Generator[Self, T, T]:
+        """Depend on `self` and return the value of handling `self`."""
+        try:
+            v = yield self
+        except MissingAbilityError:
+            raise MissingAbilityError(self) from None
+        return v