gitarsenal-cli 1.9.64 → 1.9.65

package/.venv_status.json

@@ -1 +1 @@
-{"created":"2025-08-14T08:14:09.791Z","packages":["modal","gitingest","requests","anthropic"],"uv_version":"uv 0.8.4 (Homebrew 2025-07-30)"}
+{"created":"2025-08-14T08:39:09.974Z","packages":["modal","gitingest","requests","anthropic"],"uv_version":"uv 0.8.4 (Homebrew 2025-07-30)"}

package/package.json

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

package/python/test_modalSandboxScript.py

@@ -197,7 +197,7 @@ def get_stored_credentials():
 
 # Now modify the create_modal_ssh_container function to use the PersistentShell
 def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_commands=None, 
-                               volume_name=None, timeout_minutes=60, ssh_password=None, interactive=False):
+                               volume_name=None, timeout_minutes=60, ssh_password=None, interactive=False, gpu_count=1):
     """Create a Modal SSH container with GPU support and tunneling"""
     
     # Use interactive mode if specified
@@ -295,7 +295,18 @@ def create_modal_ssh_container(gpu_type, repo_url=None, repo_name=None, setup_co
         gpu_type = 'A10G'
     
     gpu_spec = gpu_configs[gpu_type]
-    print(f"🚀 Creating SSH container with {gpu_spec['gpu']} GPU ({gpu_spec['memory']}GB VRAM)")
+    
+    # Configure GPU string for Modal (support multiple GPUs)
+    if gpu_count > 1:
+        modal_gpu_spec = f"{gpu_spec['gpu']}:{gpu_count}"
+        total_memory = gpu_spec['memory'] * gpu_count
+        print(f"🚀 Creating SSH container with {gpu_count}x {gpu_spec['gpu']} GPUs ({total_memory}GB total VRAM)")
+    else:
+        modal_gpu_spec = gpu_spec['gpu']
+        print(f"🚀 Creating SSH container with {gpu_spec['gpu']} GPU ({gpu_spec['memory']}GB VRAM)")
+    
+    # Store the modal GPU specification for the decorator
+    gpu_spec['modal_gpu'] = modal_gpu_spec
     
     # Generate or use provided SSH password
     if not ssh_password:
@@ -394,7 +405,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'],  # Use the user-selected GPU type
+        gpu=gpu_spec['modal_gpu'],  # Use the user-selected GPU type and count
         serialized=True,
         volumes=volumes_config if volumes_config else None,
     )
@@ -1262,6 +1273,12 @@ def show_usage_examples():
     print("│   --volume-name my-persistent-volume                             │")
     print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
     
+    print("With Multiple GPUs")
+    print("┌────────────────────────────────────────────────────────────────────────────────────┐")
+    print("│ gitarsenal --gpu A100-80GB --gpu-count 4 \\                       │")
+    print("│   --repo-url https://github.com/username/repo.git                │")
+    print("└────────────────────────────────────────────────────────────────────────────────────┘\n")
+    
     print("With GitIngest API (default)")
     print("┌────────────────────────────────────────────────────────────────────────────────────┐")
     print("│ gitarsenal --gpu A10G --repo-url https://github.com/username/repo.git             │")
@@ -1286,6 +1303,7 @@ def show_usage_examples():
     
     print("Available GPU Options:")
     print("  T4, L4, A10G, A100-40GB, A100-80GB, L40S, H100, H200, B200")
+    print("  Use --gpu-count to specify multiple GPUs (1-8)")
     print()
     print("Authentication Behavior:")
     print("  • First time: Interactive registration/login required")
@@ -1977,6 +1995,7 @@ if __name__ == "__main__":
     parser.add_argument('--proxy-url', help='URL of the proxy server')
     parser.add_argument('--proxy-api-key', help='API key for the proxy server')
     parser.add_argument('--gpu', default='A10G', help='GPU type to use')
+    parser.add_argument('--gpu-count', type=int, default=1, help='Number of GPUs to use (default: 1)')
     parser.add_argument('--repo-url', help='Repository URL')
     
     # Authentication-related arguments
@@ -2022,6 +2041,8 @@ if __name__ == "__main__":
         print("│ 9. B200      │ 141GB   │")
         print("└──────────────┴─────────┘")
         print("Use --gpu <type> to specify a GPU type")
+        print("Use --gpu-count <number> to specify multiple GPUs (1-8)")
+        print("\nExample: --gpu A100-80GB --gpu-count 4 (for 4x A100-80GB GPUs)")
         sys.exit(0)
     
     # If no arguments or only --show-examples is provided, show usage examples
@@ -2091,7 +2112,11 @@ if __name__ == "__main__":
     # Display configuration after GPU selection
     print("\n📋 Container Configuration:")
     print(f"Repository URL: {args.repo_url or 'Not specified'}")
-    print(f"GPU Type: {gpu_type}")
+    gpu_count = getattr(args, 'gpu_count', 1)
+    if gpu_count > 1:
+        print(f"GPU Type: {gpu_count}x {gpu_type}")
+    else:
+        print(f"GPU Type: {gpu_type}")
     print(f"Volume: {args.volume_name or 'None'}")
     if args.use_api:
         print("Setup Commands: Auto-detect from repository")
@@ -2142,6 +2167,24 @@ if __name__ == "__main__":
                 print("\n🛑 Setup cancelled.")
                 sys.exit(1)
         
+        # Ask about GPU count if not specified
+        gpu_count = getattr(args, 'gpu_count', 1)
+        if not hasattr(args, 'gpu_count') or args.gpu_count == 1:
+            try:
+                gpu_count_input = input("? How many GPUs do you need? (1-8, default: 1): ").strip()
+                if gpu_count_input:
+                    try:
+                        gpu_count = int(gpu_count_input)
+                        if gpu_count < 1 or gpu_count > 8:
+                            print("⚠️ GPU count must be between 1 and 8. Using default: 1")
+                            gpu_count = 1
+                    except ValueError:
+                        print("⚠️ Invalid GPU count. Using default: 1")
+                        gpu_count = 1
+            except KeyboardInterrupt:
+                print("\n🛑 Setup cancelled.")
+                sys.exit(1)
+        
         # Ask about setup commands
         use_gitingest = args.use_gitingest and not args.no_gitingest
         if not args.use_api and not args.setup_commands and not args.setup_commands_json:
@@ -2156,6 +2199,7 @@ if __name__ == "__main__":
         # Update args with interactive values
         args.repo_url = repo_url
         args.volume_name = volume_name
+        args.gpu_count = gpu_count
         args.use_gitingest = use_gitingest
     
     try:    
@@ -2256,14 +2300,11 @@ if __name__ == "__main__":
             volume_name=args.volume_name,
             timeout_minutes=args.timeout,
             ssh_password=ssh_password,
-            interactive=args.interactive
+            interactive=args.interactive,
+            gpu_count=getattr(args, 'gpu_count', 1)
         )
     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()