@questionbase/deskfree 0.3.0-alpha.29 → 0.3.0-alpha.30

package/dist/index.js

@@ -4117,6 +4117,7 @@ var CLOSE_MAX_RETRIES = 3;
 var DeskFreeStreamingSession = class {
   client;
   log;
+  sendViaWs;
   messageId = null;
   currentText = "";
   closed = false;
@@ -4126,9 +4127,10 @@ var DeskFreeStreamingSession = class {
   queue = Promise.resolve();
   /** Full accumulated text (currentText + pendingDelta) */
   fullText = "";
-  constructor(client, log) {
+  constructor(client, log, sendViaWs) {
     this.client = client;
     this.log = log;
+    this.sendViaWs = sendViaWs;
   }
   /**
    * Send the first chunk as a new message.
@@ -4228,13 +4230,13 @@ var DeskFreeStreamingSession = class {
     if (!delta || !this.messageId || this.closed) return;
     this.pendingDelta = "";
     const fullContent = this.fullText;
+    const messageId = this.messageId;
     this.queue = this.queue.then(async () => {
       try {
-        await this.client.streamChunk({
-          messageId: this.messageId,
-          delta,
-          fullContent
-        });
+        const sentViaWs = this.sendViaWs?.(messageId, delta, fullContent);
+        if (!sentViaWs) {
+          await this.client.streamChunk({ messageId, delta, fullContent });
+        }
         this.currentText = fullContent;
         this.lastUpdateTime = Date.now();
       } catch (err) {
@@ -4619,7 +4621,11 @@ async function deliverMessageToAgent(ctx, message, client) {
         if (kind === "block") {
           accumulatedText += text;
           if (!streamingSession) {
-            streamingSession = new DeskFreeStreamingSession(client, log);
+            streamingSession = new DeskFreeStreamingSession(
+              client,
+              log,
+              sendWsStreamChunk
+            );
             log.info(
               `Streaming reply to DeskFree (task: ${taskId ?? "none"}): "${accumulatedText.slice(0, 80)}..."`
             );
@@ -4743,6 +4749,17 @@ var BACKOFF_FACTOR = 1.8;
 var HEALTH_LOG_INTERVAL_MS = 30 * 60 * 1e3;
 var MAX_CONSECUTIVE_POLL_FAILURES = 5;
 var deliveredMessageIds = /* @__PURE__ */ new Set();
+function sendWsStreamChunk(messageId, delta, fullContent) {
+  if (!activeWs || activeWs.readyState !== wrapper_default.OPEN) return false;
+  try {
+    activeWs.send(
+      JSON.stringify({ action: "streamChunk", messageId, delta, fullContent })
+    );
+    return true;
+  } catch {
+    return false;
+  }
+}
 var healthState = /* @__PURE__ */ new Map();
 function initializeHealth(accountId) {
   if (!healthState.has(accountId)) {
@@ -5100,7 +5117,7 @@ async function runWebSocketConnection(opts) {
     ws.on("close", (code, reason) => {
       cleanup();
       isConnected = false;
-      activeWs = null;
+      if (activeWs === ws) activeWs = null;
       ctx.setStatus({ running: false, lastStopAt: Date.now() });
       if (code === 1e3) {
         log.info(`WebSocket closed normally: ${code} ${reason.toString()}`);
@@ -5150,6 +5167,7 @@ async function runWebSocketConnection(opts) {
     ws.on("error", (err) => {
       cleanup();
       isConnected = false;
+      if (activeWs === ws) activeWs = null;
       const errorMessage = err instanceof Error ? err.message : String(err);
       log.error(`WebSocket error: ${errorMessage}`);
       reportError("error", `WebSocket error: ${errorMessage}`, {