baseten-bdn 0.1.0__tar.gz

baseten_bdn-0.1.0/.github/workflows/ci.yml

@@ -0,0 +1,36 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+
+jobs:
+  check:
+    strategy:
+      matrix:
+        os: [ubuntu-latest, macos-latest, windows-latest]
+        python-version: ["3.14"]
+        include:
+          - os: ubuntu-latest
+            python-version: "3.11"
+    runs-on: ${{ matrix.os }}
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: astral-sh/setup-uv@v6
+
+      - uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+          allow-prereleases: true
+
+      - run: uv sync
+
+      - name: Lint
+        run: |
+          uv run poe lint
+          uv run poe typecheck
+
+      - name: Test
+        run: uv run poe test

baseten_bdn-0.1.0/.github/workflows/publish.yml

@@ -0,0 +1,20 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+    permissions:
+      id-token: write
+      contents: read
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: astral-sh/setup-uv@v6
+
+      - run: uv build
+
+      - uses: pypa/gh-action-pypi-publish@release/v1

baseten_bdn-0.1.0/.gitignore

@@ -0,0 +1,6 @@
+.pytest_cache
+.ruff_cache
+.venv
+__pycache__
+dist
+.ai/

baseten_bdn-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Baseten
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

baseten_bdn-0.1.0/PKG-INFO

