@livekit/agents 1.0.36-dev.0 → 1.0.37

package/dist/utils/ws_transport.test.cjs

@@ -1,212 +0,0 @@
-"use strict";
-var import_vitest = require("vitest");
-var import_ws = require("ws");
-var import_ws_transport = require("./ws_transport.cjs");
-(0, import_vitest.describe)("webSocketStream", () => {
-  (0, import_vitest.describe)("readable stream", () => {
-    (0, import_vitest.it)("receives messages from the WebSocket", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      wss.on("connection", (serverWs) => {
-        serverWs.send("hello");
-        serverWs.send("world");
-        serverWs.close();
-      });
-      const { readable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
-      const reader = readable.getReader();
-      const messages = [];
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) break;
-          messages.push(Buffer.from(value).toString());
-        }
-      } finally {
-        reader.releaseLock();
-      }
-      (0, import_vitest.expect)(messages).toEqual(["hello", "world"]);
-      wss.close();
-    });
-    (0, import_vitest.it)("handles binary messages", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      const binaryData = new Uint8Array([1, 2, 3, 4, 5]);
-      wss.on("connection", (serverWs) => {
-        serverWs.send(binaryData);
-        serverWs.close();
-      });
-      const { readable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
-      const reader = readable.getReader();
-      const chunks = [];
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) break;
-          chunks.push(new Uint8Array(value));
-        }
-      } finally {
-        reader.releaseLock();
-      }
-      (0, import_vitest.expect)(chunks).toHaveLength(1);
-      (0, import_vitest.expect)(Array.from(chunks[0])).toEqual([1, 2, 3, 4, 5]);
-      wss.close();
-    });
-    (0, import_vitest.it)("handles empty stream when connection closes immediately", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      wss.on("connection", (serverWs) => {
-        serverWs.close();
-      });
-      const { readable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
-      const reader = readable.getReader();
-      const chunks = [];
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) break;
-          chunks.push(value);
-        }
-      } finally {
-        reader.releaseLock();
-      }
-      (0, import_vitest.expect)(chunks).toEqual([]);
-      wss.close();
-    });
-  });
-  (0, import_vitest.describe)("writable stream", () => {
-    (0, import_vitest.it)("sends messages through the WebSocket", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      const ws = new import_ws.WebSocket(`ws://localhost:${port}`);
-      const connected = new Promise((resolve) => {
-        ws.on("open", resolve);
-      });
-      const messagesReceived = [];
-      const serverClosed = new Promise((resolve) => {
-        wss.on("connection", (serverWs) => {
-          serverWs.on("message", (data) => {
-            messagesReceived.push(data.toString());
-          });
-          serverWs.on("close", resolve);
-        });
-      });
-      await connected;
-      const { writable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
-      const writer = writable.getWriter();
-      await writer.write(new TextEncoder().encode("hello"));
-      await writer.write(new TextEncoder().encode("world"));
-      await writer.close();
-      await serverClosed;
-      (0, import_vitest.expect)(messagesReceived).toEqual(["hello", "world"]);
-      wss.close();
-    });
-    (0, import_vitest.it)("sends binary data through the WebSocket", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      const chunksReceived = [];
-      const serverClosed = new Promise((resolve) => {
-        wss.on("connection", (serverWs) => {
-          serverWs.on("message", (data) => {
-            chunksReceived.push(Buffer.from(data));
-          });
-          serverWs.on("close", resolve);
-        });
-      });
-      const { writable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
-      const writer = writable.getWriter();
-      const binaryData = new Uint8Array([10, 20, 30, 40, 50]);
-      await writer.write(binaryData);
-      await writer.close();
-      await serverClosed;
-      (0, import_vitest.expect)(chunksReceived).toHaveLength(1);
-      (0, import_vitest.expect)(Array.from(chunksReceived[0])).toEqual([10, 20, 30, 40, 50]);
-      wss.close();
-    });
-    (0, import_vitest.it)("buffers writes if readyState is CONNECTING", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      const { writable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
-      const writer = writable.getWriter();
-      const messagesReceived = [];
-      const serverClosed = new Promise((resolve) => {
-        wss.on("connection", (serverWs) => {
-          serverWs.on("message", (data) => {
-            messagesReceived.push(data.toString());
-          });
-          serverWs.on("close", resolve);
-        });
-      });
-      await writer.write(new TextEncoder().encode("buffered message"));
-      await writer.close();
-      await serverClosed;
-      (0, import_vitest.expect)(messagesReceived).toEqual(["buffered message"]);
-      wss.close();
-    });
-  });
-  (0, import_vitest.describe)("bidirectional communication", () => {
-    (0, import_vitest.it)("supports echo pattern with readable and writable", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      wss.on("connection", (serverWs) => {
-        serverWs.on("message", (data) => {
-          serverWs.send(data);
-        });
-      });
-      const { readable, writable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
-      const writer = writable.getWriter();
-      const reader = readable.getReader();
-      await writer.write(new TextEncoder().encode("ping1"));
-      await writer.write(new TextEncoder().encode("ping2"));
-      const { value: response1 } = await reader.read();
-      const { value: response2 } = await reader.read();
-      (0, import_vitest.expect)(Buffer.from(response1).toString()).toBe("ping1");
-      (0, import_vitest.expect)(Buffer.from(response2).toString()).toBe("ping2");
-      reader.releaseLock();
-      await writer.close();
-      wss.close();
-    });
-  });
-  (0, import_vitest.describe)("error handling", () => {
-    (0, import_vitest.it)("readable stream ends when WebSocket closes unexpectedly", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new import_ws.WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      wss.on("connection", (serverWs) => {
-        serverWs.send("before close");
-        serverWs.terminate();
-      });
-      const { readable } = (0, import_ws_transport.webSocketStream)(`ws://localhost:${port}`);
-      const reader = readable.getReader();
-      const chunks = [];
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) break;
-          chunks.push(Buffer.from(value).toString());
-        }
-      } catch (error) {
-        console.error(error);
-      } finally {
-        reader.releaseLock();
-      }
-      (0, import_vitest.expect)(chunks).toContain("before close");
-      wss.close();
-    });
-  });
-});
-//# sourceMappingURL=ws_transport.test.cjs.map

