@fonoster/autopilot 0.7.53 → 0.7.54

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,8 +1,8 @@
1
1
  import { AutopilotParams } from "./types";
2
2
  declare class Autopilot {
3
- private params;
4
- private actor;
5
- private vadWorker;
3
+ private readonly params;
4
+ private readonly actor;
5
+ private readonly vadWorker;
6
6
  constructor(params: AutopilotParams);
7
7
  start(): void;
8
8
  stop(): void;
package/dist/envs.d.ts CHANGED
@@ -7,3 +7,4 @@ export declare const NODE_ENV: string;
7
7
  export declare const SKIP_IDENTITY: boolean;
8
8
  export declare const UNSTRUCTURED_API_KEY: string | undefined;
9
9
  export declare const UNSTRUCTURED_API_URL: string;
10
+ export declare const SILERO_VAD_MODEL_PATH: string;
package/dist/envs.js CHANGED
@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
3
3
  return (mod && mod.__esModule) ? mod : { "default": mod };
4
4
  };
5
5
  Object.defineProperty(exports, "__esModule", { value: true });
6
- exports.UNSTRUCTURED_API_URL = exports.UNSTRUCTURED_API_KEY = exports.SKIP_IDENTITY = exports.NODE_ENV = exports.KNOWLEDGE_BASE_ENABLED = exports.AWS_S3_SECRET_ACCESS_KEY = exports.AWS_S3_REGION = exports.AWS_S3_ENDPOINT = exports.AWS_S3_ACCESS_KEY_ID = void 0;
6
+ exports.SILERO_VAD_MODEL_PATH = exports.UNSTRUCTURED_API_URL = exports.UNSTRUCTURED_API_KEY = exports.SKIP_IDENTITY = exports.NODE_ENV = exports.KNOWLEDGE_BASE_ENABLED = exports.AWS_S3_SECRET_ACCESS_KEY = exports.AWS_S3_REGION = exports.AWS_S3_ENDPOINT = exports.AWS_S3_ACCESS_KEY_ID = void 0;
7
7
  /*
8
8
  * Copyright (C) 2024 by Fonoster Inc (https://fonoster.com)
9
9
  * http://github.com/fonoster/fonoster
@@ -38,6 +38,7 @@ exports.NODE_ENV = e.NODE_ENV || "production";
38
38
  exports.SKIP_IDENTITY = e.SKIP_IDENTITY === "true";
39
39
  exports.UNSTRUCTURED_API_KEY = e.UNSTRUCTURED_API_KEY;
40
40
  exports.UNSTRUCTURED_API_URL = e.UNSTRUCTURED_API_URL ?? "https://api.unstructuredapp.io/general/v0/general";
41
+ exports.SILERO_VAD_MODEL_PATH = e.SILERO_VAD_MODEL_PATH ?? (0, path_1.join)(__dirname, "..", "silero_vad.onnx");
41
42
  if (exports.KNOWLEDGE_BASE_ENABLED) {
42
43
  (0, common_1.assertEnvsAreSet)([
43
44
  "AWS_S3_ACCESS_KEY_ID",
@@ -1,14 +1,14 @@
1
1
  import { Vad } from "./types";
2
2
  declare class SileroVad implements Vad {
3
3
  private vad;
4
- private params;
4
+ private readonly params;
5
5
  constructor(params: {
6
- pathToModel?: string;
6
+ pathToModel: string;
7
7
  activationThreshold: number;
8
8
  deactivationThreshold: number;
9
9
  debounceFrames: number;
10
10
  });
11
- pathToModel?: string;
11
+ pathToModel: string;
12
12
  activationThreshold: number;
13
13
  deactivationThreshold: number;
14
14
  debounceFrames: number;
@@ -1,15 +1,14 @@
1
1
  import { ONNXRuntimeAPI, SpeechProbabilities } from "./types";
2
2
  declare class SileroVadModel {
3
- private ort;
4
- private pathToModel;
5
- _session: any;
6
- _h: unknown;
7
- _c: unknown;
8
- _sr: unknown;
3
+ private readonly ort;
4
+ private readonly pathToModel;
5
+ private _session;
6
+ private _state;
7
+ private _sr;
9
8
  constructor(ort: ONNXRuntimeAPI, pathToModel: string);
10
- static new: (ort: ONNXRuntimeAPI, pathToModel: string) => Promise<SileroVadModel>;
9
+ static readonly new: (ort: ONNXRuntimeAPI, pathToModel: string) => Promise<SileroVadModel>;
11
10
  init(): Promise<void>;
11
+ resetState: () => void;
12
12
  process(audioFrame: Float32Array): Promise<SpeechProbabilities>;
13
- resetState(): void;
14
13
  }
15
14
  export { SileroVadModel };
@@ -21,16 +21,23 @@ exports.SileroVadModel = void 0;
21
21
  * limitations under the License.
22
22
  */
