gitarsenal-cli 1.9.96 → 1.9.97

package/python/test_modalSandboxScript.py

@@ -1,3 +1,11 @@
+#!/usr/bin/env python3
+
+"""
+Modal Sandbox Script for GitArsenal
+--------------------------------
+This script creates a Modal sandbox and sets up a repository.
+"""
+
 import os
 import sys
 import time
@@ -10,7 +18,15 @@ import secrets
 import string
 import argparse
 from pathlib import Path
-import modal
+
+# Try to import modal, but don't attempt to install it
+try:
+    import modal
+except ImportError:
+    print("❌ Error: Modal package not found.")
+    print("Please install the required packages manually with:")
+    print("pip install modal python-dotenv e2b-code-interpreter")
+    sys.exit(1)
 
 
 def generate_random_password(length=16):
@@ -284,7 +300,7 @@ def ssh_container_function(ssh_password=None, repo_url=None, repo_name=None, set
             
             # Call Agent directly as subprocess with real-time output
             claude_prompt = f"clone, setup and run {repo_url}. At the end of the setup process, print the ordered list of every shell command that actually ran successfully (exclude any commands that returned non-zero exit codes). Show each command exactly as executed, one per line."
-            print(f"🚀 Executing the task: \"{claude_prompt}\"")
+            # print(f"🚀 Executing the task: \"{claude_prompt}\"")
             print("\n" + "="*60)
             print("🎉 AGENT OUTPUT (LIVE)")
             print("="*60)
@@ -834,12 +850,24 @@ def show_usage_examples():
     print("│ gitarsenal --auth                         # Interactive auth management         │")
     print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
     
+    print("Sandbox Provider Options")
+    print("┌────────────────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --sandbox-provider modal      # Use Modal sandbox (default, GPU support) │")
+    print("│ gitarsenal --sandbox-provider e2b        # Use E2B sandbox (faster startup)        │")
+    print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
+    
     print("Basic Container Creation with Agent")
     print("┌────────────────────────────────────────────────────────────────────────┐")
     print("│ gitarsenal --gpu A10G --repo https://github.com/username/repo.git  │")
     print("│ # Agent will intelligently clone and setup the repository  │")
     print("└────────────────────────────────────────────────────────────────────────┘\n")
     
+    print("With E2B Sandbox Provider")
+    print("┌────────────────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --sandbox-provider e2b --repo https://github.com/username/repo.git    │")
+    print("│ # Uses E2B sandbox instead of Modal (no GPU but faster startup)      │")
+    print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
+    
     print("With Persistent Storage")
     print("┌────────────────────────────────────────────────────────────────────────────────────┐")
     print("│ gitarsenal --gpu A10G --repo https://github.com/username/repo.git \\  │")
@@ -873,6 +901,10 @@ def show_usage_examples():
     print("  T4, L4, A10G, A100-40GB, A100-80GB, L40S, H100, H200, B200")
     print("  Use --gpu-count to specify multiple GPUs (1-8)")
     print()
+    print("Sandbox Provider Options:")
+    print("  modal: GPU support, persistent volumes (default)")
+    print("  e2b: Faster startup, no GPU support")
+    print()
     print("Authentication Behavior:")
     print("  • First time: Interactive registration/login required")
     print("  • Subsequent runs: Automatic login with stored session")
@@ -881,6 +913,7 @@ def show_usage_examples():
     print("GPU Selection Behavior:")
     print("  • With --gpu: Uses specified GPU without prompting")
     print("  • Without --gpu: Shows interactive GPU selection menu")
+    print("  • Not applicable for E2B sandbox provider")
     print()
     print("Repository Setup Behavior:")
     print("  • With --repo Agent intelligently clones and sets up repository")
@@ -888,9 +921,12 @@ def show_usage_examples():
     print("  • Legacy --setup-commands: Only used when --repo not provided")
     print()
     print("Examples:")
-    print("  # Intelligent repository setup (recommended):")
+    print("  # Intelligent repository setup with Modal (recommended):")
     print("  gitarsenal --gpu A10G --repo https://github.com/username/repo.git")
     print()
+    print("  # Using E2B sandbox provider:")
+    print("  gitarsenal --sandbox-provider e2b --repo https://github.com/username/repo.git")
+    print()
     print("  # Development mode (skip authentication):")
     print("  gitarsenal --skip-auth --gpu A10G --repo https://github.com/username/repo.git")
     print()
@@ -1099,7 +1135,7 @@ if __name__ == "__main__":
     
     # Initialize tokens (import here to avoid container import issues)
     from fetch_modal_tokens import get_tokens
-    token_id, token_secret, openai_api_key, anthropic_api_key, openrouter_api_key, groq_api_key = get_tokens()
+    token_id, token_secret, openai_api_key, e2b_api_key, anthropic_api_key, openrouter_api_key, groq_api_key = get_tokens()
 
     # Check if we got valid tokens
     if token_id is None or token_secret is None:
@@ -1383,7 +1419,7 @@ if __name__ == "__main__":
         else:
             ssh_password = generate_random_password()
             print(f"🔑 Generated random SSH password: {ssh_password}")
-        
+    
         # Extract repository name from URL if not provided
         repo_name = args.repo_name
         if not repo_name and args.repo_url:

package/scripts/setup_e2b.js

@@ -0,0 +1,162 @@
+#!/usr/bin/env node
+
+const fs = require('fs');
+const path = require('path');
+const inquirer = require('inquirer');
+const chalk = require('chalk');
+const os = require('os');
+
+async function setupE2B() {
+  console.log(chalk.blue('🔧 E2B Sandbox Setup'));
+  console.log(chalk.gray('This script will help you set up the E2B API key for GitArsenal.'));
+  console.log(chalk.gray('You can get your API key by signing up at https://e2b.dev'));
+  console.log();
+
+  // Check if E2B_API_KEY is already set in environment
+  const existingKey = process.env.E2B_API_KEY;
+  if (existingKey) {
+    console.log(chalk.green('✅ E2B API key is already set in your environment.'));
+    console.log(chalk.gray(`Key: ${existingKey.substring(0, 4)}...${existingKey.substring(existingKey.length - 4)}`));
+    
+    const { updateKey } = await inquirer.prompt([
+      {
+        type: 'confirm',
+        name: 'updateKey',
+        message: 'Do you want to update your E2B API key?',
+        default: false
+      }
+    ]);
+    
+    if (!updateKey) {
+      console.log(chalk.green('✅ Using existing E2B API key.'));
+      return;
+    }
+  }
+
+  // Prompt for E2B API key
+  const { apiKey } = await inquirer.prompt([
+    {
+      type: 'password',
+      name: 'apiKey',
+      message: 'Enter your E2B API key:',
+      mask: '*',
+      validate: (input) => {
+        if (!input.trim()) {
+          return 'API key is required';
+        }
+        return true;
+      }
+    }
+  ]);
+
+  // Determine where to store the API key
+  const { storeLocation } = await inquirer.prompt([
+    {
+      type: 'list',
+      name: 'storeLocation',
+      message: 'Where would you like to store the API key?',
+      choices: [
+        { name: 'Environment variable (this session only)', value: 'env' },
+        { name: '.env file in current directory', value: 'local' },
+        { name: '.env file in home directory', value: 'home' }
+      ],
+      default: 'env'
+    }
+  ]);
+
+  // Store the API key based on user's choice
+  switch (storeLocation) {
+    case 'env':
+      process.env.E2B_API_KEY = apiKey;
+      console.log(chalk.green('✅ E2B API key set for this session.'));
+      console.log(chalk.yellow('⚠️ This setting will be lost when you close your terminal.'));
+      console.log(chalk.gray('To make it permanent, add the following to your shell profile:'));
+      console.log(chalk.cyan(`export E2B_API_KEY="${apiKey}"`));
+      break;
+      
+    case 'local':
+      try {
+        const envFile = path.join(process.cwd(), '.env');
+        let content = '';
+        
+        if (fs.existsSync(envFile)) {
+          content = fs.readFileSync(envFile, 'utf8');
+          // Replace existing E2B_API_KEY if it exists
+          if (content.match(/^E2B_API_KEY=.*/m)) {
+            content = content.replace(/^E2B_API_KEY=.*/m, `E2B_API_KEY="${apiKey}"`);
+          } else {
+            content += `\nE2B_API_KEY="${apiKey}"\n`;
+          }
+        } else {
+          content = `E2B_API_KEY="${apiKey}"\n`;
+        }
+        
+        fs.writeFileSync(envFile, content);
+        console.log(chalk.green(`✅ E2B API key saved to ${envFile}`));
+      } catch (error) {
+        console.error(chalk.red(`❌ Failed to save API key to .env file: ${error.message}`));
+      }
+      break;
+      
+    case 'home':
+      try {
+        const homeDir = os.homedir();
+        const envFile = path.join(homeDir, '.env');
+        let content = '';
+        
+        if (fs.existsSync(envFile)) {
+          content = fs.readFileSync(envFile, 'utf8');
+          // Replace existing E2B_API_KEY if it exists
+          if (content.match(/^E2B_API_KEY=.*/m)) {
+            content = content.replace(/^E2B_API_KEY=.*/m, `E2B_API_KEY="${apiKey}"`);
+          } else {
+            content += `\nE2B_API_KEY="${apiKey}"\n`;
+          }
+        } else {
+          content = `E2B_API_KEY="${apiKey}"\n`;
+        }
+        
+        fs.writeFileSync(envFile, content);
+        console.log(chalk.green(`✅ E2B API key saved to ${envFile}`));
+      } catch (error) {
+        console.error(chalk.red(`❌ Failed to save API key to home .env file: ${error.message}`));
+      }
+      break;
+  }
+
+  // Also store in GitArsenal credentials
+  try {
+    const gitarsenalDir = path.join(os.homedir(), '.gitarsenal');
+    if (!fs.existsSync(gitarsenalDir)) {
+      fs.mkdirSync(gitarsenalDir, { recursive: true });
+    }
+    
+    const credentialsFile = path.join(gitarsenalDir, 'credentials.json');
+    let credentials = {};
+    
+    if (fs.existsSync(credentialsFile)) {
+      try {
+        credentials = JSON.parse(fs.readFileSync(credentialsFile, 'utf8'));
+      } catch (error) {
+        console.log(chalk.yellow(`⚠️ Could not read existing credentials file: ${error.message}`));
+      }
+    }
+    
+    credentials['e2b'] = apiKey;
+    fs.writeFileSync(credentialsFile, JSON.stringify(credentials, null, 2));
+    console.log(chalk.green('✅ E2B API key also saved to GitArsenal credentials.'));
+    
+  } catch (error) {
+    console.error(chalk.red(`❌ Failed to save API key to GitArsenal credentials: ${error.message}`));
+  }
+
+  console.log(chalk.green('\n🎉 E2B setup complete! You can now use the E2B sandbox provider.'));
+  console.log(chalk.cyan('To use E2B sandbox, run:'));
+  console.log(chalk.white('  gitarsenal --sandbox-provider e2b --repo <repository-url>'));
+}
+
+// Run the setup
+setupE2B().catch(error => {
+  console.error(chalk.red(`❌ Error: ${error.message}`));
+  process.exit(1);
+}); 

package/kill_claude/.claude/settings.local.json

@@ -1,9 +0,0 @@
-{
-  "permissions": {
-    "allow": [
-      "Bash(python test:*)",
-      "Bash(python:*)"
-    ],
-    "defaultMode": "acceptEdits"
-  }
-}