mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-24 22:17:32 +00:00
f274bbb0c8
### Description
Add an API for users to get version of current package. example usage:
```js
import { env } from 'onnxruntime-node';
console.log(env.versions.node); // output "1.16.0"
```
```js
import { env } from 'onnxruntime-web';
console.log(env.versions.web); // output "1.16.0"
console.log(env.versions.common); // output "1.16.0"
console.log(env.versions.node); // output "undefined"
```
#16156
32 lines
949 B
TypeScript
32 lines
949 B
TypeScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
import {Env} from './env';
|
|
import {version} from './version';
|
|
|
|
type LogLevelType = Env['logLevel'];
|
|
|
|
let logLevelValue: Required<LogLevelType> = 'warning';
|
|
|
|
export const env: Env = {
|
|
wasm: {} as Env.WebAssemblyFlags,
|
|
webgl: {} as Env.WebGLFlags,
|
|
webgpu: {} as Env.WebGpuFlags,
|
|
versions: {common: version},
|
|
|
|
set logLevel(value: LogLevelType) {
|
|
if (value === undefined) {
|
|
return;
|
|
}
|
|
if (typeof value !== 'string' || ['verbose', 'info', 'warning', 'error', 'fatal'].indexOf(value) === -1) {
|
|
throw new Error(`Unsupported logging level: ${value}`);
|
|
}
|
|
logLevelValue = value;
|
|
},
|
|
get logLevel(): Required<LogLevelType> {
|
|
return logLevelValue;
|
|
},
|
|
};
|
|
|
|
// set property 'logLevel' so that they can be correctly transferred to worker by `postMessage()`.
|
|
Object.defineProperty(env, 'logLevel', {enumerable: true});
|