@tiflis-io/tiflis-code-workstation 0.3.28 → 0.3.29

package/dist/main.js

@@ -5113,9 +5113,10 @@ var MessageBroadcasterImpl = class {
   }
   /**
    * Broadcasts a message to all clients subscribed to a session (by session ID string).
-   * Sends targeted messages to each subscriber individually via forward.to_device.
+   * Sends targeted messages to each subscriber in parallel with timeout.
+   * Prevents slow clients from blocking others.
    */
-  broadcastToSubscribers(sessionId, message) {
+  async broadcastToSubscribers(sessionId, message) {
     const session = new SessionId(sessionId);
     const subscribers = this.deps.clientRegistry.getSubscribers(session);
     const authenticatedSubscribers = subscribers.filter(
@@ -5124,9 +5125,44 @@ var MessageBroadcasterImpl = class {
     if (authenticatedSubscribers.length === 0) {
       return;
     }
-    for (const client of authenticatedSubscribers) {
-      this.deps.tunnelClient.sendToDevice(client.deviceId.value, message);
-    }
+    const SEND_TIMEOUT_MS = 2e3;
+    const sendPromises = authenticatedSubscribers.map(async (client) => {
+      try {
+        await Promise.race([
+          new Promise((resolve2, reject) => {
+            const sent = this.deps.tunnelClient.sendToDevice(
+              client.deviceId.value,
+              message
+            );
+            if (sent) {
+              resolve2();
+            } else {
+              reject(new Error(`sendToDevice returned false for ${client.deviceId.value}`));
+            }
+          }),
+          new Promise(
+            (_, reject) => setTimeout(
+              () => reject(new Error(`Send timeout for ${client.deviceId.value}`)),
+              SEND_TIMEOUT_MS
+            )
+          )
+        ]);
+        this.logger.debug(
+          { deviceId: client.deviceId.value, sessionId },
+          "Message sent to subscriber"
+        );
+      } catch (error) {
+        this.logger.warn(
+          {
+            deviceId: client.deviceId.value,
+            sessionId,
+            error: error instanceof Error ? error.message : String(error)
+          },
+          "Failed to send to subscriber (timeout or error)"
+        );
+      }
+    });
+    await Promise.allSettled(sendPromises);
   }
 };
 
@@ -10132,7 +10168,7 @@ async function bootstrap() {
                 message_id: messageId
               }
             };
-            messageBroadcaster.broadcastToSubscribers(
+            await messageBroadcaster.broadcastToSubscribers(
               sessionId,
               JSON.stringify(errorEvent)
             );
@@ -10168,7 +10204,7 @@ async function bootstrap() {
                 from_device_id: deviceId
               }
             };
-            messageBroadcaster.broadcastToSubscribers(
+            await messageBroadcaster.broadcastToSubscribers(
               sessionId,
               JSON.stringify(transcriptionEvent)
             );
@@ -10259,7 +10295,7 @@ async function bootstrap() {
                 timestamp: Date.now()
               }
             };
-            messageBroadcaster.broadcastToSubscribers(
+            await messageBroadcaster.broadcastToSubscribers(
               sessionId,
               JSON.stringify(errorEvent)
             );
@@ -10387,14 +10423,20 @@ async function bootstrap() {
               is_complete: true
             }
           };
-          messageBroadcaster.broadcastToSubscribers(
+          void messageBroadcaster.broadcastToSubscribers(
             sessionId,
             JSON.stringify(cancelOutput)
-          );
-          logger.info(
-            { sessionId },
-            "Broadcasted cancel message to subscribers"
-          );
+          ).then(() => {
+            logger.info(
+              { sessionId },
+              "Broadcasted cancel message to subscribers"
+            );
+          }).catch((error) => {
+            logger.error(
+              { sessionId, error },
+              "Failed to broadcast cancel message"
+            );
+          });
           chatHistoryService.saveAgentMessage(sessionId, "assistant", "", [
             cancelBlock
           ]);
@@ -10880,9 +10922,13 @@ async function bootstrap() {
     subscriptionRepository,
     logger
   });
+  let supervisorMessageSequence = 0;
+  const agentMessageSequences = /* @__PURE__ */ new Map();
   const broadcaster = messageBroadcaster;
   const agentMessageAccumulator = /* @__PURE__ */ new Map();
   const agentStreamingMessageIds = /* @__PURE__ */ new Map();
