From 32dc19d9015890d46e345424b69f8bb15df8fcac Mon Sep 17 00:00:00 2001
From: kallebysantos <kalleby_santos@hotmail.com>
Date: Sun, 8 Dec 2024 23:05:24 +0000
Subject: [PATCH] Ensuring that 'Exposed ORT' implements necessary functions

- Adding checkings for 'Tensor' and 'InferenceSession' members of the
exposed custom ort.
---
 src/backends/onnx.js | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/backends/onnx.js b/src/backends/onnx.js
index 8d2a2c17c..e01fd9107 100644
--- a/src/backends/onnx.js
+++ b/src/backends/onnx.js
@@ -55,10 +55,24 @@ const supportedDevices = [];
 let defaultDevices;
 let ONNX;
 
+// If the JS runtime exposes their own ONNX runtime, use it
 if (apis.IS_EXPOSED_RUNTIME_ENV) {
-    // If the JS runtime exposes their own ONNX runtime, use it
-    ONNX = globalThis[apis.EXPOSED_RUNTIME_SYMBOL];
+    const onnxruntime = globalThis[apis.EXPOSED_RUNTIME_SYMBOL];
 
+    // ensure that the runtime implements the necessary functions
+    // consider use array map if need to check more required members.
+    if (!Object.hasOwn(onnxruntime, 'Tensor')) {
+        throw new Error(`Invalid "globalThis[${String(apis.EXPOSED_RUNTIME_SYMBOL)}]" definition. Missing required exported member "Tensor".`)
+    }
+
+    if (!Object.hasOwn(onnxruntime, 'InferenceSession')) {
+        throw new Error(`Invalid "globalThis[${String(apis.EXPOSED_RUNTIME_SYMBOL)}]" definition. Missing required exported member "InferenceSession".`)
+    }
+    if(!Object.hasOwn(onnxruntime?.InferenceSession, 'create')) {
+        throw new Error(`Invalid "globalThis[${String(apis.EXPOSED_RUNTIME_SYMBOL)}].InferenceSession" definition. Missing required exported member "InferenceSession.create".`)
+    }
+
+    ONNX = onnxruntime;
 } else if (apis.IS_NODE_ENV) {
     ONNX = ONNX_NODE.default ?? ONNX_NODE;