test(examples): e2e test for examples (#80)

Signed-off-by: Akihiko Kuroda <akihikokuroda2020@gmail.com>

.github/workflows/examples-tests.yml

@@ -0,0 +1,55 @@
+name: E2E Examples
+
+on:
+  push:
+    branches: ["main"]
+    paths-ignore:
+      - "**/*.md"
+  workflow_dispatch:
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  test:
+    name: Tests
+    timeout-minutes: 40
+    runs-on: ubuntu-latest
+
+    strategy:
+      fail-fast: false
+      matrix:
+        node-version: [18.x]
+
+    steps:
+      - uses: actions/checkout@v4
+      - name: Enable Corepack
+        run: corepack enable
+      - name: Use Node.js ${{ matrix.node-version }}
+        uses: actions/setup-node@v4
+        with:
+          node-version: ${{ matrix.node-version }}
+          cache: "yarn"
+      - name: Install dependencies
+        run: yarn install --immutable
+      - name: Install ollama
+        run: curl -fsSL https://ollama.com/install.sh | sh
+      - name: Run ollama
+        run: |
+          ollama serve &
+          ollama pull llama3.1
+      - name: Call ollama API
+        run: |
+          curl -d '{"model": "llama3.1:latest", "stream": false, "prompt":"Whatever I say, asnwer with Yes"}' http://localhost:11434/api/generate
+      - name: Example Tests
+        env:
+          GENAI_API_KEY: ${{ secrets.GENAI_API_KEY }}
+          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+          # TODO: enable WatsonX later
+          # WATSONX_API_KEY: ${{ secrets.WATSONX_API_KEY }}
+          # WATSONX_PROJECT_ID: ${{ secrets.WATSONX_PROJECT_ID }}
+          # WATSONX_SPACE_ID: ${{ secrets.WATSONX_SPACE_ID }}
+          # WATSONX_DEPLOYMENT_ID: ${{ secrets.WATSONX_DEPLOYMENT_ID }}
+        run: |
+          yarn test:examples

examples/vitest.examples.config.ts

@@ -0,0 +1,25 @@
+import { defineConfig } from "vitest/config";
+import tsConfigPaths from "vite-tsconfig-paths";
+import packageJson from "../package.json";
+
+export default defineConfig({
+  test: {
+    globals: true,
+    passWithNoTests: true,
+    testTimeout: 120 * 1000,
+    printConsoleTrace: true,
+    setupFiles: ["./tests/setup.examples.ts"],
+    deps: {
+      interopDefault: false,
+    },
+    maxConcurrency: 10,
+  },
+  define: {
+    __LIBRARY_VERSION: JSON.stringify(packageJson.version),
+  },
+  plugins: [
+    tsConfigPaths({
+      projects: ["tsconfig.json"],
+    }),
+  ],
+});

package.json

@@ -130,6 +130,8 @@
     "test:e2e:watch": "vitest watch tests",
     "test:all": "vitest run",
     "test:watch": "vitest watch",
+    "test:examples": "vitest --config ./examples/vitest.examples.config.ts run tests/examples",
+    "test:examples:watch": "vitest --config ./examples/vitest.examples.config.ts run tests/examples",
     "prepare": "husky",
     "copyright": "./scripts/copyright.sh",
     "release": "release-it",
@@ -200,6 +202,7 @@
     "@types/eslint": "^9.6.1",
     "@types/eslint-config-prettier": "^6.11.3",
     "@types/eslint__js": "^8.42.3",
+    "@types/glob": "^8.1.0",
     "@types/js-yaml": "^4.0.9",
     "@types/mustache": "^4",
     "@types/needle": "^3.3.0",

tests/examples/examples.test.ts

@@ -0,0 +1,73 @@
+/**
+ * Copyright 2024 IBM Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { expect } from "vitest";
+import { exec } from "child_process";
+import { glob } from "glob";
+import { promisify } from "util";
+import { isTruthy } from "remeda";
+import { hasEnv } from "@/internals/env.js";
+
+const execAsync = promisify(exec);
+const includePattern = process.env.INCLUDE_PATTERN || `./examples/**/*.ts`;
+const excludePattern = process.env.EXCLUDE_PATTERN || ``;
+
+const exclude: string[] = [
+  !hasEnv("WATSONX_API_KEY") && [
+    "examples/llms/text.ts",
+    "examples/llms/providers/watsonx_verbose.ts",
+    "examples/llms/providers/watsonx.ts",
+  ],
+  !hasEnv("GROQ_API_KEY") && ["examples/agents/sql.ts", "examples/llms/providers/groq.ts"],
+  !hasEnv("OPENAI_API_KEY") && ["agents/bee_reusable.ts", "examples/llms/providers/openai.ts"],
+  !hasEnv("IBM_VLLM_URL") && ["examples/llms/providers/ibm-vllm.ts"],
+  !hasEnv("COHERE_API_KEY") && ["examples/llms/providers/langchain.ts"],
+  ["examples/llms/providers/bam.ts", "examples/llms/providers/bam_verbose.ts"],
+]
+  .filter(isTruthy)
+  .flat(); // list of examples that are excluded
+
+describe("E2E Examples", async () => {
+  const exampleFiles = await glob(includePattern, {
+    cwd: process.cwd(),
+    dot: false,
+    realpath: true,
+    ignore: [exclude, excludePattern].flat(),
+  });
+
+  for (const example of exampleFiles) {
+    it.concurrent(`Run ${example}`, async () => {
+      await execAsync(`yarn start -- ${example} <<< "Hello world"`)
+        .then((stdout) => {
+          // eslint-disable-next-line no-console
+          console.log({
+            path: example,
+            result: stdout.stdout,
+            error: stdout.stderr,
+          });
+          expect(stdout.stderr).toBeFalsy();
+        })
+        .catch((error) => {
+          // eslint-disable-next-line no-console
+          console.log({
+            path: example,
+            errorCode: error.code,
+          });
+          expect(error.code).toBe(0);
+        });
+    });
+  }
+});

tests/setup.examples.ts

@@ -0,0 +1,36 @@
+/**
+ * Copyright 2024 IBM Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import dotenv from "dotenv";
+import { FrameworkError } from "@/errors.js";
+dotenv.config();
+dotenv.config({
+  path: ".env.test",
+  override: true,
+});
+dotenv.config({
+  path: ".env.test.local",
+  override: true,
+});
+
+expect.addSnapshotSerializer({
+  serialize(val: FrameworkError): string {
+    return val.explain();
+  },
+  test(val): boolean {
+    return val && val instanceof FrameworkError;
+  },
+});

yarn.lock

@@ -2118,6 +2118,16 @@ __metadata:
   languageName: node
   linkType: hard
 
+"@types/glob@npm:^8.1.0":
+  version: 8.1.0
+  resolution: "@types/glob@npm:8.1.0"
+  dependencies:
+    "@types/minimatch": "npm:^5.1.2"
+    "@types/node": "npm:*"
+  checksum: 10c0/ded07aa0d7a1caf3c47b85e262be82989ccd7933b4a14712b79c82fd45a239249811d9fc3a135b3e9457afa163e74a297033d7245b0dc63cd3d032f3906b053f
+  languageName: node
+  linkType: hard
+
 "@types/js-yaml@npm:^4.0.9":
   version: 4.0.9
   resolution: "@types/js-yaml@npm:4.0.9"
@@ -2166,6 +2176,13 @@ __metadata:
   languageName: node
   linkType: hard
 
+"@types/minimatch@npm:^5.1.2":
+  version: 5.1.2
+  resolution: "@types/minimatch@npm:5.1.2"
+  checksum: 10c0/83cf1c11748891b714e129de0585af4c55dd4c2cafb1f1d5233d79246e5e1e19d1b5ad9e8db449667b3ffa2b6c80125c429dbee1054e9efb45758dbc4e118562
+  languageName: node
+  linkType: hard
+
 "@types/ms@npm:*":
   version: 0.7.34
   resolution: "@types/ms@npm:0.7.34"
@@ -2888,6 +2905,7 @@ __metadata:
     "@types/eslint": "npm:^9.6.1"
     "@types/eslint-config-prettier": "npm:^6.11.3"
     "@types/eslint__js": "npm:^8.42.3"
+    "@types/glob": "npm:^8.1.0"
     "@types/js-yaml": "npm:^4.0.9"
     "@types/mustache": "npm:^4"
     "@types/needle": "npm:^3.3.0"