gitarsenal-cli 1.7.7 → 1.7.9

package/bin/gitarsenal.js

@@ -41,7 +41,7 @@ program
 const containerCmd = program
   .command('container')
   .description('Create a container with a GitHub repository')
-  .option('-g, --gpu <type>', 'GPU type (A10G, A100, H100, T4, L4, L40S, V100)', 'A10G')
+  .option('-g, --gpu <type>', 'GPU type (A10G, A100, H100, T4, L4, L40S, V100)')
   .option('-r, --repo-url <url>', 'GitHub repository URL')
   .option('-v, --volume-name <n>', 'Name of persistent volume')
   .option('-s, --setup-commands <commands...>', 'Setup commands to run in the container')
@@ -92,7 +92,7 @@ keysCmd
 // For backward compatibility, support running without a subcommand
 program
   .option('-r, --repo <url>', 'GitHub repository URL')
-  .option('-g, --gpu <type>', 'GPU type (A10G, A100, H100, T4, L4, L40S, V100)', 'A10G')
+  .option('-g, --gpu <type>', 'GPU type (A10G, A100, H100, T4, L4, L40S, V100)')
   .option('-v, --volume <n>', 'Name of persistent volume')
   .option('-y, --yes', 'Skip confirmation prompts')
   .option('-m, --manual', 'Disable automatic setup command detection')
@@ -154,13 +154,15 @@ async function runContainerCommand(options) {
           name: 'gpuType',
           message: 'Select GPU type:',
           choices: [
-            { name: 'A10G (24GB VRAM)', value: 'A10G' },
-            { name: 'A100 (40GB VRAM)', value: 'A100' },
-            { name: 'H100 (80GB VRAM)', value: 'H100' },
             { name: 'T4 (16GB VRAM)', value: 'T4' },
             { name: 'L4 (24GB VRAM)', value: 'L4' },
+            { name: 'A10G (24GB VRAM)', value: 'A10G' },
+            { name: 'A100-40 (40GB VRAM)', value: 'A100-40' },
+            { name: 'A100-80 (80GB VRAM)', value: 'A100-80' },
             { name: 'L40S (48GB VRAM)', value: 'L40S' },
-            { name: 'V100 (16GB VRAM)', value: 'V100' }
+            { name: 'H100 (80GB VRAM)', value: 'H100' },
+            { name: 'H200 (141GB VRAM)', value: 'H200' },
+            { name: 'B200 (141GB VRAM)', value: 'B200' }
           ],
           default: 'A10G'
         }

package/package.json

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

package/python/test_dynamic_commands.py

