alphared 0.1.0__tar.gz

alphared-0.1.0/PKG-INFO

@@ -0,0 +1,52 @@
+Metadata-Version: 2.4
+Name: alphared
+Version: 0.1.0
+Summary: A lightweight SDK for logging ML experiments to AlphaRed.
+Author: AlphaRed Team
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: requests
+Requires-Dist: psutil
+Provides-Extra: gpu
+Requires-Dist: pynvml; extra == "gpu"
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# AlphaRed SDK
+
+A lightweight, professional SDK for logging machine learning experiments to the AlphaRed platform. 
+
+## Features
+- 🚀 **Real-time Metrics**: Log loss, accuracy, and custom metrics in the background.
+- 📊 **Auto-Telemetry**: Automatically track CPU, RAM, and Disk metrics.
+- ⚡ **Non-blocking**: Multi-threaded logging ensures your training script speed isn't affected.
+
+## Installation
+```bash
+pip install alphared
+```
+
+## Quick Start
+```python
+import alphared
+
+# Initialize the project
+alphared.init(project="my-mnist-project")
+
+# Your training loop
+for epoch in range(10):
+    # ... training logic ...
+    alphared.log({"epoch": epoch, "loss": 0.5, "accuracy": 0.85})
+
+# Finalize the run
+alphared.finish()
+```

alphared-0.1.0/README.md

@@ -0,0 +1,29 @@
+# AlphaRed SDK
+
+A lightweight, professional SDK for logging machine learning experiments to the AlphaRed platform. 
+
+## Features
+- 🚀 **Real-time Metrics**: Log loss, accuracy, and custom metrics in the background.
+- 📊 **Auto-Telemetry**: Automatically track CPU, RAM, and Disk metrics.
+- ⚡ **Non-blocking**: Multi-threaded logging ensures your training script speed isn't affected.
+
+## Installation
+```bash
+pip install alphared
+```
+
+## Quick Start
+```python
+import alphared
+
+# Initialize the project
+alphared.init(project="my-mnist-project")
+
+# Your training loop
+for epoch in range(10):
+    # ... training logic ...
+    alphared.log({"epoch": epoch, "loss": 0.5, "accuracy": 0.85})
+
+# Finalize the run
+alphared.finish()
+```

alphared-0.1.0/alphared/__init__.py

@@ -0,0 +1,32 @@
+from .session import RunSession
+
+_global_run = None
+
+def init(project=None, name=None, api_key=None, host="http://localhost:8000"):
+    """
+    Initializes a new experiment run with AlphaRed.
+    
+    Args:
+        project (str): Name of the project.
+        name (str): Optional name for this specific run.
+        api_key (str): Your AlphaRed API Key.
+        host (str): The backend host URL.
+    """
+    global _global_run
+    _global_run = RunSession(project=project, name=name, api_key=api_key, host=host)
+    _global_run.start()
+    return _global_run
+
+def log(metrics: dict):
+    """Logs metrics to the active global run."""
+    if _global_run is None:
+        print("⚠️ [AlphaRed] Warning: You must call alphared.init() before logging. Metric ignored.")
+        return
+    _global_run.log(metrics)
+
+def finish():
+    """Finalizes the active run."""
+    global _global_run
+    if _global_run:
+        _global_run.finish()
+        _global_run = None

alphared-0.1.0/alphared/session.py

