rnow 0.2.4__py3-none-any.whl

rnow/__init__.py

@@ -0,0 +1,5 @@
+"""
+ReinforceNow CLI - Command-line interface for ReinforceNow RLHF platform.
+"""
+
+__version__ = "0.8.2"

rnow/__main__.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python
+"""Entry point for running rnow as a module."""
+
+from rnow.cli.main import main
+
+if __name__ == "__main__":
+    main()

rnow/cli/__init__.py

@@ -0,0 +1,6 @@
+# reinforcenow/cli/__init__.py
+# CLI package exports
+
+from rnow.cli.main import cli
+
+__all__ = ["cli"]

rnow/cli/auth.py

@@ -0,0 +1,67 @@
+# reinforcenow/cli/auth.py
+
+import json
+from pathlib import Path
+
+import click
+
+# Simple home directory paths
+DATA_DIR = Path.home() / ".reinforcenow"
+CREDS_FILE = DATA_DIR / "credentials.json"
+CONFIG_FILE = DATA_DIR / "config.json"
+
+
+def is_authenticated() -> bool:
+    """Check if authenticated."""
+    try:
+        with open(CREDS_FILE) as f:
+            return "api_key" in json.load(f)
+    except (FileNotFoundError, json.JSONDecodeError, KeyError):
+        return False
+
+
+def get_auth_headers() -> dict[str, str]:
+    """Get auth headers."""
+    try:
+        with open(CREDS_FILE) as f:
+            creds = json.load(f)
+            return {
+                "Content-Type": "application/json",
+                "Authorization": f"Bearer {creds['api_key']}",
+            }
+    except (FileNotFoundError, json.JSONDecodeError, KeyError):
+        raise click.ClickException("Not authenticated. Run 'reinforcenow login'")
+
+
+def get_active_org_from_config() -> str | None:
+    """Get active organization."""
+    try:
+        with open(CONFIG_FILE) as f:
+            return json.load(f).get("active_organization_id")
+    except (FileNotFoundError, json.JSONDecodeError):
+        return None
+
+
+def set_active_organization(org_id: str) -> None:
+    """Set active organization."""
+    DATA_DIR.mkdir(parents=True, exist_ok=True)
+
+    try:
+        with open(CONFIG_FILE) as f:
+            config = json.load(f)
+    except (FileNotFoundError, json.JSONDecodeError):
+        config = {}
+
+    config["active_organization_id"] = org_id
+
+    with open(CONFIG_FILE, "w") as f:
+        json.dump(config, f, indent=2)
+
+
+def logout() -> None:
+    """Remove credentials."""
+    if CREDS_FILE.exists():
+        CREDS_FILE.unlink()
+        click.echo("✓ Logged out")
+    else:
+        click.echo("Not logged in")

rnow/cli/blob.py

@@ -0,0 +1,98 @@
+# reinforcenow/cli/blob.py
+"""Vercel Blob upload support for large files."""
+
+from pathlib import Path
+
+import requests
+
+from rnow.cli import auth
+
+# Size threshold for blob uploads (4MB to stay under 4.5MB limit)
+MAX_INLINE_BYTES = 4 * 1024 * 1024
+
+BLOB_API_URL = "https://blob.vercel-storage.com"
+BLOB_API_VERSION = "7"
+
+
+def request_blob_client_token(base_url: str, pathname: str) -> str:
+    """
+    Request a client upload token from the backend.
+    This token allows direct upload to Vercel Blob.
+    """
+    headers = auth.get_auth_headers()
+    headers["Content-Type"] = "application/json"
+
+    payload = {
+        "type": "blob.generate-client-token",
+        "payload": {
+            "pathname": pathname,
+            "callbackUrl": f"{base_url}/dataset/upload",
+        },
+    }
+
+    resp = requests.post(
+        f"{base_url}/dataset/upload",
+        headers=headers,
+        json=payload,
+        timeout=30,
+    )
+    resp.raise_for_status()
+    data = resp.json()
+
+    if data.get("type") != "blob.generate-client-token":
+        raise RuntimeError(f"Unexpected response from blob token endpoint: {data}")
+
+    client_token = data.get("clientToken")
+    if not client_token:
+        raise RuntimeError("No clientToken returned from blob token endpoint")
+
+    return client_token
+
+
+def upload_file_to_blob(base_url: str, local_path: Path, blob_pathname: str) -> dict:
+    """
+    Upload a file directly to Vercel Blob using a client token.
+    Returns the blob JSON (contains url, pathname, etc).
+    """
+    client_token = request_blob_client_token(base_url, blob_pathname)
+
+    url = f"{BLOB_API_URL}/{blob_pathname.lstrip('/')}"
+    headers = {
+        "Authorization": f"Bearer {client_token}",
+        "x-api-version": BLOB_API_VERSION,
+        "x-content-type": "application/jsonl",
+    }
+
+    with open(local_path, "rb") as f:
+        resp = requests.put(url, headers=headers, data=f, timeout=300)
+
+    resp.raise_for_status()
+    return resp.json()
+
+
+def maybe_upload_to_blob(
+    base_url: str,
+    file_path: Path,
+    dataset_id: str,
+) -> tuple[str | None, dict | None]:
+    """
+    Check if file needs blob upload and handle it.
+
+    Returns:
+        (inline_contents, blob_info)
+        - If small: inline_contents is file content, blob_info is None
+        - If large: inline_contents is None, blob_info has url/pathname
+    """
+    size = file_path.stat().st_size
+
+    if size <= MAX_INLINE_BYTES:
+        # Small file - return contents for inline upload
+        return None, None
+
+    # Large file - upload to blob
+    import uuid
+
+    blob_pathname = f"datasets/{dataset_id}/{uuid.uuid4().hex[:8]}-{file_path.name}"
+
+    blob = upload_file_to_blob(base_url, file_path, blob_pathname)
+    return None, blob