aicodinggym-cli 0.4.0__tar.gz → 0.5.1__tar.gz

{aicodinggym_cli-0.4.0 → aicodinggym_cli-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aicodinggym-cli
-Version: 0.4.0
+Version: 0.5.1
 Summary: CLI tool for AI Coding Gym platform
 Author-email: AICodingGym Team <datasmithlab@gmail.com>
 License-Expression: MIT

{aicodinggym_cli-0.4.0 → aicodinggym_cli-0.5.1}/aicodinggym_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aicodinggym-cli
-Version: 0.4.0
+Version: 0.5.1
 Summary: CLI tool for AI Coding Gym platform
 Author-email: AICodingGym Team <datasmithlab@gmail.com>
 License-Expression: MIT

{aicodinggym_cli-0.4.0 → aicodinggym_cli-0.5.1}/api.py

@@ -1,6 +1,8 @@
 """HTTP API client for the AI Coding Gym backend at aicodinggym.com."""
 
+import gzip
 import os
+from pathlib import Path
 
 import requests
 
@@ -123,15 +125,17 @@ def mlebench_download_file(url: str, dest_path: str, timeout: int = 300) -> None
 def mlebench_submit_csv(user_id: str, competition_id: str, csv_path: str) -> dict:
     """Upload a prediction CSV for an MLE-bench competition."""
     try:
+        csv_name = Path(csv_path).name
         with open(csv_path, "rb") as f:
-            resp = requests.post(
-                f"{API_BASE}/competitions/{competition_id}/submit",
-                data={"user_id": user_id, "competition_id": competition_id},
-                files={"file": (f.name, f, "text/csv")},
-                timeout=60,
-            )
-            resp.raise_for_status()
-            return resp.json()
+            compressed = gzip.compress(f.read())
+        resp = requests.post(
+            f"{API_BASE}/competitions/{competition_id}/submit",
+            data={"user_id": user_id, "competition_id": competition_id},
+            files={"file": (csv_name + ".gz", compressed, "application/gzip")},
+            timeout=120,
+        )
+        resp.raise_for_status()
+        return resp.json()
     except requests.ConnectionError:
         raise APIError(
             f"Cannot connect to {API_BASE}.\n"

{aicodinggym_cli-0.4.0 → aicodinggym_cli-0.5.1}/cli.py

@@ -22,11 +22,13 @@ CODE REVIEW WORKFLOW:
     aicodinggym cr submit sentry-0001 -f review.md
 """
 
+import json
 import os
 import platform
 import re
 import subprocess
 import sys
+import urllib.request
 from datetime import datetime
 from pathlib import Path
 
@@ -78,6 +80,81 @@ def _warn(msg: str) -> None:
     click.echo(f"Warning: {msg}", err=True)
 
 
+_GYM_ENV_API = "https://api.github.com/repos/AICodingGym/gym-environment/contents"
+_GYM_ENV_SKIP = {"README.md"}
+
+
+def _install_gym_environment(dest: Path) -> None:
+    """Download gym-environment files into dest and add them to .gitignore."""
+    try:
+        req = urllib.request.Request(_GYM_ENV_API, headers={"Accept": "application/vnd.github.v3+json"})
+        with urllib.request.urlopen(req, timeout=15) as resp:
+            entries = json.loads(resp.read())
+    except Exception as e:
+        _warn(f"Could not fetch gym-environment file list: {e}")
+        return
+
+    downloaded: list[str] = []
+
+    for entry in entries:
+        name = entry.get("name", "")
+        if name in _GYM_ENV_SKIP:
+            continue
+        etype = entry.get("type")
+
+        if etype == "file":
+            url = entry.get("download_url")
+            if not url:
+                continue
+            try:
+                with urllib.request.urlopen(url, timeout=15) as r:
+                    (dest / name).write_bytes(r.read())
+                downloaded.append(name)
+            except Exception as e:
+                _warn(f"Failed to download {name}: {e}")
+
+        elif etype == "dir":
+            # Fetch subdirectory contents recursively (one level deep)
+            try:
+                sub_req = urllib.request.Request(
+                    f"{_GYM_ENV_API}/{name}",
+                    headers={"Accept": "application/vnd.github.v3+json"},
+                )
+                with urllib.request.urlopen(sub_req, timeout=15) as r:
+                    sub_entries = json.loads(r.read())
+            except Exception as e:
+                _warn(f"Failed to list directory {name}: {e}")
+                continue
+
+            sub_dir = dest / name
+            sub_dir.mkdir(parents=True, exist_ok=True)
+            for sub in sub_entries:
+                sub_name = sub.get("name", "")
+                sub_url = sub.get("download_url")
+                if sub.get("type") != "file" or not sub_url:
+                    continue
+                try:
+                    with urllib.request.urlopen(sub_url, timeout=15) as r:
+                        (sub_dir / sub_name).write_bytes(r.read())
+                except Exception as e:
+                    _warn(f"Failed to download {name}/{sub_name}: {e}")
+            downloaded.append(name)
+
+    if not downloaded:
+        return
+
+    # Append to .gitignore
+    gitignore = dest / ".gitignore"
+    existing = gitignore.read_text(encoding="utf-8") if gitignore.exists() else ""
+    existing_lines = set(existing.splitlines())
+    new_entries = [f for f in downloaded if f not in existing_lines and f"/{f}" not in existing_lines]
+    if new_entries:
+        block = "\n# gym-environment\n" + "\n".join(new_entries) + "\n"
+        with open(gitignore, "a", encoding="utf-8", newline="\n") as fh:
+            fh.write(block)
+
+
+
 def _resolve_user_id(config: dict, user_id: str | None) -> str:
     """Resolve user_id from argument or config, with helpful error."""
     if user_id:
@@ -334,6 +411,8 @@ def configure(user_id: str, workspace_dir: str | None):
         }
         save_config(config)
 
+        _install_gym_environment(Path(resolved_workspace))
+
         click.echo(
             f"\nConfiguration saved successfully!\n"
             f"\n"
@@ -449,6 +528,8 @@ def swe_fetch(problem_id: str, user_id: str | None, workspace_dir: str | None):
     if not success:
         _error(msg)
 
+    _install_gym_environment(workspace / problem_id)
+
     click.echo(
         f"\nSuccessfully fetched problem: {problem_id}\n"
         f"\n"
@@ -940,6 +1021,8 @@ def cr_fetch(problem_id: str, user_id: str | None, workspace_dir: str | None):
     if not success:
         _error(msg)
 
+    _install_gym_environment(workspace / problem_id)
+
     problem_dir = workspace / problem_id
 
     # Generate diff.patch
@@ -1099,6 +1182,8 @@ def mle_download(competition_id: str, user_id: str | None, workspace_dir: str |
     except APIError as e:
         _error(str(e))
 
+    _install_gym_environment(workspace / competition_id)
+
     click.echo(
         f"\nDataset downloaded to: {dest_path}\n"
         f"\nNext step: train your model and submit predictions with:\n"

{aicodinggym_cli-0.4.0 → aicodinggym_cli-0.5.1}/git_ops.py

@@ -236,8 +236,12 @@ def add_commit_push(problem_dir: str, branch: str, key_path: Path,
     """
     pdir = Path(problem_dir)
 
-    # Stage all changes except .github
-    result = run_git_command(["git", "add", "-A", "--", ".", ":(exclude).github"], str(pdir))
+    # Stage all changes except dotfiles/dotdirs and markdown files
+    result = run_git_command([
+        "git", "add", "-A", "--", ".",
+        ":(exclude).*",
+        ":(exclude)*.md",
+    ], str(pdir))
     if result.returncode != 0:
         return False, f"Git add failed:\n{result.stderr}", ""
 

{aicodinggym_cli-0.4.0 → aicodinggym_cli-0.5.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aicodinggym-cli"
-version = "0.4.0"
+version = "0.5.1"
 description = "CLI tool for AI Coding Gym platform"
 readme = "README.md"
 requires-python = ">=3.10"