@@ -0,0 +1,94 @@
+Metadata-Version: 2.5
+Name: baseten-bdn
+Version: 0.1.0
+Summary: Python SDK for working with BDN inside and outside Baseten
+Project-URL: Homepage, https://github.com/basetenlabs/baseten-bdn
+Project-URL: Issues, https://github.com/basetenlabs/baseten-bdn/issues
+License-Expression: MIT
+License-File: LICENSE
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Requires-Python: >=3.11
+Requires-Dist: backports-zstd>=1.7.0; python_full_version < '3.14'
+Requires-Dist: baseten<1,>=0.11
+Requires-Dist: blake3>=1.0.9
+Requires-Dist: httpx>=0.27
+Requires-Dist: pydantic>=2.0
+Description-Content-Type: text/markdown
+
+# baseten-bdn
+
+Python SDK for working with BDN inside and outside Baseten.
+
+This package is under initial development. It installs as `baseten-bdn` and
+imports as `baseten.bdn`. The `baseten` package is a namespace shared with
+other Baseten SDKs such as `baseten` and `baseten-loops`.
+
+## Installation
+
+```
+pip install baseten-bdn
+```
+
+## Hot Load
+
+Inside a model pod that is opted in to BDN Hot Load, `baseten.bdn.hotload`
+attaches BDN volumes below `/bdn/mounts` at runtime. An attach returns once the
+directory is readable.
+
+```python
+from baseten.bdn.hotload import HotLoadClient
+
+with HotLoadClient() as hotload:
+    adapter = hotload.attach("bdn:adapters/sql-lora@b3:2547…", target="sql-lora")
+
+load_adapter(adapter.path)  # /bdn/mounts/sql-lora
+```
+
+`AsyncHotLoadClient` offers the same methods as coroutines. Attachments belong
+to the pod and stay mounted after the client is closed; call `detach` to free a
+target name for reuse.
+
+## Volumes
+
+From any machine with a Baseten API key, `baseten.bdn.volumes` reads a volume
+version's manifest or pulls a version into a local directory. The client mints a short-lived token through the Baseten API,
+resolves the ref, and reads the volume's objects directly from storage with
+the credentials it is given, verifying every object against its recorded
+digest.
+
+```python
+from baseten.bdn.volumes import VolumeClient
+
+with VolumeClient(api_key="...") as volumes:
+    result = volumes.pull("bdn:loops/sampler-abc123:step-100", "./checkpoint")
+
+print(result.version_ref, result.file_count, result.bytes_written)
+```
+
+Refs are `bdn:<namespace>/<volume>` with an optional `:<tag>` or `@<digest>`
+selector and an optional `/path` inside the version; `VolumeRef` parses and
+renders them. A path on the ref, or `include=[...]`, narrows a pull to those
+entries without moving them. A pull is staged next to `dest_dir` and renamed
+into place once complete, so a failed pull leaves nothing behind; pass
+`overwrite=True` to write into an existing directory in place. Pulling is
+supported on Linux and macOS.
+
+## Development
+
+Requires [uv](https://docs.astral.sh/uv/).
+
+```
+uv sync
+uv run poe lint
+uv run poe typecheck
+uv run poe test
+```
+
+Run `uv run poe format` to apply formatting and lint fixes.
+
+## License
+
+MIT. See [LICENSE](LICENSE).

baseten_bdn-0.1.0/README.md

@@ -0,0 +1,74 @@
+# baseten-bdn
+
+Python SDK for working with BDN inside and outside Baseten.
+
+This package is under initial development. It installs as `baseten-bdn` and
+imports as `baseten.bdn`. The `baseten` package is a namespace shared with
+other Baseten SDKs such as `baseten` and `baseten-loops`.
+
+## Installation
+
+```
+pip install baseten-bdn
+```
+
+## Hot Load
+
+Inside a model pod that is opted in to BDN Hot Load, `baseten.bdn.hotload`
+attaches BDN volumes below `/bdn/mounts` at runtime. An attach returns once the
+directory is readable.
+
+```python
+from baseten.bdn.hotload import HotLoadClient
+
+with HotLoadClient() as hotload:
+    adapter = hotload.attach("bdn:adapters/sql-lora@b3:2547…", target="sql-lora")
+
+load_adapter(adapter.path)  # /bdn/mounts/sql-lora
+```
+
+`AsyncHotLoadClient` offers the same methods as coroutines. Attachments belong
+to the pod and stay mounted after the client is closed; call `detach` to free a
+target name for reuse.
+
+## Volumes
+
+From any machine with a Baseten API key, `baseten.bdn.volumes` reads a volume
+version's manifest or pulls a version into a local directory. The client mints a short-lived token through the Baseten API,
+resolves the ref, and reads the volume's objects directly from storage with
+the credentials it is given, verifying every object against its recorded
+digest.
+
+```python
+from baseten.bdn.volumes import VolumeClient
+
+with VolumeClient(api_key="...") as volumes:
+    result = volumes.pull("bdn:loops/sampler-abc123:step-100", "./checkpoint")
+
+print(result.version_ref, result.file_count, result.bytes_written)
+```
+
+Refs are `bdn:<namespace>/<volume>` with an optional `:<tag>` or `@<digest>`
+selector and an optional `/path` inside the version; `VolumeRef` parses and
+renders them. A path on the ref, or `include=[...]`, narrows a pull to those
+entries without moving them. A pull is staged next to `dest_dir` and renamed
+into place once complete, so a failed pull leaves nothing behind; pass
+`overwrite=True` to write into an existing directory in place. Pulling is
+supported on Linux and macOS.
+
+## Development
+
+Requires [uv](https://docs.astral.sh/uv/).
+
+```
+uv sync
+uv run poe lint
+uv run poe typecheck
+uv run poe test
+```
+
+Run `uv run poe format` to apply formatting and lint fixes.
+
+## License
+
+MIT. See [LICENSE](LICENSE).

baseten_bdn-0.1.0/baseten/bdn/__init__.py

@@ -0,0 +1,6 @@
+"""Baseten BDN SDK.
+
+Tools for working with BDN volumes inside and outside Baseten.
+"""
+
+__all__: list[str] = []

baseten_bdn-0.1.0/baseten/bdn/_useragent.py

@@ -0,0 +1,12 @@
+from __future__ import annotations
+
+import functools
+from importlib.metadata import PackageNotFoundError, version
+
+
+@functools.cache
+def user_agent() -> str:
+    try:
+        return f"baseten-bdn/{version('baseten-bdn')}"
+    except PackageNotFoundError:
+        return "baseten-bdn/unknown"

baseten_bdn-0.1.0/baseten/bdn/hotload/__init__.py

@@ -0,0 +1,35 @@
+"""Client for the pod-local BDN Hot Load API.
+
+Use :class:`HotLoadClient` or :class:`AsyncHotLoadClient` from inside an
+opted-in model pod to attach BDN volumes below ``/bdn/mounts`` at runtime.
+"""
+
+from baseten.bdn.hotload._client import (
+    AsyncHotLoadClient,
+    HotLoadClient,
+    HotLoadClientOptions,
+)
+from baseten.bdn.hotload._models import (
+    AttachmentState,
+    HotLoadAPIError,
+    HotLoadAttachError,
+    HotLoadConnectionError,
+    HotLoadError,
+    HotLoadProtocolError,
+    HotLoadTimeoutError,
+    VolumeAttachment,
+)
+
+__all__ = [
+    "AsyncHotLoadClient",
+    "AttachmentState",
+    "HotLoadAPIError",
+    "HotLoadAttachError",
+    "HotLoadClient",
+    "HotLoadClientOptions",
+    "HotLoadConnectionError",
+    "HotLoadError",
+    "HotLoadProtocolError",
+    "HotLoadTimeoutError",
+    "VolumeAttachment",
+]