gitarsenal-cli 1.9.10 → 1.9.12

package/.venv_status.json

@@ -1 +1 @@
-{"created":"2025-08-04T14:17:42.697Z","packages":["modal","gitingest","requests"],"uv_version":"uv 0.8.4 (Homebrew 2025-07-30)"}
+{"created":"2025-08-05T14:22:25.927Z","packages":["modal","gitingest","requests"],"uv_version":"uv 0.8.4 (Homebrew 2025-07-30)"}

package/bin/gitarsenal.js

@@ -347,6 +347,7 @@ program.parse(process.argv);
 
 async function runContainerCommand(options) {
   try {
+    console.log(chalk.blue('🔍 DEBUG: runContainerCommand called with options:'), options);
     // If show-examples flag is set, just show examples and exit
     if (options.showExamples) {
       await runContainer({
@@ -419,7 +420,7 @@ async function runContainerCommand(options) {
     }
 
     // Prompt for persistent volume
-    if (!volumeName) {
+    if (!volumeName && !skipConfirmation) {
       const volumeAnswers = await inquirer.prompt([
         {
           type: 'confirm',
@@ -439,6 +440,9 @@ async function runContainerCommand(options) {
       if (volumeAnswers.useVolume) {
         volumeName = volumeAnswers.volumeName;
       }
+    } else if (!volumeName && skipConfirmation) {
+      // If --yes flag is used and no volume specified, use default
+      volumeName = 'gitarsenal-volume';
     }
 
     // Ask about setup command detection if not specified via CLI
@@ -453,6 +457,9 @@ async function runContainerCommand(options) {
       ]);
       
       useApi = apiAnswers.useApi;
+    } else if (options.yes) {
+      // If --yes flag is used, default to using API for setup command detection
+      useApi = true;
     }
 
     // Only prompt for custom commands if auto-detection is disabled and no commands provided
@@ -485,6 +492,7 @@ async function runContainerCommand(options) {
     }
 
     // Confirm settings (configuration will be shown by Python script after GPU selection)
+    console.log(chalk.gray(`🔍 Debug: skipConfirmation = ${skipConfirmation}, options.yes = ${options.yes}`));
     if (!skipConfirmation) {
       const confirmAnswers = await inquirer.prompt([
         {
@@ -506,12 +514,14 @@ async function runContainerCommand(options) {
 
     // Run the container
     try {
+      console.log(chalk.gray(`🔍 Debug: skipConfirmation = ${skipConfirmation}`));
       await runContainer({
         repoUrl,
         gpuType,
         volumeName,
         setupCommands,
-        useApi
+        useApi,
+        yes: skipConfirmation
       });
       
     } catch (containerError) {

package/lib/sandbox.js

@@ -43,7 +43,8 @@ async function runContainer(options) {
     volumeName, 
     setupCommands = [], 
     useApi = true,
-    showExamples = false
+    showExamples = false,
+    yes = false
   } = options;
   
   // Get the path to the Python script
@@ -100,6 +101,12 @@ async function runContainer(options) {
     args.push('--use-api');
   }
   
+  // Add --yes flag to skip confirmation prompts
+  if (yes) {
+    args.push('--yes');
+    console.log(chalk.gray(`🔍 Debug: Adding --yes flag to Python script`));
+  }
+  
   // Handle manual setup commands if provided
   if (setupCommands.length > 0) {
     // Create a temporary file to store setup commands
@@ -110,6 +117,7 @@ async function runContainer(options) {
   
   // Log the command being executed
   console.log(chalk.dim(`\nExecuting: python ${args.join(' ')}`));
+  console.log(chalk.gray(`🔍 Debug: yes parameter = ${yes}`));
   
   // Run the Python script without spinner to avoid terminal interference
   console.log(chalk.dim('Launching container...'));

package/package.json

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

package/python/documentation.py

@@ -0,0 +1,76 @@
+def show_usage_examples():
+    """Display usage examples for the script."""
+    print("Usage Examples\n")
+    
+    print("🔐 Authentication Commands")
+    print("┌────────────────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --register                    # Register new account                  │")
+    print("│ gitarsenal --login                        # Login to existing account            │")
+    print("│ gitarsenal --logout                       # Logout from account                  │")
+    print("│ gitarsenal --user-info                    # Show current user information       │")
+    print("│ gitarsenal --change-password              # Change password                     │")
+    print("│ gitarsenal --delete-account               # Delete account                      │")
+    print("│ gitarsenal --store-api-key openai        # Store OpenAI API key                │")
+    print("│ gitarsenal --auth                         # Interactive auth management         │")
+    print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
+    
+    print("Basic Container Creation")
+    print("┌────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --gpu A10G --repo-url https://github.com/username/repo.git  │")
+    print("└────────────────────────────────────────────────────────────────────────┘\n")
+    
+    print("With Setup Commands")
+    print("┌────────────────────────────────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --gpu A100 --repo-url https://github.com/username/repo.git \\           │")
+    print("│   --setup-commands \"pip install -r requirements.txt\" \"python setup.py install\"  │")
+    print("└────────────────────────────────────────────────────────────────────────────────────────────────────┘\n")
+    
+    print("With Persistent Storage")
+    print("┌────────────────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --gpu A10G --repo-url https://github.com/username/repo.git \\  │")
+    print("│   --volume-name my-persistent-volume                             │")
+    print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
+    
+    print("With GitIngest API (default)")
+    print("┌────────────────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --gpu A10G --repo-url https://github.com/username/repo.git             │")
+    print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
+    
+    print("Without GitIngest API")
+    print("┌────────────────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --gpu A10G --repo-url https://github.com/username/repo.git \\  │")
+    print("│   --no-gitingest                                                │")
+    print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
+    
+    print("With Original API")
+    print("┌────────────────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --gpu A10G --repo-url https://github.com/username/repo.git \\  │")
+    print("│   --use-api                                                      │")
+    print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
+    
+    print("Development Mode (Skip Authentication)")
+    print("┌────────────────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --skip-auth --gpu A10G --repo-url https://github.com/username/repo.git │")
+    print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
+    
+    print("Available GPU Options:")
+    print("  T4, L4, A10G, A100-40GB, A100-80GB, L40S, H100, H200, B200")
+    print()
+    print("Authentication Behavior:")
+    print("  • First time: Interactive registration/login required")
+    print("  • Subsequent runs: Automatic login with stored session")
+    print("  • Use --skip-auth for development (bypasses auth)")
+    print()
+    print("GPU Selection Behavior:")
+    print("  • With --gpu: Uses specified GPU without prompting")
+    print("  • Without --gpu: Shows interactive GPU selection menu")
+    print()
+    print("Examples:")
+    print("  # First time setup (will prompt for registration):")
+    print("  gitarsenal --gpu A10G --repo-url https://github.com/username/repo.git")
+    print()
+    print("  # Subsequent runs (automatic login):")
+    print("  gitarsenal --repo-url https://github.com/username/repo.git")
+    print()
+    print("  # Development mode (skip authentication):")
+    print("  gitarsenal --skip-auth --repo-url https://github.com/username/repo.git")

package/python/test_modalSandboxScript.py

@@ -31,6 +31,7 @@ parser.add_argument('--gpu', default='A10G', help='GPU type to use')
 parser.add_argument('--repo-url', help='Repository URL')
 parser.add_argument('--volume-name', help='Volume name')
 parser.add_argument('--use-api', action='store_true', help='Use API to fetch setup commands')
+parser.add_argument('--yes', action='store_true', help='Skip confirmation prompts')
 
 # Parse only known args to avoid conflicts with other arguments
 args, unknown = parser.parse_known_args()
@@ -520,7 +521,7 @@ class PersistentShell:
             client = openai.OpenAI(api_key=api_key)
             
             response = client.chat.completions.create(
-                model="gpt-4o-mini",
+                model="gpt-4.1",
                 messages=[
                     {"role": "system", "content": "You are a helpful assistant that suggests alternative commands that don't require user input."},
                     {"role": "user", "content": prompt}
@@ -573,20 +574,10 @@ class PersistentShell:
             return output.strip()
         return self.working_dir
     
-    def get_virtual_env(self):
-        """Get the currently activated virtual environment path."""
-        return self.virtual_env_path
-    
     def is_in_venv(self):
         """Check if we're currently in a virtual environment."""
         return self.virtual_env_path is not None and self.virtual_env_path != ""
     
-    def get_venv_name(self):
-        """Get the name of the current virtual environment if active."""
-        if self.is_in_venv():
-            return os.path.basename(self.virtual_env_path)
-        return None
-    
     def exec(self, *args, **kwargs):
         """Compatibility method to make PersistentShell work with call_openai_for_debug."""
         # Convert exec call to execute method
@@ -1128,7 +1119,7 @@ class CommandListManager:
             print("🔍 Analyzing command list for optimizations...")
             
             response = client.chat.completions.create(
-                model="gpt-4o-mini",  # Use a more capable model for this complex task
+                model="gpt-4.1",  # Use a more capable model for this complex task
                 messages=[
                     {"role": "system", "content": "You are a helpful assistant that analyzes and optimizes command lists."},
                     {"role": "user", "content": prompt}
@@ -1246,7 +1237,6 @@ class CommandListManager:
             print(f"⚠️ Error analyzing command list: {e}")
             return False
 
-
 # Import the fetch_modal_tokens module
 # print("🔄 Fetching tokens from proxy server...")
 from fetch_modal_tokens import get_tokens
@@ -1666,13 +1656,9 @@ IMPORTANT GUIDELINES:
 
 Do not provide any explanations, just the exact command to run.
 """
-    
-    # Prepare the API request payload
-    # print("🔍 DEBUG: Preparing API request...")
-    
-    # Try to use GPT-4 first, but fall back to other models if needed
+
     models_to_try = [
-        "gpt-4o-mini",        # First choice: GPT-4o (most widely available)
+        "gpt-4.1",        # First choice: GPT-4o (most widely available)
     ]
     
     # Check if we have a preferred model in environment
@@ -1680,19 +1666,12 @@ Do not provide any explanations, just the exact command to run.
     if preferred_model:
         # Insert the preferred model at the beginning of the list
         models_to_try.insert(0, preferred_model)
-        # print(f"✅ Using preferred model from environment: {preferred_model}")
         
     # Remove duplicates while preserving order
     models_to_try = list(dict.fromkeys(models_to_try))
-    # print(f"🔍 DEBUG: Models to try: {models_to_try}")
     
     # Function to make the API call with a specific model
     def try_api_call(model_name, retries=2, backoff_factor=1.5):
-        # print(f"🔍 DEBUG: Attempting API call with model: {model_name}")
-        # print(f"🔍 DEBUG: API key available: {'Yes' if api_key else 'No'}")
-        # if api_key:
-        #     print(f"🔍 DEBUG: API key length: {len(api_key)}")
-        #     print(f"🔍 DEBUG: API key starts with: {api_key[:10]}...")
         
         payload = {
             "model": model_name,
@@ -1795,7 +1774,6 @@ Do not provide any explanations, just the exact command to run.
     for model in models_to_try:
         result, error = try_api_call(model)
         if result:
-            # print(f"✅ Successfully got response from {model}")
             break
         else:
             print(f"⚠️ Failed to get response from {model}: {error}")
@@ -1959,7 +1937,7 @@ Provide fixes for all {len(failed_commands)} failed commands:"""
     }
     
     payload = {
-        "model": "gpt-4o-mini",  # Use a more capable model for batch analysis
+        "model": "gpt-4.1",  # Use a more capable model for batch analysis
         "messages": [
             {"role": "system", "content": "You are a debugging assistant. Analyze failed commands and provide specific fix commands. Return only the fix commands and reasons in the specified format."},
             {"role": "user", "content": prompt}
@@ -2016,43 +1994,6 @@ Provide fixes for all {len(failed_commands)} failed commands:"""
         print(f"❌ Error during batch debugging: {e}")
         return []
 
-def prompt_for_hf_token():
-    """Prompt user for Hugging Face token when needed"""
-    # Try to use credentials manager first
-    try:
-        from credentials_manager import CredentialsManager
-        credentials_manager = CredentialsManager()
-        token = credentials_manager.get_huggingface_token()
-        if token:
-            return token
-    except ImportError:
-        # Fall back to direct input if credentials_manager is not available
-        pass
-    
-    # Traditional direct input method as fallback
-    print("\n" + "="*60)
-    print("🔑 HUGGING FACE TOKEN REQUIRED")
-    print("="*60)
-    print("The training script requires a valid Hugging Face token.")
-    print("You can get your token from: https://huggingface.co/settings/tokens")
-    print("📝 Please paste your Hugging Face token below:")
-    print("   (Your input will be hidden for security)")
-    print("-" * 60)
-    
-    try:
-        token = getpass.getpass("HF Token: ").strip()
-        if not token:
-            print("❌ No token provided.")
-            return None
-        print("✅ Token received successfully!")
-        return token
-    except KeyboardInterrupt:
-        print("\n❌ Token input cancelled by user.")
-        return None
-    except Exception as e:
-        print(f"❌ Error getting token: {e}")
-        return None
-
 def generate_random_password(length=16):
     """Generate a random password for SSH access"""
     alphabet = string.ascii_letters + string.digits + "!@#$%^&*"
@@ -2067,11 +2008,6 @@ def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_co
     
     # Use interactive mode if specified
     if interactive:
-        # If GPU type is not specified, prompt for it
-        if not gpu_type:
-            gpu_type = prompt_for_gpu()
-        else:
-            print(f"✅ Using provided GPU type: {gpu_type}")
             
         # If repo URL is not specified, prompt for it
         if not repo_url:
@@ -2748,7 +2684,6 @@ def fetch_setup_commands_from_api(repo_url):
                 print(f"⚠️ GitIngest CLI failed with exit code {result.returncode}")
                 print(f"⚠️ Error output: {result.stderr}")
                 print("Falling back to basic analysis")
-                gitingest_data = generate_basic_repo_analysis_from_url(repo_url)
             else:
                 print(f"✅ GitIngest analysis completed successfully")
                 
@@ -2832,13 +2767,13 @@ def fetch_setup_commands_from_api(repo_url):
                                 print(f"📁 Processed GitIngest data saved to: {processed_file}")
                     except FileNotFoundError:
                         print(f"⚠️ Output file not found at {output_file}")
-                        gitingest_data = generate_basic_repo_analysis_from_url(repo_url)
+                       
                 except Exception as e:
                     print(f"⚠️ Error reading GitIngest output: {e}")
-                    gitingest_data = generate_basic_repo_analysis_from_url(repo_url)
+     
         else:
             # Fall back to basic analysis if gitingest CLI is not available
-            gitingest_data = generate_basic_repo_analysis_from_url(repo_url)
+            gitingest_data = "{}"
         
         # Prepare the request payload with GitIngest data
         payload = {
@@ -3072,171 +3007,6 @@ def generate_fallback_commands(gitingest_data):
     
     return fixed_commands
 
-def generate_basic_repo_analysis_from_url(repo_url):
-    """Generate basic repository analysis data from a repository URL."""
-    import tempfile
-    import subprocess
-    import os
-    import shutil
-    
-    # Create a temporary directory for cloning
-    temp_dir = tempfile.mkdtemp(prefix="repo_basic_analysis_")
-    
-    try:
-        print(f"📥 Cloning repository to {temp_dir} for basic analysis...")
-        clone_result = subprocess.run(
-            ["git", "clone", "--depth", "1", repo_url, temp_dir],
-            capture_output=True,
-            text=True
-        )
-        
-        if clone_result.returncode != 0:
-            print(f"❌ Failed to clone repository: {clone_result.stderr}")
-            return {
-                "system_info": {
-                    "platform": "linux",
-                    "python_version": "3.10",
-                    "detected_language": "Unknown",
-                    "detected_technologies": [],
-                    "file_count": 0,
-                    "repo_stars": 0,
-                    "repo_forks": 0,
-                    "primary_package_manager": "Unknown",
-                    "complexity_level": "low"
-                },
-                "repository_analysis": {
-                    "summary": f"Repository analysis for {repo_url}",
-                    "tree": "Failed to clone repository",
-                    "content_preview": "No content available"
-                },
-                "success": False
-            }
-        
-        print(f"✅ Repository cloned successfully for basic analysis")
-        
-        # Use the existing generate_basic_repo_analysis function
-        return generate_basic_repo_analysis(temp_dir)
-    finally:
-        # Clean up the temporary directory
-        print(f"🧹 Cleaning up temporary directory for basic analysis...")
-        shutil.rmtree(temp_dir, ignore_errors=True)
-
-def generate_basic_repo_analysis(repo_dir):
-    """Generate basic repository analysis when GitIngest is not available."""
-    import os
-    import subprocess
-    
-    # Detect language and technologies based on file extensions
-    file_extensions = {}
-    file_count = 0
-    
-    for root, _, files in os.walk(repo_dir):
-        for file in files:
-            file_count += 1
-            ext = os.path.splitext(file)[1].lower()
-            if ext:
-                file_extensions[ext] = file_extensions.get(ext, 0) + 1
-    
-    # Determine primary language
-    language_map = {
-        '.py': 'Python',
-        '.js': 'JavaScript',
-        '.ts': 'TypeScript',
-        '.jsx': 'JavaScript',
-        '.tsx': 'TypeScript',
-        '.java': 'Java',
-        '.cpp': 'C++',
-        '.c': 'C',
-        '.go': 'Go',
-        '.rs': 'Rust',
-        '.rb': 'Ruby',
-        '.php': 'PHP',
-        '.swift': 'Swift',
-        '.kt': 'Kotlin',
-        '.cs': 'C#'
-    }
-    
-    # Count files by language
-    language_counts = {}
-    for ext, count in file_extensions.items():
-        if ext in language_map:
-            lang = language_map[ext]
-            language_counts[lang] = language_counts.get(lang, 0) + count
-    
-    # Determine primary language
-    primary_language = max(language_counts.items(), key=lambda x: x[1])[0] if language_counts else "Unknown"
-    
-    # Detect package managers
-    package_managers = []
-    package_files = {
-        'requirements.txt': 'pip',
-        'setup.py': 'pip',
-        'pyproject.toml': 'pip',
-        'package.json': 'npm',
-        'yarn.lock': 'yarn',
-        'pnpm-lock.yaml': 'pnpm',
-        'Cargo.toml': 'cargo',
-        'go.mod': 'go',
-        'Gemfile': 'bundler',
-        'pom.xml': 'maven',
-        'build.gradle': 'gradle',
-        'composer.json': 'composer'
-    }
-    
-    for file, manager in package_files.items():
-        if os.path.exists(os.path.join(repo_dir, file)):
-            package_managers.append(manager)
-    
-    primary_package_manager = package_managers[0] if package_managers else "Unknown"
-    
-    # Get README content
-    readme_content = ""
-    for readme_name in ['README.md', 'README', 'README.txt', 'readme.md']:
-        readme_path = os.path.join(repo_dir, readme_name)
-        if os.path.exists(readme_path):
-            with open(readme_path, 'r', encoding='utf-8', errors='ignore') as f:
-                readme_content = f.read()
-            break
-    
-    # Try to get repository info
-    repo_info = {}
-    try:
-        # Get remote origin URL
-        cmd = ["git", "config", "--get", "remote.origin.url"]
-        result = subprocess.run(cmd, cwd=repo_dir, capture_output=True, text=True)
-        if result.returncode == 0:
-            repo_info["url"] = result.stdout.strip()
-        
-        # Get commit count as a proxy for activity
-        cmd = ["git", "rev-list", "--count", "HEAD"]
-        result = subprocess.run(cmd, cwd=repo_dir, capture_output=True, text=True)
-        if result.returncode == 0:
-            repo_info["commit_count"] = int(result.stdout.strip())
-    except Exception:
-        pass
-    
-    # Build the analysis data
-    return {
-        "system_info": {
-            "platform": "linux",  # Assuming Linux for container environment
-            "python_version": "3.10",  # Common Python version
-            "detected_language": primary_language,
-            "detected_technologies": list(language_counts.keys()),
-            "file_count": file_count,
-            "repo_stars": repo_info.get("stars", 0),
-            "repo_forks": repo_info.get("forks", 0),
-            "primary_package_manager": primary_package_manager,
-            "complexity_level": "medium"  # Default assumption
-        },
-        "repository_analysis": {
-            "summary": f"Repository analysis for {repo_dir}",
-            "readme_content": readme_content[:5000] if readme_content else "No README found",
-            "package_managers": package_managers,
-            "file_extensions": list(file_extensions.keys())
-        },
-        "success": True
-    }
-
 def fix_setup_commands(commands):
     """Fix setup commands by removing placeholders and comments."""
     fixed_commands = []
@@ -3746,152 +3516,6 @@ def get_setup_commands_from_gitingest(repo_url):
     print("❌ All API endpoints failed")
     return generate_fallback_commands(gitingest_data)
 
-def prompt_for_gpu():
-    """
-    Prompt the user to select a GPU type from available options using arrow keys.
-    Returns the selected GPU type.
-    """
-    import sys
-    import tty
-    import termios
-    
-    print("\n🔧 GPU Selection Required")
-    print("No GPU type was specified with --gpu flag.")
-    print("Please select a GPU type for your container:")
-    
-    # Define available GPU types and their specifications
-    gpu_specs = {
-        'T4': {'gpu': 'T4', 'memory': '16GB'},
-        'L4': {'gpu': 'L4', 'memory': '24GB'},
-        'A10G': {'gpu': 'A10G', 'memory': '24GB'},
-        'A100-40': {'gpu': 'A100-40GB', 'memory': '40GB'},
-        'A100-80': {'gpu': 'A100-80GB', 'memory': '80GB'},
-        'L40S': {'gpu': 'L40S', 'memory': '48GB'},
-        'H100': {'gpu': 'H100', 'memory': '80GB'},
-        'H200': {'gpu': 'H200', 'memory': '141GB'},
-        'B200': {'gpu': 'B200', 'memory': '141GB'}
-    }
-    
-    # Create a list of options
-    options = list(gpu_specs.keys())
-    selected_index = 2  # Default to A10G (index 2)
-    
-    def get_key():
-        """Get a single keypress from the user."""
-        fd = sys.stdin.fileno()
-        old_settings = termios.tcgetattr(fd)
-        try:
-            tty.setraw(sys.stdin.fileno())
-            ch = sys.stdin.read(1)
-            if ch == '\x1b':  # Escape sequence
-                ch2 = sys.stdin.read(1)
-                if ch2 == '[':
-                    ch3 = sys.stdin.read(1)
-                    if ch3 == 'A':
-                        return 'UP'
-                    elif ch3 == 'B':
-                        return 'DOWN'
-            elif ch == '\r' or ch == '\n':
-                return 'ENTER'
-            elif ch == '\x03':  # Ctrl+C
-                return 'CTRL_C'
-            return ch
-        finally:
-            termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
-    
-    def display_menu():
-        """Display the GPU selection menu with current selection highlighted."""
-        print("\n📊 Available GPU Options:")
-        print("┌──────────────┬─────────┐")
-        print("│   GPU Type   │  VRAM   │")
-        print("├──────────────┼─────────┤")
-        
-        for i, gpu_type in enumerate(options):
-            specs = gpu_specs[gpu_type]
-            # Calculate proper spacing for alignment
-            number_part = f"{i+1}."
-            if i == selected_index:
-                prefix = "> "
-                suffix = " ←"
-            else:
-                prefix = "  "
-                suffix = ""
-            
-            # Ensure consistent width for GPU type column
-            gpu_display = f"{prefix}{number_part} {gpu_type}"
-            gpu_padded = f"{gpu_display:<12}"  # Fixed width for GPU column
-            
-            print(f"│ {gpu_padded} │ {specs['memory']:<7} │{suffix}")
-        
-        print("└──────────────┴─────────┘")
-        print("Use ↑/↓ arrows to select, Enter to confirm, Ctrl+C to cancel")
-    
-    # Clear screen and show initial menu
-    print("\033[2J\033[H", end="")  # Clear screen and move cursor to top
-    display_menu()
-    
-    while True:
-        try:
-            key = get_key()
-            
-            if key == 'UP':
-                selected_index = (selected_index - 1) % len(options)
-                print("\033[2J\033[H", end="")  # Clear screen
-                display_menu()
-            elif key == 'DOWN':
-                selected_index = (selected_index + 1) % len(options)
-                print("\033[2J\033[H", end="")  # Clear screen
-                display_menu()
-            elif key == 'ENTER':
-                selected_gpu = options[selected_index]
-                print(f"\n✅ Selected GPU: {selected_gpu}")
-                return selected_gpu
-            elif key == 'CTRL_C':
-                print("\n🛑 Selection cancelled.")
-                sys.exit(1)
-                
-        except KeyboardInterrupt:
-            print("\n🛑 Selection cancelled.")
-            sys.exit(1)
-        except Exception as e:
-            print(f"\n❌ Error with interactive menu: {e}")
-            print("🔄 Falling back to simple text input...")
-            # Fall back to simple input method
-            try:
-                print("\n📊 Available GPU Options:")
-                for i, gpu_type in enumerate(options, 1):
-                    specs = gpu_specs[gpu_type]
-                    print(f"  {i}. {gpu_type} ({specs['memory']})")
-                print(f"  Default: A10G")
-                
-                choice = input("\n🔍 Select GPU type (number or name, default is A10G): ").strip()
-                if not choice:
-                    print("✅ Using default GPU: A10G")
-                    return "A10G"
-                if choice.isdigit():
-                    index = int(choice) - 1
-                    if 0 <= index < len(options):
-                        selected = options[index]
-                        print(f"✅ Selected GPU: {selected}")
-                        return selected
-                    else:
-                        print(f"⚠️ Invalid number. Using default: A10G")
-                        return "A10G"
-                elif choice in options:
-                    print(f"✅ Selected GPU: {choice}")
-                    return choice
-                else:
-                    print(f"⚠️ Invalid choice '{choice}'. Using default: A10G")
-                    return "A10G"
-            except KeyboardInterrupt:
-                print("\n🛑 Selection cancelled.")
-                sys.exit(1)
-            except Exception as fallback_error:
-                print(f"❌ Error in fallback input: {fallback_error}")
-                print("✅ Using default GPU: A10G")
-                return "A10G"
-
-
 
 def preprocess_commands_with_llm(setup_commands, stored_credentials, api_key=None):
     """
@@ -4235,11 +3859,6 @@ if __name__ == "__main__":
         _handle_auth_commands(auth_manager, args)
         sys.exit(0)
     
-    # If --list-gpus is specified, just show GPU options and exit
-    if args.list_gpus:
-        prompt_for_gpu()
-        sys.exit(0)
-    
     # If no arguments or only --show-examples is provided, show usage examples
     if len(sys.argv) == 1 or args.show_examples:
         show_usage_examples()
@@ -4298,7 +3917,6 @@ if __name__ == "__main__":
     else:
         print("\n📋 No GPU type specified with --gpu flag.")
         print("🔄 Prompting for GPU selection...")
-        gpu_type = prompt_for_gpu()
         args.gpu = gpu_type
     
     # Display configuration after GPU selection
@@ -4314,14 +3932,17 @@ if __name__ == "__main__":
         print("Setup Commands: Auto-detect from repository")
     
     # Confirm settings
-    try:
-        proceed = input("Proceed with these settings? (Y/n): ").strip().lower()
-        if proceed in ('n', 'no'):
-            print("🛑 Operation cancelled by user.")
+    if not args.yes:
+        try:
+            proceed = input("Proceed with these settings? (Y/n): ").strip().lower()
+            if proceed in ('n', 'no'):
+                print("🛑 Operation cancelled by user.")
+                sys.exit(0)
+        except KeyboardInterrupt:
+            print("\n🛑 Operation cancelled by user.")
             sys.exit(0)
-    except KeyboardInterrupt:
-        print("\n🛑 Operation cancelled by user.")
-        sys.exit(0)
+    else:
+        print("✅ Skipping confirmation prompt (--yes flag used)")
     
     # Interactive mode or missing required arguments
     if args.interactive or not args.repo_url or not args.volume_name:
@@ -4491,12 +4112,8 @@ if __name__ == "__main__":
             interactive=args.interactive
         )
     except KeyboardInterrupt:
-        # print("\n\n🛑 Execution interrupted")
-        # print("🧹 Cleaning up resources...")
         cleanup_modal_token()
         sys.exit(1)
     except Exception as e:
-        # print(f"\n❌ Error: {e}")
-        # print("🧹 Cleaning up resources...")
         cleanup_modal_token()
         sys.exit(1)

package/test_modalSandboxScript.py

@@ -31,6 +31,7 @@ parser.add_argument('--gpu', default='A10G', help='GPU type to use')
 parser.add_argument('--repo-url', help='Repository URL')
 parser.add_argument('--volume-name', help='Volume name')
 parser.add_argument('--use-api', action='store_true', help='Use API to fetch setup commands')
+parser.add_argument('--yes', action='store_true', help='Skip confirmation prompts')
 
 # Parse only known args to avoid conflicts with other arguments
 args, unknown = parser.parse_known_args()
@@ -520,7 +521,7 @@ class PersistentShell:
             client = openai.OpenAI(api_key=api_key)
             
             response = client.chat.completions.create(
-                model="gpt-4o-mini",
+                model="gpt-4.1",
                 messages=[
                     {"role": "system", "content": "You are a helpful assistant that suggests alternative commands that don't require user input."},
                     {"role": "user", "content": prompt}
@@ -573,20 +574,10 @@ class PersistentShell:
             return output.strip()
         return self.working_dir
     
-    def get_virtual_env(self):
-        """Get the currently activated virtual environment path."""
-        return self.virtual_env_path
-    
     def is_in_venv(self):
         """Check if we're currently in a virtual environment."""
         return self.virtual_env_path is not None and self.virtual_env_path != ""
     
-    def get_venv_name(self):
-        """Get the name of the current virtual environment if active."""
-        if self.is_in_venv():
-            return os.path.basename(self.virtual_env_path)
-        return None
-    
     def exec(self, *args, **kwargs):
         """Compatibility method to make PersistentShell work with call_openai_for_debug."""
         # Convert exec call to execute method
@@ -1128,7 +1119,7 @@ class CommandListManager:
             print("🔍 Analyzing command list for optimizations...")
             
             response = client.chat.completions.create(
-                model="gpt-4o-mini",  # Use a more capable model for this complex task
+                model="gpt-4.1",  # Use a more capable model for this complex task
                 messages=[
                     {"role": "system", "content": "You are a helpful assistant that analyzes and optimizes command lists."},
                     {"role": "user", "content": prompt}
@@ -1246,7 +1237,6 @@ class CommandListManager:
             print(f"⚠️ Error analyzing command list: {e}")
             return False
 
-
 # Import the fetch_modal_tokens module
 # print("🔄 Fetching tokens from proxy server...")
 from fetch_modal_tokens import get_tokens
@@ -1666,13 +1656,9 @@ IMPORTANT GUIDELINES:
 
 Do not provide any explanations, just the exact command to run.
 """
-    
-    # Prepare the API request payload
-    # print("🔍 DEBUG: Preparing API request...")
-    
-    # Try to use GPT-4 first, but fall back to other models if needed
+
     models_to_try = [
-        "gpt-4o-mini",        # First choice: GPT-4o (most widely available)
+        "gpt-4.1",        # First choice: GPT-4o (most widely available)
     ]
     
     # Check if we have a preferred model in environment
@@ -1680,19 +1666,12 @@ Do not provide any explanations, just the exact command to run.
     if preferred_model:
         # Insert the preferred model at the beginning of the list
         models_to_try.insert(0, preferred_model)
-        # print(f"✅ Using preferred model from environment: {preferred_model}")
         
     # Remove duplicates while preserving order
     models_to_try = list(dict.fromkeys(models_to_try))
-    # print(f"🔍 DEBUG: Models to try: {models_to_try}")
     
     # Function to make the API call with a specific model
     def try_api_call(model_name, retries=2, backoff_factor=1.5):
-        # print(f"🔍 DEBUG: Attempting API call with model: {model_name}")
-        # print(f"🔍 DEBUG: API key available: {'Yes' if api_key else 'No'}")
-        # if api_key:
-        #     print(f"🔍 DEBUG: API key length: {len(api_key)}")
-        #     print(f"🔍 DEBUG: API key starts with: {api_key[:10]}...")
         
         payload = {
             "model": model_name,
@@ -1795,7 +1774,6 @@ Do not provide any explanations, just the exact command to run.
     for model in models_to_try:
         result, error = try_api_call(model)
         if result:
-            # print(f"✅ Successfully got response from {model}")
             break
         else:
             print(f"⚠️ Failed to get response from {model}: {error}")
@@ -1959,7 +1937,7 @@ Provide fixes for all {len(failed_commands)} failed commands:"""
     }
     
     payload = {
-        "model": "gpt-4o-mini",  # Use a more capable model for batch analysis
+        "model": "gpt-4.1",  # Use a more capable model for batch analysis
         "messages": [
             {"role": "system", "content": "You are a debugging assistant. Analyze failed commands and provide specific fix commands. Return only the fix commands and reasons in the specified format."},
             {"role": "user", "content": prompt}
@@ -2016,43 +1994,6 @@ Provide fixes for all {len(failed_commands)} failed commands:"""
         print(f"❌ Error during batch debugging: {e}")
         return []
 
-def prompt_for_hf_token():
-    """Prompt user for Hugging Face token when needed"""
-    # Try to use credentials manager first
-    try:
-        from credentials_manager import CredentialsManager
-        credentials_manager = CredentialsManager()
-        token = credentials_manager.get_huggingface_token()
-        if token:
-            return token
-    except ImportError:
-        # Fall back to direct input if credentials_manager is not available
-        pass
-    
-    # Traditional direct input method as fallback
-    print("\n" + "="*60)
-    print("🔑 HUGGING FACE TOKEN REQUIRED")
-    print("="*60)
-    print("The training script requires a valid Hugging Face token.")
-    print("You can get your token from: https://huggingface.co/settings/tokens")
-    print("📝 Please paste your Hugging Face token below:")
-    print("   (Your input will be hidden for security)")
-    print("-" * 60)
-    
-    try:
-        token = getpass.getpass("HF Token: ").strip()
-        if not token:
-            print("❌ No token provided.")
-            return None
-        print("✅ Token received successfully!")
-        return token
-    except KeyboardInterrupt:
-        print("\n❌ Token input cancelled by user.")
-        return None
-    except Exception as e:
-        print(f"❌ Error getting token: {e}")
-        return None
-
 def generate_random_password(length=16):
     """Generate a random password for SSH access"""
     alphabet = string.ascii_letters + string.digits + "!@#$%^&*"
@@ -2067,11 +2008,6 @@ def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_co
     
     # Use interactive mode if specified
     if interactive:
-        # If GPU type is not specified, prompt for it
-        if not gpu_type:
-            gpu_type = prompt_for_gpu()
-        else:
-            print(f"✅ Using provided GPU type: {gpu_type}")
             
         # If repo URL is not specified, prompt for it
         if not repo_url:
@@ -2748,7 +2684,6 @@ def fetch_setup_commands_from_api(repo_url):
                 print(f"⚠️ GitIngest CLI failed with exit code {result.returncode}")
                 print(f"⚠️ Error output: {result.stderr}")
                 print("Falling back to basic analysis")
-                gitingest_data = generate_basic_repo_analysis_from_url(repo_url)
             else:
                 print(f"✅ GitIngest analysis completed successfully")
                 
@@ -2832,13 +2767,13 @@ def fetch_setup_commands_from_api(repo_url):
                                 print(f"📁 Processed GitIngest data saved to: {processed_file}")
                     except FileNotFoundError:
                         print(f"⚠️ Output file not found at {output_file}")
-                        gitingest_data = generate_basic_repo_analysis_from_url(repo_url)
+                       
                 except Exception as e:
                     print(f"⚠️ Error reading GitIngest output: {e}")
-                    gitingest_data = generate_basic_repo_analysis_from_url(repo_url)
+     
         else:
             # Fall back to basic analysis if gitingest CLI is not available
-            gitingest_data = generate_basic_repo_analysis_from_url(repo_url)
+            gitingest_data = "{}"
         
         # Prepare the request payload with GitIngest data
         payload = {
@@ -3072,171 +3007,6 @@ def generate_fallback_commands(gitingest_data):
     
     return fixed_commands
 
-def generate_basic_repo_analysis_from_url(repo_url):
-    """Generate basic repository analysis data from a repository URL."""
-    import tempfile
-    import subprocess
-    import os
-    import shutil
-    
-    # Create a temporary directory for cloning
-    temp_dir = tempfile.mkdtemp(prefix="repo_basic_analysis_")
-    
-    try:
-        print(f"📥 Cloning repository to {temp_dir} for basic analysis...")
-        clone_result = subprocess.run(
-            ["git", "clone", "--depth", "1", repo_url, temp_dir],
-            capture_output=True,
-            text=True
-        )
-        
-        if clone_result.returncode != 0:
-            print(f"❌ Failed to clone repository: {clone_result.stderr}")
-            return {
-                "system_info": {
-                    "platform": "linux",
-                    "python_version": "3.10",
-                    "detected_language": "Unknown",
-                    "detected_technologies": [],
-                    "file_count": 0,
-                    "repo_stars": 0,
-                    "repo_forks": 0,
-                    "primary_package_manager": "Unknown",
-                    "complexity_level": "low"
-                },
-                "repository_analysis": {
-                    "summary": f"Repository analysis for {repo_url}",
-                    "tree": "Failed to clone repository",
-                    "content_preview": "No content available"
-                },
-                "success": False
-            }
-        
-        print(f"✅ Repository cloned successfully for basic analysis")
-        
-        # Use the existing generate_basic_repo_analysis function
-        return generate_basic_repo_analysis(temp_dir)
-    finally:
-        # Clean up the temporary directory
-        print(f"🧹 Cleaning up temporary directory for basic analysis...")
-        shutil.rmtree(temp_dir, ignore_errors=True)
-
-def generate_basic_repo_analysis(repo_dir):
-    """Generate basic repository analysis when GitIngest is not available."""
-    import os
-    import subprocess
-    
-    # Detect language and technologies based on file extensions
-    file_extensions = {}
-    file_count = 0
-    
-    for root, _, files in os.walk(repo_dir):
-        for file in files:
-            file_count += 1
-            ext = os.path.splitext(file)[1].lower()
-            if ext:
-                file_extensions[ext] = file_extensions.get(ext, 0) + 1
-    
-    # Determine primary language
-    language_map = {
-        '.py': 'Python',
-        '.js': 'JavaScript',
-        '.ts': 'TypeScript',
-        '.jsx': 'JavaScript',
-        '.tsx': 'TypeScript',
-        '.java': 'Java',
-        '.cpp': 'C++',
-        '.c': 'C',
-        '.go': 'Go',
-        '.rs': 'Rust',
-        '.rb': 'Ruby',
-        '.php': 'PHP',
-        '.swift': 'Swift',
-        '.kt': 'Kotlin',
-        '.cs': 'C#'
-    }
-    
-    # Count files by language
-    language_counts = {}
-    for ext, count in file_extensions.items():
-        if ext in language_map:
-            lang = language_map[ext]
-            language_counts[lang] = language_counts.get(lang, 0) + count
-    
-    # Determine primary language
-    primary_language = max(language_counts.items(), key=lambda x: x[1])[0] if language_counts else "Unknown"
-    
-    # Detect package managers
-    package_managers = []
-    package_files = {
-        'requirements.txt': 'pip',
-        'setup.py': 'pip',
-        'pyproject.toml': 'pip',
-        'package.json': 'npm',
-        'yarn.lock': 'yarn',
-        'pnpm-lock.yaml': 'pnpm',
-        'Cargo.toml': 'cargo',
-        'go.mod': 'go',
-        'Gemfile': 'bundler',
-        'pom.xml': 'maven',
-        'build.gradle': 'gradle',
-        'composer.json': 'composer'
-    }
-    
-    for file, manager in package_files.items():
-        if os.path.exists(os.path.join(repo_dir, file)):
-            package_managers.append(manager)
-    
-    primary_package_manager = package_managers[0] if package_managers else "Unknown"
-    
-    # Get README content
-    readme_content = ""
-    for readme_name in ['README.md', 'README', 'README.txt', 'readme.md']:
-        readme_path = os.path.join(repo_dir, readme_name)
-        if os.path.exists(readme_path):
-            with open(readme_path, 'r', encoding='utf-8', errors='ignore') as f:
-                readme_content = f.read()
-            break
-    
-    # Try to get repository info
-    repo_info = {}
-    try:
-        # Get remote origin URL
-        cmd = ["git", "config", "--get", "remote.origin.url"]
-        result = subprocess.run(cmd, cwd=repo_dir, capture_output=True, text=True)
-        if result.returncode == 0:
-            repo_info["url"] = result.stdout.strip()
-        
-        # Get commit count as a proxy for activity
-        cmd = ["git", "rev-list", "--count", "HEAD"]
-        result = subprocess.run(cmd, cwd=repo_dir, capture_output=True, text=True)
-        if result.returncode == 0:
-            repo_info["commit_count"] = int(result.stdout.strip())
-    except Exception:
-        pass
-    
-    # Build the analysis data
-    return {
-        "system_info": {
-            "platform": "linux",  # Assuming Linux for container environment
-            "python_version": "3.10",  # Common Python version
-            "detected_language": primary_language,
-            "detected_technologies": list(language_counts.keys()),
-            "file_count": file_count,
-            "repo_stars": repo_info.get("stars", 0),
-            "repo_forks": repo_info.get("forks", 0),
-            "primary_package_manager": primary_package_manager,
-            "complexity_level": "medium"  # Default assumption
-        },
-        "repository_analysis": {
-            "summary": f"Repository analysis for {repo_dir}",
-            "readme_content": readme_content[:5000] if readme_content else "No README found",
-            "package_managers": package_managers,
-            "file_extensions": list(file_extensions.keys())
-        },
-        "success": True
-    }
-
 def fix_setup_commands(commands):
     """Fix setup commands by removing placeholders and comments."""
     fixed_commands = []
@@ -3746,152 +3516,6 @@ def get_setup_commands_from_gitingest(repo_url):
     print("❌ All API endpoints failed")
     return generate_fallback_commands(gitingest_data)
 
-def prompt_for_gpu():
-    """
-    Prompt the user to select a GPU type from available options using arrow keys.
-    Returns the selected GPU type.
-    """
-    import sys
-    import tty
-    import termios
-    
-    print("\n🔧 GPU Selection Required")
-    print("No GPU type was specified with --gpu flag.")
-    print("Please select a GPU type for your container:")
-    
-    # Define available GPU types and their specifications
-    gpu_specs = {
-        'T4': {'gpu': 'T4', 'memory': '16GB'},
-        'L4': {'gpu': 'L4', 'memory': '24GB'},
-        'A10G': {'gpu': 'A10G', 'memory': '24GB'},
-        'A100-40': {'gpu': 'A100-40GB', 'memory': '40GB'},
-        'A100-80': {'gpu': 'A100-80GB', 'memory': '80GB'},
-        'L40S': {'gpu': 'L40S', 'memory': '48GB'},
-        'H100': {'gpu': 'H100', 'memory': '80GB'},
-        'H200': {'gpu': 'H200', 'memory': '141GB'},
-        'B200': {'gpu': 'B200', 'memory': '141GB'}
-    }
-    
-    # Create a list of options
-    options = list(gpu_specs.keys())
-    selected_index = 2  # Default to A10G (index 2)
-    
-    def get_key():
-        """Get a single keypress from the user."""
-        fd = sys.stdin.fileno()
-        old_settings = termios.tcgetattr(fd)
-        try:
-            tty.setraw(sys.stdin.fileno())
-            ch = sys.stdin.read(1)
-            if ch == '\x1b':  # Escape sequence
-                ch2 = sys.stdin.read(1)
-                if ch2 == '[':
-                    ch3 = sys.stdin.read(1)
-                    if ch3 == 'A':
-                        return 'UP'
-                    elif ch3 == 'B':
-                        return 'DOWN'
-            elif ch == '\r' or ch == '\n':
-                return 'ENTER'
-            elif ch == '\x03':  # Ctrl+C
-                return 'CTRL_C'
-            return ch
-        finally:
-            termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
-    
-    def display_menu():
-        """Display the GPU selection menu with current selection highlighted."""
-        print("\n📊 Available GPU Options:")
-        print("┌──────────────┬─────────┐")
-        print("│   GPU Type   │  VRAM   │")
-        print("├──────────────┼─────────┤")
-        
-        for i, gpu_type in enumerate(options):
-            specs = gpu_specs[gpu_type]
-            # Calculate proper spacing for alignment
-            number_part = f"{i+1}."
-            if i == selected_index:
-                prefix = "> "
-                suffix = " ←"
-            else:
-                prefix = "  "
-                suffix = ""
-            
-            # Ensure consistent width for GPU type column
-            gpu_display = f"{prefix}{number_part} {gpu_type}"
-            gpu_padded = f"{gpu_display:<12}"  # Fixed width for GPU column
-            
-            print(f"│ {gpu_padded} │ {specs['memory']:<7} │{suffix}")
-        
-        print("└──────────────┴─────────┘")
-        print("Use ↑/↓ arrows to select, Enter to confirm, Ctrl+C to cancel")
-    
-    # Clear screen and show initial menu
-    print("\033[2J\033[H", end="")  # Clear screen and move cursor to top
-    display_menu()
-    
-    while True:
-        try:
-            key = get_key()
-            
-            if key == 'UP':
-                selected_index = (selected_index - 1) % len(options)
-                print("\033[2J\033[H", end="")  # Clear screen
-                display_menu()
-            elif key == 'DOWN':
-                selected_index = (selected_index + 1) % len(options)
-                print("\033[2J\033[H", end="")  # Clear screen
-                display_menu()
-            elif key == 'ENTER':
-                selected_gpu = options[selected_index]
-                print(f"\n✅ Selected GPU: {selected_gpu}")
-                return selected_gpu
-            elif key == 'CTRL_C':
-                print("\n🛑 Selection cancelled.")
-                sys.exit(1)
-                
-        except KeyboardInterrupt:
-            print("\n🛑 Selection cancelled.")
-            sys.exit(1)
-        except Exception as e:
-            print(f"\n❌ Error with interactive menu: {e}")
-            print("🔄 Falling back to simple text input...")
-            # Fall back to simple input method
-            try:
-                print("\n📊 Available GPU Options:")
-                for i, gpu_type in enumerate(options, 1):
-                    specs = gpu_specs[gpu_type]
-                    print(f"  {i}. {gpu_type} ({specs['memory']})")
-                print(f"  Default: A10G")
-                
-                choice = input("\n🔍 Select GPU type (number or name, default is A10G): ").strip()
-                if not choice:
-                    print("✅ Using default GPU: A10G")
-                    return "A10G"
-                if choice.isdigit():
-                    index = int(choice) - 1
-                    if 0 <= index < len(options):
-                        selected = options[index]
-                        print(f"✅ Selected GPU: {selected}")
-                        return selected
-                    else:
-                        print(f"⚠️ Invalid number. Using default: A10G")
-                        return "A10G"
-                elif choice in options:
-                    print(f"✅ Selected GPU: {choice}")
-                    return choice
-                else:
-                    print(f"⚠️ Invalid choice '{choice}'. Using default: A10G")
-                    return "A10G"
-            except KeyboardInterrupt:
-                print("\n🛑 Selection cancelled.")
-                sys.exit(1)
-            except Exception as fallback_error:
-                print(f"❌ Error in fallback input: {fallback_error}")
-                print("✅ Using default GPU: A10G")
-                return "A10G"
-
-
 
 def preprocess_commands_with_llm(setup_commands, stored_credentials, api_key=None):
     """
@@ -4198,7 +3822,6 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument('--gpu', type=str, help='GPU type (e.g., A10G, T4, A100-80GB). If not provided, will prompt for GPU selection.')
     parser.add_argument('--repo-url', type=str, help='Repository URL to clone')
-    parser.add_argument('--repo-name', type=str, help='Repository name override')
     parser.add_argument('--setup-commands', type=str, nargs='+', help='Setup commands to run (deprecated)')
     parser.add_argument('--setup-commands-json', type=str, help='Setup commands as JSON array')
     parser.add_argument('--commands-file', type=str, help='Path to file containing setup commands (one per line)')
@@ -4235,11 +3858,6 @@ if __name__ == "__main__":
         _handle_auth_commands(auth_manager, args)
         sys.exit(0)
     
-    # If --list-gpus is specified, just show GPU options and exit
-    if args.list_gpus:
-        prompt_for_gpu()
-        sys.exit(0)
-    
     # If no arguments or only --show-examples is provided, show usage examples
     if len(sys.argv) == 1 or args.show_examples:
         show_usage_examples()
@@ -4298,7 +3916,6 @@ if __name__ == "__main__":
     else:
         print("\n📋 No GPU type specified with --gpu flag.")
         print("🔄 Prompting for GPU selection...")
-        gpu_type = prompt_for_gpu()
         args.gpu = gpu_type
     
     # Display configuration after GPU selection
@@ -4314,14 +3931,17 @@ if __name__ == "__main__":
         print("Setup Commands: Auto-detect from repository")
     
     # Confirm settings
-    try:
-        proceed = input("Proceed with these settings? (Y/n): ").strip().lower()
-        if proceed in ('n', 'no'):
-            print("🛑 Operation cancelled by user.")
+    if not args.yes:
+        try:
+            proceed = input("Proceed with these settings? (Y/n): ").strip().lower()
+            if proceed in ('n', 'no'):
+                print("🛑 Operation cancelled by user.")
+                sys.exit(0)
+        except KeyboardInterrupt:
+            print("\n🛑 Operation cancelled by user.")
             sys.exit(0)
-    except KeyboardInterrupt:
-        print("\n🛑 Operation cancelled by user.")
-        sys.exit(0)
+    else:
+        print("✅ Skipping confirmation prompt (--yes flag used)")
     
     # Interactive mode or missing required arguments
     if args.interactive or not args.repo_url or not args.volume_name:

package/eval_repos.py

@@ -1 +0,0 @@
-