umarise-wandb 0.1.0__tar.gz

umarise_wandb-0.1.0/PKG-INFO

@@ -0,0 +1,23 @@
+Metadata-Version: 2.4
+Name: umarise-wandb
+Version: 0.1.0
+Summary: Anchor W&B artifacts to Bitcoin. Zero-touch provenance for ML experiments.
+Author-email: Umarise <partners@umarise.com>
+License: Unlicense
+Project-URL: Homepage, https://umarise.com
+Project-URL: Documentation, https://umarise.com/api-reference
+Project-URL: Repository, https://github.com/AnchoringTrust/umarise-wandb
+Project-URL: Case Study, https://umarise.com/case/ai-code-generation
+Keywords: umarise,wandb,weights-and-biases,anchoring,bitcoin,proof-of-existence,opentimestamps,provenance,ai-audit
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Security :: Cryptography
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: umarise-core-sdk>=1.1.0
+Requires-Dist: wandb>=0.15
+
+# umarise-wandb

umarise_wandb-0.1.0/README.md

@@ -0,0 +1 @@
+# umarise-wandb

umarise_wandb-0.1.0/pyproject.toml

@@ -0,0 +1,28 @@
+[build-system]
+requires = ["setuptools>=68.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "umarise-wandb"
+version = "0.1.0"
+description = "Anchor W&B artifacts to Bitcoin. Zero-touch provenance for ML experiments."
+readme = "README.md"
+license = { text = "Unlicense" }
+requires-python = ">=3.8"
+authors = [{ name = "Umarise", email = "partners@umarise.com" }]
+keywords = ["umarise", "wandb", "weights-and-biases", "anchoring", "bitcoin", "proof-of-existence", "opentimestamps", "provenance", "ai-audit"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Science/Research",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Security :: Cryptography",
+    "Programming Language :: Python :: 3",
+]
+dependencies = ["umarise-core-sdk>=1.1.0", "wandb>=0.15"]
+
+[project.urls]
+Homepage = "https://umarise.com"
+Documentation = "https://umarise.com/api-reference"
+Repository = "https://github.com/AnchoringTrust/umarise-wandb"
+"Case Study" = "https://umarise.com/case/ai-code-generation"

umarise_wandb-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

umarise_wandb-0.1.0/umarise_wandb/__init__.py

@@ -0,0 +1,13 @@
+"""
+umarise-wandb: Anchor W&B artifacts to Bitcoin.
+
+Usage:
+    import umarise_wandb
+    umarise_wandb.enable()
+"""
+
+from umarise_wandb.anchor import anchor_artifact, anchor_logged_artifact
+from umarise_wandb.callback import AnchorCallback, enable
+
+__version__ = "0.1.0"
+__all__ = ["anchor_artifact", "anchor_logged_artifact", "AnchorCallback", "enable"]

umarise_wandb-0.1.0/umarise_wandb/anchor.py

@@ -0,0 +1,51 @@
+"""Core anchoring functions for W&B artifacts."""
+
+import hashlib
+import os
+from typing import Optional
+
+import wandb
+from umarise import UmariseCore
+
+
+def _hash_file(path: str) -> str:
+    h = hashlib.sha256()
+    with open(path, "rb") as f:
+        for chunk in iter(lambda: f.read(8192), b""):
+            h.update(chunk)
+    return f"sha256:{h.hexdigest()}"
+
+
+def anchor_artifact(artifact: wandb.Artifact, api_key: Optional[str] = None) -> list:
+    client = UmariseCore(api_key=api_key)
+    results = []
+    for entry in artifact.manifest.entries.values():
+        if entry.local_path and os.path.exists(entry.local_path):
+            file_hash = _hash_file(entry.local_path)
+            result = client.attest(hash=file_hash)
+            result["artifact_name"] = artifact.name
+            result["entry_path"] = entry.path
+            results.append(result)
+            if wandb.run:
+                wandb.log({"umarise/origin_id": result.get("origin_id"), "umarise/anchored": True})
+    return results
+
+
+def anchor_logged_artifact(artifact_name: str, version: str = "latest", api_key: Optional[str] = None) -> list:
+    if not wandb.run:
+        raise RuntimeError("No active W&B run. Call wandb.init() first.")
+    ref = f"{artifact_name}:{version}" if ":" not in artifact_name else artifact_name
+    artifact = wandb.run.use_artifact(ref)
+    artifact_dir = artifact.download()
+    client = UmariseCore(api_key=api_key)
+    results = []
+    for root, _, files in os.walk(artifact_dir):
+        for fname in files:
+            fpath = os.path.join(root, fname)
+            file_hash = _hash_file(fpath)
+            result = client.attest(hash=file_hash)
+            result["file"] = os.path.relpath(fpath, artifact_dir)
+            results.append(result)
+    if results:
+        wandb.log({"umarise/artifact_count": len(results), "umarise/anchored": True})
+    return results

umarise_wandb-0.1.0/umarise_wandb/callback.py

@@ -0,0 +1,65 @@
+"""Auto-anchor callback for W&B."""
+
+import threading
+from typing import Optional
+
+import wandb
+from umarise_wandb.anchor import anchor_artifact as _anchor
+
+
+class AnchorCallback:
+    def __init__(self, api_key: Optional[str] = None):
+        self.api_key = api_key
+        self._lock = threading.Lock()
+
+    def on_artifact_log(self, artifact: wandb.Artifact, **kwargs):
+        with self._lock:
+            try:
+                _anchor(artifact, api_key=self.api_key)
+            except Exception:
+                pass
+
+
+_original_log_artifact = None
+_enabled = False
+
+
+def enable(api_key: Optional[str] = None):
+    global _original_log_artifact, _enabled
+    if _enabled:
+        return
+    from umarise import UmariseCore
+    import hashlib
+    import os
+    client = UmariseCore(api_key=api_key)
+    _original_log_artifact = wandb.Run.log_artifact
+
+    def _patched_log_artifact(self, artifact, *args, **kwargs):
+        result = _original_log_artifact(self, artifact, *args, **kwargs)
+        try:
+            if hasattr(artifact, 'manifest') and artifact.manifest:
+                for entry in artifact.manifest.entries.values():
+                    if entry.local_path and os.path.exists(entry.local_path):
+                        h = hashlib.sha256()
+                        with open(entry.local_path, "rb") as f:
+                            for chunk in iter(lambda: f.read(8192), b""):
+                                h.update(chunk)
+                        file_hash = f"sha256:{h.hexdigest()}"
+                        anchor_result = client.attest(hash=file_hash)
+                        if wandb.run:
+                            wandb.run.summary["umarise_anchored"] = True
+                            wandb.run.summary["umarise_last_origin_id"] = anchor_result.get("origin_id")
+        except Exception:
+            pass
+        return result
+
+    wandb.Run.log_artifact = _patched_log_artifact
+    _enabled = True
+
+
+def disable():
+    global _original_log_artifact, _enabled
+    if _original_log_artifact:
+        wandb.Run.log_artifact = _original_log_artifact
+        _original_log_artifact = None
+    _enabled = False

umarise_wandb-0.1.0/umarise_wandb.egg-info/PKG-INFO

@@ -0,0 +1,23 @@
+Metadata-Version: 2.4
+Name: umarise-wandb
+Version: 0.1.0
+Summary: Anchor W&B artifacts to Bitcoin. Zero-touch provenance for ML experiments.
+Author-email: Umarise <partners@umarise.com>
+License: Unlicense
+Project-URL: Homepage, https://umarise.com
+Project-URL: Documentation, https://umarise.com/api-reference
+Project-URL: Repository, https://github.com/AnchoringTrust/umarise-wandb
+Project-URL: Case Study, https://umarise.com/case/ai-code-generation
+Keywords: umarise,wandb,weights-and-biases,anchoring,bitcoin,proof-of-existence,opentimestamps,provenance,ai-audit
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Security :: Cryptography
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: umarise-core-sdk>=1.1.0
+Requires-Dist: wandb>=0.15
+
+# umarise-wandb

umarise_wandb-0.1.0/umarise_wandb.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.md
+pyproject.toml
+umarise_wandb/__init__.py
+umarise_wandb/anchor.py
+umarise_wandb/callback.py
+umarise_wandb.egg-info/PKG-INFO
+umarise_wandb.egg-info/SOURCES.txt
+umarise_wandb.egg-info/dependency_links.txt
+umarise_wandb.egg-info/requires.txt
+umarise_wandb.egg-info/top_level.txt

umarise_wandb-0.1.0/umarise_wandb.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

umarise_wandb-0.1.0/umarise_wandb.egg-info/requires.txt

@@ -0,0 +1,2 @@
+umarise-core-sdk>=1.1.0
+wandb>=0.15

umarise_wandb-0.1.0/umarise_wandb.egg-info/top_level.txt

@@ -0,0 +1 @@
+umarise_wandb