@livekit/agents 0.1.0

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.
Files changed (114) hide show
  1. package/.turbo/turbo-build.log +4 -0
  2. package/api-extractor.json +20 -0
  3. package/dist/cli.d.ts +3 -0
  4. package/dist/cli.d.ts.map +1 -0
  5. package/dist/cli.js +58 -0
  6. package/dist/cli.js.map +1 -0
  7. package/dist/generator.d.ts +12 -0
  8. package/dist/generator.d.ts.map +1 -0
  9. package/dist/generator.js +8 -0
  10. package/dist/generator.js.map +1 -0
  11. package/dist/http_server.d.ts +11 -0
  12. package/dist/http_server.d.ts.map +1 -0
  13. package/dist/http_server.js +45 -0
  14. package/dist/http_server.js.map +1 -0
  15. package/dist/index.d.ts +14 -0
  16. package/dist/index.d.ts.map +1 -0
  17. package/dist/index.js +17 -0
  18. package/dist/index.js.map +1 -0
  19. package/dist/ipc/job_main.d.ts +5 -0
  20. package/dist/ipc/job_main.d.ts.map +1 -0
  21. package/dist/ipc/job_main.js +77 -0
  22. package/dist/ipc/job_main.js.map +1 -0
  23. package/dist/ipc/job_process.d.ts +22 -0
  24. package/dist/ipc/job_process.d.ts.map +1 -0
  25. package/dist/ipc/job_process.js +73 -0
  26. package/dist/ipc/job_process.js.map +1 -0
  27. package/dist/ipc/protocol.d.ts +40 -0
  28. package/dist/ipc/protocol.d.ts.map +1 -0
  29. package/dist/ipc/protocol.js +14 -0
  30. package/dist/ipc/protocol.js.map +1 -0
  31. package/dist/job_context.d.ts +16 -0
  32. package/dist/job_context.d.ts.map +1 -0
  33. package/dist/job_context.js +31 -0
  34. package/dist/job_context.js.map +1 -0
  35. package/dist/job_request.d.ts +42 -0
  36. package/dist/job_request.d.ts.map +1 -0
  37. package/dist/job_request.js +79 -0
  38. package/dist/job_request.js.map +1 -0
  39. package/dist/log.d.ts +2 -0
  40. package/dist/log.d.ts.map +1 -0
  41. package/dist/log.js +13 -0
  42. package/dist/log.js.map +1 -0
  43. package/dist/plugin.d.ts +10 -0
  44. package/dist/plugin.d.ts.map +1 -0
  45. package/dist/plugin.js +22 -0
  46. package/dist/plugin.js.map +1 -0
  47. package/dist/stt/index.d.ts +3 -0
  48. package/dist/stt/index.d.ts.map +1 -0
  49. package/dist/stt/index.js +6 -0
  50. package/dist/stt/index.js.map +1 -0
  51. package/dist/stt/stream_adapter.d.ts +28 -0
  52. package/dist/stt/stream_adapter.d.ts.map +1 -0
  53. package/dist/stt/stream_adapter.js +82 -0
  54. package/dist/stt/stream_adapter.js.map +1 -0
  55. package/dist/stt/stt.d.ts +34 -0
  56. package/dist/stt/stt.d.ts.map +1 -0
  57. package/dist/stt/stt.js +30 -0
  58. package/dist/stt/stt.js.map +1 -0
  59. package/dist/tokenize.d.ts +15 -0
  60. package/dist/tokenize.d.ts.map +1 -0
  61. package/dist/tokenize.js +12 -0
  62. package/dist/tokenize.js.map +1 -0
  63. package/dist/tts/index.d.ts +4 -0
  64. package/dist/tts/index.d.ts.map +1 -0
  65. package/dist/tts/index.js +7 -0
  66. package/dist/tts/index.js.map +1 -0
  67. package/dist/tts/stream_adapter.d.ts +26 -0
  68. package/dist/tts/stream_adapter.d.ts.map +1 -0
  69. package/dist/tts/stream_adapter.js +77 -0
  70. package/dist/tts/stream_adapter.js.map +1 -0
  71. package/dist/tts/tts.d.ts +37 -0
  72. package/dist/tts/tts.d.ts.map +1 -0
  73. package/dist/tts/tts.js +48 -0
  74. package/dist/tts/tts.js.map +1 -0
  75. package/dist/utils.d.ts +4 -0
  76. package/dist/utils.d.ts.map +1 -0
  77. package/dist/utils.js +29 -0
  78. package/dist/utils.js.map +1 -0
  79. package/dist/vad.d.ts +28 -0
  80. package/dist/vad.d.ts.map +1 -0
  81. package/dist/vad.js +14 -0
  82. package/dist/vad.js.map +1 -0
  83. package/dist/version.d.ts +2 -0
  84. package/dist/version.d.ts.map +1 -0
  85. package/dist/version.js +5 -0
  86. package/dist/version.js.map +1 -0
  87. package/dist/worker.d.ts +84 -0
  88. package/dist/worker.d.ts.map +1 -0
  89. package/dist/worker.js +296 -0
  90. package/dist/worker.js.map +1 -0
  91. package/package.json +31 -0
  92. package/src/cli.ts +79 -0
  93. package/src/generator.ts +18 -0
  94. package/src/http_server.ts +47 -0
  95. package/src/index.ts +18 -0
  96. package/src/ipc/job_main.ts +83 -0
  97. package/src/ipc/job_process.ts +96 -0
  98. package/src/ipc/protocol.ts +51 -0
  99. package/src/job_context.ts +49 -0
  100. package/src/job_request.ts +118 -0
  101. package/src/log.ts +13 -0
  102. package/src/plugin.ts +28 -0
  103. package/src/stt/index.ts +6 -0
  104. package/src/stt/stream_adapter.ts +104 -0
  105. package/src/stt/stt.ts +58 -0
  106. package/src/tokenize.ts +22 -0
  107. package/src/tts/index.ts +23 -0
  108. package/src/tts/stream_adapter.ts +92 -0
  109. package/src/tts/tts.ts +78 -0
  110. package/src/utils.ts +37 -0
  111. package/src/vad.ts +42 -0
  112. package/src/version.ts +5 -0
  113. package/src/worker.ts +384 -0
  114. package/tsconfig.json +10 -0