23
23
  const fs_1 = require("fs");
24
+ function getNewState(ortInstance) {
25
+ const zeroes = Array(2 * 128).fill(0);
26
+ return new ortInstance.Tensor("float32", zeroes, [2, 1, 128]);
27
+ }
24
28
  class SileroVadModel {
25
29
  constructor(ort, pathToModel) {
26
30
  this.ort = ort;
27
31
  this.pathToModel = pathToModel;
32
+ this.resetState = () => {
33
+ this._state = getNewState(this.ort);
34
+ };
28
35
  }
29
36
  async init() {
30
37
  const modelArrayBuffer = (0, fs_1.readFileSync)(this.pathToModel).buffer;
31
38
  this._session = await this.ort.InferenceSession.create(modelArrayBuffer);
32
39
  this._sr = new this.ort.Tensor("int64", [16000n]);
33
- this.resetState();
40
+ this._state = getNewState(this.ort);
34
41
  }
35
42
  async process(audioFrame) {
36
43
  const t = new this.ort.Tensor("float32", audioFrame, [
@@ -39,22 +46,15 @@ class SileroVadModel {
39
46
  ]);
40
47
  const inputs = {
41
48
  input: t,
42
- h: this._h,
43
- c: this._c,
49
+ state: this._state,
44
50
  sr: this._sr
45
51
  };
46
52
  const out = await this._session.run(inputs);
47
- this._h = out.hn;
48
- this._c = out.cn;
49
- const [isSpeech] = out.output.data;
53
+ this._state = out["stateN"];
54
+ const [isSpeech] = out["output"].data;
50
55
  const notSpeech = 1 - isSpeech;
51
56
  return { notSpeech, isSpeech };
52
57
  }
53
- resetState() {
54
- const zeroes = Array(2 * 64).fill(0);
55
- this._h = new this.ort.Tensor("float32", zeroes, [2, 1, 64]);
56
- this._c = new this.ort.Tensor("float32", zeroes, [2, 1, 64]);
57
- }
58
58
  }
59
59
  exports.SileroVadModel = SileroVadModel;
60
60
  _a = SileroVadModel;
@@ -1,5 +1,5 @@
1
1
  declare function makeVad(params: {
2
- pathToModel?: string;
2
+ pathToModel: string;
3
3
  activationThreshold: number;
4
4
  deactivationThreshold: number;
5
5
  debounceFrames: number;
@@ -34,36 +34,15 @@ var __importStar = (this && this.__importStar) || (function () {
34
34
  })();
35
35
  Object.defineProperty(exports, "__esModule", { value: true });
36
36
  exports.makeVad = makeVad;
37
- /* eslint-disable no-loops/no-loops */
38
- /*
39
- * Copyright (C) 2024 by Fonoster Inc (https://fonoster.com)
40
- * http://github.com/fonoster/fonoster
41
- *
42
- * This file is part of Fonoster
43
- *
44
- * Licensed under the MIT License (the "License");
45
- * you may not use this file except in compliance with
46
- * the License. You may obtain a copy of the License at
47
- *
48
- * https://opensource.org/licenses/MIT
49
- *
50
- * Unless required by applicable law or agreed to in writing, software
51
- * distributed under the License is distributed on an "AS IS" BASIS,
52
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
53
- * See the License for the specific language governing permissions and
54
- * limitations under the License.
55
- */
56
- const path_1 = require("path");
57
37
  const logger_1 = require("@fonoster/logger");
58
38
  const ort = __importStar(require("onnxruntime-node"));
59
39
  const chunkToFloat32Array_1 = require("./chunkToFloat32Array");
60
40
  const SileroVadModel_1 = require("./SileroVadModel");
61
41
  const logger = (0, logger_1.getLogger)({ service: "autopilot", filePath: __filename });
62
- const BUFFER_SIZE = 16000;
42
+ const BUFFER_SIZE = 512;
63
43
  async function makeVad(params) {
64
44
  const { pathToModel, activationThreshold, deactivationThreshold, debounceFrames } = params;
65
- const effectivePath = pathToModel || (0, path_1.join)(__dirname, "..", "..", "silero_vad.onnx");
66
- const silero = await SileroVadModel_1.SileroVadModel.new(ort, effectivePath);
45
+ const silero = await SileroVadModel_1.SileroVadModel.new(ort, pathToModel);
67
46
  let audioBuffer = [];
68
47
  let isSpeechActive = false;
69
48
  let consecutiveSpeechFrames = 0;
@@ -11,9 +11,9 @@ type ONNXRuntimeAPI = {
11
11
  create(modelArrayBuffer: ArrayBuffer): Promise<unknown>;
12
12
  };
13
13
  Tensor: {
14
- new (type: "int64", dims: [16000n]): unknown;
15
- new (type: "float32", data: number[], dims: [2, 1, 64]): unknown;
16
- new (type: "float32", data: Float32Array, dims: [1, number]): unknown;
14
+ new (type: "int64", data: BigInt[]): unknown;
15
+ new (type: "int64", data: BigInt[], dims: [1]): unknown;
16
+ new (type: "float32", data: Float32Array | number[], dims: [2, 1, 128]): unknown;
17
17
  new (type: "float32", data: Float32Array, dims: [1, number]): unknown;
18
18
  };
19
19
  };
package/dist/vadWorker.js CHANGED
@@ -20,7 +20,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
20
20
  */
21
21
  const worker_threads_1 = require("worker_threads");
22
22
  const SileroVad_1 = require("./vad/SileroVad");
23
- const vad = new SileroVad_1.SileroVad(worker_threads_1.workerData);
23
+ const envs_1 = require("./envs");
24
+ const vad = new SileroVad_1.SileroVad({
25
+ ...worker_threads_1.workerData,
26
+ pathToModel: envs_1.SILERO_VAD_MODEL_PATH
27
+ });
24
28
  vad.init().then(() => {
25
29
  worker_threads_1.parentPort?.on("message", (chunk) => {
26
30
  vad.processChunk(chunk, (voiceActivity) => {
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@fonoster/autopilot",
3
- "version": "0.7.53",
3
+ "version": "0.7.54",
4
4
  "description": "Voice AI for the Fonoster platform",
5
5
  "author": "Pedro Sanders <psanders@fonoster.com>",
6
6
  "homepage": "https://github.com/fonoster/fonoster#readme",
@@ -36,10 +36,10 @@
36
36
  },
37
37
  "dependencies": {
38
38
  "@aws-sdk/client-s3": "^3.712.0",
39
- "@fonoster/common": "^0.7.53",
40
- "@fonoster/logger": "^0.7.53",
41
- "@fonoster/types": "^0.7.53",
42
- "@fonoster/voice": "^0.7.53",
39
+ "@fonoster/common": "^0.7.54",
40
+ "@fonoster/logger": "^0.7.54",
41
+ "@fonoster/types": "^0.7.54",
42
+ "@fonoster/voice": "^0.7.54",
43
43
  "@langchain/community": "^0.3.19",
44
44
  "@langchain/core": "^0.3.23",
45
45
  "@langchain/groq": "^0.1.2",
@@ -57,5 +57,5 @@
57
57
  "devDependencies": {
58
58
  "typescript": "^5.5.4"
59
59
  },
60
- "gitHead": "62f407f61199e5e35a5a5af1a70dd055ce127547"
60
+ "gitHead": "f762eb57b3e07b679bbb2194d48abaeaf75e88ef"
61
61
  }
package/silero_vad.onnx CHANGED
Binary file