@@ -0,0 +1,147 @@
+#!/usr/bin/env python3
+"""
+Test script for the dynamic command list functionality.
+This demonstrates how the CommandListManager works with setup commands.
+"""
+
+import sys
+import os
+import time
+
+# Add the current directory to the path so we can import from test_modalSandboxScript
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from test_modalSandboxScript import CommandListManager
+
+def simulate_command_execution(command, success_rate=0.8):
+    """Simulate command execution with a given success rate."""
+    import random
+    time.sleep(0.1)  # Simulate execution time
+    
+    success = random.random() < success_rate
+    stdout = f"Simulated output for: {command}"
+    stderr = "" if success else f"Simulated error for: {command}"
+    
+    return success, stdout, stderr
+
+def test_dynamic_command_list():
+    """Test the dynamic command list functionality."""
+    print("🧪 Testing Dynamic Command List Manager")
+    print("=" * 60)
+    
+    # Initialize with some setup commands
+    initial_commands = [
+        "echo 'Starting setup...'",
+        "pwd",
+        "ls -la",
+        "which python",
+        "python --version"
+    ]
+    
+    print(f"📋 Initial commands: {len(initial_commands)}")
+    for i, cmd in enumerate(initial_commands, 1):
+        print(f"  {i}. {cmd}")
+    
+    # Create command list manager
+    cmd_manager = CommandListManager(initial_commands)
+    
+    print(f"\n🔄 Starting command execution simulation...")
+    
+    # Simulate command execution
+    while cmd_manager.has_pending_commands():
+        next_cmd, cmd_type = cmd_manager.get_next_command()
+        
+        if not next_cmd:
+            break
+        
+        # Print current status
+        cmd_manager.print_status()
+        
+        if cmd_type == 'main':
+            cmd_text = next_cmd['command']
+            cmd_index = next_cmd['index']
+            
+            print(f"\n📋 Executing: {cmd_text}")
+            
+            # Simulate execution
+            start_time = time.time()
+            success, stdout, stderr = simulate_command_execution(cmd_text, success_rate=0.7)
+            execution_time = time.time() - start_time
+            
+            # Mark as executed
+            cmd_manager.mark_command_executed(
+                cmd_index, 'main', success, stdout, stderr, execution_time
+            )
+            
+            # If command failed, simulate adding a fix
+            if not success:
+                print(f"⚠️ Command failed, adding simulated fix...")
+                
+                # Simulate LLM suggesting a fix
+                fix_command = f"echo 'Fix for: {cmd_text}'"
+                fix_index = cmd_manager.add_suggested_fix(cmd_text, fix_command, "Simulated LLM fix")
+                
+                print(f"🔧 Added fix: {fix_command}")
+        
+        elif cmd_type == 'fix':
+            cmd_text = next_cmd['fix_command']
+            cmd_index = next_cmd['index']
+            
+            print(f"\n🔧 Executing fix: {cmd_text}")
+            
+            # Simulate execution
+            start_time = time.time()
+            success, stdout, stderr = simulate_command_execution(cmd_text, success_rate=0.9)
+            execution_time = time.time() - start_time
+            
+            # Mark as executed
+            cmd_manager.mark_command_executed(
+                cmd_index, 'fix', success, stdout, stderr, execution_time
+            )
+    
+    # Demonstrate dynamic command addition
+    print(f"\n📋 Adding dynamic commands...")
+    cmd_manager.add_command_dynamically("echo 'Dynamic command 1'", priority='normal')
+    cmd_manager.add_command_dynamically("echo 'High priority command'", priority='high')
+    cmd_manager.add_command_dynamically("echo 'Dynamic command 2'", priority='normal')
+    
+    # Execute the new commands
+    print(f"\n🔄 Executing dynamic commands...")
+    while cmd_manager.has_pending_commands():
+        next_cmd, cmd_type = cmd_manager.get_next_command()
+        
+        if not next_cmd:
+            break
+        
+        cmd_text = next_cmd['command']
+        cmd_index = next_cmd['index']
+        
+        print(f"📋 Executing dynamic command: {cmd_text}")
+        
+        # Simulate execution
+        start_time = time.time()
+        success, stdout, stderr = simulate_command_execution(cmd_text, success_rate=0.9)
+        execution_time = time.time() - start_time
+        
+        # Mark as executed
+        cmd_manager.mark_command_executed(
+            cmd_index, 'main', success, stdout, stderr, execution_time
+        )
+    
+    # Print final status
+    print(f"\n" + "=" * 60)
+    print("🎉 FINAL STATUS")
+    print("=" * 60)
+    cmd_manager.print_status()
+    
+    # Show all commands with their status
+    print(f"\n📋 All Commands Summary:")
+    all_commands = cmd_manager.get_all_commands()
+    for i, cmd in enumerate(all_commands, 1):
+        status_icon = "✅" if cmd['status'] == 'success' else "❌" if cmd['status'] == 'failed' else "⏳"
+        cmd_type = cmd.get('type', 'main')
+        cmd_text = cmd.get('command', cmd.get('fix_command', 'Unknown command'))
+        print(f"  {i}. {status_icon} [{cmd_type}] {cmd_text}")
+
+if __name__ == "__main__":
+    test_dynamic_command_list()