@@ -0,0 +1,79 @@
1
+ import { EventEmitter } from 'events';
2
+ import { log } from './log.js';
3
+ class AnsweredError extends Error {
4
+ constructor() {
5
+ super();
6
+ this.name = 'AnsweredError';
7
+ this.message = 'request already answered';
8
+ }
9
+ }
10
+ var AutoDisconnect;
11
+ (function (AutoDisconnect) {
12
+ AutoDisconnect[AutoDisconnect["ROOM_EMPTY"] = 0] = "ROOM_EMPTY";
13
+ AutoDisconnect[AutoDisconnect["PUBLISHER_LEFT"] = 1] = "PUBLISHER_LEFT";
14
+ AutoDisconnect[AutoDisconnect["NONE"] = 2] = "NONE";
15
+ })(AutoDisconnect || (AutoDisconnect = {}));
16
+ export var AutoSubscribe;
17
+ (function (AutoSubscribe) {
18
+ AutoSubscribe[AutoSubscribe["SUBSCRIBE_ALL"] = 0] = "SUBSCRIBE_ALL";
19
+ AutoSubscribe[AutoSubscribe["SUBSCRIBE_NONE"] = 1] = "SUBSCRIBE_NONE";
20
+ AutoSubscribe[AutoSubscribe["VIDEO_ONLY"] = 2] = "VIDEO_ONLY";
21
+ AutoSubscribe[AutoSubscribe["AUDIO_ONLY"] = 3] = "AUDIO_ONLY";
22
+ })(AutoSubscribe || (AutoSubscribe = {}));
23
+ export class JobRequest {
24
+ #job;
25
+ #answered = false;
26
+ tx;
27
+ logger = log.child({ job: this.job });
28
+ constructor(job, tx) {
29
+ this.#job = job;
30
+ this.tx = tx;
31
+ }
32
+ get id() {
33
+ return this.#job.id;
34
+ }
35
+ get job() {
36
+ return this.#job;
37
+ }
38
+ get room() {
39
+ return this.#job.room;
40
+ }
41
+ get publisher() {
42
+ return this.#job.participant;
43
+ }
44
+ get answered() {
45
+ return this.#answered;
46
+ }
47
+ async reject() {
48
+ if (this.#answered) {
49
+ throw new AnsweredError();
50
+ }
51
+ this.#answered = true;
52
+ this.tx.emit('recv', { avail: false, data: undefined });
53
+ this.logger.info('rejected job', this.id);
54
+ }
55
+ async accept(entry, autoSubscribe = AutoSubscribe.SUBSCRIBE_ALL, autoDisconnect = AutoDisconnect.ROOM_EMPTY, name = '', identity = '', metadata = '') {
56
+ if (this.#answered) {
57
+ throw new AnsweredError();
58
+ }
59
+ this.#answered = true;
60
+ const assign = new EventEmitter();
61
+ assign.on('error', (e) => {
62
+ throw e;
63
+ });
64
+ if (identity === '')
65
+ identity = 'agent-' + this.id;
66
+ const data = {
67
+ entry,
68
+ autoSubscribe,
69
+ autoDisconnect,
70
+ name,
71
+ identity,
72
+ metadata,
73
+ assign,
74
+ };
75
+ this.tx.emit('recv', { avail: true, data });
76
+ this.logger.info('accepted job', this.id);
77
+ }
78
+ }
79
+ //# sourceMappingURL=job_request.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"job_request.js","sourceRoot":"","sources":["../src/job_request.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAE/B,MAAM,aAAc,SAAQ,KAAK;IAC/B;QACE,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,IAAI,GAAG,eAAe,CAAC;QAC5B,IAAI,CAAC,OAAO,GAAG,0BAA0B,CAAC;IAC5C,CAAC;CACF;AAED,IAAK,cAIJ;AAJD,WAAK,cAAc;IACjB,+DAAU,CAAA;IACV,uEAAc,CAAA;IACd,mDAAI,CAAA;AACN,CAAC,EAJI,cAAc,KAAd,cAAc,QAIlB;AAED,MAAM,CAAN,IAAY,aAKX;AALD,WAAY,aAAa;IACvB,mEAAa,CAAA;IACb,qEAAc,CAAA;IACd,6DAAU,CAAA;IACV,6DAAU,CAAA;AACZ,CAAC,EALW,aAAa,KAAb,aAAa,QAKxB;AAiBD,MAAM,OAAO,UAAU;IACrB,IAAI,CAAM;IACV,SAAS,GAAG,KAAK,CAAC;IAClB,EAAE,CAAe;IACjB,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAEtC,YAAY,GAAQ,EAAE,EAAgB;QACpC,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC;QAChB,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;IACf,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACtB,CAAC;IAED,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IACxB,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC;IAC/B,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,MAAM;QACV,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,IAAI,aAAa,EAAE,CAAC;QAC5B,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAc,CAAC,CAAC;QACpE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,MAAM,CACV,KAAa,EACb,gBAA+B,aAAa,CAAC,aAAa,EAC1D,iBAAiC,cAAc,CAAC,UAAU,EAC1D,OAAe,EAAE,EACjB,WAAmB,EAAE,EACrB,WAAmB,EAAE;QAErB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,IAAI,aAAa,EAAE,CAAC;QAC5B,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAEtB,MAAM,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;QAClC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;YACvB,MAAM,CAAC,CAAC;QACV,CAAC,CAAC,CAAC;QAEH,IAAI,QAAQ,KAAK,EAAE;YAAE,QAAQ,GAAG,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC;QAEnD,MAAM,IAAI,GAAe;YACvB,KAAK;YACL,aAAa;YACb,cAAc;YACd,IAAI;YACJ,QAAQ;YACR,QAAQ;YACR,MAAM;SACP,CAAC;QAEF,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAc,CAAC,CAAC;QAExD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IAC5C,CAAC;CACF"}
package/dist/log.d.ts ADDED
@@ -0,0 +1,2 @@
1
+ export declare const log: import("pino").Logger<never>;
2
+ //# sourceMappingURL=log.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"log.d.ts","sourceRoot":"","sources":["../src/log.ts"],"names":[],"mappings":"AAKA,eAAO,MAAM,GAAG,8BAOd,CAAC"}
package/dist/log.js ADDED
@@ -0,0 +1,13 @@
1
+ // SPDX-FileCopyrightText: 2024 LiveKit, Inc.
2
+ //
3
+ // SPDX-License-Identifier: Apache-2.0
4
+ import { pino } from 'pino';
5
+ export const log = pino({
6
+ transport: {
7
+ target: 'pino-pretty',
8
+ options: {
9
+ colorize: true,
10
+ },
11
+ },
12
+ });
13
+ //# sourceMappingURL=log.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"log.js","sourceRoot":"","sources":["../src/log.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAC7C,EAAE;AACF,sCAAsC;AACtC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAE5B,MAAM,CAAC,MAAM,GAAG,GAAG,IAAI,CAAC;IACtB,SAAS,EAAE;QACT,MAAM,EAAE,aAAa;QACrB,OAAO,EAAE;YACP,QAAQ,EAAE,IAAI;SACf;KACF;CACF,CAAC,CAAC"}
@@ -0,0 +1,10 @@
1
+ export declare abstract class Plugin {
2
+ #private;
3
+ registeredPlugins: Plugin[];
4
+ constructor(title: string, version: string);
5
+ static registerPlugins(plugin: Plugin): void;
6
+ abstract downloadFiles(): void;
7
+ get title(): string;
8
+ get version(): string;
9
+ }
10
+ //# sourceMappingURL=plugin.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"plugin.d.ts","sourceRoot":"","sources":["../src/plugin.ts"],"names":[],"mappings":"AAIA,8BAAsB,MAAM;;IAC1B,iBAAiB,EAAE,MAAM,EAAE,CAAM;gBAIrB,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM;WAK5B,eAAe,CAAC,MAAM,EAAE,MAAM;IAI5C,QAAQ,CAAC,aAAa,IAAI,IAAI;IAE9B,IAAI,KAAK,IAAI,MAAM,CAElB;IAED,IAAI,OAAO,IAAI,MAAM,CAEpB;CACF"}
package/dist/plugin.js ADDED
@@ -0,0 +1,22 @@
1
+ // SPDX-FileCopyrightText: 2024 LiveKit, Inc.
2
+ //
3
+ // SPDX-License-Identifier: Apache-2.0
4
+ export class Plugin {
5
+ registeredPlugins = [];
6
+ #title;
7
+ #version;
8
+ constructor(title, version) {
9
+ this.#title = title;
10
+ this.#version = version;
11
+ }
12
+ static registerPlugins(plugin) {
13
+ plugin.registeredPlugins.push(plugin);
14
+ }
15
+ get title() {
16
+ return this.#title;
17
+ }
18
+ get version() {
19
+ return this.#version;
20
+ }
21
+ }
22
+ //# sourceMappingURL=plugin.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"plugin.js","sourceRoot":"","sources":["../src/plugin.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAC7C,EAAE;AACF,sCAAsC;AAEtC,MAAM,OAAgB,MAAM;IAC1B,iBAAiB,GAAa,EAAE,CAAC;IACjC,MAAM,CAAS;IACf,QAAQ,CAAS;IAEjB,YAAY,KAAa,EAAE,OAAe;QACxC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IAC1B,CAAC;IAEM,MAAM,CAAC,eAAe,CAAC,MAAc;QAC1C,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC;IAID,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;CACF"}
@@ -0,0 +1,3 @@
1
+ export { STT, SpeechEvent, SpeechEventType, SpeechStream } from './stt.js';
2
+ export { StreamAdapter, StreamAdapterWrapper } from './stream_adapter.js';
3
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/stt/index.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC"}
@@ -0,0 +1,6 @@
1
+ // SPDX-FileCopyrightText: 2024 LiveKit, Inc.
2
+ //
3
+ // SPDX-License-Identifier: Apache-2.0
4
+ export { STT, SpeechEvent, SpeechEventType, SpeechStream } from './stt.js';
5
+ export { StreamAdapter, StreamAdapterWrapper } from './stream_adapter.js';
6
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/stt/index.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAC7C,EAAE;AACF,sCAAsC;AAEtC,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC"}
@@ -0,0 +1,28 @@
1
+ import type { AudioFrame } from '@livekit/rtc-node';
2
+ import { type AudioBuffer } from '../utils.js';
3
+ import { type VADStream } from '../vad.js';
4
+ import { STT, SpeechEvent, SpeechStream } from './stt.js';
5
+ export declare class StreamAdapterWrapper extends SpeechStream {
6
+ closed: boolean;
7
+ stt: STT;
8
+ vadStream: VADStream;
9
+ eventQueue: (SpeechEvent | undefined)[];
10
+ language?: string;
11
+ task: {
12
+ run: Promise<void>;
13
+ cancel: () => void;
14
+ };
15
+ constructor(stt: STT, vadStream: VADStream, language?: string | undefined);
16
+ run(reject: (arg: Error) => void): Promise<void>;
17
+ pushFrame(frame: AudioFrame): void;
18
+ close(wait?: boolean): Promise<void>;
19
+ next(): IteratorResult<SpeechEvent>;
20
+ }
21
+ export declare class StreamAdapter extends STT {
22
+ stt: STT;
23
+ vadStream: VADStream;
24
+ constructor(stt: STT, vadStream: VADStream);
25
+ recognize(buffer: AudioBuffer, language?: string | undefined): Promise<SpeechEvent>;
26
+ stream(language?: string | undefined): StreamAdapterWrapper;
27
+ }
28
+ //# sourceMappingURL=stream_adapter.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"stream_adapter.d.ts","sourceRoot":"","sources":["../../src/stt/stream_adapter.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,KAAK,WAAW,EAAe,MAAM,aAAa,CAAC;AAC5D,OAAO,EAAgB,KAAK,SAAS,EAAE,MAAM,WAAW,CAAC;AACzD,OAAO,EAAE,GAAG,EAAE,WAAW,EAAmB,YAAY,EAAE,MAAM,UAAU,CAAC;AAE3E,qBAAa,oBAAqB,SAAQ,YAAY;IACpD,MAAM,EAAE,OAAO,CAAC;IAChB,GAAG,EAAE,GAAG,CAAC;IACT,SAAS,EAAE,SAAS,CAAC;IACrB,UAAU,EAAE,CAAC,WAAW,GAAG,SAAS,CAAC,EAAE,CAAC;IACxC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE;QACJ,GAAG,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;QACnB,MAAM,EAAE,MAAM,IAAI,CAAC;KACpB,CAAC;gBAEU,GAAG,EAAE,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,GAAE,MAAM,GAAG,SAAqB;IAe9E,GAAG,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,KAAK,KAAK,IAAI;IAoBtC,SAAS,CAAC,KAAK,EAAE,UAAU;IAQrB,KAAK,CAAC,IAAI,GAAE,OAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAWhD,IAAI,IAAI,cAAc,CAAC,WAAW,CAAC;CAQpC;AAED,qBAAa,aAAc,SAAQ,GAAG;IACpC,GAAG,EAAE,GAAG,CAAC;IACT,SAAS,EAAE,SAAS,CAAC;gBAET,GAAG,EAAE,GAAG,EAAE,SAAS,EAAE,SAAS;IAMpC,SAAS,CACb,MAAM,EAAE,WAAW,EACnB,QAAQ,GAAE,MAAM,GAAG,SAAqB,GACvC,OAAO,CAAC,WAAW,CAAC;IAIvB,MAAM,CAAC,QAAQ,GAAE,MAAM,GAAG,SAAqB;CAGhD"}
@@ -0,0 +1,82 @@
1
+ import { mergeFrames } from '../utils.js';
2
+ import { VADEventType } from '../vad.js';
3
+ import { STT, SpeechEvent, SpeechEventType, SpeechStream } from './stt.js';
4
+ export class StreamAdapterWrapper extends SpeechStream {
5
+ closed;
6
+ stt;
7
+ vadStream;
8
+ eventQueue;
9
+ language;
10
+ task;
11
+ constructor(stt, vadStream, language = undefined) {
12
+ super();
13
+ this.closed = false;
14
+ this.stt = stt;
15
+ this.vadStream = vadStream;
16
+ this.eventQueue = [];
17
+ this.language = language;
18
+ this.task = {
19
+ run: new Promise((_, reject) => {
20
+ this.run(reject);
21
+ }),
22
+ cancel: () => { },
23
+ };
24
+ }
25
+ async run(reject) {
26
+ this.task.cancel = () => {
27
+ this.closed = true;
28
+ reject(new Error('cancelled'));
29
+ };
30
+ for (const event of this.vadStream) {
31
+ if (event.type == VADEventType.START_OF_SPEECH) {
32
+ const startEvent = new SpeechEvent(SpeechEventType.START_OF_SPEECH);
33
+ this.eventQueue.push(startEvent);
34
+ }
35
+ else if (event.type == VADEventType.END_OF_SPEECH) {
36
+ const mergedFrames = mergeFrames(event.speech);
37
+ const endEvent = await this.stt.recognize(mergedFrames, this.language);
38
+ this.eventQueue.push(endEvent);
39
+ }
40
+ }
41
+ this.eventQueue.push(undefined);
42
+ }
43
+ pushFrame(frame) {
44
+ if (this.closed) {
45
+ throw new TypeError('cannot push frame to closed stream');
46
+ }
47
+ this.vadStream.pushFrame(frame);
48
+ }
49
+ async close(wait = true) {
50
+ this.closed = true;
51
+ if (!wait) {
52
+ this.task.cancel();
53
+ }
54
+ await this.vadStream.close(wait);
55
+ await this.task.run;
56
+ }
57
+ next() {
58
+ const item = this.eventQueue.shift();
59
+ if (item) {
60
+ return { done: false, value: item };
61
+ }
62
+ else {
63
+ return { done: true, value: undefined };
64
+ }
65
+ }
66
+ }
67
+ export class StreamAdapter extends STT {
68
+ stt;
69
+ vadStream;
70
+ constructor(stt, vadStream) {
71
+ super(true);
72
+ this.stt = stt;
73
+ this.vadStream = vadStream;
74
+ }
75
+ async recognize(buffer, language = undefined) {
76
+ return await this.stt.recognize(buffer, language);
77
+ }
78
+ stream(language = undefined) {
79
+ return new StreamAdapterWrapper(this.stt, this.vadStream, language);
80
+ }
81
+ }
82
+ //# sourceMappingURL=stream_adapter.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"stream_adapter.js","sourceRoot":"","sources":["../../src/stt/stream_adapter.ts"],"names":[],"mappings":"AAIA,OAAO,EAAoB,WAAW,EAAE,MAAM,aAAa,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAkB,MAAM,WAAW,CAAC;AACzD,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAE3E,MAAM,OAAO,oBAAqB,SAAQ,YAAY;IACpD,MAAM,CAAU;IAChB,GAAG,CAAM;IACT,SAAS,CAAY;IACrB,UAAU,CAA8B;IACxC,QAAQ,CAAU;IAClB,IAAI,CAGF;IAEF,YAAY,GAAQ,EAAE,SAAoB,EAAE,WAA+B,SAAS;QAClF,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,IAAI,GAAG;YACV,GAAG,EAAE,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBAC7B,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACnB,CAAC,CAAC;YACF,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC;SACjB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAA4B;QACpC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,GAAG,EAAE;YACtB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;YACnB,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC;QAEF,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnC,IAAI,KAAK,CAAC,IAAI,IAAI,YAAY,CAAC,eAAe,EAAE,CAAC;gBAC/C,MAAM,UAAU,GAAG,IAAI,WAAW,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;gBACpE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACnC,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;gBACpD,MAAM,YAAY,GAAG,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC/C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACvE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAED,SAAS,CAAC,KAAiB;QACzB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,SAAS,CAAC,oCAAoC,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAAgB,IAAI;QAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QAEnB,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACrB,CAAC;QAED,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;IACtB,CAAC;IAED,IAAI;QACF,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACrC,IAAI,IAAI,EAAE,CAAC;YACT,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACtC,CAAC;aAAM,CAAC;YACN,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAC1C,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,aAAc,SAAQ,GAAG;IACpC,GAAG,CAAM;IACT,SAAS,CAAY;IAErB,YAAY,GAAQ,EAAE,SAAoB;QACxC,KAAK,CAAC,IAAI,CAAC,CAAC;QACZ,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,SAAS,CACb,MAAmB,EACnB,WAA+B,SAAS;QAExC,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,CAAC,WAA+B,SAAS;QAC7C,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IACtE,CAAC;CACF"}
@@ -0,0 +1,34 @@
1
+ import type { AudioFrame } from '@livekit/rtc-node';
2
+ import type { AudioBuffer } from '../utils.js';
3
+ export declare enum SpeechEventType {
4
+ START_OF_SPEECH = 0,
5
+ INTERIM_TRANSCRIPT = 1,
6
+ FINAL_TRANSCRIPT = 2,
7
+ END_OF_SPEECH = 3
8
+ }
9
+ export interface SpeechData {
10
+ language: string;
11
+ text: string;
12
+ startTime: number;
13
+ endTime: number;
14
+ confidence: number;
15
+ }
16
+ export declare class SpeechEvent {
17
+ type: SpeechEventType;
18
+ alternatives: SpeechData[];
19
+ constructor(type: SpeechEventType, alternatives?: SpeechData[]);
20
+ }
21
+ export declare abstract class SpeechStream implements IterableIterator<SpeechEvent> {
22
+ abstract pushFrame(token: AudioFrame): void;
23
+ abstract close(wait: boolean): Promise<void>;
24
+ abstract next(): IteratorResult<SpeechEvent>;
25
+ [Symbol.iterator](): SpeechStream;
26
+ }
27
+ export declare abstract class STT {
28
+ #private;
29
+ constructor(streamingSupported: boolean);
30
+ abstract recognize(buffer: AudioBuffer, language?: string): Promise<SpeechEvent>;
31
+ abstract stream(language: string | undefined): SpeechStream;
32
+ get streamingSupported(): boolean;
33
+ }
34
+ //# sourceMappingURL=stt.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"stt.d.ts","sourceRoot":"","sources":["../../src/stt/stt.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE/C,oBAAY,eAAe;IACzB,eAAe,IAAI;IACnB,kBAAkB,IAAI;IACtB,gBAAgB,IAAI;IACpB,aAAa,IAAI;CAClB;AAED,MAAM,WAAW,UAAU;IACzB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,WAAW;IACtB,IAAI,EAAE,eAAe,CAAC;IACtB,YAAY,EAAE,UAAU,EAAE,CAAC;gBAEf,IAAI,EAAE,eAAe,EAAE,YAAY,GAAE,UAAU,EAAO;CAInE;AAED,8BAAsB,YAAa,YAAW,gBAAgB,CAAC,WAAW,CAAC;IACzE,QAAQ,CAAC,SAAS,CAAC,KAAK,EAAE,UAAU,GAAG,IAAI;IAE3C,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAE5C,QAAQ,CAAC,IAAI,IAAI,cAAc,CAAC,WAAW,CAAC;IAE5C,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,YAAY;CAGlC;AAED,8BAAsB,GAAG;;gBAGX,kBAAkB,EAAE,OAAO;IAIvC,QAAQ,CAAC,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAEhF,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,GAAG,YAAY;IAE3D,IAAI,kBAAkB,IAAI,OAAO,CAEhC;CACF"}
@@ -0,0 +1,30 @@
1
+ export var SpeechEventType;
2
+ (function (SpeechEventType) {
3
+ SpeechEventType[SpeechEventType["START_OF_SPEECH"] = 0] = "START_OF_SPEECH";
4
+ SpeechEventType[SpeechEventType["INTERIM_TRANSCRIPT"] = 1] = "INTERIM_TRANSCRIPT";
5
+ SpeechEventType[SpeechEventType["FINAL_TRANSCRIPT"] = 2] = "FINAL_TRANSCRIPT";
6
+ SpeechEventType[SpeechEventType["END_OF_SPEECH"] = 3] = "END_OF_SPEECH";
7
+ })(SpeechEventType || (SpeechEventType = {}));
8
+ export class SpeechEvent {
9
+ type;
10
+ alternatives;
11
+ constructor(type, alternatives = []) {
12
+ this.type = type;
13
+ this.alternatives = alternatives;
14
+ }
15
+ }
16
+ export class SpeechStream {
17
+ [Symbol.iterator]() {
18
+ return this;
19
+ }
20
+ }
21
+ export class STT {
22
+ #streamingSupported;
23
+ constructor(streamingSupported) {
24
+ this.#streamingSupported = streamingSupported;
25
+ }
26
+ get streamingSupported() {
27
+ return this.#streamingSupported;
28
+ }
29
+ }
30
+ //# sourceMappingURL=stt.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"stt.js","sourceRoot":"","sources":["../../src/stt/stt.ts"],"names":[],"mappings":"AAMA,MAAM,CAAN,IAAY,eAKX;AALD,WAAY,eAAe;IACzB,2EAAmB,CAAA;IACnB,iFAAsB,CAAA;IACtB,6EAAoB,CAAA;IACpB,uEAAiB,CAAA;AACnB,CAAC,EALW,eAAe,KAAf,eAAe,QAK1B;AAUD,MAAM,OAAO,WAAW;IACtB,IAAI,CAAkB;IACtB,YAAY,CAAe;IAE3B,YAAY,IAAqB,EAAE,eAA6B,EAAE;QAChE,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;CACF;AAED,MAAM,OAAgB,YAAY;IAOhC,CAAC,MAAM,CAAC,QAAQ,CAAC;QACf,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAED,MAAM,OAAgB,GAAG;IACvB,mBAAmB,CAAU;IAE7B,YAAY,kBAA2B;QACrC,IAAI,CAAC,mBAAmB,GAAG,kBAAkB,CAAC;IAChD,CAAC;IAMD,IAAI,kBAAkB;QACpB,OAAO,IAAI,CAAC,mBAAmB,CAAC;IAClC,CAAC;CACF"}
@@ -0,0 +1,15 @@
1
+ export interface SegmentedSentence {
2
+ text: string;
3
+ }
4
+ export declare abstract class SentenceTokenizer {
5
+ abstract tokenize(text: string, language?: string): SegmentedSentence[];
6
+ abstract stream(language: string | undefined): SentenceStream;
7
+ }
8
+ export declare abstract class SentenceStream implements IterableIterator<SegmentedSentence> {
9
+ abstract pushText(text: string): void;
10
+ abstract flush(): Promise<void>;
11
+ close(): Promise<void>;
12
+ abstract next(): IteratorResult<SegmentedSentence>;
13
+ [Symbol.iterator](): SentenceStream;
14
+ }
15
+ //# sourceMappingURL=tokenize.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tokenize.d.ts","sourceRoot":"","sources":["../src/tokenize.ts"],"names":[],"mappings":"AAIA,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;CACd;AAED,8BAAsB,iBAAiB;IACrC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,iBAAiB,EAAE;IACvE,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,GAAG,cAAc;CAC9D;AAED,8BAAsB,cAAe,YAAW,gBAAgB,CAAC,iBAAiB,CAAC;IACjF,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IACrC,QAAQ,CAAC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IACzB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAC5B,QAAQ,CAAC,IAAI,IAAI,cAAc,CAAC,iBAAiB,CAAC;IAClD,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,cAAc;CAGpC"}
@@ -0,0 +1,12 @@
1
+ // SPDX-FileCopyrightText: 2024 LiveKit, Inc.
2
+ //
3
+ // SPDX-License-Identifier: Apache-2.0
4
+ export class SentenceTokenizer {
5
+ }
6
+ export class SentenceStream {
7
+ async close() { }
8
+ [Symbol.iterator]() {
9
+ return this;
10
+ }
11
+ }
12
+ //# sourceMappingURL=tokenize.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tokenize.js","sourceRoot":"","sources":["../src/tokenize.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAC7C,EAAE;AACF,sCAAsC;AAMtC,MAAM,OAAgB,iBAAiB;CAGtC;AAED,MAAM,OAAgB,cAAc;IAGlC,KAAK,CAAC,KAAK,KAAmB,CAAC;IAE/B,CAAC,MAAM,CAAC,QAAQ,CAAC;QACf,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
@@ -0,0 +1,4 @@
1
+ import { StreamAdapter, StreamAdapterWrapper } from './stream_adapter.js';
2
+ import { ChunkedStream, SynthesisEvent, SynthesisEventType, SynthesizeStream, type SynthesizedAudio, TTS } from './tts.js';
3
+ export { TTS, SynthesisEvent, SynthesisEventType, SynthesizedAudio, SynthesizeStream, StreamAdapter, StreamAdapterWrapper, ChunkedStream, };
4
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tts/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC1E,OAAO,EACL,aAAa,EACb,cAAc,EACd,kBAAkB,EAClB,gBAAgB,EAChB,KAAK,gBAAgB,EACrB,GAAG,EACJ,MAAM,UAAU,CAAC;AAElB,OAAO,EACL,GAAG,EACH,cAAc,EACd,kBAAkB,EAClB,gBAAgB,EAChB,gBAAgB,EAChB,aAAa,EACb,oBAAoB,EACpB,aAAa,GACd,CAAC"}
@@ -0,0 +1,7 @@
1
+ // SPDX-FileCopyrightText: 2024 LiveKit, Inc.
2
+ //
3
+ // SPDX-License-Identifier: Apache-2.0
4
+ import { StreamAdapter, StreamAdapterWrapper } from './stream_adapter.js';
5
+ import { ChunkedStream, SynthesisEvent, SynthesisEventType, SynthesizeStream, TTS, } from './tts.js';
6
+ export { TTS, SynthesisEvent, SynthesisEventType, SynthesizeStream, StreamAdapter, StreamAdapterWrapper, ChunkedStream, };
7
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tts/index.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAC7C,EAAE;AACF,sCAAsC;AACtC,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC1E,OAAO,EACL,aAAa,EACb,cAAc,EACd,kBAAkB,EAClB,gBAAgB,EAEhB,GAAG,GACJ,MAAM,UAAU,CAAC;AAElB,OAAO,EACL,GAAG,EACH,cAAc,EACd,kBAAkB,EAElB,gBAAgB,EAChB,aAAa,EACb,oBAAoB,EACpB,aAAa,GACd,CAAC"}
@@ -0,0 +1,26 @@
1
+ import type { SentenceStream, SentenceTokenizer } from '../tokenize.js';
2
+ import { ChunkedStream, SynthesisEvent, SynthesizeStream, TTS } from './tts.js';
3
+ export declare class StreamAdapterWrapper extends SynthesizeStream {
4
+ closed: boolean;
5
+ tts: TTS;
6
+ sentenceStream: SentenceStream;
7
+ eventQueue: (SynthesisEvent | undefined)[];
8
+ task: {
9
+ run: Promise<void>;
10
+ cancel: () => void;
11
+ };
12
+ constructor(tts: TTS, sentenceStream: SentenceStream);
13
+ run(reject: (arg: Error) => void): Promise<void>;
14
+ pushText(token: string): void;
15
+ flush(): Promise<void>;
16
+ next(): IteratorResult<SynthesisEvent>;
17
+ close(): Promise<void>;
18
+ }
19
+ export declare class StreamAdapter extends TTS {
20
+ tts: TTS;
21
+ tokenizer: SentenceTokenizer;
22
+ constructor(tts: TTS, tokenizer: SentenceTokenizer);
23
+ synthesize(text: string): Promise<ChunkedStream>;
24
+ stream(): StreamAdapterWrapper;
25
+ }
26
+ //# sourceMappingURL=stream_adapter.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"stream_adapter.d.ts","sourceRoot":"","sources":["../../src/tts/stream_adapter.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACxE,OAAO,EAAE,aAAa,EAAE,cAAc,EAAsB,gBAAgB,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAEpG,qBAAa,oBAAqB,SAAQ,gBAAgB;IACxD,MAAM,EAAE,OAAO,CAAC;IAChB,GAAG,EAAE,GAAG,CAAC;IACT,cAAc,EAAE,cAAc,CAAC;IAC/B,UAAU,EAAE,CAAC,cAAc,GAAG,SAAS,CAAC,EAAE,CAAC;IAC3C,IAAI,EAAE;QACJ,GAAG,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;QACnB,MAAM,EAAE,MAAM,IAAI,CAAC;KACpB,CAAC;gBAEU,GAAG,EAAE,GAAG,EAAE,cAAc,EAAE,cAAc;IAc9C,GAAG,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,KAAK,KAAK,IAAI;IAiBtC,QAAQ,CAAC,KAAK,EAAE,MAAM;IAIhB,KAAK;IAIX,IAAI,IAAI,cAAc,CAAC,cAAc,CAAC;IAShC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAQ7B;AAED,qBAAa,aAAc,SAAQ,GAAG;IACpC,GAAG,EAAE,GAAG,CAAC;IACT,SAAS,EAAE,iBAAiB,CAAC;gBAEjB,GAAG,EAAE,GAAG,EAAE,SAAS,EAAE,iBAAiB;IAMlD,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAIhD,MAAM;CAGP"}
@@ -0,0 +1,77 @@
1
+ import { SynthesisEvent, SynthesisEventType, SynthesizeStream, TTS } from './tts.js';
2
+ export class StreamAdapterWrapper extends SynthesizeStream {
3
+ closed;
4
+ tts;
5
+ sentenceStream;
6
+ eventQueue;
7
+ task;
8
+ constructor(tts, sentenceStream) {
9
+ super();
10
+ this.closed = false;
11
+ this.tts = tts;
12
+ this.sentenceStream = sentenceStream;
13
+ this.eventQueue = [];
14
+ this.task = {
15
+ run: new Promise((_, reject) => {
16
+ this.run(reject);
17
+ }),
18
+ cancel: () => { },
19
+ };
20
+ }
21
+ async run(reject) {
22
+ while (!this.closed) {
23
+ this.task.cancel = () => {
24
+ this.closed = true;
25
+ reject(new Error('cancelled'));
26
+ };
27
+ for await (const sentence of this.sentenceStream) {
28
+ const audio = await this.tts.synthesize(sentence.text).then((data) => data.next());
29
+ if (!audio.done) {
30
+ this.eventQueue.push(new SynthesisEvent(SynthesisEventType.STARTED));
31
+ this.eventQueue.push(new SynthesisEvent(SynthesisEventType.AUDIO, audio.value));
32
+ this.eventQueue.push(new SynthesisEvent(SynthesisEventType.FINISHED));
33
+ }
34
+ }
35
+ }
36
+ }
37
+ pushText(token) {
38
+ this.sentenceStream.pushText(token);
39
+ }
40
+ async flush() {
41
+ await this.sentenceStream.flush();
42
+ }
43
+ next() {
44
+ const event = this.eventQueue.shift();
45
+ if (event) {
46
+ return { done: false, value: event };
47
+ }
48
+ else {
49
+ return { done: true, value: undefined };
50
+ }
51
+ }
52
+ async close() {
53
+ this.task.cancel();
54
+ try {
55
+ await this.task.run;
56
+ }
57
+ finally {
58
+ this.eventQueue.push(undefined);
59
+ }
60
+ }
61
+ }
62
+ export class StreamAdapter extends TTS {
63
+ tts;
64
+ tokenizer;
65
+ constructor(tts, tokenizer) {
66
+ super(true);
67
+ this.tts = tts;
68
+ this.tokenizer = tokenizer;
69
+ }
70
+ synthesize(text) {
71
+ return this.tts.synthesize(text);
72
+ }
73
+ stream() {
74
+ return new StreamAdapterWrapper(this.tts, this.tokenizer.stream(undefined));
75
+ }
76
+ }
77
+ //# sourceMappingURL=stream_adapter.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"stream_adapter.js","sourceRoot":"","sources":["../../src/tts/stream_adapter.ts"],"names":[],"mappings":"AAIA,OAAO,EAAiB,cAAc,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAEpG,MAAM,OAAO,oBAAqB,SAAQ,gBAAgB;IACxD,MAAM,CAAU;IAChB,GAAG,CAAM;IACT,cAAc,CAAiB;IAC/B,UAAU,CAAiC;IAC3C,IAAI,CAGF;IAEF,YAAY,GAAQ,EAAE,cAA8B;QAClD,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,IAAI,GAAG;YACV,GAAG,EAAE,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBAC7B,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACnB,CAAC,CAAC;YACF,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC;SACjB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAA4B;QACpC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACpB,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,GAAG,EAAE;gBACtB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;gBACnB,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YACjC,CAAC,CAAC;YACF,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACjD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBACnF,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;oBAChB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC;oBACrE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,kBAAkB,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;oBAChF,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,KAAa;QACpB,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;IACpC,CAAC;IAED,IAAI;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACtC,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACnB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;QACtB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,aAAc,SAAQ,GAAG;IACpC,GAAG,CAAM;IACT,SAAS,CAAoB;IAE7B,YAAY,GAAQ,EAAE,SAA4B;QAChD,KAAK,CAAC,IAAI,CAAC,CAAC;QACZ,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,UAAU,CAAC,IAAY;QACrB,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAC9E,CAAC;CACF"}
@@ -0,0 +1,37 @@
1
+ import type { AudioFrame } from '@livekit/rtc-node';
2
+ export interface SynthesizedAudio {
3
+ text: string;
4
+ data: AudioFrame;
5
+ }
6
+ export declare enum SynthesisEventType {
7
+ STARTED = 0,
8
+ AUDIO = 1,
9
+ FINISHED = 2
10
+ }
11
+ export declare class SynthesisEvent {
12
+ type: SynthesisEventType;
13
+ audio?: SynthesizedAudio;
14
+ constructor(type: SynthesisEventType, audio?: SynthesizedAudio | undefined);
15
+ }
16
+ export declare abstract class SynthesizeStream implements IterableIterator<SynthesisEvent> {
17
+ abstract pushText(token?: string): void;
18
+ markSegmentEnd(): void;
19
+ abstract close(wait: boolean): Promise<void>;
20
+ abstract next(): IteratorResult<SynthesisEvent>;
21
+ [Symbol.iterator](): SynthesizeStream;
22
+ }
23
+ export declare abstract class TTS {
24
+ #private;
25
+ constructor(streamingSupported: boolean);
26
+ abstract synthesize(text: string): Promise<ChunkedStream>;
27
+ abstract stream(): SynthesizeStream;
28
+ get streamingSupported(): boolean;
29
+ }
30
+ export declare abstract class ChunkedStream implements AsyncIterableIterator<SynthesizedAudio> {
31
+ collect(): Promise<AudioFrame>;
32
+ abstract close(): Promise<void>;
33
+ abstract next(): Promise<IteratorResult<SynthesizedAudio>>;
34
+ [Symbol.iterator](): ChunkedStream;
35
+ [Symbol.asyncIterator](): ChunkedStream;
36
+ }
37
+ //# sourceMappingURL=tts.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tts.d.ts","sourceRoot":"","sources":["../../src/tts/tts.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAGpD,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,UAAU,CAAC;CAClB;AAED,oBAAY,kBAAkB;IAC5B,OAAO,IAAI;IACX,KAAK,IAAI;IACT,QAAQ,IAAI;CACb;AAED,qBAAa,cAAc;IACzB,IAAI,EAAE,kBAAkB,CAAC;IACzB,KAAK,CAAC,EAAE,gBAAgB,CAAC;gBAEb,IAAI,EAAE,kBAAkB,EAAE,KAAK,GAAE,gBAAgB,GAAG,SAAqB;CAItF;AAED,8BAAsB,gBAAiB,YAAW,gBAAgB,CAAC,cAAc,CAAC;IAChF,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI;IAEvC,cAAc;IAId,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAC5C,QAAQ,CAAC,IAAI,IAAI,cAAc,CAAC,cAAc,CAAC;IAE/C,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,gBAAgB;CAGtC;AAED,8BAAsB,GAAG;;gBAGX,kBAAkB,EAAE,OAAO;IAIvC,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAEzD,QAAQ,CAAC,MAAM,IAAI,gBAAgB;IAEnC,IAAI,kBAAkB,IAAI,OAAO,CAEhC;CACF;AAED,8BAAsB,aAAc,YAAW,qBAAqB,CAAC,gBAAgB,CAAC;IAC9E,OAAO,IAAI,OAAO,CAAC,UAAU,CAAC;IAQpC,QAAQ,CAAC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAC/B,QAAQ,CAAC,IAAI,IAAI,OAAO,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;IAE1D,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,aAAa;IAIlC,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa;CAGxC"}
@@ -0,0 +1,48 @@
1
+ import { mergeFrames } from '../utils.js';
2
+ export var SynthesisEventType;
3
+ (function (SynthesisEventType) {
4
+ SynthesisEventType[SynthesisEventType["STARTED"] = 0] = "STARTED";
5
+ SynthesisEventType[SynthesisEventType["AUDIO"] = 1] = "AUDIO";
6
+ SynthesisEventType[SynthesisEventType["FINISHED"] = 2] = "FINISHED";
7
+ })(SynthesisEventType || (SynthesisEventType = {}));
8
+ export class SynthesisEvent {
9
+ type;
10
+ audio;
11
+ constructor(type, audio = undefined) {
12
+ this.type = type;
13
+ this.audio = audio;
14
+ }
15
+ }
16
+ export class SynthesizeStream {
17
+ markSegmentEnd() {
18
+ this.pushText(undefined);
19
+ }
20
+ [Symbol.iterator]() {
21
+ return this;
22
+ }
23
+ }
24
+ export class TTS {
25
+ #streamingSupported;
26
+ constructor(streamingSupported) {
27
+ this.#streamingSupported = streamingSupported;
28
+ }
29
+ get streamingSupported() {
30
+ return this.#streamingSupported;
31
+ }
32
+ }
33
+ export class ChunkedStream {
34
+ async collect() {
35
+ const frames = [];
36
+ for await (const ev of this) {
37
+ frames.push(ev.data);
38
+ }
39
+ return mergeFrames(frames);
40
+ }
41
+ [Symbol.iterator]() {
42
+ return this;
43
+ }
44
+ [Symbol.asyncIterator]() {
45
+ return this;
46
+ }
47
+ }
48
+ //# sourceMappingURL=tts.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tts.js","sourceRoot":"","sources":["../../src/tts/tts.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAO1C,MAAM,CAAN,IAAY,kBAIX;AAJD,WAAY,kBAAkB;IAC5B,iEAAW,CAAA;IACX,6DAAS,CAAA;IACT,mEAAY,CAAA;AACd,CAAC,EAJW,kBAAkB,KAAlB,kBAAkB,QAI7B;AAED,MAAM,OAAO,cAAc;IACzB,IAAI,CAAqB;IACzB,KAAK,CAAoB;IAEzB,YAAY,IAAwB,EAAE,QAAsC,SAAS;QACnF,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AAED,MAAM,OAAgB,gBAAgB;IAGpC,cAAc;QACZ,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC3B,CAAC;IAKD,CAAC,MAAM,CAAC,QAAQ,CAAC;QACf,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAED,MAAM,OAAgB,GAAG;IACvB,mBAAmB,CAAU;IAE7B,YAAY,kBAA2B;QACrC,IAAI,CAAC,mBAAmB,GAAG,kBAAkB,CAAC;IAChD,CAAC;IAMD,IAAI,kBAAkB;QACpB,OAAO,IAAI,CAAC,mBAAmB,CAAC;IAClC,CAAC;CACF;AAED,MAAM,OAAgB,aAAa;IACjC,KAAK,CAAC,OAAO;QACX,MAAM,MAAM,GAAG,EAAE,CAAC;QAClB,IAAI,KAAK,EAAE,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,WAAW,CAAC,MAAM,CAAC,CAAC;IAC7B,CAAC;IAKD,CAAC,MAAM,CAAC,QAAQ,CAAC;QACf,OAAO,IAAI,CAAC;IACd,CAAC;IAED,CAAC,MAAM,CAAC,aAAa,CAAC;QACpB,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}