open-agents-ai 0.138.57 → 0.138.58

package/dist/index.js

@@ -40238,8 +40238,16 @@ Error: ${err2 instanceof Error ? err2.message : String(err2)}`);
                 if (torchCheck === "cpu") {
                   renderWarning("GPU detected but PyTorch is CPU-only. Reinstalling with CUDA support in background...");
                   try {
-                    await this.asyncShell(`${JSON.stringify(venvPy)} -m pip install --quiet --force-reinstall torch torchaudio --index-url https://download.pytorch.org/whl/cu124`, 6e5);
-                    renderInfo("PyTorch reinstalled with CUDA GPU support.");
+                    const detectScript = join50(__dirname, "../../../../scripts/detect-torch.py");
+                    let pipArgs = `torch torchaudio --index-url https://download.pytorch.org/whl/cu124`;
+                    try {
+                      const args = await this.asyncShell(`python3 ${JSON.stringify(detectScript)} --pip-args`, 1e4);
+                      if (args.trim())
+                        pipArgs = args.trim();
+                    } catch {
+                    }
+                    await this.asyncShell(`${JSON.stringify(venvPy)} -m pip install --quiet --force-reinstall ${pipArgs}`, 6e5);
+                    renderInfo("PyTorch reinstalled with GPU support.");
                   } catch {
                   }
                 }
@@ -40262,32 +40270,39 @@ Error: ${err2 instanceof Error ? err2.message : String(err2)}`);
             throw new Error(`Failed to create venv: ${err instanceof Error ? err.message : String(err)}`);
           }
         }
-        let hasCuda = false;
-        try {
-          await this.asyncShell("nvidia-smi", 5e3);
-          hasCuda = true;
-        } catch {
-        }
-        if (hasCuda) {
-          renderInfo("  Installing PyTorch (CUDA GPU)...");
+        {
+          const detectScript = join50(__dirname, "../../../../scripts/detect-torch.py");
+          let pipArgsStr = "torch torchaudio";
+          let torchDesc = "unknown platform";
           try {
-            await this.asyncShell(`${JSON.stringify(venvPy)} -m pip install --quiet torch torchaudio --index-url https://download.pytorch.org/whl/cu124`, 6e5);
+            const detectResult = await this.asyncShell(`${py} ${JSON.stringify(detectScript)} --json`, 15e3);
+            const spec = JSON.parse(detectResult.trim());
+            pipArgsStr = spec.pip_args_str || "torch torchaudio";
+            torchDesc = spec.description || `${spec.platform} ${spec.arch} ${spec.accelerator}`;
           } catch {
             try {
-              await this.asyncShell(`${JSON.stringify(venvPy)} -m pip install --quiet torch torchaudio`, 6e5);
-            } catch (err2) {
-              throw new Error(`Failed to install PyTorch (CUDA): ${err2 instanceof Error ? err2.message : String(err2)}`);
+              await this.asyncShell("nvidia-smi", 5e3);
+              pipArgsStr = "torch torchaudio --index-url https://download.pytorch.org/whl/cu124";
+              torchDesc = "CUDA (fallback)";
+            } catch {
+              const arch = process.arch;
+              if (arch === "arm64" || arch === "arm") {
+                pipArgsStr = "torch torchaudio";
+                torchDesc = "ARM CPU (generic PyPI)";
+              } else {
+                pipArgsStr = "torch torchaudio --index-url https://download.pytorch.org/whl/cpu";
+                torchDesc = "x86_64 CPU";
+              }
             }
           }
-        } else {
-          renderInfo("  Installing PyTorch (CPU \u2014 no NVIDIA GPU detected)...");
+          renderInfo(`  Installing PyTorch (${torchDesc})...`);
           try {
-            await this.asyncShell(`${JSON.stringify(venvPy)} -m pip install --quiet torch torchaudio --index-url https://download.pytorch.org/whl/cpu`, 6e5);
+            await this.asyncShell(`${JSON.stringify(venvPy)} -m pip install --quiet ${pipArgsStr}`, 6e5);
           } catch {
             try {
               await this.asyncShell(`${JSON.stringify(venvPy)} -m pip install --quiet torch torchaudio`, 6e5);
             } catch (err2) {
-              throw new Error(`Failed to install PyTorch: ${err2 instanceof Error ? err2.message : String(err2)}`);
+              throw new Error(`Failed to install PyTorch (${torchDesc}): ${err2 instanceof Error ? err2.message : String(err2)}`);
             }
           }
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "open-agents-ai",
-  "version": "0.138.57",
+  "version": "0.138.58",
   "description": "AI coding agent powered by open-source models (Ollama/vLLM) — interactive TUI with agentic tool-calling loop",
   "type": "module",
   "main": "./dist/index.js",