@livekit/agents-plugin-rime 1.0.1

package/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

package/README.md

@@ -0,0 +1,15 @@
+<!--
+SPDX-FileCopyrightText: 2024 LiveKit, Inc.
+
+SPDX-License-Identifier: Apache-2.0
+-->
+# Rime plugin for LiveKit Agents
+
+The Agents Framework is designed for building realtime, programmable
+participants that run on servers. Use it to create conversational, multi-modal
+voice agents that can see, hear, and understand.
+
+This package contains the Rime plugin, which provides high-quality text-to-speech (TTS) capabilities for voice synthesis. Refer to the
+[documentation](https://docs.livekit.io/agents/overview/) for information on how
+See the [repository](https://github.com/livekit/agents-js) for more information
+about the framework as a whole.

package/dist/index.cjs

@@ -0,0 +1,36 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __reExport = (target, mod, secondTarget) => (__copyProps(target, mod, "default"), secondTarget && __copyProps(secondTarget, mod, "default"));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var index_exports = {};
+module.exports = __toCommonJS(index_exports);
+var import_agents = require("@livekit/agents");
+__reExport(index_exports, require("./models.cjs"), module.exports);
+__reExport(index_exports, require("./tts.cjs"), module.exports);
+class RimePlugin extends import_agents.Plugin {
+  constructor() {
+    super({
+      title: "rime",
+      version: "0.1.0",
+      package: "@livekit/agents-plugin-rime"
+    });
+  }
+}
+import_agents.Plugin.registerPlugin(new RimePlugin());
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ...require("./models.cjs"),
+  ...require("./tts.cjs")
+});
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2025 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { Plugin } from '@livekit/agents';\n\nexport * from './models.js';\nexport * from './tts.js';\n\nclass RimePlugin extends Plugin {\n  constructor() {\n    super({\n      title: 'rime',\n      version: '0.1.0',\n      package: '@livekit/agents-plugin-rime',\n    });\n  }\n}\n\nPlugin.registerPlugin(new RimePlugin());\n"],"mappings":";;;;;;;;;;;;;;;AAAA;AAAA;AAGA,oBAAuB;AAEvB,0BAAc,wBALd;AAMA,0BAAc,qBANd;AAQA,MAAM,mBAAmB,qBAAO;AAAA,EAC9B,cAAc;AACZ,UAAM;AAAA,MACJ,OAAO;AAAA,MACP,SAAS;AAAA,MACT,SAAS;AAAA,IACX,CAAC;AAAA,EACH;AACF;AAEA,qBAAO,eAAe,IAAI,WAAW,CAAC;","names":[]}

package/dist/index.d.cts

@@ -0,0 +1,3 @@
+export * from './models.js';
+export * from './tts.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export * from './models.js';
+export * from './tts.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAKA,cAAc,aAAa,CAAC;AAC5B,cAAc,UAAU,CAAC"}

package/dist/index.js

@@ -0,0 +1,14 @@
+import { Plugin } from "@livekit/agents";
+export * from "./models.js";
+export * from "./tts.js";
+class RimePlugin extends Plugin {
+  constructor() {
+    super({
+      title: "rime",
+      version: "0.1.0",
+      package: "@livekit/agents-plugin-rime"
+    });
+  }
+}
+Plugin.registerPlugin(new RimePlugin());
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2025 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { Plugin } from '@livekit/agents';\n\nexport * from './models.js';\nexport * from './tts.js';\n\nclass RimePlugin extends Plugin {\n  constructor() {\n    super({\n      title: 'rime',\n      version: '0.1.0',\n      package: '@livekit/agents-plugin-rime',\n    });\n  }\n}\n\nPlugin.registerPlugin(new RimePlugin());\n"],"mappings":"AAGA,SAAS,cAAc;AAEvB,cAAc;AACd,cAAc;AAEd,MAAM,mBAAmB,OAAO;AAAA,EAC9B,cAAc;AACZ,UAAM;AAAA,MACJ,OAAO;AAAA,MACP,SAAS;AAAA,MACT,SAAS;AAAA,IACX,CAAC;AAAA,EACH;AACF;AAEA,OAAO,eAAe,IAAI,WAAW,CAAC;","names":[]}

package/dist/models.cjs

@@ -0,0 +1,17 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var models_exports = {};
+module.exports = __toCommonJS(models_exports);
+//# sourceMappingURL=models.cjs.map

package/dist/models.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/models.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\n\n/** Supported Rime AI TTS models */\nexport type TTSModels = 'arcana' | 'mistv2';\n\n/** Supported default languages for Rime AI TTS */\nexport type DefaultLanguages = 'eng' | 'spa' | 'fra' | 'ger';\n"],"mappings":";;;;;;;;;;;;;;AAAA;AAAA;","names":[]}

package/dist/models.d.cts

@@ -0,0 +1,5 @@
+/** Supported Rime AI TTS models */
+export type TTSModels = 'arcana' | 'mistv2';
+/** Supported default languages for Rime AI TTS */
+export type DefaultLanguages = 'eng' | 'spa' | 'fra' | 'ger';
+//# sourceMappingURL=models.d.ts.map

package/dist/models.d.ts

@@ -0,0 +1,5 @@
+/** Supported Rime AI TTS models */
+export type TTSModels = 'arcana' | 'mistv2';
+/** Supported default languages for Rime AI TTS */
+export type DefaultLanguages = 'eng' | 'spa' | 'fra' | 'ger';
+//# sourceMappingURL=models.d.ts.map

package/dist/models.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../src/models.ts"],"names":[],"mappings":"AAIA,mCAAmC;AACnC,MAAM,MAAM,SAAS,GAAG,QAAQ,GAAG,QAAQ,CAAC;AAE5C,kDAAkD;AAClD,MAAM,MAAM,gBAAgB,GAAG,KAAK,GAAG,KAAK,GAAG,KAAK,GAAG,KAAK,CAAC"}

package/dist/models.js

@@ -0,0 +1 @@
+//# sourceMappingURL=models.js.map

package/dist/models.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/tts.cjs

