Add greedy_causal_lm js running into sample tests

Signed-off-by: Kirill Suvorov <kirill.suvorov@intel.com>

Author: Retribution98

.github/workflows/linux.yml

@@ -465,6 +465,13 @@ jobs:
           pattern: "{${{ needs.openvino_download.outputs.ov_artifact_name }},genai_archive_${{ matrix.build-type }},genai_samples_${{ matrix.build-type }},genai_wheels}"
           path: ${{ env.INSTALL_DIR }}
           merge-multiple: true
+      
+      - name: Download GenAI JS Bildings Artifacts
+        uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8
+        with:
+          name: genai_nodejs_bindings
+          path: ${{ env.SRC_DIR }}/src/js/bin
+          merge-multiple: true
 
       - name: Extract Artifacts
         run: |
@@ -484,11 +491,24 @@ jobs:
           requirements_files: "${{ env.SRC_DIR }}/samples/requirements.txt"
           local_wheel_dir: ${{ env.INSTALL_DIR }}/wheels
 
+      - name: Setup Node
+        uses: actions/setup-node@1d0ff469b7ec7b3cb9d8673fde0c81c44821de2a # v4.2.0
+        with:
+          node-version: 21
+
+      - name: Install npm package tests dependencies
+        working-directory: ${{ env.SRC_DIR }}/samples/js/text_generation
+        run: |
+          npm install --verbose
+          rm -rf node_modules/openvino-node/bin
+          cp -R ${{ env.INSTALL_DIR }}/openvino_js_package node_modules/openvino-node/bin
+
       - name: Test Samples (Python and C++)
         run: python -m pytest -vs ${{ env.SRC_DIR }}/${{ matrix.test.cmd }} -m "${{ matrix.test.marker }}"
         env:
           LD_LIBRARY_PATH: "${{ env.INSTALL_DIR }}/runtime/lib/intel64:${{ env.INSTALL_DIR }}/runtime/3rdparty/tbb/lib:$LD_LIBRARY_PATH" # Required for C++ samples
           SAMPLES_PY_DIR: "${{ env.INSTALL_DIR }}/samples/python"
+          SAMPLES_JS_DIR: "${{ env.INSTALL_DIR }}/samples/js"
           SAMPLES_CPP_DIR: "${{ env.INSTALL_DIR }}/samples_bin"
           SAMPLES_C_DIR: "${{ env.INSTALL_DIR }}/samples_bin"
 

samples/js/text_generation/greedy_causal_lm.js

@@ -0,0 +1,27 @@
+import { LLMPipeline } from 'openvino-genai-node';
+
+main();
+
+async function main() {
+    const modelPath = process.argv[2];
+    const prompt = process.argv[3];
+    
+    if (!modelPath) {
+        console.error('Please specify path to model directory\n'
+                    + 'Run command must be: `node chat_sample.js *path_to_model_dir* *prompt*`');
+        process.exit(1);
+    }
+    if (!modelPath) {
+        console.error('Please specify prompt\n'
+                      + 'Run command must be: `node chat_sample.js *path_to_model_dir* *prompt*`');
+        process.exit(1);
+    }
+
+    const device = 'CPU'; // GPU can be used as well
+    const pipe = await LLMPipeline(modelPath, device);
+
+    const config = { 'max_new_tokens': 100 };
+    const result = await pipe.generate(prompt, config);
+
+    console.log(result);
+}

tests/python_tests/samples/conftest.py

@@ -104,6 +104,7 @@
 SAMPLES_PY_DIR = os.environ.get("SAMPLES_PY_DIR", os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../samples/python")))
 SAMPLES_CPP_DIR = os.environ.get("SAMPLES_CPP_DIR", os.getcwd())
 SAMPLES_C_DIR = os.environ.get("SAMPLES_CPP_DIR", os.getcwd())
+SAMPLES_JS_DIR = os.environ.get("SAMPLES_JS_DIR", os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../samples/js")))
 
 @pytest.fixture(scope="session", autouse=True)
 def setup_and_teardown(request, tmp_path_factory):

tests/python_tests/samples/test_greedy_causal_lm.py

@@ -5,7 +5,7 @@
 import pytest
 import sys
 
-from conftest import logger, MODELS, SAMPLES_PY_DIR, SAMPLES_CPP_DIR, SAMPLES_C_DIR
+from conftest import logger, MODELS, SAMPLES_PY_DIR, SAMPLES_CPP_DIR, SAMPLES_C_DIR, SAMPLES_JS_DIR
 from test_utils import run_sample
 
 class TestGreedyCausalLM:
@@ -41,9 +41,15 @@ def test_sample_greedy_causal_lm(self, request, convert_model, sample_args):
         c_command =[c_sample, convert_model, sample_args]
         c_result = run_sample(c_command)
 
+        # Test JS sample
+        js_sample = os.path.join(SAMPLES_JS_DIR, "text_generation/greedy_causal_lm.js")
+        js_command =['node', js_sample, convert_model, sample_args]
+        js_result = run_sample(js_command)
+
         # Compare results
         assert py_result.stdout == cpp_result.stdout, f"Results should match"
         assert cpp_result.stdout == c_result.stdout, f"Results should match"
+        assert c_result.stdout == js_result.stdout, f"Results should match"
 
         model_name = request.node.callspec.params['convert_model']
         model = MODELS[model_name]
@@ -61,4 +67,4 @@ def test_sample_greedy_causal_lm(self, request, convert_model, sample_args):
 
             idx = cpp_predictions.find(ref)
             assert -1 != idx, f'Missing "{ref=}" from predictions'
-            cpp_predictions = cpp_predictions[:idx] + cpp_predictions[idx + len(ref):]
+            cpp_predictions = cpp_predictions[:idx] + cpp_predictions[idx + len(ref):]