Add proxy agent for node fetch

Signed-off-by: Junze Wu <[email protected]>
microsoft · Dec 31, 2024 · c0e0d81 · c0e0d81
1 parent 8687011
commit c0e0d81
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 3 deletions.
diff --git a/js/node/package-lock.json b/js/node/package-lock.json
diff --git a/js/node/package.json b/js/node/package.json
@@ -16,7 +16,8 @@
   "version": "1.21.0",
   "dependencies": {
     "onnxruntime-common": "file:../common",
-    "tar": "^7.0.1"
+    "tar": "^7.0.1",
+    "undici": "^6.21.0"
   },
   "scripts": {
     "postinstall": "node ./script/install",

diff --git a/js/node/script/install.js b/js/node/script/install.js
@@ -23,6 +23,7 @@ const path = require('path');
 const tar = require('tar');
 const { execFileSync } = require('child_process');
 const { Readable } = require('stream');
+const { ProxyAgent } = require('undici');
 
 // commandline flag:
 // --onnxruntime-node-install-cuda         Force install the CUDA EP binaries. Try to detect the CUDA version.
@@ -79,8 +80,12 @@ To use ONNX Runtime Node.js binding with CUDA v11 support, please follow the man
     ORT_VERSION
   }.tgz`,
 }[INSTALL_CUDA_FLAG || tryGetCudaVersion()];
+
+const proxyUrl = process.env.HTTPS_PROXY || process.env.HTTP_PROXY;
+const proxyAgent = new ProxyAgent(proxyUrl);
+
 console.log(`Downloading "${artifactUrl}"...`);
-fetch(artifactUrl).then((res) => {
+fetch(artifactUrl, { dispatcher: proxyAgent }).then((res) => {
   if (!res.ok) {
     throw new Error(`Failed to download the binaries: ${res.status} ${res.statusText}.