arkhai-vms 0.2.0__tar.gz

arkhai_vms-0.2.0/.gitignore

@@ -0,0 +1,232 @@
+.vscode/
+
+# Project-local scratch files, private notes, and generated diagnostics.
+.scm-local/
+
+# Visible transfer staging for snap-confined tools such as Multipass.
+scm-clean-room-transfer/
+
+# Legacy local rehearsal params (private keys, RPC URLs, etc.).
+.rehearsal-params.local.md
+.rehearsal-params.local.*
+
+### Python ###
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+*.log.[0-9]*
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.env.*
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+### Python Patch ###
+# Poetry local configuration file - https://python-poetry.org/docs/configuration/#local-configuration
+poetry.toml
+
+# ruff
+.ruff_cache/
+
+# LSP config files
+pyrightconfig.json
+
+# Infra
+infra/*/*.tfvars
+infra/*/*.tfstate
+infra/*/*.tfstate.backup
+infra/*/.terraform*
+infra/*/*.plan
+*.db
+# PufferRL Experiments
+experiments/
+wandb/
+!.env.sample
+!.env.docker-compose
+!.env.*.docker-compose
+!.env.*.docker-compose
+!.env.docker-compose
+.cursor/
+
+# Cloud Function
+node_modules
+
+# Puffer artifacts — ignore any resources/ dir or symlink at any depth
+**/resources
+pr-message-puffer.md
+dev-env/state/state.json
+shared-env/*
+helm/Chart.lock
+helm/charts/*.tgz
+helm/.port-forward.pids
+provisioning-service/settings.local.toml
+provisioning-service/src/config/config-local.yml
+.snapshot/*
+
+# Per-env deployment scratch dirs (carry keys + per-env configs)
+deploy-base-sepolia/
+
+# Claude Code per-project state; shared skills are repository-owned.
+.claude/*
+!.claude/skills/
+!.claude/skills/**
+
+# Local assistant session exports
+omp-session-*.html
+pi-session-*.html
+
+# Local-only mise overrides (machine-specific paths).
+mise.local.toml

arkhai_vms-0.2.0/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.5
+Name: arkhai-vms
+Version: 0.2.0
+Summary: Shared VM-domain models and helpers used by Arkhai VM buyer and storefront packages.
+License-Expression: MIT
+Requires-Python: <3.14,>=3.12
+Requires-Dist: arkhai-core>=0.2.0
+Requires-Dist: pydantic>=2
+Provides-Extra: storefront
+Requires-Dist: arkhai-core-storefront>=0.3.0; extra == 'storefront'

arkhai_vms-0.2.0/pyproject.toml

@@ -0,0 +1,34 @@
+[project]
+name = "arkhai-vms"
+version = "0.2.0"
+license = "MIT"
+description = "Shared VM-domain models and helpers used by Arkhai VM buyer and storefront packages."
+requires-python = ">=3.12,<3.14"
+dependencies = [
+    "pydantic>=2",
+    "arkhai-core>=0.2.0",
+]
+
+[project.optional-dependencies]
+storefront = [
+    "arkhai-core-storefront>=0.3.0",
+]
+
+[project.entry-points."market.storefront_domains"]
+vms = "arkhai_vms.domain_runtime:VM_MARKET_DOMAIN"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/arkhai_vms"]
+
+[tool.pytest.ini_options]
+pythonpath = ["src"]
+
+[dependency-groups]
+dev = [
+    "arkhai-core-storefront>=0.3.0",
+    "pytest>=8.0",
+]

arkhai_vms-0.2.0/src/arkhai_vms/__init__.py

@@ -0,0 +1,51 @@
+"""Shared VM-domain helpers used across buyer and storefront packages."""
+
+from arkhai_vms.compute_requirements import (
+    DIMENSION_KEYS,
+    DISK_GB_DIMENSION,
+    GPU_COUNT_DIMENSION,
+    RAM_GB_DIMENSION,
+    VCPU_COUNT_DIMENSION,
+)
+from arkhai_vms.provision_terms import (
+    VM_PROVISION_KIND,
+    VmProvisionTerms,
+    make_vm_provision_terms,
+    normalize_vm_provision_terms,
+    provision_compute_resource,
+    provision_duration_seconds,
+    provision_payload,
+    provision_ssh_public_key,
+    provision_start_utc,
+)
+from arkhai_vms.schema import (
+    VmListing,
+    VmMaterialization,
+    VmMessage,
+    VmReceipt,
+    VmResult,
+    VmTerms,
+)
+
+__all__ = [
+    "DIMENSION_KEYS",
+    "DISK_GB_DIMENSION",
+    "GPU_COUNT_DIMENSION",
+    "RAM_GB_DIMENSION",
+    "VCPU_COUNT_DIMENSION",
+    "VM_PROVISION_KIND",
+    "VmListing",
+    "VmMaterialization",
+    "VmMessage",
+    "VmProvisionTerms",
+    "VmReceipt",
+    "VmResult",
+    "VmTerms",
+    "make_vm_provision_terms",
+    "normalize_vm_provision_terms",
+    "provision_compute_resource",
+    "provision_duration_seconds",
+    "provision_payload",
+    "provision_ssh_public_key",
+    "provision_start_utc",
+]

arkhai_vms-0.2.0/src/arkhai_vms/compute_requirements.py

@@ -0,0 +1,25 @@
+"""Canonical VM capacity-dimension vocabulary.
+
+These names and units are part of the VM domain contract. Translation into a
+specific provisioning playbook belongs to that provider's requirement delegate.
+"""
+
+GPU_COUNT_DIMENSION = "gpu_count"
+VCPU_COUNT_DIMENSION = "vcpu_count"
+RAM_GB_DIMENSION = "ram_gb"
+DISK_GB_DIMENSION = "disk_gb"
+
+DIMENSION_KEYS: tuple[str, ...] = (
+    GPU_COUNT_DIMENSION,
+    VCPU_COUNT_DIMENSION,
+    RAM_GB_DIMENSION,
+    DISK_GB_DIMENSION,
+)
+
+__all__ = [
+    "DIMENSION_KEYS",
+    "DISK_GB_DIMENSION",
+    "GPU_COUNT_DIMENSION",
+    "RAM_GB_DIMENSION",
+    "VCPU_COUNT_DIMENSION",
+]

arkhai_vms-0.2.0/src/arkhai_vms/domain_runtime.py

@@ -0,0 +1,77 @@
+"""VM implementation of the core market-domain contract."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from market_core import (
+    MARKET_DOMAIN_CONTRACT_VERSION,
+    DomainCapability,
+    DomainIdentity,
+    ImmutableCodecCapability,
+    ImmutablePublicationCapability,
+    MarketDomainContract,
+)
+
+from .provision_terms import VM_PROVISION_KIND
+from .schema import (
+    VmListing,
+    VmMaterialization,
+    VmMessage,
+    VmReceipt,
+    VmResult,
+    VmTerms,
+)
+
+
+def _normalize_listing(value: Any) -> VmListing:
+    return VmListing.model_validate(value)
+
+
+def _normalize_message(value: Any) -> VmMessage:
+    return VmMessage.model_validate(value)
+
+
+def _normalize_terms(value: Any) -> VmTerms:
+    return VmTerms.model_validate(value)
+
+
+def _normalize_materialization(value: Any) -> VmMaterialization:
+    return VmMaterialization.model_validate(value)
+
+
+def _normalize_receipt(value: Any) -> VmReceipt:
+    return VmReceipt.model_validate(value)
+
+
+def _normalize_result(value: Any) -> VmResult:
+    return VmResult.model_validate(value)
+
+
+def _publication_source(**kwargs: Any) -> Any:
+    from .storefront_adapter import vm_publication_adapter
+
+    return vm_publication_adapter(**kwargs)
+
+
+VM_MARKET_DOMAIN = MarketDomainContract(
+    identity=DomainIdentity(VM_PROVISION_KIND),
+    contract_version=MARKET_DOMAIN_CONTRACT_VERSION,
+    codecs=ImmutableCodecCapability(
+        normalize_listing=_normalize_listing,
+        normalize_message=_normalize_message,
+        normalize_terms=_normalize_terms,
+        normalize_materialization=_normalize_materialization,
+        normalize_receipt=_normalize_receipt,
+        normalize_result=_normalize_result,
+    ),
+    declared_capabilities=frozenset({DomainCapability.PUBLICATION}),
+    publication=ImmutablePublicationCapability(
+        source_factory=_publication_source,
+    ),
+)
+
+
+def market_domain() -> MarketDomainContract:
+    """Return the VM market-domain contract."""
+    return VM_MARKET_DOMAIN

arkhai_vms-0.2.0/src/arkhai_vms/provision_terms.py

@@ -0,0 +1,133 @@
+"""VM provision-term construction."""
+
+from typing import Any, Literal
+
+from pydantic import BaseModel, ConfigDict, Field, field_validator
+
+
+VM_PROVISION_KIND = "compute.v1"
+VM_PROVISION_VERSION = 1
+
+
+class VmProvisionPayload(BaseModel):
+    """Domain-owned payload schema for VM provision envelope version 1."""
+
+    model_config = ConfigDict(extra="forbid")
+
+    duration_seconds: int = Field(gt=0)
+    ssh_public_key: str = ""
+    start_utc: str | None = None
+    compute_resource: dict[str, Any] | None = None
+
+
+class VmProvisionTerms(BaseModel):
+    """VM provision envelope with independently versioned payload."""
+
+    model_config = ConfigDict(extra="forbid")
+
+    kind: Literal["compute.v1"] = VM_PROVISION_KIND
+    version: Literal[1] = VM_PROVISION_VERSION
+    payload: dict[str, Any]
+
+    @field_validator("payload")
+    @classmethod
+    def _validate_payload(cls, value: dict[str, Any]) -> dict[str, Any]:
+        return VmProvisionPayload.model_validate(value).model_dump(
+            exclude_none=True,
+        )
+
+    @property
+    def duration_seconds(self) -> int | None:
+        raw = self.payload.get("duration_seconds")
+        return int(raw) if raw is not None else None
+
+    @property
+    def start_utc(self) -> str | None:
+        raw = self.payload.get("start_utc")
+        if raw is None:
+            return None
+        text = str(raw).strip()
+        return text or None
+
+    @property
+    def ssh_public_key(self) -> str:
+        raw = self.payload.get("ssh_public_key")
+        return raw if isinstance(raw, str) else ""
+
+    @property
+    def compute_resource(self) -> dict[str, Any] | None:
+        raw = self.payload.get("compute_resource")
+        return raw if isinstance(raw, dict) else None
+
+
+def make_vm_provision_terms(
+    *,
+    duration_seconds: int,
+    ssh_public_key: str,
+    start_utc: str | None = None,
+    compute_resource: dict[str, Any] | None = None,
+) -> VmProvisionTerms:
+    payload: dict[str, Any] = {
+        "duration_seconds": int(duration_seconds),
+        "ssh_public_key": ssh_public_key,
+    }
+    if start_utc is not None:
+        payload["start_utc"] = str(start_utc)
+    if compute_resource is not None:
+        payload["compute_resource"] = compute_resource
+    return VmProvisionTerms(payload=payload)
+
+def normalize_vm_provision_terms(terms: Any) -> VmProvisionTerms:
+    """Validate a wire envelope, VM payload, or durably decoded VM message."""
+
+    if isinstance(terms, dict) and "payload" in terms:
+        return VmProvisionTerms.model_validate(terms)
+    durable = dict(terms) if isinstance(terms, dict) else terms
+    if isinstance(durable, dict):
+        kind = durable.pop("kind", VM_PROVISION_KIND)
+        if kind != VM_PROVISION_KIND:
+            raise ValueError("durable VM provision terms have an unsupported kind")
+        durable.pop("settlement_selection", None)
+    payload = VmProvisionPayload.model_validate(durable).model_dump(exclude_none=True)
+    return VmProvisionTerms(payload=payload)
+
+
+# ---------------------------------------------------------------------------
+# compute.v1 payload accessors
+# ---------------------------------------------------------------------------
+# Wire-received provision terms arrive as the core opaque carrier
+# (market_core.schemas.ProvisionTerms) or a plain dict; the core model
+# does not interpret the payload. These functions are the VM domain's
+# interpretation of the compute.v1 payload shape, accepting any carrier
+# with a ``payload`` attribute or key.
+
+
+def provision_payload(terms: Any) -> dict[str, Any]:
+    if isinstance(terms, dict):
+        raw = terms.get("payload")
+    else:
+        raw = getattr(terms, "payload", None)
+    return raw if isinstance(raw, dict) else {}
+
+
+def provision_duration_seconds(terms: Any) -> int | None:
+    raw = provision_payload(terms).get("duration_seconds")
+    return int(raw) if raw is not None else None
+
+
+def provision_start_utc(terms: Any) -> str | None:
+    raw = provision_payload(terms).get("start_utc")
+    if raw is None:
+        return None
+    text = str(raw).strip()
+    return text or None
+
+
+def provision_ssh_public_key(terms: Any) -> str:
+    raw = provision_payload(terms).get("ssh_public_key")
+    return raw if isinstance(raw, str) else ""
+
+
+def provision_compute_resource(terms: Any) -> dict[str, Any] | None:
+    raw = provision_payload(terms).get("compute_resource")
+    return raw if isinstance(raw, dict) else None