gitarsenal-cli 1.7.5 → 1.7.7

package/ascii_banner.txt

@@ -1,12 +1,12 @@
-   ╭─────────────────────────────────────────────────────────────────────────────────────────────────────────╮
-   │                                                                                                         │
-   │  ██████╗ ██╗████████╗ █████╗ ██████╗ ███████╗███████╗███╗   ██╗ █████╗ ██╗          ██████╗██╗     ██╗  │
-   │ ██╔════╝ ██║╚══██╔══╝██╔══██╗██╔══██╗██╔════╝██╔════╝████╗  ██║██╔══██╗██║         ██╔════╝██║     ██║  │
-   │ ██║  ███╗██║   ██║   ███████║██████╔╝███████╗█████╗  ██╔██╗ ██║███████║██║         ██║     ██║     ██║  │
-   │ ██║   ██║██║   ██║   ██╔══██║██╔══██╗╚════██║██╔══╝  ██║╚██╗██║██╔══██║██║         ██║     ██║     ██║  │
-   │ ╚██████╔╝██║   ██║   ██║  ██║██║  ██║███████║███████╗██║ ╚████║██║  ██║███████╗    ╚██████╗███████╗██║  │
-   │  ╚═════╝ ╚═╝   ╚═╝   ╚═╝  ╚═╝╚═╝  ╚═╝╚══════╝╚══════╝╚═╝  ╚═══╝╚═╝  ╚═╝╚══════╝     ╚═════╝╚══════╝╚═╝  │
-   │                                                                                                         │
-   │                             GPU-Accelerated Development Environments                                    │
-   │                                                                                                         │
-   ╰─────────────────────────────────────────────────────────────────────────────────────────────────────────╯ 
+   ╭────────────────────────────────────────────────────────────────────────────────╮
+   │                                                                                │
+   │  ██████╗ ██╗████████╗ █████╗ ██████╗ ███████╗███████╗███╗   ██╗ █████╗ ██╗     │
+   │ ██╔════╝ ██║╚══██╔══╝██╔══██╗██╔══██╗██╔════╝██╔════╝████╗  ██║██╔══██╗██║     │
+   │ ██║  ███╗██║   ██║   ███████║██████╔╝███████╗█████╗  ██╔██╗ ██║███████║██║     │
+   │ ██║   ██║██║   ██║   ██╔══██║██╔══██╗╚════██║██╔══╝  ██║╚██╗██║██╔══██║██║     │
+   │ ╚██████╔╝██║   ██║   ██║  ██║██║  ██║███████║███████╗██║ ╚████║██║  ██║███████╗│
+   │  ╚═════╝ ╚═╝   ╚═╝   ╚═╝  ╚═╝╚═╝  ╚═╝╚══════╝╚══════╝╚═╝  ╚═══╝╚═╝  ╚═╝╚══════╝│
+   │                                                                                │
+   │                  GPU-Accelerated Development Environments                      │
+   │                                                                                │
+   ╰────────────────────────────────────────────────────────────────────────────────╯ 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gitarsenal-cli",
-  "version": "1.7.5",
+  "version": "1.7.7",
   "description": "CLI tool for creating Modal sandboxes with GitHub repositories",
   "main": "index.js",
   "bin": {

package/python/test_modalSandboxScript.py

@@ -1400,15 +1400,6 @@ def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_co
     modal_token = os.environ.get("MODAL_TOKEN_ID")
     print(f"  - token in env: {'Yes' if modal_token else 'No'}")
     print(f"  - Token length: {len(modal_token) if modal_token else 'N/A'}")
-    
-    # Verify we can create a Modal app
-    try:
-        print("🔍 Testing app creation...")
-        app = modal.App(app_name)
-        print("✅ Created app successfully")
-    except Exception as e:
-        print(f"❌ Error creating app: {e}")
-        return None
 
     # Create SSH-enabled image
     try:
@@ -1416,14 +1407,14 @@ def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_co
         
         # Use a more stable CUDA base image and avoid problematic packages
         ssh_image = (
-            # modal.Image.from_registry("nvidia/cuda:12.4.0-runtime-ubuntu22.04", add_python="3.11")
+            # modal.Image.from_registry("nvidia/cuda:12.4.0-devel-ubuntu22.04", add_python="3.11")
             modal.Image.debian_slim()
             .apt_install(
                 "openssh-server", "sudo", "curl", "wget", "vim", "htop", "git", 
                 "python3", "python3-pip", "build-essential", "tmux", "screen", "nano",
                 "gpg", "ca-certificates", "software-properties-common"
             )
-            .pip_install("uv", "modal", "requests", "openai")  # Remove problematic CUDA packages
+            .uv_pip_install("uv", "modal", "requests", "openai")  # Remove problematic CUDA packages
             .run_commands(
                 # Create SSH directory
                 "mkdir -p /var/run/sshd",
@@ -1468,8 +1459,7 @@ def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_co
     # Define the SSH container function (remove image from decorator)
     @app.function(
         timeout=timeout_minutes * 60,  # Convert to seconds
-        gpu=gpu_spec['gpu'],
-        cpu=2,
+        gpu="A10G",
         serialized=True,
         volumes=volumes_config if volumes_config else None,
     )

package/test_modalSandboxScript.py

@@ -1400,15 +1400,6 @@ def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_co
     modal_token = os.environ.get("MODAL_TOKEN_ID")
     print(f"  - token in env: {'Yes' if modal_token else 'No'}")
     print(f"  - Token length: {len(modal_token) if modal_token else 'N/A'}")
-    
-    # Verify we can create a Modal app
-    try:
-        print("🔍 Testing app creation...")
-        app = modal.App(app_name)
-        print("✅ Created app successfully")
-    except Exception as e:
-        print(f"❌ Error creating app: {e}")
-        return None
 
     # Create SSH-enabled image
     try:
@@ -1416,14 +1407,14 @@ def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_co
         
         # Use a more stable CUDA base image and avoid problematic packages
         ssh_image = (
-            # modal.Image.from_registry("nvidia/cuda:12.4.0-runtime-ubuntu22.04", add_python="3.11")
+            # modal.Image.from_registry("nvidia/cuda:12.4.0-devel-ubuntu22.04", add_python="3.11")
             modal.Image.debian_slim()
             .apt_install(
                 "openssh-server", "sudo", "curl", "wget", "vim", "htop", "git", 
                 "python3", "python3-pip", "build-essential", "tmux", "screen", "nano",
                 "gpg", "ca-certificates", "software-properties-common"
             )
-            .pip_install("uv", "modal", "requests", "openai")  # Remove problematic CUDA packages
+            .uv_pip_install("uv", "modal", "requests", "openai")  # Remove problematic CUDA packages
             .run_commands(
                 # Create SSH directory
                 "mkdir -p /var/run/sshd",
@@ -1468,8 +1459,7 @@ def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_co
     # Define the SSH container function (remove image from decorator)
     @app.function(
         timeout=timeout_minutes * 60,  # Convert to seconds
-        gpu=gpu_spec['gpu'],
-        cpu=2,
+        gpu="A10G",
         serialized=True,
         volumes=volumes_config if volumes_config else None,
     )

package/python/cuda_image_options.py

@@ -1,124 +0,0 @@
-#!/usr/bin/env python3
-"""
-Alternative CUDA image options for GitArsenal CLI
-These images are more stable and less likely to cause segmentation faults
-"""
-
-import modal
-
-def get_stable_cuda_image():
-    """
-    Get a stable CUDA image that's less likely to cause segmentation faults
-    """
-    return modal.Image.from_registry("nvidia/cuda:11.8.0-runtime-ubuntu22.04", add_python="3.11")
-
-def get_lightweight_cuda_image():
-    """
-    Get a lightweight CUDA image for basic GPU operations
-    """
-    return modal.Image.from_registry("nvidia/cuda:11.8.0-base-ubuntu22.04", add_python="3.11")
-
-def get_latest_stable_cuda_image():
-    """
-    Get the latest stable CUDA image (12.1 instead of 12.4)
-    """
-    return modal.Image.from_registry("nvidia/cuda:12.1.0-runtime-ubuntu22.04", add_python="3.11")
-
-def get_minimal_cuda_image():
-    """
-    Get a minimal CUDA image with just the essentials
-    """
-    return modal.Image.from_registry("nvidia/cuda:11.8.0-minimal-ubuntu22.04", add_python="3.11")
-
-def get_custom_cuda_image():
-    """
-    Create a custom CUDA image with specific optimizations
-    """
-    return (
-        modal.Image.from_registry("nvidia/cuda:11.8.0-runtime-ubuntu22.04", add_python="3.11")
-        .apt_install(
-            "openssh-server", "sudo", "curl", "wget", "vim", "htop", "git", 
-            "python3", "python3-pip", "build-essential", "tmux", "screen", "nano",
-            "gpg", "ca-certificates", "software-properties-common"
-        )
-        .pip_install("uv", "modal", "requests", "openai")
-        .run_commands(
-            # SSH setup
-            "mkdir -p /var/run/sshd",
-            "mkdir -p /root/.ssh",
-            "chmod 700 /root/.ssh",
-            "sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config",
-            "sed -i 's/#PasswordAuthentication yes/PasswordAuthentication yes/' /etc/ssh/sshd_config",
-            "sed -i 's/#PubkeyAuthentication yes/PubkeyAuthentication yes/' /etc/ssh/sshd_config",
-            "echo 'ClientAliveInterval 60' >> /etc/ssh/sshd_config",
-            "echo 'ClientAliveCountMax 3' >> /etc/ssh/sshd_config",
-            "ssh-keygen -A",
-            
-            # GPU compatibility
-            "echo 'export CUDA_VISIBLE_DEVICES=0' >> /root/.bashrc",
-            "echo 'export NVIDIA_VISIBLE_DEVICES=all' >> /root/.bashrc",
-            "echo 'export NVIDIA_DRIVER_CAPABILITIES=compute,utility' >> /root/.bashrc",
-            
-            # Bash prompt
-            "echo 'export PS1=\"\\[\\e[1;32m\\]modal:\\[\\e[1;34m\\]\\w\\[\\e[0m\\]$ \"' >> /root/.bashrc",
-        )
-    )
-
-# Image selection based on use case
-CUDA_IMAGE_OPTIONS = {
-    "stable": get_stable_cuda_image,
-    "lightweight": get_lightweight_cuda_image,
-    "latest": get_latest_stable_cuda_image,
-    "minimal": get_minimal_cuda_image,
-    "custom": get_custom_cuda_image,
-    "default": lambda: modal.Image.debian_slim()  # No CUDA, most stable
-}
-
-def get_cuda_image(option="default"):
-    """
-    Get a CUDA image based on the specified option
-    
-    Args:
-        option (str): One of "stable", "lightweight", "latest", "minimal", "custom", "default"
-    
-    Returns:
-        modal.Image: The selected CUDA image
-    """
-    if option not in CUDA_IMAGE_OPTIONS:
-        print(f"⚠️ Unknown CUDA image option: {option}. Using default.")
-        option = "default"
-    
-    return CUDA_IMAGE_OPTIONS[option]()
-
-def test_cuda_image_stability(image_func, name):
-    """
-    Test the stability of a CUDA image
-    
-    Args:
-        image_func: Function that returns a modal.Image
-        name (str): Name of the image for logging
-    
-    Returns:
-        bool: True if image builds successfully
-    """
-    try:
-        print(f"🧪 Testing {name} CUDA image...")
-        image = image_func()
-        print(f"✅ {name} image created successfully")
-        return True
-    except Exception as e:
-        print(f"❌ {name} image failed: {e}")
-        return False
-
-if __name__ == "__main__":
-    print("🧪 Testing CUDA image stability...")
-    
-    for name, image_func in CUDA_IMAGE_OPTIONS.items():
-        test_cuda_image_stability(image_func, name)
-    
-    print("\n📋 CUDA Image Recommendations:")
-    print("• For maximum stability: Use 'default' (no CUDA)")
-    print("• For basic GPU operations: Use 'stable' (CUDA 11.8 runtime)")
-    print("• For minimal GPU support: Use 'minimal' (CUDA 11.8 minimal)")
-    print("• For latest features: Use 'latest' (CUDA 12.1 runtime)")
-    print("• For custom setup: Use 'custom' (CUDA 11.8 with SSH)")