gitarsenal-cli 1.5.7 → 1.5.8

package/package.json

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

package/python/fetch_modal_tokens.py

@@ -144,7 +144,7 @@ def get_tokens():
     
     # If we couldn't fetch from the server, try to get default tokens from gitarsenal.dev
     if not token_id or not token_secret:
-        print("⚠️ Proxy server failed, trying to fetch default tokens from gitarsenal.dev")
+        # print("⚠️ Proxy server failed, trying to fetch default tokens from gitarsenal.dev")
         token_id, token_secret, openai_api_key = fetch_default_tokens_from_gitarsenal()
     
     # If we still don't have tokens, we can't proceed

package/python/test_modalSandboxScript.py

@@ -1487,7 +1487,7 @@ def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_co
             print(f"⚙️ Running {len(setup_commands)} setup commands...")
             
             # Define a helper function for running commands with LLM debugging
-            def run_command_with_basic_error_handling(cmd, show_output=True, retry_count=0, max_retries=3):
+            def run_command_with_basic_error_handling(cmd, show_output=True, retry_count=0, max_retries=2):
                 """Execute a command with LLM debugging enabled"""
                 print(f"🔧 Executing: {cmd}")
                 try:
@@ -2273,7 +2273,7 @@ def create_ssh_container_function(gpu_type="a10g", timeout_minutes=60, volume=No
     
     # Create SSH-enabled image
     ssh_image = (
-        modal.Image.debian_slim()
+        modal.Image.from_registry("nvidia/cuda:12.4.0-devel-ubuntu22.04", add_python="3.10")
         .apt_install(
             "openssh-server", "sudo", "curl", "wget", "vim", "htop", "git", 
             "python3", "python3-pip", "build-essential", "tmux", "screen", "nano",

package/python/test_nvcc.py

@@ -0,0 +1,80 @@
+import subprocess
+import time
+import modal
+
+app = modal.App()
+
+# Use NVIDIA CUDA development image with full toolkit including nvcc
+cuda_version = "12.8.1"  # should be no greater than host CUDA version
+flavor = "devel"  # includes full CUDA toolkit with nvcc
+operating_sys = "ubuntu24.04"
+tag = f"{cuda_version}-{flavor}-{operating_sys}"
+
+image = (
+    modal.Image.from_registry(f"nvidia/cuda:{tag}", add_python="3.12")
+    .entrypoint([])  # remove verbose logging by base image on entry
+    .apt_install("openssh-server", "vim", "nano", "htop", "git")  # SSH and useful tools
+    .run_commands("mkdir -p /run/sshd")  # SSH daemon requires this directory
+    .add_local_file("~/.ssh/id_rsa.pub", "/root/.ssh/authorized_keys", copy=True)
+    .run_commands("chmod 700 /root/.ssh && chmod 600 /root/.ssh/authorized_keys")
+)
+
+@app.function(
+    image=image, 
+    gpu="any",  # Request GPU access
+    timeout=3600,  # Keep alive for 1 hour
+    cpu=2,
+    memory=8192
+)
+def start_cuda_container():
+    """
+    Start a CUDA container with nvcc compiler and SSH access.
+    """
+    
+    # Start SSH daemon
+    print("Starting SSH daemon...")
+    subprocess.Popen(["/usr/sbin/sshd", "-D", "-e"])
+    
+    # Verify CUDA installation
+    print("Verifying CUDA installation...")
+    try:
+        nvcc_output = subprocess.check_output(["nvcc", "--version"], text=True)
+        print("NVCC version:")
+        print(nvcc_output)
+    except Exception as e:
+        print(f"Error checking nvcc: {e}")
+    
+    try:
+        nvidia_smi_output = subprocess.check_output(["nvidia-smi"], text=True)
+        print("GPU status:")
+        print(nvidia_smi_output)
+    except Exception as e:
+        print(f"Error checking nvidia-smi: {e}")
+    
+    # Set up port forwarding for SSH
+    with modal.forward(port=22, unencrypted=True) as tunnel:
+        hostname, port = tunnel.tcp_socket
+        connection_cmd = f'ssh -p {port} root@{hostname}'
+        
+        print("=" * 60)
+        print("🚀 CUDA Container Ready!")
+        print("=" * 60)
+        print(f"SSH into container: {connection_cmd}")
+        print(f"Hostname: {hostname}")
+        print(f"Port: {port}")
+        print("=" * 60)
+        print("Available tools:")
+        print("- nvcc (CUDA compiler)")
+        print("- nvidia-smi (GPU monitoring)")
+        print("- vim, nano, git, htop")
+        print("- Python 3.12")
+        print("=" * 60)
+        
+        # Keep the container alive
+        print("Container running for 1 hour...")
+        time.sleep(3600)
+
+# Start the container immediately
+if __name__ == "__main__":
+    print("Starting CUDA container with nvcc...")
+    start_cuda_container.remote()