Merge pull request #25 from Atome-FE/feature/ci

fix: ci cross compile failure report

.github/workflows/llama-build.yml

@@ -4,7 +4,6 @@ on:
     branches:
       - main
       - develop
-      - 18-zsh-illegal-hardware-instruction
   pull_request:
     branches:
       - master

package.json

@@ -49,16 +49,17 @@
     "@types/semver": "^7.3.13",
     "axios": "*",
     "glob": "^9.3.4",
+    "langchain": "^0.0.56",
     "rimraf": "^4.4.1",
     "semver": "^7.3.8",
     "tsup": "^6.7.0",
     "tsx": "^3.12.6",
     "typescript": "^5.0.4",
-    "vitest": "^0.29.8",
-    "langchain": "^0.0.56"
+    "vitest": "^0.29.8"
   },
   "dependencies": {
-    "@llama-node/cli": "0.0.29"
+    "@llama-node/cli": "0.0.29",
+    "p-all": "^5.0.0"
   },
   "optionalDependencies": {
     "@llama-node/core": "0.0.29",
@@ -69,4 +70,4 @@
     "@llama-node/core": "0.0.29",
     "@llama-node/llama-cpp": "0.0.29"
   }
-}
+}

packages/core/package.json

@@ -23,7 +23,7 @@
   },
   "scripts": {
     "prepublish": "npm run artifacts",
-    "cross-compile": "rimraf @llama-node && tsx scripts/cross-compile.ts",
+    "cross-compile": "rimraf @llama-node && tsx scripts/cross-compile.mts",
     "artifacts": "tsx scripts/artifacts.ts",
     "build": "napi build --platform --release",
     "build:debug": "napi build --platform",

packages/core/scripts/cross-compile.ts → packages/core/scripts/cross-compile.mts

@@ -1,4 +1,5 @@
 import { exec } from "child_process";
+import pAll from "p-all";
 
 const getTargets = () => {
     switch (process.platform) {
@@ -16,15 +17,33 @@ const getTargets = () => {
     }
 };
 
-const compile = () => {
+const compile = async () => {
     const targets = getTargets();
-    targets.forEach((target) => {
+    const promises = targets.map((target) => {
         const buildProcess = exec(
             `rustup target add ${target} && napi build --platform --target ${target} --release`
         );
         buildProcess.stdout?.pipe(process.stdout);
         buildProcess.stderr?.pipe(process.stderr);
+
+        return () =>
+            new Promise<boolean>((resolve, reject) => {
+                buildProcess.on("close", (code) => {
+                    if (code !== 0) {
+                        reject(code);
+                    } else {
+                        resolve(true);
+                    }
+                });
+            });
     });
+
+    try {
+        await pAll(promises, { concurrency: 1 });
+    } catch (error) {
+        console.error(error);
+        process.exit(1);
+    }
 };
 
 compile();