bladekit 4.5.0__py3-none-any.whl

bladekit/__init__.py

@@ -0,0 +1,219 @@
+"""Blade AI Fabric SDK — run Python functions on the GPU fleet.
+
+Public surface::
+
+    import bladekit
+
+    app = bladekit.app("demo")                 # same as bladekit.App("demo")
+
+    @app.function(gpu="RTX-A4500", image="fabric/torch-2.x")
+    def embed(texts):
+        ...
+
+    result = embed.remote(["hello"])           # blocking
+    handle = embed.spawn(["world"])            # non-blocking
+    value = handle.get(timeout=120)
+
+    # batch fan-out
+    for result in embed.map(["hello", "world"]):
+        print(result)
+
+    # cron schedule
+    @app.function(gpu="RTX-A4500", schedule=bladekit.Cron("0 3 * * *"))
+    def nightly(): ...
+
+    # imperative sandbox (for agents, code interpreters, LLM-generated code)
+    with bladekit.Sandbox.create(image="fabric/base-cuda12", gpu="RTX-A4500") as sb:
+        proc = sb.exec("python", "-c", "print('hello')")
+        print(proc.stdout.read())
+        proc.wait()
+
+Import is intentionally cheap: heavy deps (``cloudpickle``, ``httpx``,
+``websockets``) are imported lazily inside the functions that need them, so
+``import bladekit`` does not pull them in at module load.
+See ``tests/test_import_light.py``.
+
+This package was called ``forge`` (distribution ``forge-shadow``) before
+PAAS-771. The ``forge-shadow`` distribution still exists as a thin
+compatibility package that installs this one plus a ``forge`` import shim and a
+``forge`` command — see the README's migration section.
+"""
+
+from __future__ import annotations
+
+from ._app import (
+    App,
+    Autoscale,
+    DeployedFunctionCall,
+    DeploymentManifest,
+    Function,
+    FunctionCall,
+    complete_session,
+)
+from ._batch import BatchedFunction, batched
+from ._webhook import Webhook, verify_webhook
+from ._bucket import CloudBucketMount
+from ._cls import enter, exit, build, parameter
+from ._preempt import load_checkpoint, on_preempt, save_checkpoint
+from ._cron import Cron
+from ._dict import Dict
+from ._image import Image
+from ._map import MapHandle
+from ._queue import Queue
+from ._sandbox import Sandbox
+from ._secret import Secret
+from ._volume import Volume
+from ._env import mirror_legacy_runtime_env as _mirror_legacy_runtime_env
+from .exceptions import (
+    APIError,
+    AuthenticationError,
+    BundleFetchFailedError,
+    ConflictError,
+    ConnectionFailed,
+    ContainerFailedError,
+    FabricError,
+    GPUQuotaExceededError,
+    OrgGPUQuotaExceededError,
+    ImageBuildError,
+    ImagePullFailedError,
+    InputTimeoutError,
+    InvalidArgumentError,
+    NodeLostError,
+    NotFoundError,
+    PayloadTooLargeError,
+    PermissionDeniedError,
+    PreemptedError,
+    RemoteError,
+    RemoteFunctionError,
+    RetriesExhaustedError,
+    RunCancelled,
+    RunError,
+    RunLostError,
+    RunnerCrashedError,
+    UnavailableError,
+    VersionMismatchError,
+    VolumeCommitError,
+    VolumeError,
+    # sandbox errors
+    SandboxDiskQuotaExceededError,
+    SandboxExecLimitError,
+    SandboxFailedError,
+    SandboxFileNotFoundError,
+    SandboxFileTooLargeError,
+    SandboxNotRunningError,
+    SandboxPathInvalidError,
+    SandboxPortInUseError,
+    SandboxTerminatedError,
+)
+
+# `bladekit.app("name")`, the documented spelling: the App class itself under a
+# lowercase name, so it returns an App, isinstance(x, bladekit.app) holds and
+# App's signature and docstring come along unchanged.
+app = App
+
+# The PyPI distribution name (the import name is bladekit).
+DISTRIBUTION = "bladekit"
+
+# Inside a container the control plane still injects FORGE_* (PAAS-771): expose
+# the user-facing ones under their documented FABRIC_* names too.
+_mirror_legacy_runtime_env()
+
+# Derive the version from the installed package metadata so it can never drift
+# from pyproject.toml (the old hardcoded literal lagged for several releases,
+# making the client's warn_if_outdated compare a stale version and warn wrongly).
+try:
+    from importlib.metadata import version as _pkg_version
+
+    __version__ = _pkg_version(DISTRIBUTION)
+except Exception:  # editable/source checkout without installed metadata
+    __version__ = "0.0.0+local"
+
+__all__ = [
+    "App",
+    "app",
+    "Autoscale",
+    "Function",
+    "FunctionCall",
+    "DeployedFunctionCall",
+    "DeploymentManifest",
+    "Webhook",
+    "verify_webhook",
+    "complete_session",
+    # dynamic batching
+    "batched",
+    "BatchedFunction",
+    # object API
+    "Image",
+    "Volume",
+    "Secret",
+    "CloudBucketMount",
+    # container lifecycle hooks + parametrized classes
+    "enter",
+    "exit",
+    "build",
+    "parameter",
+    # preemption hook + checkpoint helpers
+    "on_preempt",
+    "save_checkpoint",
+    "load_checkpoint",
+    # object API
+    "Cron",
+    "Dict",
+    "MapHandle",
+    "Queue",
+    "__version__",
+    # exceptions
+    "FabricError",
+    "APIError",
+    "AuthenticationError",
+    "PermissionDeniedError",
+    "NotFoundError",
+    "InvalidArgumentError",
+    "ConflictError",
+    "PayloadTooLargeError",
+    "UnavailableError",
+    "ConnectionFailed",
+    "RunError",
+    "NodeLostError",
+    "ContainerFailedError",
+    "ImagePullFailedError",
+    "RunnerCrashedError",
+    "BundleFetchFailedError",
+    "VersionMismatchError",
+    "RunCancelled",
+    "RunLostError",
+    "RemoteError",
+    "RemoteFunctionError",
+    # exceptions
+    "ImageBuildError",
+    "VolumeError",
+    "VolumeCommitError",
+    # exceptions
+    "PreemptedError",
+    "InputTimeoutError",
+    "RetriesExhaustedError",
+    "GPUQuotaExceededError",
+    "OrgGPUQuotaExceededError",
+    # sandbox
+    "Sandbox",
+    # exceptions
+    "SandboxDiskQuotaExceededError",
+    "SandboxExecLimitError",
+    "SandboxFailedError",
+    "SandboxFileNotFoundError",
+    "SandboxFileTooLargeError",
+    "SandboxNotRunningError",
+    "SandboxPathInvalidError",
+    "SandboxPortInUseError",
+    "SandboxTerminatedError",
+]
+
+
+def __getattr__(name: str):
+    # Deprecated aliases (ForgeError -> FabricError) live in exceptions; resolving
+    # them there emits the DeprecationWarning and returns the same class object.
+    from . import exceptions as _exceptions
+
+    if name in _exceptions.DEPRECATED_ALIASES:
+        return _exceptions.resolve_deprecated(name)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")