+  const STREAMING_STATE_GRACE_PERIOD_MS = 1e4;
+  const agentCleanupTimeouts = /* @__PURE__ */ new Map();
   const getOrCreateAgentStreamingMessageId = (sessionId) => {
     let messageId = agentStreamingMessageIds.get(sessionId);
     if (!messageId) {
@@ -10892,7 +10938,16 @@ async function bootstrap() {
     return messageId;
   };
   const clearAgentStreamingMessageId = (sessionId) => {
-    agentStreamingMessageIds.delete(sessionId);
+    const existingTimeout = agentCleanupTimeouts.get(sessionId);
+    if (existingTimeout) {
+      clearTimeout(existingTimeout);
+    }
+    const timeout = setTimeout(() => {
+      agentStreamingMessageIds.delete(sessionId);
+      agentCleanupTimeouts.delete(sessionId);
+      logger.debug({ sessionId }, "Agent streaming state cleaned up after grace period");
+    }, STREAMING_STATE_GRACE_PERIOD_MS);
+    agentCleanupTimeouts.set(sessionId, timeout);
   };
   agentSessionManager.on(
     "blocks",
@@ -10966,10 +11021,14 @@ async function bootstrap() {
       }
       const mergedBlocks = mergeToolBlocks(accumulatedBlocks);
       const fullAccumulatedText = mergedBlocks.filter((b) => b.block_type === "text").map((b) => b.content).join("\n");
+      const currentSequence = agentMessageSequences.get(sessionId) ?? 0;
+      const newSequence = currentSequence + 1;
+      agentMessageSequences.set(sessionId, newSequence);
       const outputEvent = {
         type: "session.output",
         session_id: sessionId,
         streaming_message_id: streamingMessageId,
+        sequence: newSequence,
         payload: {
           content_type: "agent",
           content: fullAccumulatedText,
@@ -10980,13 +11039,19 @@ async function bootstrap() {
           is_complete: isComplete
         }
       };
-      broadcaster.broadcastToSubscribers(
+      await broadcaster.broadcastToSubscribers(
         sessionId,
         JSON.stringify(outputEvent)
       );
       if (isComplete) {
         clearAgentStreamingMessageId(sessionId);
-        agentMessageAccumulator.delete(sessionId);
+        setTimeout(() => {
+          agentMessageAccumulator.delete(sessionId);
+          logger.debug(
+            { sessionId, messageId: streamingMessageId },
+            "Agent message accumulator cleaned up after grace period"
+          );
+        }, STREAMING_STATE_GRACE_PERIOD_MS);
       }
       if (isComplete && fullTextContent.length > 0) {
         const pendingVoiceCommand = pendingAgentVoiceCommands.get(sessionId);
@@ -11045,7 +11110,7 @@ async function bootstrap() {
                 from_device_id: originDeviceId
               }
             };
-            broadcaster.broadcastToSubscribers(
+            await broadcaster.broadcastToSubscribers(
               sessionId,
               JSON.stringify(voiceOutputEvent)
             );
@@ -11090,6 +11155,7 @@ async function bootstrap() {
       const outputEvent = {
         type: "supervisor.output",
         streaming_message_id: streamingMessageId,
+        sequence: ++supervisorMessageSequence,
         payload: {
           content_type: "supervisor",
           content: textContent,
@@ -11101,7 +11167,13 @@ async function bootstrap() {
       const message = JSON.stringify(outputEvent);
       broadcaster.broadcastToAll(message);
       if (isComplete) {
-        supervisorMessageAccumulator.clear();
+        setTimeout(() => {
+          supervisorMessageAccumulator.clear();
+          logger.debug(
+            { messageId: streamingMessageId },
+            "Supervisor message accumulator cleaned up after grace period"
+          );
+        }, STREAMING_STATE_GRACE_PERIOD_MS);
       }
       if (isComplete && finalOutput && finalOutput.length > 0) {
         chatHistoryService.saveSupervisorMessage(
@@ -11230,21 +11302,23 @@ async function bootstrap() {
       batchIntervalMs: env.TERMINAL_BATCH_INTERVAL_MS,
       maxBatchSize: env.TERMINAL_BATCH_MAX_SIZE,
       onFlush: (batchedData) => {
-        const outputMessage = session.addOutputToBuffer(batchedData);
-        const outputEvent = {
-          type: "session.output",
-          session_id: sessionId.value,
-          payload: {
-            content_type: "terminal",
-            content: batchedData,
-            timestamp: outputMessage.timestamp,
-            sequence: outputMessage.sequence
-          }
-        };
-        broadcaster.broadcastToSubscribers(
-          sessionId.value,
-          JSON.stringify(outputEvent)
-        );
+        void (async () => {
+          const outputMessage = session.addOutputToBuffer(batchedData);
+          const outputEvent = {
+            type: "session.output",
+            session_id: sessionId.value,
+            payload: {
+              content_type: "terminal",
+              content: batchedData,
+              timestamp: outputMessage.timestamp,
+              sequence: outputMessage.sequence
+            }
+          };
+          await broadcaster.broadcastToSubscribers(
+            sessionId.value,
+            JSON.stringify(outputEvent)
+          );
+        })();
       }
     });
     session.onOutput((data) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tiflis-io/tiflis-code-workstation",
-  "version": "0.3.28",
+  "version": "0.3.29",
   "description": "Workstation server for tiflis-code - manages agent sessions and terminal access",
   "author": "Roman Barinov <rbarinov@gmail.com>",
   "license": "FSL-1.1-NC",