package/dist/utils/ws_transport.test.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/utils/ws_transport.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { describe, expect, it } from 'vitest';\nimport { WebSocket, WebSocketServer } from 'ws';\nimport { webSocketStream } from './ws_transport.js';\n\ndescribe('webSocketStream', () => {\n  describe('readable stream', () => {\n    it('receives messages from the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send('hello');\n        serverWs.send('world');\n        serverWs.close();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const messages: string[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          messages.push(Buffer.from(value).toString());\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(messages).toEqual(['hello', 'world']);\n\n      wss.close();\n    });\n\n    it('handles binary messages', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const binaryData = new Uint8Array([1, 2, 3, 4, 5]);\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send(binaryData);\n        serverWs.close();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: Uint8Array[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(new Uint8Array(value));\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(chunks).toHaveLength(1);\n      expect(Array.from(chunks[0]!)).toEqual([1, 2, 3, 4, 5]);\n\n      wss.close();\n    });\n\n    it('handles empty stream when connection closes immediately', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.close();\n      });\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: Uint8Array[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(value);\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(chunks).toEqual([]);\n\n      wss.close();\n    });\n  });\n\n  describe('writable stream', () => {\n    it('sends messages through the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n      const ws = new WebSocket(`ws://localhost:${port}`);\n\n      const connected = new Promise<void>((resolve) => {\n        ws.on('open', resolve);\n      });\n\n      const messagesReceived: string[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            messagesReceived.push(data.toString());\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      await connected;\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      await writer.write(new TextEncoder().encode('hello'));\n      await writer.write(new TextEncoder().encode('world'));\n      await writer.close();\n\n      await serverClosed;\n\n      expect(messagesReceived).toEqual(['hello', 'world']);\n\n      wss.close();\n    });\n\n    it('sends binary data through the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const chunksReceived: Buffer[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            chunksReceived.push(Buffer.from(data as Buffer));\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      const binaryData = new Uint8Array([10, 20, 30, 40, 50]);\n      await writer.write(binaryData);\n      await writer.close();\n\n      await serverClosed;\n\n      expect(chunksReceived).toHaveLength(1);\n      expect(Array.from(chunksReceived[0]!)).toEqual([10, 20, 30, 40, 50]);\n\n      wss.close();\n    });\n\n    it('buffers writes if readyState is CONNECTING', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      const messagesReceived: string[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            messagesReceived.push(data.toString());\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      // These writes should be buffered\n      await writer.write(new TextEncoder().encode('buffered message'));\n      await writer.close();\n\n      await serverClosed;\n\n      expect(messagesReceived).toEqual(['buffered message']);\n\n      wss.close();\n    });\n  });\n\n  describe('bidirectional communication', () => {\n    it('supports echo pattern with readable and writable', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      // Server echoes messages back\n      wss.on('connection', (serverWs) => {\n        serverWs.on('message', (data) => {\n          serverWs.send(data);\n        });\n      });\n\n      const { readable, writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n      const reader = readable.getReader();\n\n      // Send messages\n      await writer.write(new TextEncoder().encode('ping1'));\n      await writer.write(new TextEncoder().encode('ping2'));\n\n      // Read echoed responses\n      const { value: response1 } = await reader.read();\n      const { value: response2 } = await reader.read();\n\n      expect(Buffer.from(response1!).toString()).toBe('ping1');\n      expect(Buffer.from(response2!).toString()).toBe('ping2');\n\n      reader.releaseLock();\n      await writer.close();\n\n      wss.close();\n    });\n  });\n\n  describe('error handling', () => {\n    it('readable stream ends when WebSocket closes unexpectedly', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send('before close');\n        // Terminate connection abruptly\n        serverWs.terminate();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: string[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(Buffer.from(value).toString());\n        }\n      } catch (error) {\n        console.error(error);\n        // Connection terminated, stream may error\n      } finally {\n        reader.releaseLock();\n      }\n\n      // Should have received the message sent before termination\n      expect(chunks).toContain('before close');\n\n      wss.close();\n    });\n  });\n});\n"],"mappings":";AAGA,oBAAqC;AACrC,gBAA2C;AAC3C,0BAAgC;AAAA,IAEhC,wBAAS,mBAAmB,MAAM;AAChC,8BAAS,mBAAmB,MAAM;AAChC,0BAAG,wCAAwC,YAAY;AACrD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,OAAO;AACrB,iBAAS,KAAK,OAAO;AACrB,iBAAS,MAAM;AAAA,MACjB,CAAC;AAED,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,WAAqB,CAAC;AAC5B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,mBAAS,KAAK,OAAO,KAAK,KAAK,EAAE,SAAS,CAAC;AAAA,QAC7C;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,gCAAO,QAAQ,EAAE,QAAQ,CAAC,SAAS,OAAO,CAAC;AAE3C,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,0BAAG,2BAA2B,YAAY;AACxC,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,aAAa,IAAI,WAAW,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,UAAU;AACxB,iBAAS,MAAM;AAAA,MACjB,CAAC;AAED,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAuB,CAAC;AAC9B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,IAAI,WAAW,KAAK,CAAC;AAAA,QACnC;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,gCAAO,MAAM,EAAE,aAAa,CAAC;AAC7B,gCAAO,MAAM,KAAK,OAAO,CAAC,CAAE,CAAC,EAAE,QAAQ,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC;AAEtD,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,0BAAG,2DAA2D,YAAY;AACxE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,MAAM;AAAA,MACjB,CAAC;AACD,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAuB,CAAC;AAC9B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,KAAK;AAAA,QACnB;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,gCAAO,MAAM,EAAE,QAAQ,CAAC,CAAC;AAEzB,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,8BAAS,mBAAmB,MAAM;AAChC,0BAAG,wCAAwC,YAAY;AACrD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AACjD,YAAM,KAAK,IAAI,oBAAU,kBAAkB,IAAI,EAAE;AAEjD,YAAM,YAAY,IAAI,QAAc,CAAC,YAAY;AAC/C,WAAG,GAAG,QAAQ,OAAO;AAAA,MACvB,CAAC;AAED,YAAM,mBAA6B,CAAC;AACpC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,6BAAiB,KAAK,KAAK,SAAS,CAAC;AAAA,UACvC,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAED,YAAM;AACN,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,gCAAO,gBAAgB,EAAE,QAAQ,CAAC,SAAS,OAAO,CAAC;AAEnD,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,0BAAG,2CAA2C,YAAY;AACxD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,iBAA2B,CAAC;AAClC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,2BAAe,KAAK,OAAO,KAAK,IAAc,CAAC;AAAA,UACjD,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAED,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,aAAa,IAAI,WAAW,CAAC,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;AACtD,YAAM,OAAO,MAAM,UAAU;AAC7B,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,gCAAO,cAAc,EAAE,aAAa,CAAC;AACrC,gCAAO,MAAM,KAAK,eAAe,CAAC,CAAE,CAAC,EAAE,QAAQ,CAAC,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;AAEnE,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,0BAAG,8CAA8C,YAAY;AAC3D,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,mBAA6B,CAAC;AACpC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,6BAAiB,KAAK,KAAK,SAAS,CAAC;AAAA,UACvC,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAGD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,kBAAkB,CAAC;AAC/D,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,gCAAO,gBAAgB,EAAE,QAAQ,CAAC,kBAAkB,CAAC;AAErD,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,8BAAS,+BAA+B,MAAM;AAC5C,0BAAG,oDAAoD,YAAY;AACjE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAGjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,mBAAS,KAAK,IAAI;AAAA,QACpB,CAAC;AAAA,MACH,CAAC;AAED,YAAM,EAAE,UAAU,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AACvE,YAAM,SAAS,SAAS,UAAU;AAClC,YAAM,SAAS,SAAS,UAAU;AAGlC,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AAGpD,YAAM,EAAE,OAAO,UAAU,IAAI,MAAM,OAAO,KAAK;AAC/C,YAAM,EAAE,OAAO,UAAU,IAAI,MAAM,OAAO,KAAK;AAE/C,gCAAO,OAAO,KAAK,SAAU,EAAE,SAAS,CAAC,EAAE,KAAK,OAAO;AACvD,gCAAO,OAAO,KAAK,SAAU,EAAE,SAAS,CAAC,EAAE,KAAK,OAAO;AAEvD,aAAO,YAAY;AACnB,YAAM,OAAO,MAAM;AAEnB,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,8BAAS,kBAAkB,MAAM;AAC/B,0BAAG,2DAA2D,YAAY;AACxE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,0BAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,cAAc;AAE5B,iBAAS,UAAU;AAAA,MACrB,CAAC;AAED,YAAM,EAAE,SAAS,QAAI,qCAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAmB,CAAC;AAC1B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,OAAO,KAAK,KAAK,EAAE,SAAS,CAAC;AAAA,QAC3C;AAAA,MACF,SAAS,OAAO;AACd,gBAAQ,MAAM,KAAK;AAAA,MAErB,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAGA,gCAAO,MAAM,EAAE,UAAU,cAAc;AAEvC,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AACH,CAAC;","names":[]}

package/dist/utils/ws_transport.test.js

@@ -1,211 +0,0 @@
-import { describe, expect, it } from "vitest";
-import { WebSocket, WebSocketServer } from "ws";
-import { webSocketStream } from "./ws_transport.js";
-describe("webSocketStream", () => {
-  describe("readable stream", () => {
-    it("receives messages from the WebSocket", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      wss.on("connection", (serverWs) => {
-        serverWs.send("hello");
-        serverWs.send("world");
-        serverWs.close();
-      });
-      const { readable } = webSocketStream(`ws://localhost:${port}`);
-      const reader = readable.getReader();
-      const messages = [];
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) break;
-          messages.push(Buffer.from(value).toString());
-        }
-      } finally {
-        reader.releaseLock();
-      }
-      expect(messages).toEqual(["hello", "world"]);
-      wss.close();
-    });
-    it("handles binary messages", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      const binaryData = new Uint8Array([1, 2, 3, 4, 5]);
-      wss.on("connection", (serverWs) => {
-        serverWs.send(binaryData);
-        serverWs.close();
-      });
-      const { readable } = webSocketStream(`ws://localhost:${port}`);
-      const reader = readable.getReader();
-      const chunks = [];
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) break;
-          chunks.push(new Uint8Array(value));
-        }
-      } finally {
-        reader.releaseLock();
-      }
-      expect(chunks).toHaveLength(1);
-      expect(Array.from(chunks[0])).toEqual([1, 2, 3, 4, 5]);
-      wss.close();
-    });
-    it("handles empty stream when connection closes immediately", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      wss.on("connection", (serverWs) => {
-        serverWs.close();
-      });
-      const { readable } = webSocketStream(`ws://localhost:${port}`);
-      const reader = readable.getReader();
-      const chunks = [];
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) break;
-          chunks.push(value);
-        }
-      } finally {
-        reader.releaseLock();
-      }
-      expect(chunks).toEqual([]);
-      wss.close();
-    });
-  });
-  describe("writable stream", () => {
-    it("sends messages through the WebSocket", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      const ws = new WebSocket(`ws://localhost:${port}`);
-      const connected = new Promise((resolve) => {
-        ws.on("open", resolve);
-      });
-      const messagesReceived = [];
-      const serverClosed = new Promise((resolve) => {
-        wss.on("connection", (serverWs) => {
-          serverWs.on("message", (data) => {
-            messagesReceived.push(data.toString());
-          });
-          serverWs.on("close", resolve);
-        });
-      });
-      await connected;
-      const { writable } = webSocketStream(`ws://localhost:${port}`);
-      const writer = writable.getWriter();
-      await writer.write(new TextEncoder().encode("hello"));
-      await writer.write(new TextEncoder().encode("world"));
-      await writer.close();
-      await serverClosed;
-      expect(messagesReceived).toEqual(["hello", "world"]);
-      wss.close();
-    });
-    it("sends binary data through the WebSocket", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      const chunksReceived = [];
-      const serverClosed = new Promise((resolve) => {
-        wss.on("connection", (serverWs) => {
-          serverWs.on("message", (data) => {
-            chunksReceived.push(Buffer.from(data));
-          });
-          serverWs.on("close", resolve);
-        });
-      });
-      const { writable } = webSocketStream(`ws://localhost:${port}`);
-      const writer = writable.getWriter();
-      const binaryData = new Uint8Array([10, 20, 30, 40, 50]);
-      await writer.write(binaryData);
-      await writer.close();
-      await serverClosed;
-      expect(chunksReceived).toHaveLength(1);
-      expect(Array.from(chunksReceived[0])).toEqual([10, 20, 30, 40, 50]);
-      wss.close();
-    });
-    it("buffers writes if readyState is CONNECTING", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      const { writable } = webSocketStream(`ws://localhost:${port}`);
-      const writer = writable.getWriter();
-      const messagesReceived = [];
-      const serverClosed = new Promise((resolve) => {
-        wss.on("connection", (serverWs) => {
-          serverWs.on("message", (data) => {
-            messagesReceived.push(data.toString());
-          });
-          serverWs.on("close", resolve);
-        });
-      });
-      await writer.write(new TextEncoder().encode("buffered message"));
-      await writer.close();
-      await serverClosed;
-      expect(messagesReceived).toEqual(["buffered message"]);
-      wss.close();
-    });
-  });
-  describe("bidirectional communication", () => {
-    it("supports echo pattern with readable and writable", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      wss.on("connection", (serverWs) => {
-        serverWs.on("message", (data) => {
-          serverWs.send(data);
-        });
-      });
-      const { readable, writable } = webSocketStream(`ws://localhost:${port}`);
-      const writer = writable.getWriter();
-      const reader = readable.getReader();
-      await writer.write(new TextEncoder().encode("ping1"));
-      await writer.write(new TextEncoder().encode("ping2"));
-      const { value: response1 } = await reader.read();
-      const { value: response2 } = await reader.read();
-      expect(Buffer.from(response1).toString()).toBe("ping1");
-      expect(Buffer.from(response2).toString()).toBe("ping2");
-      reader.releaseLock();
-      await writer.close();
-      wss.close();
-    });
-  });
-  describe("error handling", () => {
-    it("readable stream ends when WebSocket closes unexpectedly", async () => {
-      const wss = await new Promise((resolve) => {
-        const server = new WebSocketServer({ port: 0 }, () => resolve(server));
-      });
-      const port = wss.address().port;
-      wss.on("connection", (serverWs) => {
-        serverWs.send("before close");
-        serverWs.terminate();
-      });
-      const { readable } = webSocketStream(`ws://localhost:${port}`);
-      const reader = readable.getReader();
-      const chunks = [];
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) break;
-          chunks.push(Buffer.from(value).toString());
-        }
-      } catch (error) {
-        console.error(error);
-      } finally {
-        reader.releaseLock();
-      }
-      expect(chunks).toContain("before close");
-      wss.close();
-    });
-  });
-});
-//# sourceMappingURL=ws_transport.test.js.map

