@livekit/agents 1.0.34 → 1.0.36-dev.0

package/dist/utils/ws_transport.d.ts

@@ -0,0 +1,9 @@
+/// <reference types="node" resolution-mode="require"/>
+/// <reference types="node" resolution-mode="require"/>
+/// <reference types="node" resolution-mode="require"/>
+export declare function webSocketStream(wsUrl: string): {
+    readable: import("stream/web").ReadableStream<any>;
+    writable: import("stream/web").WritableStream<any>;
+    close: (code?: number | undefined, data?: string | Buffer | undefined) => void;
+};
+//# sourceMappingURL=ws_transport.d.ts.map

package/dist/utils/ws_transport.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws_transport.d.ts","sourceRoot":"","sources":["../../src/utils/ws_transport.ts"],"names":[],"mappings":";;;AAGA,wBAAgB,eAAe,CAAC,KAAK,EAAE,MAAM;;;;EAkB5C"}

package/dist/utils/ws_transport.js

@@ -0,0 +1,17 @@
+import { Readable, Writable } from "node:stream";
+import WebSocket, { createWebSocketStream } from "ws";
+function webSocketStream(wsUrl) {
+  const ws = new WebSocket(wsUrl);
+  const duplex = createWebSocketStream(ws);
+  duplex.on("error", console.error);
+  duplex.on("end", () => {
+    duplex.end();
+  });
+  const writable = Writable.toWeb(duplex);
+  const readable = Readable.toWeb(duplex);
+  return { readable, writable, close: ws.close };
+}
+export {
+  webSocketStream
+};
+//# sourceMappingURL=ws_transport.js.map

package/dist/utils/ws_transport.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/utils/ws_transport.ts"],"sourcesContent":["import { Readable, Writable } from 'node:stream';\nimport WebSocket, { createWebSocketStream } from 'ws';\n\nexport function webSocketStream(wsUrl: string) {\n  const ws = new WebSocket(wsUrl);\n  const duplex = createWebSocketStream(ws);\n  duplex.on('error', console.error);\n\n  // End the write side when the read side ends to properly close the stream.\n  // This is needed because Readable.toWeb() waits for both sides of the duplex\n  // to close before signaling done on the ReadableStream.\n  duplex.on('end', () => {\n    duplex.end();\n  });\n\n  // Convert the writable side\n  const writable = Writable.toWeb(duplex);\n  // Convert the readable side\n  const readable = Readable.toWeb(duplex);\n\n  return { readable, writable, close: ws.close };\n}\n"],"mappings":"AAAA,SAAS,UAAU,gBAAgB;AACnC,OAAO,aAAa,6BAA6B;AAE1C,SAAS,gBAAgB,OAAe;AAC7C,QAAM,KAAK,IAAI,UAAU,KAAK;AAC9B,QAAM,SAAS,sBAAsB,EAAE;AACvC,SAAO,GAAG,SAAS,QAAQ,KAAK;AAKhC,SAAO,GAAG,OAAO,MAAM;AACrB,WAAO,IAAI;AAAA,EACb,CAAC;AAGD,QAAM,WAAW,SAAS,MAAM,MAAM;AAEtC,QAAM,WAAW,SAAS,MAAM,MAAM;AAEtC,SAAO,EAAE,UAAU,UAAU,OAAO,GAAG,MAAM;AAC/C;","names":[]}

package/dist/utils/ws_transport.test.cjs

@@ -0,0 +1,212 @@
+"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

@@ -0,0 +1 @@
+{"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

@@ -0,0 +1,211 @@
+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

@@ -0,0 +1 @@
+{"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/dist/voice/agent_activity.cjs

@@ -36,6 +36,7 @@ var import_telemetry = require("../telemetry/index.cjs");
 var import_word = require("../tokenize/basic/word.cjs");
 var import_tts = require("../tts/tts.cjs");
 var import_utils = require("../utils.cjs");
+var import_interruption = require("../inference/interruption/interruption.cjs");
 var import_vad = require("../vad.cjs");
 var import_agent = require("./agent.cjs");
 var import_agent_session = require("./agent_session.cjs");
@@ -69,6 +70,24 @@ class AgentActivity {
   /** @internal */
   _mainTask;
   _userTurnCompletedTask;
+  /**
+   * Notify that agent started speaking.
+   * This enables interruption detection in AudioRecognition.
+   * @internal
+   */
+  notifyAgentSpeechStarted() {
+    var _a;
+    (_a = this.audioRecognition) == null ? void 0 : _a.onStartOfAgentSpeech();
+  }
+  /**
+   * Notify that agent stopped speaking.
+   * This disables interruption detection in AudioRecognition.
+   * @internal
+   */
+  notifyAgentSpeechEnded() {
+    var _a;
+    (_a = this.audioRecognition) == null ? void 0 : _a.onEndOfAgentSpeech();
+  }
   constructor(agent, agentSession) {
     this.agent = agent;
     this.agentSession = agentSession;
@@ -201,6 +220,7 @@ class AgentActivity {
         // Disable stt node if stt is not provided
         stt: this.stt ? (...args) => this.agent.sttNode(...args) : void 0,
         vad: this.vad,
+        interruptionDetector: this.agentSession.interruptionDetector,
         turnDetector: typeof this.turnDetection === "string" ? void 0 : this.turnDetection,
         turnDetectionMode: this.turnDetectionMode,
         minEndpointingDelay: this.agentSession.options.minEndpointingDelay,
@@ -494,6 +514,35 @@ class AgentActivity {
       this._currentSpeech.interrupt();
     }
   }
+  onInterruption(ev) {
+    var _a, _b;
+    if (ev.type !== import_interruption.InterruptionEventType.INTERRUPTION) {
+      return;
+    }
+    this.logger.info(
+      {
+        probability: ev.probability,
+        detectionDelay: ev.detectionDelay,
+        totalDuration: ev.totalDuration
+      },
+      "adaptive interruption detected"
+    );
+    if (this.turnDetection === "manual" || this.turnDetection === "realtime_llm") {
+      return;
+    }
+    if (this.llm instanceof import_llm.RealtimeModel && this.llm.capabilities.turnDetection) {
+      return;
+    }
+    (_a = this.realtimeSession) == null ? void 0 : _a.startUserActivity();
+    if (this._currentSpeech && !this._currentSpeech.interrupted && this._currentSpeech.allowInterruptions) {
+      this.logger.info(
+        { "speech id": this._currentSpeech.id },
+        "speech interrupted by adaptive interruption detector"
+      );
+      (_b = this.realtimeSession) == null ? void 0 : _b.interrupt();
+      this._currentSpeech.interrupt();
+    }
+  }
   onInterimTranscript(ev) {
     if (this.llm instanceof import_llm.RealtimeModel && this.llm.capabilities.userTranscription) {
       return;