auraagent 0.1.0__tar.gz

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

@@ -0,0 +1,36 @@
+name: Publish to GitHub Packages
+
+on:
+  push:
+    tags:
+      - "v*.*.*"
+
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Install build tools
+        run: pip install hatchling build twine
+
+      - name: Build package
+        run: python -m build
+
+      - name: Publish to GitHub Packages (PyPI)
+        env:
+          TWINE_USERNAME: __token__
+          TWINE_PASSWORD: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          twine upload \
+            --repository-url https://pypi.pkg.github.com/tombouVincentA/ \
+            dist/*

auraagent-0.1.0/.gitignore

@@ -0,0 +1,47 @@
+# Python
+__pycache__/
+*.py[cod]
+*.pyo
+*.pyd
+*.so
+*.egg
+*.egg-info/
+dist/
+build/
+.eggs/
+.venv/
+venv/
+env/
+
+# Tests & coverage
+.coverage
+.coverage.*
+htmlcov/
+.pytest_cache/
+.mypy_cache/
+.ruff_cache/
+
+# Aura config locale (contient des tokens — NE JAMAIS COMMITER)
+.aura.config
+*.aura.config
+.env
+.env.*
+
+# PyInstaller
+installer/dist/
+installer/build/
+installer/*.spec
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Logs
+*.log
+logs/

auraagent-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 CEVIA
+
+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.

auraagent-0.1.0/PKG-INFO

@@ -0,0 +1,140 @@
+Metadata-Version: 2.4
+Name: auraagent
+Version: 0.1.0
+Summary: Aura — Carbon tracking and monitoring toolkit
+Project-URL: Homepage, https://cevia.io
+Project-URL: Repository, https://github.com/tombouVincentA/auraagent
+Project-URL: Documentation, https://docs.cevia.io/aura
+Project-URL: Issues, https://github.com/tombouVincentA/auraagent/issues
+Author-email: CEVIA <info@cevia.ai>
+License: MIT License
+        
+        Copyright (c) 2025 CEVIA
+        
+        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.
+License-File: LICENSE
+Keywords: ai,carbon,emissions,sustainability,tracking
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: System :: Monitoring
+Requires-Python: >=3.9
+Requires-Dist: click>=8.1.0
+Requires-Dist: codecarbon>=2.4.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: requests>=2.31.0
+Requires-Dist: rich>=13.0.0
+Provides-Extra: all
+Requires-Dist: aura[installer,web]; extra == 'all'
+Provides-Extra: dev
+Requires-Dist: httpx>=0.27.0; extra == 'dev'
+Requires-Dist: mypy>=1.9.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23.0; extra == 'dev'
+Requires-Dist: pytest-cov>=4.0.0; extra == 'dev'
+Requires-Dist: pytest>=8.0.0; extra == 'dev'
+Requires-Dist: ruff>=0.3.0; extra == 'dev'
+Provides-Extra: installer
+Requires-Dist: pyinstaller>=6.0.0; extra == 'installer'
+Provides-Extra: web
+Requires-Dist: fastapi>=0.110.0; extra == 'web'
+Requires-Dist: httpx>=0.27.0; extra == 'web'
+Requires-Dist: jinja2>=3.1.0; extra == 'web'
+Requires-Dist: python-multipart>=0.0.9; extra == 'web'
+Requires-Dist: uvicorn[standard]>=0.27.0; extra == 'web'
+Requires-Dist: websockets>=12.0; extra == 'web'
+Description-Content-Type: text/markdown
+
+# Aura
+
+Carbon tracking and monitoring toolkit for AI engineers.
+
+## Composants
+
+| Composant | Description |
+|-----------|-------------|
+| `aura.carbon` | Mesure d'émissions CO2 (wrapper CodeCarbon) |
+| `aura.web` | Interface web locale (FastAPI) |
+| `aura.core` | Utilitaires partagés (config, auth, models) |
+
+## Installation
+
+```bash
+# Depuis GitHub Packages
+pip install aura --index-url https://pip.pkg.github.com/cevia
+
+# Avec le composant web
+pip install "aura[web]" --index-url https://pip.pkg.github.com/cevia
+```
+
+## Usage rapide
+
+```python
+from aura.carbon import AuraCarbon
+
+# Context manager
+with AuraCarbon() as tracker:
+    train_model()
+
+# Decorator
+from aura.carbon import track_emissions
+
+@track_emissions()
+def train_model():
+    pass
+```
+
+## Interface web locale
+
+```bash
+aura-web
+# Ouvre http://localhost:8765 dans le navigateur
+```
+
+## Structure
+
+```
+aura/
+├── carbon/          # Mesure CO2
+│   ├── tracker.py   # AuraCarbon (wrapper CodeCarbon)
+│   ├── output.py    # AuraOutput (envoi vers serveur)
+│   └── decorators.py
+├── web/             # Interface locale
+│   ├── main.py      # Point d'entrée (FastAPI + navigateur)
+│   ├── app.py       # Application FastAPI
+│   ├── routers/     # Endpoints
+│   ├── services/    # Logique métier
+│   └── templates/   # HTML
+└── core/            # Partagé
+    ├── config.py    # ~/.aura.config
+    ├── auth.py      # Auth serveur
+    ├── models.py    # Modèles Pydantic
+    └── exceptions.py
+```
+
+## Licence
+
+MIT — CEVIA

auraagent-0.1.0/README.md

@@ -0,0 +1,70 @@
+# Aura
+
+Carbon tracking and monitoring toolkit for AI engineers.
+
+## Composants
+
+| Composant | Description |
+|-----------|-------------|
+| `aura.carbon` | Mesure d'émissions CO2 (wrapper CodeCarbon) |
+| `aura.web` | Interface web locale (FastAPI) |
+| `aura.core` | Utilitaires partagés (config, auth, models) |
+
+## Installation
+
+```bash
+# Depuis GitHub Packages
+pip install aura --index-url https://pip.pkg.github.com/cevia
+
+# Avec le composant web
+pip install "aura[web]" --index-url https://pip.pkg.github.com/cevia
+```
+
+## Usage rapide
+
+```python
+from aura.carbon import AuraCarbon
+
+# Context manager
+with AuraCarbon() as tracker:
+    train_model()
+
+# Decorator
+from aura.carbon import track_emissions
+
+@track_emissions()
+def train_model():
+    pass
+```
+
+## Interface web locale
+
+```bash
+aura-web
+# Ouvre http://localhost:8765 dans le navigateur
+```
+
+## Structure
+
+```
+aura/
+├── carbon/          # Mesure CO2
+│   ├── tracker.py   # AuraCarbon (wrapper CodeCarbon)
+│   ├── output.py    # AuraOutput (envoi vers serveur)
+│   └── decorators.py
+├── web/             # Interface locale
+│   ├── main.py      # Point d'entrée (FastAPI + navigateur)
+│   ├── app.py       # Application FastAPI
+│   ├── routers/     # Endpoints
+│   ├── services/    # Logique métier
+│   └── templates/   # HTML
+└── core/            # Partagé
+    ├── config.py    # ~/.aura.config
+    ├── auth.py      # Auth serveur
+    ├── models.py    # Modèles Pydantic
+    └── exceptions.py
+```
+
+## Licence
+
+MIT — CEVIA

auraagent-0.1.0/aura/__init__.py

@@ -0,0 +1,30 @@
+"""
+Aura — Carbon tracking and monitoring toolkit.
+
+Composants disponibles :
+    - aura.carbon  : Mesure d'émissions CO2 (wrapper CodeCarbon)
+    - aura.web     : Interface web locale (FastAPI)
+    - aura.core    : Utilitaires partagés (config, auth, logging)
+
+Usage rapide :
+    from aura.carbon import AuraCarbon
+
+    tracker = AuraCarbon()
+    tracker.start()
+    # ... ton code ...
+    tracker.stop()
+"""
+
+__version__ = "0.1.0"
+__author__ = "CEVIA"
+__email__ = "contact@cevia.io"
+
+# Exports publics principaux
+from aura.carbon import AuraCarbon
+from aura.carbon.decorators import track_emissions
+
+__all__ = [
+    "AuraCarbon",
+    "track_emissions",
+    "__version__",
+]

auraagent-0.1.0/aura/carbon/__init__.py

@@ -0,0 +1,33 @@
+"""
+aura.carbon — Composant de mesure d'émissions CO2.
+
+Wrapper autour de CodeCarbon avec sortie vers le serveur Aura/Django.
+
+Usage :
+    from aura.carbon import AuraCarbon
+
+    # Context manager
+    with AuraCarbon() as tracker:
+        train_model()
+
+    # Decorator
+    from aura.carbon import track_emissions
+
+    @track_emissions()
+    def train_model():
+        pass
+
+    # Start/Stop explicite
+    tracker = AuraCarbon()
+    tracker.start()
+    train_model()
+    emissions = tracker.stop()
+"""
+
+from aura.carbon.tracker import AuraCarbon
+from aura.carbon.decorators import track_emissions
+
+__all__ = [
+    "AuraCarbon",
+    "track_emissions",
+]

auraagent-0.1.0/aura/carbon/decorators.py

@@ -0,0 +1,54 @@
+"""
+Décorateurs Aura pour le tracking d'émissions.
+
+Usage :
+    from aura.carbon import track_emissions
+
+    @track_emissions()
+    def train_model():
+        pass
+
+    @track_emissions(project_name="GPT Training", measure_power_secs=5)
+    def fine_tune(model, dataset):
+        pass
+"""
+
+import functools
+from typing import Optional, Callable, Any
+
+from aura.carbon.tracker import AuraCarbon
+
+
+def track_emissions(
+    project_name: Optional[str] = None,
+    measure_power_secs: float = 15,
+    save_to_file: bool = False,
+    **kwargs,
+) -> Callable:
+    """
+    Décorateur pour tracker les émissions d'une fonction.
+
+    Args:
+        project_name: Nom du projet (défaut : depuis ~/.aura.config).
+        measure_power_secs: Fréquence de mesure en secondes.
+        save_to_file: Sauvegarder aussi dans un CSV local.
+        **kwargs: Autres paramètres transmis à AuraCarbon / EmissionsTracker.
+
+    Example :
+        @track_emissions(project_name="Training", measure_power_secs=5)
+        def train():
+            ...
+    """
+    def decorator(func: Callable) -> Callable:
+        @functools.wraps(func)
+        def wrapper(*args, **func_kwargs) -> Any:
+            tracker = AuraCarbon(
+                project_name=project_name or func.__name__,
+                measure_power_secs=measure_power_secs,
+                save_to_file=save_to_file,
+                **kwargs,
+            )
+            with tracker:
+                return func(*args, **func_kwargs)
+        return wrapper
+    return decorator

auraagent-0.1.0/aura/carbon/output.py

@@ -0,0 +1,129 @@
+"""
+AuraOutput — Sortie HTTP vers le serveur Aura/Django.
+
+Hérite de BaseOutput (CodeCarbon) et envoie les données
+d'émissions à l'endpoint /api/emissions/ du serveur.
+"""
+
+import requests
+from typing import Optional
+
+from codecarbon.output_methods.base_output import BaseOutput
+from codecarbon.output_methods.emissions_data import EmissionsData
+
+from aura.core.config import AuraConfig
+from aura.core.exceptions import AuraAPIError
+
+
+class AuraOutput(BaseOutput):
+    """
+    Output CodeCarbon qui envoie les données vers le serveur Aura.
+
+    Appelé automatiquement par CodeCarbon à chaque :
+        - live_out() : toutes les api_call_interval mesures (temps réel)
+        - out()      : à stop() ou flush()
+    """
+
+    def __init__(self, config: Optional[AuraConfig] = None):
+        """
+        Args:
+            config: AuraConfig chargée. Si None, chargée automatiquement
+                    depuis ~/.aura.config.
+        """
+        if config is None:
+            config = AuraConfig().load()
+
+        self._endpoint = f"{config.api_endpoint.rstrip('/')}/emissions/"
+        self._headers = {
+            "Authorization": f"Token {config.api_key}",
+            "Content-Type": "application/json",
+        }
+
+    # ─── Interface BaseOutput ─────────────────────────────────────────────────
+
+    def out(self, total: EmissionsData, delta: EmissionsData) -> None:
+        """Appelé à stop() ou flush() — envoie les données cumulatives."""
+        self._send(total)
+
+    def live_out(self, total: EmissionsData, delta: EmissionsData) -> None:
+        """Appelé périodiquement pendant le run — envoie le delta."""
+        self._send(delta)
+
+    def task_out(self, data, experiment_name: str) -> None:
+        """Appelé à stop() si des tâches ont été trackées."""
+        # Optionnel : envoyer les tâches séparément
+        pass
+
+    def exit(self) -> None:
+        """Cleanup à la fin du tracking."""
+        pass
+
+    # ─── Envoi HTTP ───────────────────────────────────────────────────────────
+
+    def _send(self, data: EmissionsData) -> None:
+        """
+        Envoie un objet EmissionsData au serveur.
+
+        En cas d'erreur réseau ou serveur, on logue sans planter
+        le code utilisateur.
+        """
+        try:
+            payload = self._to_dict(data)
+            response = requests.post(
+                self._endpoint,
+                json=payload,
+                headers=self._headers,
+                timeout=10,
+            )
+            if not response.ok:
+                print(
+                    f"[Aura] Avertissement : l'envoi des émissions a échoué "
+                    f"({response.status_code})"
+                )
+        except requests.ConnectionError:
+            print("[Aura] Avertissement : impossible de joindre le serveur.")
+        except requests.Timeout:
+            print("[Aura] Avertissement : timeout lors de l'envoi des émissions.")
+        except Exception as e:
+            print(f"[Aura] Avertissement : erreur inattendue lors de l'envoi : {e}")
+
+    @staticmethod
+    def _to_dict(data: EmissionsData) -> dict:
+        """Convertit EmissionsData en dict JSON-sérialisable."""
+        return {
+            "run_id": str(data.run_id),
+            "project_name": data.project_name,
+            "experiment_id": str(data.experiment_id) if data.experiment_id else None,
+            "timestamp": data.timestamp,
+            "duration": data.duration,
+            "emissions": data.emissions,
+            "emissions_rate": data.emissions_rate,
+            "energy_consumed": data.energy_consumed,
+            "cpu_power": data.cpu_power,
+            "gpu_power": data.gpu_power,
+            "ram_power": data.ram_power,
+            "cpu_energy": data.cpu_energy,
+            "gpu_energy": data.gpu_energy,
+            "ram_energy": data.ram_energy,
+            "cpu_utilization_percent": getattr(data, "cpu_utilization_percent", None),
+            "gpu_utilization_percent": getattr(data, "gpu_utilization_percent", None),
+            "ram_utilization_percent": getattr(data, "ram_utilization_percent", None),
+            "ram_used_gb": getattr(data, "ram_used_gb", None),
+            "country_name": data.country_name,
+            "country_iso_code": data.country_iso_code,
+            "region": data.region,
+            "cloud_provider": data.cloud_provider,
+            "cloud_region": data.cloud_region,
+            "on_cloud": data.on_cloud,
+            "os": data.os,
+            "python_version": data.python_version,
+            "codecarbon_version": data.codecarbon_version,
+            "cpu_model": data.cpu_model,
+            "cpu_count": data.cpu_count,
+            "gpu_model": data.gpu_model,
+            "gpu_count": data.gpu_count,
+            "ram_total_size": data.ram_total_size,
+            "tracking_mode": data.tracking_mode,
+            "pue": data.pue,
+            "wue": getattr(data, "wue", 0.0),
+        }