package/dist/utils/ws_transport.test.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/utils/ws_transport.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { describe, expect, it } from 'vitest';\nimport { WebSocket, WebSocketServer } from 'ws';\nimport { webSocketStream } from './ws_transport.js';\n\ndescribe('webSocketStream', () => {\n  describe('readable stream', () => {\n    it('receives messages from the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send('hello');\n        serverWs.send('world');\n        serverWs.close();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const messages: string[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          messages.push(Buffer.from(value).toString());\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(messages).toEqual(['hello', 'world']);\n\n      wss.close();\n    });\n\n    it('handles binary messages', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const binaryData = new Uint8Array([1, 2, 3, 4, 5]);\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send(binaryData);\n        serverWs.close();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: Uint8Array[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(new Uint8Array(value));\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(chunks).toHaveLength(1);\n      expect(Array.from(chunks[0]!)).toEqual([1, 2, 3, 4, 5]);\n\n      wss.close();\n    });\n\n    it('handles empty stream when connection closes immediately', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.close();\n      });\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: Uint8Array[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(value);\n        }\n      } finally {\n        reader.releaseLock();\n      }\n\n      expect(chunks).toEqual([]);\n\n      wss.close();\n    });\n  });\n\n  describe('writable stream', () => {\n    it('sends messages through the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n      const ws = new WebSocket(`ws://localhost:${port}`);\n\n      const connected = new Promise<void>((resolve) => {\n        ws.on('open', resolve);\n      });\n\n      const messagesReceived: string[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            messagesReceived.push(data.toString());\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      await connected;\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      await writer.write(new TextEncoder().encode('hello'));\n      await writer.write(new TextEncoder().encode('world'));\n      await writer.close();\n\n      await serverClosed;\n\n      expect(messagesReceived).toEqual(['hello', 'world']);\n\n      wss.close();\n    });\n\n    it('sends binary data through the WebSocket', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const chunksReceived: Buffer[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            chunksReceived.push(Buffer.from(data as Buffer));\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      const binaryData = new Uint8Array([10, 20, 30, 40, 50]);\n      await writer.write(binaryData);\n      await writer.close();\n\n      await serverClosed;\n\n      expect(chunksReceived).toHaveLength(1);\n      expect(Array.from(chunksReceived[0]!)).toEqual([10, 20, 30, 40, 50]);\n\n      wss.close();\n    });\n\n    it('buffers writes if readyState is CONNECTING', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      const { writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n\n      const messagesReceived: string[] = [];\n      const serverClosed = new Promise<void>((resolve) => {\n        wss.on('connection', (serverWs) => {\n          serverWs.on('message', (data) => {\n            messagesReceived.push(data.toString());\n          });\n          serverWs.on('close', resolve);\n        });\n      });\n\n      // These writes should be buffered\n      await writer.write(new TextEncoder().encode('buffered message'));\n      await writer.close();\n\n      await serverClosed;\n\n      expect(messagesReceived).toEqual(['buffered message']);\n\n      wss.close();\n    });\n  });\n\n  describe('bidirectional communication', () => {\n    it('supports echo pattern with readable and writable', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      // Server echoes messages back\n      wss.on('connection', (serverWs) => {\n        serverWs.on('message', (data) => {\n          serverWs.send(data);\n        });\n      });\n\n      const { readable, writable } = webSocketStream(`ws://localhost:${port}`);\n      const writer = writable.getWriter();\n      const reader = readable.getReader();\n\n      // Send messages\n      await writer.write(new TextEncoder().encode('ping1'));\n      await writer.write(new TextEncoder().encode('ping2'));\n\n      // Read echoed responses\n      const { value: response1 } = await reader.read();\n      const { value: response2 } = await reader.read();\n\n      expect(Buffer.from(response1!).toString()).toBe('ping1');\n      expect(Buffer.from(response2!).toString()).toBe('ping2');\n\n      reader.releaseLock();\n      await writer.close();\n\n      wss.close();\n    });\n  });\n\n  describe('error handling', () => {\n    it('readable stream ends when WebSocket closes unexpectedly', async () => {\n      const wss = await new Promise<WebSocketServer>((resolve) => {\n        const server: WebSocketServer = new WebSocketServer({ port: 0 }, () => resolve(server));\n      });\n\n      const port = (wss.address() as { port: number }).port;\n\n      wss.on('connection', (serverWs) => {\n        serverWs.send('before close');\n        // Terminate connection abruptly\n        serverWs.terminate();\n      });\n\n      const { readable } = webSocketStream(`ws://localhost:${port}`);\n      const reader = readable.getReader();\n\n      const chunks: string[] = [];\n      try {\n        while (true) {\n          const { done, value } = await reader.read();\n          if (done) break;\n          chunks.push(Buffer.from(value).toString());\n        }\n      } catch (error) {\n        console.error(error);\n        // Connection terminated, stream may error\n      } finally {\n        reader.releaseLock();\n      }\n\n      // Should have received the message sent before termination\n      expect(chunks).toContain('before close');\n\n      wss.close();\n    });\n  });\n});\n"],"mappings":"AAGA,SAAS,UAAU,QAAQ,UAAU;AACrC,SAAS,WAAW,uBAAuB;AAC3C,SAAS,uBAAuB;AAEhC,SAAS,mBAAmB,MAAM;AAChC,WAAS,mBAAmB,MAAM;AAChC,OAAG,wCAAwC,YAAY;AACrD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,OAAO;AACrB,iBAAS,KAAK,OAAO;AACrB,iBAAS,MAAM;AAAA,MACjB,CAAC;AAED,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,WAAqB,CAAC;AAC5B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,mBAAS,KAAK,OAAO,KAAK,KAAK,EAAE,SAAS,CAAC;AAAA,QAC7C;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,aAAO,QAAQ,EAAE,QAAQ,CAAC,SAAS,OAAO,CAAC;AAE3C,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,OAAG,2BAA2B,YAAY;AACxC,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,aAAa,IAAI,WAAW,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,UAAU;AACxB,iBAAS,MAAM;AAAA,MACjB,CAAC;AAED,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAuB,CAAC;AAC9B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,IAAI,WAAW,KAAK,CAAC;AAAA,QACnC;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,aAAO,MAAM,EAAE,aAAa,CAAC;AAC7B,aAAO,MAAM,KAAK,OAAO,CAAC,CAAE,CAAC,EAAE,QAAQ,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC;AAEtD,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,OAAG,2DAA2D,YAAY;AACxE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,MAAM;AAAA,MACjB,CAAC;AACD,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAuB,CAAC;AAC9B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,KAAK;AAAA,QACnB;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,aAAO,MAAM,EAAE,QAAQ,CAAC,CAAC;AAEzB,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,WAAS,mBAAmB,MAAM;AAChC,OAAG,wCAAwC,YAAY;AACrD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AACjD,YAAM,KAAK,IAAI,UAAU,kBAAkB,IAAI,EAAE;AAEjD,YAAM,YAAY,IAAI,QAAc,CAAC,YAAY;AAC/C,WAAG,GAAG,QAAQ,OAAO;AAAA,MACvB,CAAC;AAED,YAAM,mBAA6B,CAAC;AACpC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,6BAAiB,KAAK,KAAK,SAAS,CAAC;AAAA,UACvC,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAED,YAAM;AACN,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,aAAO,gBAAgB,EAAE,QAAQ,CAAC,SAAS,OAAO,CAAC;AAEnD,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,OAAG,2CAA2C,YAAY;AACxD,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,iBAA2B,CAAC;AAClC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,2BAAe,KAAK,OAAO,KAAK,IAAc,CAAC;AAAA,UACjD,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAED,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,aAAa,IAAI,WAAW,CAAC,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;AACtD,YAAM,OAAO,MAAM,UAAU;AAC7B,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,aAAO,cAAc,EAAE,aAAa,CAAC;AACrC,aAAO,MAAM,KAAK,eAAe,CAAC,CAAE,CAAC,EAAE,QAAQ,CAAC,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;AAEnE,UAAI,MAAM;AAAA,IACZ,CAAC;AAED,OAAG,8CAA8C,YAAY;AAC3D,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,mBAA6B,CAAC;AACpC,YAAM,eAAe,IAAI,QAAc,CAAC,YAAY;AAClD,YAAI,GAAG,cAAc,CAAC,aAAa;AACjC,mBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,6BAAiB,KAAK,KAAK,SAAS,CAAC;AAAA,UACvC,CAAC;AACD,mBAAS,GAAG,SAAS,OAAO;AAAA,QAC9B,CAAC;AAAA,MACH,CAAC;AAGD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,kBAAkB,CAAC;AAC/D,YAAM,OAAO,MAAM;AAEnB,YAAM;AAEN,aAAO,gBAAgB,EAAE,QAAQ,CAAC,kBAAkB,CAAC;AAErD,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,WAAS,+BAA+B,MAAM;AAC5C,OAAG,oDAAoD,YAAY;AACjE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAGjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,GAAG,WAAW,CAAC,SAAS;AAC/B,mBAAS,KAAK,IAAI;AAAA,QACpB,CAAC;AAAA,MACH,CAAC;AAED,YAAM,EAAE,UAAU,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AACvE,YAAM,SAAS,SAAS,UAAU;AAClC,YAAM,SAAS,SAAS,UAAU;AAGlC,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AACpD,YAAM,OAAO,MAAM,IAAI,YAAY,EAAE,OAAO,OAAO,CAAC;AAGpD,YAAM,EAAE,OAAO,UAAU,IAAI,MAAM,OAAO,KAAK;AAC/C,YAAM,EAAE,OAAO,UAAU,IAAI,MAAM,OAAO,KAAK;AAE/C,aAAO,OAAO,KAAK,SAAU,EAAE,SAAS,CAAC,EAAE,KAAK,OAAO;AACvD,aAAO,OAAO,KAAK,SAAU,EAAE,SAAS,CAAC,EAAE,KAAK,OAAO;AAEvD,aAAO,YAAY;AACnB,YAAM,OAAO,MAAM;AAEnB,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AAED,WAAS,kBAAkB,MAAM;AAC/B,OAAG,2DAA2D,YAAY;AACxE,YAAM,MAAM,MAAM,IAAI,QAAyB,CAAC,YAAY;AAC1D,cAAM,SAA0B,IAAI,gBAAgB,EAAE,MAAM,EAAE,GAAG,MAAM,QAAQ,MAAM,CAAC;AAAA,MACxF,CAAC;AAED,YAAM,OAAQ,IAAI,QAAQ,EAAuB;AAEjD,UAAI,GAAG,cAAc,CAAC,aAAa;AACjC,iBAAS,KAAK,cAAc;AAE5B,iBAAS,UAAU;AAAA,MACrB,CAAC;AAED,YAAM,EAAE,SAAS,IAAI,gBAAgB,kBAAkB,IAAI,EAAE;AAC7D,YAAM,SAAS,SAAS,UAAU;AAElC,YAAM,SAAmB,CAAC;AAC1B,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,KAAM;AACV,iBAAO,KAAK,OAAO,KAAK,KAAK,EAAE,SAAS,CAAC;AAAA,QAC3C;AAAA,MACF,SAAS,OAAO;AACd,gBAAQ,MAAM,KAAK;AAAA,MAErB,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAGA,aAAO,MAAM,EAAE,UAAU,cAAc;AAEvC,UAAI,MAAM;AAAA,IACZ,CAAC;AAAA,EACH,CAAC;AACH,CAAC;","names":[]}

package/src/inference/interruption/AdaptiveInterruptionDetector.ts

@@ -1,166 +0,0 @@
-import type { TypedEventEmitter } from '@livekit/typed-emitter';
-import EventEmitter from 'events';
-import { type ReadableStream, TransformStream } from 'stream/web';
-import { InterruptionStreamBase } from './InterruptionStream.js';
-import {
-  DEFAULT_BASE_URL,
-  FRAMES_PER_SECOND,
-  SAMPLE_RATE,
-  interruptionOptionDefaults,
-} from './defaults.js';
-import {
-  type InterruptionDetectionError,
-  type InterruptionEvent,
-  InterruptionEventType,
-} from './interruption.js';
-
-type InterruptionCallbacks = {
-  interruptionDetected: () => void;
-  overlapSpeechDetected: () => void;
-  error: (error: InterruptionDetectionError) => void;
-};
-
-export interface InterruptionOptions {
-  sampleRate: number;
-  threshold: number;
-  minFrames: number;
-  maxAudioDuration: number;
-  audioPrefixDuration: number;
-  detectionInterval: number;
-  inferenceTimeout: number;
-  minInterruptionDuration: number;
-  baseUrl: string;
-  apiKey: string;
-  apiSecret: string;
-  useProxy: boolean;
-}
-
-export type AdaptiveInterruptionDetectorOptions = Partial<InterruptionOptions>;
-
-export class AdaptiveInterruptionDetector extends (EventEmitter as new () => TypedEventEmitter<InterruptionCallbacks>) {
-  options: InterruptionOptions;
-  private label: string;
-  private streams: WeakSet<object>; // TODO: Union of InterruptionHttpStream | InterruptionWebSocketStream
-
-  constructor(options: AdaptiveInterruptionDetectorOptions = {}) {
-    super();
-
-    const {
-      maxAudioDuration,
-      baseUrl,
-      apiKey,
-      apiSecret,
-      useProxy: useProxyArg,
-      audioPrefixDuration,
-      threshold,
-      detectionInterval,
-      inferenceTimeout,
-      minInterruptionDuration,
-    } = { ...interruptionOptionDefaults, ...options };
-
-    if (maxAudioDuration > 3.0) {
-      throw new Error('maxAudioDuration must be less than or equal to 3.0 seconds');
-    }
-
-    const lkBaseUrl = baseUrl ?? process.env.LIVEKIT_REMOTE_EOT_URL ?? DEFAULT_BASE_URL;
-    let lkApiKey = apiKey ?? '';
-    let lkApiSecret = apiSecret ?? '';
-    let useProxy: boolean;
-
-    // use LiveKit credentials if using the default base URL (inference)
-    if (lkBaseUrl === DEFAULT_BASE_URL) {
-      lkApiKey =
-        apiKey ?? process.env.LIVEKIT_INFERENCE_API_KEY ?? process.env.LIVEKIT_API_KEY ?? '';
-      if (!lkApiKey) {
-        throw new Error(
-          'apiKey is required, either as argument or set LIVEKIT_API_KEY environmental variable',
-        );
-      }
-
-      lkApiSecret =
-        apiSecret ??
-        process.env.LIVEKIT_INFERENCE_API_SECRET ??
-        process.env.LIVEKIT_API_SECRET ??
-        '';
-      if (!lkApiSecret) {
-        throw new Error(
-          'apiSecret is required, either as argument or set LIVEKIT_API_SECRET environmental variable',
-        );
-      }
-
-      useProxy = true;
-    } else {
-      useProxy = useProxyArg ?? false;
-    }
-
-    this.options = {
-      sampleRate: SAMPLE_RATE,
-      threshold,
-      minFrames: Math.ceil(minInterruptionDuration * FRAMES_PER_SECOND),
-      maxAudioDuration,
-      audioPrefixDuration,
-      detectionInterval,
-      inferenceTimeout,
-      baseUrl: lkBaseUrl,
-      apiKey: lkApiKey,
-      apiSecret: lkApiSecret,
-      useProxy,
-      minInterruptionDuration,
-    };
-
-    this.label = `${this.constructor.name}`;
-    this.streams = new WeakSet();
-
-    console.info('adaptive interruption detector initialized', {
-      baseUrl: this.options.baseUrl,
-      detectionInterval: this.options.detectionInterval,
-      audioPrefixDuration: this.options.audioPrefixDuration,
-      maxAudioDuration: this.options.maxAudioDuration,
-      minFrames: this.options.minFrames,
-      threshold: this.options.threshold,
-      inferenceTimeout: this.options.inferenceTimeout,
-      useProxy: this.options.useProxy,
-    });
-  }
-
-  /**
-   * Creates a new InterruptionStreamBase for internal use.
-   * The stream can receive audio frames and sentinels via pushFrame().
-   * Use this when you need direct access to the stream for pushing frames.
-   */
-  createStream(): InterruptionStreamBase {
-    const stream = new InterruptionStreamBase(this, {});
-    this.streams.add(stream);
-    return stream;
-  }
-
-  /**
-   * Creates a new interruption stream and returns a ReadableStream of InterruptionEvents.
-   * This is a convenience method for consuming interruption events without needing
-   * to manage the underlying stream directly.
-   */
-  stream(): ReadableStream<InterruptionEvent> {
-    const httpStream = this.createStream();
-    const transformer = new TransformStream<InterruptionEvent, InterruptionEvent>({
-      transform: (chunk, controller) => {
-        if (chunk.type === InterruptionEventType.INTERRUPTION) {
-          this.emit('interruptionDetected'); // TODO payload
-        } else if (chunk.type === InterruptionEventType.OVERLAP_SPEECH_ENDED) {
-          this.emit('overlapSpeechDetected'); // TODO payload
-        }
-        controller.enqueue(chunk);
-      },
-    });
-    const stream = httpStream.stream.pipeThrough(transformer);
-    return stream;
-  }
-
-  updateOptions(options: { threshold?: number; minInterruptionDuration?: number }): void {
-    if (options.threshold !== undefined) {
-      this.options.threshold = options.threshold;
-    }
-    if (options.minInterruptionDuration !== undefined) {
-      this.options.minFrames = Math.ceil(options.minInterruptionDuration * FRAMES_PER_SECOND);
-    }
-  }
-}