@@ -0,0 +1,100 @@
+import threading
+import time
+import requests
+import queue
+import uuid
+from .utils.system_telemetry import get_system_metrics
+
+class RunSession:
+    def __init__(self, project, name=None, api_key=None, host="http://localhost:8000"):
+        self.project = project
+        self.name = name or f"run-{uuid.uuid4().hex[:6]}"
+        self.api_key = api_key
+        self.host = host.rstrip('/')
+        self.run_id = None
+        self.running = False
+        self._queue = queue.Queue()
+        self._thread = None
+        self._telemetry_thread = None
+
+    def start(self):
+        """Initializes the run on the backend and starts background threads."""
+        try:
+            response = requests.post(
+                f"{self.host}/api/sdk/init",
+                json={
+                    "project": self.project,
+                    "name": self.name,
+                    "api_key": self.api_key
+                },
+                timeout=5
+            )
+            if response.status_code == 200:
+                data = response.json()
+                self.run_id = data.get("run_id")
+                self.running = True
+                
+                # Start background log consumer
+                self._thread = threading.Thread(target=self._worker, daemon=True)
+                self._thread.start()
+                
+                # Start background telemetry
+                self._telemetry_thread = threading.Thread(target=self._telemetry_worker, daemon=True)
+                self._telemetry_thread.start()
+                
+                print(f"🚀 [AlphaRed] Run initialized: {self.run_id}")
+                print(f"📊 [AlphaRed] View dashboard at: {self.host}/runs/{self.run_id}")
+            else:
+                print(f"❌ [AlphaRed] Failed to initialize run: {response.text}")
+        except Exception as e:
+            print(f"❌ [AlphaRed] Error connecting to backend: {e}")
+
+    def log(self, metrics):
+        """Adds metrics to the processing queue."""
+        if not self.running:
+            return
+        self._queue.put({"type": "metrics", "data": metrics, "timestamp": time.time()})
+
+    def finish(self):
+        """Stops the session and flushes pending data."""
+        self.running = False
+        if self._thread:
+            self._thread.join(timeout=2)
+        print(f"✨ [AlphaRed] Run {self.run_id} completed.")
+
+    def _worker(self):
+        """Consumes metrics from the queue and sends them to the backend."""
+        while self.running or not self._queue.empty():
+            try:
+                item = self._queue.get(timeout=1)
+                self._send_payload(item)
+                self._queue.task_done()
+            except queue.Empty:
+                continue
+            except Exception as e:
+                print(f"⚠️ [AlphaRed] Log error: {e}")
+
+    def _telemetry_worker(self):
+        """Gathers system telemetry (CPU, GPU) every 5 seconds."""
+        while self.running:
+            try:
+                metrics = get_system_metrics()
+                self._send_payload({"type": "telemetry", "data": metrics, "timestamp": time.time()})
+                time.sleep(5)
+            except Exception as e:
+                print(f"⚠️ [AlphaRed] Telemetry error: {e}")
+                time.sleep(10)
+
+    def _send_payload(self, payload):
+        """Sends data to the backend."""
+        if not self.run_id:
+            return
+        try:
+            requests.post(
+                f"{self.host}/api/sdk/runs/{self.run_id}/log",
+                json=payload,
+                headers={"X-API-KEY": self.api_key} if self.api_key else {},
+                timeout=5
+            )
+        except Exception:
+            pass # Silent failure to not interrupt user training

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

@@ -0,0 +1,52 @@
+Metadata-Version: 2.4
+Name: alphared
+Version: 0.1.0
+Summary: A lightweight SDK for logging ML experiments to AlphaRed.
+Author: AlphaRed Team
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: requests
+Requires-Dist: psutil
+Provides-Extra: gpu
+Requires-Dist: pynvml; extra == "gpu"
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# AlphaRed SDK
+
+A lightweight, professional SDK for logging machine learning experiments to the AlphaRed platform. 
+
+## Features
+- 🚀 **Real-time Metrics**: Log loss, accuracy, and custom metrics in the background.
+- 📊 **Auto-Telemetry**: Automatically track CPU, RAM, and Disk metrics.
+- ⚡ **Non-blocking**: Multi-threaded logging ensures your training script speed isn't affected.
+
+## Installation
+```bash
+pip install alphared
+```
+
+## Quick Start
+```python
+import alphared
+
+# Initialize the project
+alphared.init(project="my-mnist-project")
+
+# Your training loop
+for epoch in range(10):
+    # ... training logic ...
+    alphared.log({"epoch": epoch, "loss": 0.5, "accuracy": 0.85})
+
+# Finalize the run
+alphared.finish()
+```

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

@@ -0,0 +1,9 @@
+README.md
+setup.py
+alphared/__init__.py
+alphared/session.py
+alphared.egg-info/PKG-INFO
+alphared.egg-info/SOURCES.txt
+alphared.egg-info/dependency_links.txt
+alphared.egg-info/requires.txt
+alphared.egg-info/top_level.txt

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

@@ -0,0 +1 @@
+

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

@@ -0,0 +1,5 @@
+requests
+psutil
+
+[gpu]
+pynvml

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

@@ -0,0 +1 @@
+alphared

alphared-0.1.0/setup.cfg

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

alphared-0.1.0/setup.py

@@ -0,0 +1,29 @@
+from setuptools import setup, find_packages
+import os
+
+# Read the README for the long description
+with open(os.path.join(os.path.dirname(__file__), "README.md"), encoding="utf-8") as f:
+    long_description = f.read()
+
+setup(
+    name="alphared",
+    version="0.1.0",
+    packages=find_packages(),
+    install_requires=[
+        "requests",
+        "psutil",
+    ],
+    extras_require={
+        "gpu": ["pynvml"],
+    },
+    author="AlphaRed Team",
+    description="A lightweight SDK for logging ML experiments to AlphaRed.",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+    ],
+    python_requires='>=3.7',
+)