@@ -0,0 +1,150 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var tts_exports = {};
+__export(tts_exports, {
+  ChunkedStream: () => ChunkedStream,
+  TTS: () => TTS
+});
+module.exports = __toCommonJS(tts_exports);
+var import_agents = require("@livekit/agents");
+const RIME_BASE_URL = "https://users.rime.ai/v1/rime-tts";
+const RIME_TTS_SAMPLE_RATE = 24e3;
+const RIME_TTS_CHANNELS = 1;
+function getSampleRate(opts) {
+  if ((opts == null ? void 0 : opts.samplingRate) && typeof opts.samplingRate === "number") {
+    return opts.samplingRate;
+  }
+  switch (opts == null ? void 0 : opts.modelId) {
+    case "arcana":
+      return 24e3;
+    case "mistv2":
+      return 16e3;
+    default:
+      return RIME_TTS_SAMPLE_RATE;
+  }
+}
+const defaultTTSOptions = {
+  modelId: "arcana",
+  speaker: "luna",
+  apiKey: process.env.RIME_API_KEY,
+  baseURL: RIME_BASE_URL
+};
+class TTS extends import_agents.tts.TTS {
+  opts;
+  label = "rime.TTS";
+  /**
+   * Create a new instance of Rime TTS.
+   *
+   * @remarks
+   * `apiKey` must be set to your Rime AI API key, either using the argument or by setting the
+   * `RIME_API_KEY` environmental variable.
+   *
+   * @param opts - Configuration options for the TTS instance
+   */
+  constructor(opts = defaultTTSOptions) {
+    const sampleRate = getSampleRate(opts);
+    super(sampleRate, RIME_TTS_CHANNELS, {
+      streaming: false
+    });
+    this.opts = { ...defaultTTSOptions, ...opts };
+    if (this.opts.apiKey === void 0) {
+      throw new Error("RIME API key is required, whether as an argument or as $RIME_API_KEY");
+    }
+  }
+  /**
+   * Update TTS options after initialization
+   *
+   * @param opts - Partial options to update
+   */
+  updateOptions(opts) {
+    this.opts = { ...this.opts, ...opts };
+  }
+  /**
+   * Synthesize text to audio using Rime AI TTS.
+   *
+   * @param text - Text to synthesize
+   * @returns A chunked stream of synthesized audio
+   */
+  synthesize(text) {
+    return new ChunkedStream(this, text, this.opts);
+  }
+  stream() {
+    throw new Error("Streaming is not supported on RimeTTS");
+  }
+}
+class ChunkedStream extends import_agents.tts.ChunkedStream {
+  label = "rime-tts.ChunkedStream";
+  opts;
+  text;
+  /**
+   * Create a new ChunkedStream instance.
+   *
+   * @param tts - The parent TTS instance
+   * @param text - Text to synthesize
+   * @param opts - TTS configuration options
+   */
+  constructor(tts2, text, opts) {
+    super(text, tts2);
+    this.text = text;
+    this.opts = opts;
+  }
+  async run() {
+    const requestId = (0, import_agents.shortuuid)();
+    const response = await fetch(`${this.opts.baseURL}`, {
+      method: "POST",
+      headers: {
+        Accept: "audio/pcm",
+        Authorization: `Bearer ${this.opts.apiKey}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({
+        ...Object.fromEntries(
+          Object.entries(this.opts).filter(([k]) => !["apiKey", "baseURL"].includes(k))
+        ),
+        text: this.text
+      })
+    });
+    if (!response.ok) {
+      throw new Error(`Rime AI TTS request failed: ${response.status} ${response.statusText}`);
+    }
+    const buffer = await response.arrayBuffer();
+    const sampleRate = getSampleRate(this.opts);
+    const audioByteStream = new import_agents.AudioByteStream(sampleRate, RIME_TTS_CHANNELS);
+    const frames = audioByteStream.write(buffer);
+    let lastFrame;
+    const sendLastFrame = (segmentId, final) => {
+      if (lastFrame) {
+        this.queue.put({ requestId, segmentId, frame: lastFrame, final });
+        lastFrame = void 0;
+      }
+    };
+    for (const frame of frames) {
+      sendLastFrame(requestId, false);
+      lastFrame = frame;
+    }
+    sendLastFrame(requestId, true);
+    this.queue.close();
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ChunkedStream,
+  TTS
+});
+//# sourceMappingURL=tts.cjs.map

package/dist/tts.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/tts.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { AudioByteStream, shortuuid, tts } from '@livekit/agents';\nimport type { AudioFrame } from '@livekit/rtc-node';\nimport type { DefaultLanguages, TTSModels } from './models.js';\n\nconst RIME_BASE_URL = 'https://users.rime.ai/v1/rime-tts';\nconst RIME_TTS_SAMPLE_RATE = 24000;\nconst RIME_TTS_CHANNELS = 1;\n\n/**\n * Get the appropriate sample rate based on TTS options.\n *\n * @param opts - Optional TTS configuration options\n * @returns The sample rate in Hz. Returns the explicit samplingRate if provided,\n *          otherwise returns model-specific defaults (24000 for arcana, 16000 for mistv2,\n *          or the default RIME_TTS_SAMPLE_RATE for other models)\n */\nfunction getSampleRate(opts?: Partial<TTSOptions>): number {\n  if (opts?.samplingRate && typeof opts.samplingRate === 'number') {\n    return opts.samplingRate;\n  }\n  switch (opts?.modelId) {\n    case 'arcana':\n      return 24000;\n    case 'mistv2':\n      return 16000;\n    default:\n      return RIME_TTS_SAMPLE_RATE;\n  }\n}\n\n/** Configuration options for Rime AI TTS */\nexport interface TTSOptions {\n  speaker: string;\n  modelId: TTSModels | string;\n  baseURL?: string;\n  apiKey?: string;\n  lang?: DefaultLanguages | string;\n  repetition_penalty?: number;\n  temperature?: number;\n  top_p?: number;\n  max_tokens?: number;\n  samplingRate?: number;\n  speedAlpha?: number;\n  pauseBetweenBrackets?: boolean;\n  phonemizeBetweenBrackets?: boolean;\n  inlineSpeedAlpha?: string;\n  noTextNormalization?: boolean;\n  saveOovs?: boolean;\n  /** Additional Rime API parameters */\n  [key: string]: string | number | boolean | undefined;\n}\n\nconst defaultTTSOptions: TTSOptions = {\n  modelId: 'arcana',\n  speaker: 'luna',\n  apiKey: process.env.RIME_API_KEY,\n  baseURL: RIME_BASE_URL,\n};\n\nexport class TTS extends tts.TTS {\n  private opts: TTSOptions;\n  label = 'rime.TTS';\n\n  /**\n   * Create a new instance of Rime TTS.\n   *\n   * @remarks\n   * `apiKey` must be set to your Rime AI API key, either using the argument or by setting the\n   * `RIME_API_KEY` environmental variable.\n   *\n   * @param opts - Configuration options for the TTS instance\n   */\n\n  constructor(opts: Partial<TTSOptions> = defaultTTSOptions) {\n    const sampleRate = getSampleRate(opts);\n    super(sampleRate, RIME_TTS_CHANNELS, {\n      streaming: false,\n    });\n\n    this.opts = { ...defaultTTSOptions, ...opts };\n    if (this.opts.apiKey === undefined) {\n      throw new Error('RIME API key is required, whether as an argument or as $RIME_API_KEY');\n    }\n  }\n\n  /**\n   * Update TTS options after initialization\n   *\n   * @param opts - Partial options to update\n   */\n  updateOptions(opts: Partial<TTSOptions>) {\n    this.opts = { ...this.opts, ...opts };\n  }\n\n  /**\n   * Synthesize text to audio using Rime AI TTS.\n   *\n   * @param text - Text to synthesize\n   * @returns A chunked stream of synthesized audio\n   */\n  synthesize(text: string): ChunkedStream {\n    return new ChunkedStream(this, text, this.opts);\n  }\n\n  stream(): tts.SynthesizeStream {\n    throw new Error('Streaming is not supported on RimeTTS');\n  }\n}\n\nexport class ChunkedStream extends tts.ChunkedStream {\n  label = 'rime-tts.ChunkedStream';\n  private opts: TTSOptions;\n  private text: string;\n\n  /**\n   * Create a new ChunkedStream instance.\n   *\n   * @param tts - The parent TTS instance\n   * @param text - Text to synthesize\n   * @param opts - TTS configuration options\n   */\n  constructor(tts: TTS, text: string, opts: TTSOptions) {\n    super(text, tts);\n    this.text = text;\n    this.opts = opts;\n  }\n\n  protected async run() {\n    const requestId = shortuuid();\n    const response = await fetch(`${this.opts.baseURL}`, {\n      method: 'POST',\n      headers: {\n        Accept: 'audio/pcm',\n        Authorization: `Bearer ${this.opts.apiKey}`,\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify({\n        ...Object.fromEntries(\n          Object.entries(this.opts).filter(([k]) => !['apiKey', 'baseURL'].includes(k)),\n        ),\n        text: this.text,\n      }),\n    });\n\n    if (!response.ok) {\n      throw new Error(`Rime AI TTS request failed: ${response.status} ${response.statusText}`);\n    }\n\n    const buffer = await response.arrayBuffer();\n    const sampleRate = getSampleRate(this.opts);\n    const audioByteStream = new AudioByteStream(sampleRate, RIME_TTS_CHANNELS);\n    const frames = audioByteStream.write(buffer);\n    let lastFrame: AudioFrame | undefined;\n    const sendLastFrame = (segmentId: string, final: boolean) => {\n      if (lastFrame) {\n        this.queue.put({ requestId, segmentId, frame: lastFrame, final });\n        lastFrame = undefined;\n      }\n    };\n\n    for (const frame of frames) {\n      sendLastFrame(requestId, false);\n      lastFrame = frame;\n    }\n    sendLastFrame(requestId, true);\n\n    this.queue.close();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAGA,oBAAgD;AAIhD,MAAM,gBAAgB;AACtB,MAAM,uBAAuB;AAC7B,MAAM,oBAAoB;AAU1B,SAAS,cAAc,MAAoC;AACzD,OAAI,6BAAM,iBAAgB,OAAO,KAAK,iBAAiB,UAAU;AAC/D,WAAO,KAAK;AAAA,EACd;AACA,UAAQ,6BAAM,SAAS;AAAA,IACrB,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT;AACE,aAAO;AAAA,EACX;AACF;AAwBA,MAAM,oBAAgC;AAAA,EACpC,SAAS;AAAA,EACT,SAAS;AAAA,EACT,QAAQ,QAAQ,IAAI;AAAA,EACpB,SAAS;AACX;AAEO,MAAM,YAAY,kBAAI,IAAI;AAAA,EACvB;AAAA,EACR,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYR,YAAY,OAA4B,mBAAmB;AACzD,UAAM,aAAa,cAAc,IAAI;AACrC,UAAM,YAAY,mBAAmB;AAAA,MACnC,WAAW;AAAA,IACb,CAAC;AAED,SAAK,OAAO,EAAE,GAAG,mBAAmB,GAAG,KAAK;AAC5C,QAAI,KAAK,KAAK,WAAW,QAAW;AAClC,YAAM,IAAI,MAAM,sEAAsE;AAAA,IACxF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,cAAc,MAA2B;AACvC,SAAK,OAAO,EAAE,GAAG,KAAK,MAAM,GAAG,KAAK;AAAA,EACtC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,WAAW,MAA6B;AACtC,WAAO,IAAI,cAAc,MAAM,MAAM,KAAK,IAAI;AAAA,EAChD;AAAA,EAEA,SAA+B;AAC7B,UAAM,IAAI,MAAM,uCAAuC;AAAA,EACzD;AACF;AAEO,MAAM,sBAAsB,kBAAI,cAAc;AAAA,EACnD,QAAQ;AAAA,EACA;AAAA,EACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASR,YAAYA,MAAU,MAAc,MAAkB;AACpD,UAAM,MAAMA,IAAG;AACf,SAAK,OAAO;AACZ,SAAK,OAAO;AAAA,EACd;AAAA,EAEA,MAAgB,MAAM;AACpB,UAAM,gBAAY,yBAAU;AAC5B,UAAM,WAAW,MAAM,MAAM,GAAG,KAAK,KAAK,OAAO,IAAI;AAAA,MACnD,QAAQ;AAAA,MACR,SAAS;AAAA,QACP,QAAQ;AAAA,QACR,eAAe,UAAU,KAAK,KAAK,MAAM;AAAA,QACzC,gBAAgB;AAAA,MAClB;AAAA,MACA,MAAM,KAAK,UAAU;AAAA,QACnB,GAAG,OAAO;AAAA,UACR,OAAO,QAAQ,KAAK,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,UAAU,SAAS,EAAE,SAAS,CAAC,CAAC;AAAA,QAC9E;AAAA,QACA,MAAM,KAAK;AAAA,MACb,CAAC;AAAA,IACH,CAAC;AAED,QAAI,CAAC,SAAS,IAAI;AAChB,YAAM,IAAI,MAAM,+BAA+B,SAAS,MAAM,IAAI,SAAS,UAAU,EAAE;AAAA,IACzF;AAEA,UAAM,SAAS,MAAM,SAAS,YAAY;AAC1C,UAAM,aAAa,cAAc,KAAK,IAAI;AAC1C,UAAM,kBAAkB,IAAI,8BAAgB,YAAY,iBAAiB;AACzE,UAAM,SAAS,gBAAgB,MAAM,MAAM;AAC3C,QAAI;AACJ,UAAM,gBAAgB,CAAC,WAAmB,UAAmB;AAC3D,UAAI,WAAW;AACb,aAAK,MAAM,IAAI,EAAE,WAAW,WAAW,OAAO,WAAW,MAAM,CAAC;AAChE,oBAAY;AAAA,MACd;AAAA,IACF;AAEA,eAAW,SAAS,QAAQ;AAC1B,oBAAc,WAAW,KAAK;AAC9B,kBAAY;AAAA,IACd;AACA,kBAAc,WAAW,IAAI;AAE7B,SAAK,MAAM,MAAM;AAAA,EACnB;AACF;","names":["tts"]}

package/dist/tts.d.cts

@@ -0,0 +1,66 @@
+import { tts } from '@livekit/agents';
+import type { DefaultLanguages, TTSModels } from './models.js';
+/** Configuration options for Rime AI TTS */
+export interface TTSOptions {
+    speaker: string;
+    modelId: TTSModels | string;
+    baseURL?: string;
+    apiKey?: string;
+    lang?: DefaultLanguages | string;
+    repetition_penalty?: number;
+    temperature?: number;
+    top_p?: number;
+    max_tokens?: number;
+    samplingRate?: number;
+    speedAlpha?: number;
+    pauseBetweenBrackets?: boolean;
+    phonemizeBetweenBrackets?: boolean;
+    inlineSpeedAlpha?: string;
+    noTextNormalization?: boolean;
+    saveOovs?: boolean;
+    /** Additional Rime API parameters */
+    [key: string]: string | number | boolean | undefined;
+}
+export declare class TTS extends tts.TTS {
+    private opts;
+    label: string;
+    /**
+     * Create a new instance of Rime TTS.
+     *
+     * @remarks
+     * `apiKey` must be set to your Rime AI API key, either using the argument or by setting the
+     * `RIME_API_KEY` environmental variable.
+     *
+     * @param opts - Configuration options for the TTS instance
+     */
+    constructor(opts?: Partial<TTSOptions>);
+    /**
+     * Update TTS options after initialization
+     *
+     * @param opts - Partial options to update
+     */
+    updateOptions(opts: Partial<TTSOptions>): void;
+    /**
+     * Synthesize text to audio using Rime AI TTS.
+     *
+     * @param text - Text to synthesize
+     * @returns A chunked stream of synthesized audio
+     */
+    synthesize(text: string): ChunkedStream;
+    stream(): tts.SynthesizeStream;
+}
+export declare class ChunkedStream extends tts.ChunkedStream {
+    label: string;
+    private opts;
+    private text;
+    /**
+     * Create a new ChunkedStream instance.
+     *
+     * @param tts - The parent TTS instance
+     * @param text - Text to synthesize
+     * @param opts - TTS configuration options
+     */
+    constructor(tts: TTS, text: string, opts: TTSOptions);
+    protected run(): Promise<void>;
+}
+//# sourceMappingURL=tts.d.ts.map

package/dist/tts.d.ts

@@ -0,0 +1,66 @@
+import { tts } from '@livekit/agents';
+import type { DefaultLanguages, TTSModels } from './models.js';
+/** Configuration options for Rime AI TTS */
+export interface TTSOptions {
+    speaker: string;
+    modelId: TTSModels | string;
+    baseURL?: string;
+    apiKey?: string;
+    lang?: DefaultLanguages | string;
+    repetition_penalty?: number;
+    temperature?: number;
+    top_p?: number;
+    max_tokens?: number;
+    samplingRate?: number;
+    speedAlpha?: number;
+    pauseBetweenBrackets?: boolean;
+    phonemizeBetweenBrackets?: boolean;
+    inlineSpeedAlpha?: string;
+    noTextNormalization?: boolean;
+    saveOovs?: boolean;
+    /** Additional Rime API parameters */
+    [key: string]: string | number | boolean | undefined;
+}
+export declare class TTS extends tts.TTS {
+    private opts;
+    label: string;
+    /**
+     * Create a new instance of Rime TTS.
+     *
+     * @remarks
+     * `apiKey` must be set to your Rime AI API key, either using the argument or by setting the
+     * `RIME_API_KEY` environmental variable.
+     *
+     * @param opts - Configuration options for the TTS instance
+     */
+    constructor(opts?: Partial<TTSOptions>);
+    /**
+     * Update TTS options after initialization
+     *
+     * @param opts - Partial options to update
+     */
+    updateOptions(opts: Partial<TTSOptions>): void;
+    /**
+     * Synthesize text to audio using Rime AI TTS.
+     *
+     * @param text - Text to synthesize
+     * @returns A chunked stream of synthesized audio
+     */
+    synthesize(text: string): ChunkedStream;
+    stream(): tts.SynthesizeStream;
+}
+export declare class ChunkedStream extends tts.ChunkedStream {
+    label: string;
+    private opts;
+    private text;
+    /**
+     * Create a new ChunkedStream instance.
+     *
+     * @param tts - The parent TTS instance
+     * @param text - Text to synthesize
+     * @param opts - TTS configuration options
+     */
+    constructor(tts: TTS, text: string, opts: TTSOptions);
+    protected run(): Promise<void>;
+}
+//# sourceMappingURL=tts.d.ts.map

package/dist/tts.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tts.d.ts","sourceRoot":"","sources":["../src/tts.ts"],"names":[],"mappings":"AAGA,OAAO,EAA8B,GAAG,EAAE,MAAM,iBAAiB,CAAC;AAElE,OAAO,KAAK,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AA4B/D,4CAA4C;AAC5C,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,SAAS,GAAG,MAAM,CAAC;IAC5B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,gBAAgB,GAAG,MAAM,CAAC;IACjC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,wBAAwB,CAAC,EAAE,OAAO,CAAC;IACnC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,qCAAqC;IACrC,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,SAAS,CAAC;CACtD;AASD,qBAAa,GAAI,SAAQ,GAAG,CAAC,GAAG;IAC9B,OAAO,CAAC,IAAI,CAAa;IACzB,KAAK,SAAc;IAEnB;;;;;;;;OAQG;gBAES,IAAI,GAAE,OAAO,CAAC,UAAU,CAAqB;IAYzD;;;;OAIG;IACH,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,UAAU,CAAC;IAIvC;;;;;OAKG;IACH,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,aAAa;IAIvC,MAAM,IAAI,GAAG,CAAC,gBAAgB;CAG/B;AAED,qBAAa,aAAc,SAAQ,GAAG,CAAC,aAAa;IAClD,KAAK,SAA4B;IACjC,OAAO,CAAC,IAAI,CAAa;IACzB,OAAO,CAAC,IAAI,CAAS;IAErB;;;;;;OAMG;gBACS,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU;cAMpC,GAAG;CAyCpB"}

package/dist/tts.js

@@ -0,0 +1,125 @@
+import { AudioByteStream, shortuuid, tts } from "@livekit/agents";
+const RIME_BASE_URL = "https://users.rime.ai/v1/rime-tts";
+const RIME_TTS_SAMPLE_RATE = 24e3;
+const RIME_TTS_CHANNELS = 1;
+function getSampleRate(opts) {
+  if ((opts == null ? void 0 : opts.samplingRate) && typeof opts.samplingRate === "number") {
+    return opts.samplingRate;
+  }
+  switch (opts == null ? void 0 : opts.modelId) {
+    case "arcana":
+      return 24e3;
+    case "mistv2":
+      return 16e3;
+    default:
+      return RIME_TTS_SAMPLE_RATE;
+  }
+}
+const defaultTTSOptions = {
+  modelId: "arcana",
+  speaker: "luna",
+  apiKey: process.env.RIME_API_KEY,
+  baseURL: RIME_BASE_URL
+};
+class TTS extends tts.TTS {
+  opts;
+  label = "rime.TTS";
+  /**
+   * Create a new instance of Rime TTS.
+   *
+   * @remarks
+   * `apiKey` must be set to your Rime AI API key, either using the argument or by setting the
+   * `RIME_API_KEY` environmental variable.
+   *
+   * @param opts - Configuration options for the TTS instance
+   */
+  constructor(opts = defaultTTSOptions) {
+    const sampleRate = getSampleRate(opts);
+    super(sampleRate, RIME_TTS_CHANNELS, {
+      streaming: false
+    });
+    this.opts = { ...defaultTTSOptions, ...opts };
+    if (this.opts.apiKey === void 0) {
+      throw new Error("RIME API key is required, whether as an argument or as $RIME_API_KEY");
+    }
+  }
+  /**
+   * Update TTS options after initialization
+   *
+   * @param opts - Partial options to update
+   */
+  updateOptions(opts) {
+    this.opts = { ...this.opts, ...opts };
+  }
+  /**
+   * Synthesize text to audio using Rime AI TTS.
+   *
+   * @param text - Text to synthesize
+   * @returns A chunked stream of synthesized audio
+   */
+  synthesize(text) {
+    return new ChunkedStream(this, text, this.opts);
+  }
+  stream() {
+    throw new Error("Streaming is not supported on RimeTTS");
+  }
+}
+class ChunkedStream extends tts.ChunkedStream {
+  label = "rime-tts.ChunkedStream";
+  opts;
+  text;
+  /**
+   * Create a new ChunkedStream instance.
+   *
+   * @param tts - The parent TTS instance
+   * @param text - Text to synthesize
+   * @param opts - TTS configuration options
+   */
+  constructor(tts2, text, opts) {
+    super(text, tts2);
+    this.text = text;
+    this.opts = opts;
+  }
+  async run() {
+    const requestId = shortuuid();
+    const response = await fetch(`${this.opts.baseURL}`, {
+      method: "POST",
+      headers: {
+        Accept: "audio/pcm",
+        Authorization: `Bearer ${this.opts.apiKey}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({
+        ...Object.fromEntries(
+          Object.entries(this.opts).filter(([k]) => !["apiKey", "baseURL"].includes(k))
+        ),
+        text: this.text
+      })
+    });
+    if (!response.ok) {
+      throw new Error(`Rime AI TTS request failed: ${response.status} ${response.statusText}`);
+    }
+    const buffer = await response.arrayBuffer();
+    const sampleRate = getSampleRate(this.opts);
+    const audioByteStream = new AudioByteStream(sampleRate, RIME_TTS_CHANNELS);
+    const frames = audioByteStream.write(buffer);
+    let lastFrame;
+    const sendLastFrame = (segmentId, final) => {
+      if (lastFrame) {
+        this.queue.put({ requestId, segmentId, frame: lastFrame, final });
+        lastFrame = void 0;
+      }
+    };
+    for (const frame of frames) {
+      sendLastFrame(requestId, false);
+      lastFrame = frame;
+    }
+    sendLastFrame(requestId, true);
+    this.queue.close();
+  }
+}
+export {
+  ChunkedStream,
+  TTS
+};
+//# sourceMappingURL=tts.js.map

package/dist/tts.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/tts.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { AudioByteStream, shortuuid, tts } from '@livekit/agents';\nimport type { AudioFrame } from '@livekit/rtc-node';\nimport type { DefaultLanguages, TTSModels } from './models.js';\n\nconst RIME_BASE_URL = 'https://users.rime.ai/v1/rime-tts';\nconst RIME_TTS_SAMPLE_RATE = 24000;\nconst RIME_TTS_CHANNELS = 1;\n\n/**\n * Get the appropriate sample rate based on TTS options.\n *\n * @param opts - Optional TTS configuration options\n * @returns The sample rate in Hz. Returns the explicit samplingRate if provided,\n *          otherwise returns model-specific defaults (24000 for arcana, 16000 for mistv2,\n *          or the default RIME_TTS_SAMPLE_RATE for other models)\n */\nfunction getSampleRate(opts?: Partial<TTSOptions>): number {\n  if (opts?.samplingRate && typeof opts.samplingRate === 'number') {\n    return opts.samplingRate;\n  }\n  switch (opts?.modelId) {\n    case 'arcana':\n      return 24000;\n    case 'mistv2':\n      return 16000;\n    default:\n      return RIME_TTS_SAMPLE_RATE;\n  }\n}\n\n/** Configuration options for Rime AI TTS */\nexport interface TTSOptions {\n  speaker: string;\n  modelId: TTSModels | string;\n  baseURL?: string;\n  apiKey?: string;\n  lang?: DefaultLanguages | string;\n  repetition_penalty?: number;\n  temperature?: number;\n  top_p?: number;\n  max_tokens?: number;\n  samplingRate?: number;\n  speedAlpha?: number;\n  pauseBetweenBrackets?: boolean;\n  phonemizeBetweenBrackets?: boolean;\n  inlineSpeedAlpha?: string;\n  noTextNormalization?: boolean;\n  saveOovs?: boolean;\n  /** Additional Rime API parameters */\n  [key: string]: string | number | boolean | undefined;\n}\n\nconst defaultTTSOptions: TTSOptions = {\n  modelId: 'arcana',\n  speaker: 'luna',\n  apiKey: process.env.RIME_API_KEY,\n  baseURL: RIME_BASE_URL,\n};\n\nexport class TTS extends tts.TTS {\n  private opts: TTSOptions;\n  label = 'rime.TTS';\n\n  /**\n   * Create a new instance of Rime TTS.\n   *\n   * @remarks\n   * `apiKey` must be set to your Rime AI API key, either using the argument or by setting the\n   * `RIME_API_KEY` environmental variable.\n   *\n   * @param opts - Configuration options for the TTS instance\n   */\n\n  constructor(opts: Partial<TTSOptions> = defaultTTSOptions) {\n    const sampleRate = getSampleRate(opts);\n    super(sampleRate, RIME_TTS_CHANNELS, {\n      streaming: false,\n    });\n\n    this.opts = { ...defaultTTSOptions, ...opts };\n    if (this.opts.apiKey === undefined) {\n      throw new Error('RIME API key is required, whether as an argument or as $RIME_API_KEY');\n    }\n  }\n\n  /**\n   * Update TTS options after initialization\n   *\n   * @param opts - Partial options to update\n   */\n  updateOptions(opts: Partial<TTSOptions>) {\n    this.opts = { ...this.opts, ...opts };\n  }\n\n  /**\n   * Synthesize text to audio using Rime AI TTS.\n   *\n   * @param text - Text to synthesize\n   * @returns A chunked stream of synthesized audio\n   */\n  synthesize(text: string): ChunkedStream {\n    return new ChunkedStream(this, text, this.opts);\n  }\n\n  stream(): tts.SynthesizeStream {\n    throw new Error('Streaming is not supported on RimeTTS');\n  }\n}\n\nexport class ChunkedStream extends tts.ChunkedStream {\n  label = 'rime-tts.ChunkedStream';\n  private opts: TTSOptions;\n  private text: string;\n\n  /**\n   * Create a new ChunkedStream instance.\n   *\n   * @param tts - The parent TTS instance\n   * @param text - Text to synthesize\n   * @param opts - TTS configuration options\n   */\n  constructor(tts: TTS, text: string, opts: TTSOptions) {\n    super(text, tts);\n    this.text = text;\n    this.opts = opts;\n  }\n\n  protected async run() {\n    const requestId = shortuuid();\n    const response = await fetch(`${this.opts.baseURL}`, {\n      method: 'POST',\n      headers: {\n        Accept: 'audio/pcm',\n        Authorization: `Bearer ${this.opts.apiKey}`,\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify({\n        ...Object.fromEntries(\n          Object.entries(this.opts).filter(([k]) => !['apiKey', 'baseURL'].includes(k)),\n        ),\n        text: this.text,\n      }),\n    });\n\n    if (!response.ok) {\n      throw new Error(`Rime AI TTS request failed: ${response.status} ${response.statusText}`);\n    }\n\n    const buffer = await response.arrayBuffer();\n    const sampleRate = getSampleRate(this.opts);\n    const audioByteStream = new AudioByteStream(sampleRate, RIME_TTS_CHANNELS);\n    const frames = audioByteStream.write(buffer);\n    let lastFrame: AudioFrame | undefined;\n    const sendLastFrame = (segmentId: string, final: boolean) => {\n      if (lastFrame) {\n        this.queue.put({ requestId, segmentId, frame: lastFrame, final });\n        lastFrame = undefined;\n      }\n    };\n\n    for (const frame of frames) {\n      sendLastFrame(requestId, false);\n      lastFrame = frame;\n    }\n    sendLastFrame(requestId, true);\n\n    this.queue.close();\n  }\n}\n"],"mappings":"AAGA,SAAS,iBAAiB,WAAW,WAAW;AAIhD,MAAM,gBAAgB;AACtB,MAAM,uBAAuB;AAC7B,MAAM,oBAAoB;AAU1B,SAAS,cAAc,MAAoC;AACzD,OAAI,6BAAM,iBAAgB,OAAO,KAAK,iBAAiB,UAAU;AAC/D,WAAO,KAAK;AAAA,EACd;AACA,UAAQ,6BAAM,SAAS;AAAA,IACrB,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT;AACE,aAAO;AAAA,EACX;AACF;AAwBA,MAAM,oBAAgC;AAAA,EACpC,SAAS;AAAA,EACT,SAAS;AAAA,EACT,QAAQ,QAAQ,IAAI;AAAA,EACpB,SAAS;AACX;AAEO,MAAM,YAAY,IAAI,IAAI;AAAA,EACvB;AAAA,EACR,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYR,YAAY,OAA4B,mBAAmB;AACzD,UAAM,aAAa,cAAc,IAAI;AACrC,UAAM,YAAY,mBAAmB;AAAA,MACnC,WAAW;AAAA,IACb,CAAC;AAED,SAAK,OAAO,EAAE,GAAG,mBAAmB,GAAG,KAAK;AAC5C,QAAI,KAAK,KAAK,WAAW,QAAW;AAClC,YAAM,IAAI,MAAM,sEAAsE;AAAA,IACxF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,cAAc,MAA2B;AACvC,SAAK,OAAO,EAAE,GAAG,KAAK,MAAM,GAAG,KAAK;AAAA,EACtC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,WAAW,MAA6B;AACtC,WAAO,IAAI,cAAc,MAAM,MAAM,KAAK,IAAI;AAAA,EAChD;AAAA,EAEA,SAA+B;AAC7B,UAAM,IAAI,MAAM,uCAAuC;AAAA,EACzD;AACF;AAEO,MAAM,sBAAsB,IAAI,cAAc;AAAA,EACnD,QAAQ;AAAA,EACA;AAAA,EACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASR,YAAYA,MAAU,MAAc,MAAkB;AACpD,UAAM,MAAMA,IAAG;AACf,SAAK,OAAO;AACZ,SAAK,OAAO;AAAA,EACd;AAAA,EAEA,MAAgB,MAAM;AACpB,UAAM,YAAY,UAAU;AAC5B,UAAM,WAAW,MAAM,MAAM,GAAG,KAAK,KAAK,OAAO,IAAI;AAAA,MACnD,QAAQ;AAAA,MACR,SAAS;AAAA,QACP,QAAQ;AAAA,QACR,eAAe,UAAU,KAAK,KAAK,MAAM;AAAA,QACzC,gBAAgB;AAAA,MAClB;AAAA,MACA,MAAM,KAAK,UAAU;AAAA,QACnB,GAAG,OAAO;AAAA,UACR,OAAO,QAAQ,KAAK,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,UAAU,SAAS,EAAE,SAAS,CAAC,CAAC;AAAA,QAC9E;AAAA,QACA,MAAM,KAAK;AAAA,MACb,CAAC;AAAA,IACH,CAAC;AAED,QAAI,CAAC,SAAS,IAAI;AAChB,YAAM,IAAI,MAAM,+BAA+B,SAAS,MAAM,IAAI,SAAS,UAAU,EAAE;AAAA,IACzF;AAEA,UAAM,SAAS,MAAM,SAAS,YAAY;AAC1C,UAAM,aAAa,cAAc,KAAK,IAAI;AAC1C,UAAM,kBAAkB,IAAI,gBAAgB,YAAY,iBAAiB;AACzE,UAAM,SAAS,gBAAgB,MAAM,MAAM;AAC3C,QAAI;AACJ,UAAM,gBAAgB,CAAC,WAAmB,UAAmB;AAC3D,UAAI,WAAW;AACb,aAAK,MAAM,IAAI,EAAE,WAAW,WAAW,OAAO,WAAW,MAAM,CAAC;AAChE,oBAAY;AAAA,MACd;AAAA,IACF;AAEA,eAAW,SAAS,QAAQ;AAC1B,oBAAc,WAAW,KAAK;AAC9B,kBAAY;AAAA,IACd;AACA,kBAAc,WAAW,IAAI;AAE7B,SAAK,MAAM,MAAM;AAAA,EACnB;AACF;","names":["tts"]}

package/dist/tts.test.cjs

@@ -0,0 +1,9 @@
+"use strict";
+var import_agents_plugin_openai = require("@livekit/agents-plugin-openai");
+var import_agents_plugins_test = require("@livekit/agents-plugins-test");
+var import_vitest = require("vitest");
+var import_tts = require("./tts.cjs");
+(0, import_vitest.describe)("Rime TTS", async () => {
+  await (0, import_agents_plugins_test.tts)(new import_tts.TTS(), new import_agents_plugin_openai.STT());
+});
+//# sourceMappingURL=tts.test.cjs.map

package/dist/tts.test.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/tts.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { STT } from '@livekit/agents-plugin-openai';\nimport { tts } from '@livekit/agents-plugins-test';\nimport { describe } from 'vitest';\nimport { TTS } from './tts.js';\n\ndescribe('Rime TTS', async () => {\n  await tts(new TTS(), new STT());\n});\n"],"mappings":";AAGA,kCAAoB;AACpB,iCAAoB;AACpB,oBAAyB;AACzB,iBAAoB;AAAA,IAEpB,wBAAS,YAAY,YAAY;AAC/B,YAAM,gCAAI,IAAI,eAAI,GAAG,IAAI,gCAAI,CAAC;AAChC,CAAC;","names":[]}

package/dist/tts.test.d.cts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=tts.test.d.ts.map

package/dist/tts.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=tts.test.d.ts.map

package/dist/tts.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tts.test.d.ts","sourceRoot":"","sources":["../src/tts.test.ts"],"names":[],"mappings":""}

package/dist/tts.test.js

@@ -0,0 +1,8 @@
+import { STT } from "@livekit/agents-plugin-openai";
+import { tts } from "@livekit/agents-plugins-test";
+import { describe } from "vitest";
+import { TTS } from "./tts.js";
+describe("Rime TTS", async () => {
+  await tts(new TTS(), new STT());
+});
+//# sourceMappingURL=tts.test.js.map

package/dist/tts.test.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/tts.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { STT } from '@livekit/agents-plugin-openai';\nimport { tts } from '@livekit/agents-plugins-test';\nimport { describe } from 'vitest';\nimport { TTS } from './tts.js';\n\ndescribe('Rime TTS', async () => {\n  await tts(new TTS(), new STT());\n});\n"],"mappings":"AAGA,SAAS,WAAW;AACpB,SAAS,WAAW;AACpB,SAAS,gBAAgB;AACzB,SAAS,WAAW;AAEpB,SAAS,YAAY,YAAY;AAC/B,QAAM,IAAI,IAAI,IAAI,GAAG,IAAI,IAAI,CAAC;AAChC,CAAC;","names":[]}

package/package.json

@@ -0,0 +1,54 @@
+{
+  "name": "@livekit/agents-plugin-rime",
+  "version": "1.0.1",
+  "description": "Rime plugin for LiveKit Node Agents",
+  "main": "dist/index.js",
+  "require": "dist/index.cjs",
+  "types": "dist/index.d.ts",
+  "exports": {
+    "import": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    },
+    "require": {
+      "types": "./dist/index.d.cts",
+      "default": "./dist/index.cjs"
+    }
+  },
+  "author": "LiveKit",
+  "type": "module",
+  "repository": "git@github.com:livekit/agents-js.git",
+  "license": "Apache-2.0",
+  "files": [
+    "dist",
+    "src",
+    "README.md"
+  ],
+  "devDependencies": {
+    "@livekit/rtc-node": "^0.13.12",
+    "@microsoft/api-extractor": "^7.35.0",
+    "@types/ws": "^8.5.10",
+    "tsup": "^8.3.5",
+    "typescript": "^5.0.0",
+    "@livekit/agents": "1.0.8",
+    "@livekit/agents-plugin-openai": "1.0.8",
+    "@livekit/agents-plugin-silero": "1.0.8",
+    "@livekit/agents-plugins-test": "1.0.8"
+  },
+  "dependencies": {
+    "ws": "^8.16.0"
+  },
+  "peerDependencies": {
+    "@livekit/rtc-node": "^0.13.12",
+    "@livekit/agents": "1.0.8"
+  },
+  "scripts": {
+    "build": "tsup --onSuccess \"pnpm build:types\"",
+    "build:types": "tsc --declaration --emitDeclarationOnly && node ../../scripts/copyDeclarationOutput.js",
+    "clean": "rm -rf dist",
+    "clean:build": "pnpm clean && pnpm build",
+    "lint": "eslint -f unix \"src/**/*.{ts,js}\"",
+    "api:check": "api-extractor run --typescript-compiler-folder ../../node_modules/typescript",
+    "api:update": "api-extractor run --local --typescript-compiler-folder ../../node_modules/typescript --verbose"
+  }
+}

package/src/index.ts

@@ -0,0 +1,19 @@
+// SPDX-FileCopyrightText: 2025 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+import { Plugin } from '@livekit/agents';
+
+export * from './models.js';
+export * from './tts.js';
+
+class RimePlugin extends Plugin {
+  constructor() {
+    super({
+      title: 'rime',
+      version: '0.1.0',
+      package: '@livekit/agents-plugin-rime',
+    });
+  }
+}
+
+Plugin.registerPlugin(new RimePlugin());

package/src/models.ts

@@ -0,0 +1,9 @@
+// SPDX-FileCopyrightText: 2024 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+
+/** Supported Rime AI TTS models */
+export type TTSModels = 'arcana' | 'mistv2';
+
+/** Supported default languages for Rime AI TTS */
+export type DefaultLanguages = 'eng' | 'spa' | 'fra' | 'ger';

package/src/tts.test.ts

@@ -0,0 +1,11 @@
+// SPDX-FileCopyrightText: 2024 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+import { STT } from '@livekit/agents-plugin-openai';
+import { tts } from '@livekit/agents-plugins-test';
+import { describe } from 'vitest';
+import { TTS } from './tts.js';
+
+describe('Rime TTS', async () => {
+  await tts(new TTS(), new STT());
+});

package/src/tts.ts

@@ -0,0 +1,172 @@
+// SPDX-FileCopyrightText: 2024 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+import { AudioByteStream, shortuuid, tts } from '@livekit/agents';
+import type { AudioFrame } from '@livekit/rtc-node';
+import type { DefaultLanguages, TTSModels } from './models.js';
+
+const RIME_BASE_URL = 'https://users.rime.ai/v1/rime-tts';
+const RIME_TTS_SAMPLE_RATE = 24000;
+const RIME_TTS_CHANNELS = 1;
+
+/**
+ * Get the appropriate sample rate based on TTS options.
+ *
+ * @param opts - Optional TTS configuration options
+ * @returns The sample rate in Hz. Returns the explicit samplingRate if provided,
+ *          otherwise returns model-specific defaults (24000 for arcana, 16000 for mistv2,
+ *          or the default RIME_TTS_SAMPLE_RATE for other models)
+ */
+function getSampleRate(opts?: Partial<TTSOptions>): number {
+  if (opts?.samplingRate && typeof opts.samplingRate === 'number') {
+    return opts.samplingRate;
+  }
+  switch (opts?.modelId) {
+    case 'arcana':
+      return 24000;
+    case 'mistv2':
+      return 16000;
+    default:
+      return RIME_TTS_SAMPLE_RATE;
+  }
+}
+
+/** Configuration options for Rime AI TTS */
+export interface TTSOptions {
+  speaker: string;
+  modelId: TTSModels | string;
+  baseURL?: string;
+  apiKey?: string;
+  lang?: DefaultLanguages | string;
+  repetition_penalty?: number;
+  temperature?: number;
+  top_p?: number;
+  max_tokens?: number;
+  samplingRate?: number;
+  speedAlpha?: number;
+  pauseBetweenBrackets?: boolean;
+  phonemizeBetweenBrackets?: boolean;
+  inlineSpeedAlpha?: string;
+  noTextNormalization?: boolean;
+  saveOovs?: boolean;
+  /** Additional Rime API parameters */
+  [key: string]: string | number | boolean | undefined;
+}
+
+const defaultTTSOptions: TTSOptions = {
+  modelId: 'arcana',
+  speaker: 'luna',
+  apiKey: process.env.RIME_API_KEY,
+  baseURL: RIME_BASE_URL,
+};
+
+export class TTS extends tts.TTS {
+  private opts: TTSOptions;
+  label = 'rime.TTS';
+
+  /**
+   * Create a new instance of Rime TTS.
+   *
+   * @remarks
+   * `apiKey` must be set to your Rime AI API key, either using the argument or by setting the
+   * `RIME_API_KEY` environmental variable.
+   *
+   * @param opts - Configuration options for the TTS instance
+   */
+
+  constructor(opts: Partial<TTSOptions> = defaultTTSOptions) {
+    const sampleRate = getSampleRate(opts);
+    super(sampleRate, RIME_TTS_CHANNELS, {
+      streaming: false,
+    });
+
+    this.opts = { ...defaultTTSOptions, ...opts };
+    if (this.opts.apiKey === undefined) {
+      throw new Error('RIME API key is required, whether as an argument or as $RIME_API_KEY');
+    }
+  }
+
+  /**
+   * Update TTS options after initialization
+   *
+   * @param opts - Partial options to update
+   */
+  updateOptions(opts: Partial<TTSOptions>) {
+    this.opts = { ...this.opts, ...opts };
+  }
+
+  /**
+   * Synthesize text to audio using Rime AI TTS.
+   *
+   * @param text - Text to synthesize
+   * @returns A chunked stream of synthesized audio
+   */
+  synthesize(text: string): ChunkedStream {
+    return new ChunkedStream(this, text, this.opts);
+  }
+
+  stream(): tts.SynthesizeStream {
+    throw new Error('Streaming is not supported on RimeTTS');
+  }
+}
+
+export class ChunkedStream extends tts.ChunkedStream {
+  label = 'rime-tts.ChunkedStream';
+  private opts: TTSOptions;
+  private text: string;
+
+  /**
+   * Create a new ChunkedStream instance.
+   *
+   * @param tts - The parent TTS instance
+   * @param text - Text to synthesize
+   * @param opts - TTS configuration options
+   */
+  constructor(tts: TTS, text: string, opts: TTSOptions) {
+    super(text, tts);
+    this.text = text;
+    this.opts = opts;
+  }
+
+  protected async run() {
+    const requestId = shortuuid();
+    const response = await fetch(`${this.opts.baseURL}`, {
+      method: 'POST',
+      headers: {
+        Accept: 'audio/pcm',
+        Authorization: `Bearer ${this.opts.apiKey}`,
+        'Content-Type': 'application/json',
+      },
+      body: JSON.stringify({
+        ...Object.fromEntries(
+          Object.entries(this.opts).filter(([k]) => !['apiKey', 'baseURL'].includes(k)),
+        ),
+        text: this.text,
+      }),
+    });
+
+    if (!response.ok) {
+      throw new Error(`Rime AI TTS request failed: ${response.status} ${response.statusText}`);
+    }
+
+    const buffer = await response.arrayBuffer();
+    const sampleRate = getSampleRate(this.opts);
+    const audioByteStream = new AudioByteStream(sampleRate, RIME_TTS_CHANNELS);
+    const frames = audioByteStream.write(buffer);
+    let lastFrame: AudioFrame | undefined;
+    const sendLastFrame = (segmentId: string, final: boolean) => {
+      if (lastFrame) {
+        this.queue.put({ requestId, segmentId, frame: lastFrame, final });
+        lastFrame = undefined;
+      }
+    };
+
+    for (const frame of frames) {
+      sendLastFrame(requestId, false);
+      lastFrame = frame;
+    }
+    sendLastFrame(requestId, true);
+
+    this.queue.close();
+  }
+}