Review feedback - thanks mike!

w3c · Jul 11, 2024 · 8e405b2 · 8e405b2
1 parent 21d1b36
commit 8e405b2
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/agent/at-driver.js b/src/agent/at-driver.js
@@ -78,8 +78,12 @@ export class ATDriver {
     const id = this._nextId++;
     const rawMessage = JSON.stringify({ id, ...command });
     this.log(AgentMessage.AT_DRIVER_COMMS, { direction: 'outbound', message: rawMessage });
-    if (this.hasClosed) throw new Error('AT-Driver connection unexpectedly closed');
-    this.socket.send(rawMessage);
+    await new Promise((resolve, reject) => {
+      this.socket.send(rawMessage, error => {
+        if (error) reject(error);
+        else resolve();
+      });
+    });
     for await (const message of this._messages()) {
       if (message.id === id) {
         if (message.error) {
@@ -109,7 +113,6 @@ export class ATDriver {
    * @returns {AsyncGenerator<string>}
    */
   async *speeches() {
-    if (this.hasClosed) throw new Error('AT-Driver connection unexpectedly closed');
     for await (const message of this._messages()) {
       if (message.method === 'interaction.capturedOutput') {
         yield message.params.data;