@hef2024/llmasaservice-ui 0.25.1 → 0.25.2

package/dist/index.mjs

@@ -5498,6 +5498,10 @@ var AIChatPanel = ({
   const queuedToolRequestsRef = useRef6(null);
   const suppressAbortHistoryUpdateRef = useRef6(false);
   const toolReplaySummariesByKeyRef = useRef6({});
+  const continuationDispatchPendingRef = useRef6(false);
+  const successfulMutationSignaturesRef = useRef6(/* @__PURE__ */ new Set());
+  const queuedDrainTimerRef = useRef6(null);
+  const queuedDrainScheduledRef = useRef6(false);
   useEffect8(() => {
     if (!initialHistory) return;
     setHistory((prev) => {
@@ -6006,6 +6010,8 @@ var AIChatPanel = ({
     return false;
   }, [customerEmailCaptureMode, emailInputSet]);
   const pendingToolRequestsRef = useRef6(pendingToolRequests);
+  const activeToolCallsRef = useRef6(activeToolCalls);
+  const turnLockRef = useRef6(false);
   const streamIdleRef = useRef6(idle);
   streamIdleRef.current = idle;
   const waitForStreamIdle = useCallback2((timeoutMs = 2500) => __async(void 0, null, function* () {
@@ -6019,6 +6025,36 @@ var AIChatPanel = ({
   useEffect8(() => {
     pendingToolRequestsRef.current = pendingToolRequests;
   }, [pendingToolRequests]);
+  useEffect8(() => {
+    activeToolCallsRef.current = activeToolCalls;
+  }, [activeToolCalls]);
+  const hasUnresolvedToolRequestsInLatestResponse = useCallback2(() => {
+    const latestResponse = String(responseRef.current || "");
+    if (!latestResponse.trim()) return false;
+    const latestRequests = extractToolRequestMatchesFromText(latestResponse);
+    if (latestRequests.length === 0) return false;
+    return latestRequests.some((request) => {
+      const signature = getToolCallSignature(request.toolName, request.callId);
+      if (!signature) return false;
+      if (handledToolCallSignaturesRef.current.has(signature)) return false;
+      if (inFlightToolCallSignaturesRef.current.has(signature)) return false;
+      return true;
+    });
+  }, [getToolCallSignature]);
+  const hasInFlightTurnWork = useCallback2(() => {
+    return toolRequestProcessingRef.current || continuationDispatchPendingRef.current || (queuedToolRequestsRef.current || []).length > 0 || (pendingToolRequestsRef.current || []).length > 0 || (activeToolCallsRef.current || []).length > 0 || !streamIdleRef.current || hasUnresolvedToolRequestsInLatestResponse();
+  }, [hasUnresolvedToolRequestsInLatestResponse]);
+  const queuePromptForLater = useCallback2((promptText) => {
+    const normalizedPrompt = String(promptText || "").trim();
+    if (!normalizedPrompt) return;
+    const nextQueuedPrompts = [...queuedPromptsRef.current, normalizedPrompt];
+    queuedPromptsRef.current = nextQueuedPrompts;
+    setQueuedPrompts(nextQueuedPrompts);
+  }, []);
+  const releaseTurnLockIfSettled = useCallback2(() => {
+    if (hasInFlightTurnWork()) return;
+    turnLockRef.current = false;
+  }, [hasInFlightTurnWork]);
   const processGivenToolRequests = useCallback2(
     (requests) => __async(void 0, null, function* () {
       var _a2, _b, _c;
@@ -6045,6 +6081,7 @@ var AIChatPanel = ({
         return;
       }
       toolRequestProcessingRef.current = true;
+      turnLockRef.current = true;
       try {
         let requestsToProcess = requests;
         if (!requestsToProcess || requestsToProcess.length === 0) {
@@ -6123,6 +6160,44 @@ ${traceSummary}` : traceSummary;
             });
           }
         }
+        const normalizeToolArgsForSignature = (value) => {
+          if (Array.isArray(value)) {
+            return value.map((item) => normalizeToolArgsForSignature(item));
+          }
+          if (value && typeof value === "object") {
+            const normalizedObject = {};
+            Object.keys(value).sort().forEach((key) => {
+              normalizedObject[key] = normalizeToolArgsForSignature(
+                value[key]
+              );
+            });
+            return normalizedObject;
+          }
+          return value != null ? value : null;
+        };
+        const getSemanticMutationSignature = (toolName, args) => {
+          const normalizedName = String(toolName || "").trim().toLowerCase();
+          const normalizedArgs = normalizeToolArgsForSignature(args);
+          return `${normalizedName}::${JSON.stringify(normalizedArgs)}`;
+        };
+        const isLikelyMutatingCall = (toolName, args) => {
+          const normalizedName = String(toolName || "").trim().toLowerCase();
+          const action = String((args == null ? void 0 : args.action) || "").trim().toLowerCase();
+          const mutatingActions = /* @__PURE__ */ new Set([
+            "add",
+            "create",
+            "materialize_from_sourcing_result",
+            "materialize",
+            "update",
+            "delete",
+            "remove",
+            "cancel",
+            "complete",
+            "move"
+          ]);
+          if (action && mutatingActions.has(action)) return true;
+          return normalizedName.includes("pipeline") || normalizedName.includes("work_item") || normalizedName.includes("candidate_pipeline") || normalizedName.includes("cron");
+        };
         const parsedToolCalls = yield Promise.all(
           requestsToProcess.map((req, index) => __async(void 0, null, function* () {
             var _a3, _b2, _c2, _d, _e, _f;
@@ -6151,27 +6226,44 @@ ${traceSummary}` : traceSummary;
             const serviceTag = typeof req.serviceTag === "string" && req.serviceTag || typeof req.groups[3] === "string" && req.groups[3] || typeof (parsedToolCall == null ? void 0 : parsedToolCall.service) === "string" && parsedToolCall.service || "";
             const callSignature = getToolCallSignature(toolName, callId);
             if (!callSignature) return null;
+            const semanticMutationSignature = getSemanticMutationSignature(toolName, args);
+            const likelyMutating = isLikelyMutatingCall(toolName, args);
             return {
               req,
               toolName,
               callId,
               args,
               serviceTag,
-              callSignature
+              callSignature,
+              semanticMutationSignature,
+              likelyMutating
             };
           }))
         );
         const toolCallBatch = parsedToolCalls.filter(Boolean);
         const seenCallSignatures = /* @__PURE__ */ new Set();
+        const seenMutationSemanticsInBatch = /* @__PURE__ */ new Set();
         const callsToRun = [];
+        const suppressedMutationDuplicates = [];
         toolCallBatch.forEach((toolCall) => {
           if (seenCallSignatures.has(toolCall.callSignature)) return;
           seenCallSignatures.add(toolCall.callSignature);
           if (handledToolCallSignaturesRef.current.has(toolCall.callSignature)) return;
           if (inFlightToolCallSignaturesRef.current.has(toolCall.callSignature)) return;
+          if (toolCall.likelyMutating && seenMutationSemanticsInBatch.has(toolCall.semanticMutationSignature)) {
+            suppressedMutationDuplicates.push(toolCall);
+            return;
+          }
+          if (toolCall.likelyMutating && successfulMutationSignaturesRef.current.has(toolCall.semanticMutationSignature)) {
+            suppressedMutationDuplicates.push(toolCall);
+            return;
+          }
+          if (toolCall.likelyMutating) {
+            seenMutationSemanticsInBatch.add(toolCall.semanticMutationSignature);
+          }
           callsToRun.push(toolCall);
         });
-        if (callsToRun.length === 0) {
+        if (callsToRun.length === 0 && suppressedMutationDuplicates.length === 0) {
           setPendingToolRequests(
             (prev) => prev.filter((request) => {
               const signature = getToolCallSignature(request.toolName, request.callId);
@@ -6183,11 +6275,14 @@ ${traceSummary}` : traceSummary;
           setIsLoading(false);
           return;
         }
-        const callsToRunSignatures = new Set(callsToRun.map((toolCall) => toolCall.callSignature));
+        const handledBatchSignatures = /* @__PURE__ */ new Set([
+          ...callsToRun.map((toolCall) => toolCall.callSignature),
+          ...suppressedMutationDuplicates.map((toolCall) => toolCall.callSignature)
+        ]);
         setPendingToolRequests(
           (prev) => prev.filter((request) => {
             const signature = getToolCallSignature(request.toolName, request.callId);
-            return !signature || !callsToRunSignatures.has(signature);
+            return !signature || !handledBatchSignatures.has(signature);
           })
         );
         callsToRun.forEach((toolCall) => {
@@ -6200,7 +6295,7 @@ ${traceSummary}` : traceSummary;
             args: toolCall.args
           }))
         );
-        const finalToolCalls = callsToRun.map((toolCall) => ({
+        const finalToolCalls = [...callsToRun, ...suppressedMutationDuplicates].map((toolCall) => ({
           id: toolCall.callId,
           type: "tool_use",
           name: toolCall.toolName,
@@ -6331,12 +6426,30 @@ ${traceSummary}` : traceSummary;
               }
             }))
           );
-          finalToolResponses = toolResponses.filter(Boolean);
+          const executedResponses = toolResponses.filter(Boolean);
+          const suppressedResponses = suppressedMutationDuplicates.map((toolCall) => ({
+            tool_call_id: toolCall.callId,
+            tool_name: toolCall.toolName,
+            result: "Duplicate mutating call suppressed: this exact action already succeeded earlier in the same turn.",
+            isError: false
+          }));
+          finalToolResponses = [...executedResponses, ...suppressedResponses];
+          callsToRun.forEach((toolCall) => {
+            const matchedResponse = executedResponses.find(
+              (response2) => (response2 == null ? void 0 : response2.tool_call_id) === toolCall.callId
+            );
+            if (!(matchedResponse == null ? void 0 : matchedResponse.isError) && toolCall.likelyMutating) {
+              successfulMutationSignaturesRef.current.add(toolCall.semanticMutationSignature);
+            }
+          });
         } finally {
           callsToRun.forEach((toolCall) => {
             inFlightToolCallSignaturesRef.current.delete(toolCall.callSignature);
             handledToolCallSignaturesRef.current.add(toolCall.callSignature);
           });
+          suppressedMutationDuplicates.forEach((toolCall) => {
+            handledToolCallSignaturesRef.current.add(toolCall.callSignature);
+          });
         }
         setActiveToolCalls([]);
         const currentLastKey = lastKeyRef.current;
@@ -6440,6 +6553,7 @@ ${traceSummary}` : traceSummary;
         if (!streamIdleRef.current) {
           setActiveToolCalls([]);
           setIsLoading(false);
+          continuationDispatchPendingRef.current = false;
           setError({
             message: "Timed out waiting for the previous stream to settle before tool continuation.",
             code: "TOOL_CONTINUATION_WAIT_TIMEOUT"
@@ -6455,6 +6569,7 @@ ${traceSummary}` : traceSummary;
         } else {
           activeStreamAppendBaseRef.current = null;
         }
+        continuationDispatchPendingRef.current = true;
         send(
           "",
           newMessages,
@@ -6472,12 +6587,14 @@ ${traceSummary}` : traceSummary;
           newController,
           void 0,
           (errorMsg) => {
+            continuationDispatchPendingRef.current = false;
             setActiveToolCalls([]);
             setIsLoading(false);
             setError({
               message: errorMsg,
               code: "TOOL_ERROR"
             });
+            releaseTurnLockIfSettled();
           }
         );
       } finally {
@@ -6488,6 +6605,8 @@ ${traceSummary}` : traceSummary;
           queueMicrotask(() => {
             void processGivenToolRequests(queued);
           });
+        } else {
+          releaseTurnLockIfSettled();
         }
       }
     }),
@@ -6506,7 +6625,8 @@ ${traceSummary}` : traceSummary;
       idle,
       stop,
       lastController,
-      waitForStreamIdle
+      waitForStreamIdle,
+      releaseTurnLockIfSettled
     ]
   );
   const handleToolApproval = useCallback2(
@@ -6531,6 +6651,8 @@ ${traceSummary}` : traceSummary;
   );
   useEffect8(() => {
     if (pendingToolRequests.length === 0) return;
+    if (!idle) return;
+    if (continuationDispatchPendingRef.current) return;
     const configuredAutoApproveTools = Array.isArray(autoApproveTools) ? new Set(
       autoApproveTools.map((toolName) => normalizeToolName(String(toolName))).filter(Boolean)
     ) : null;
@@ -6559,6 +6681,7 @@ ${traceSummary}` : traceSummary;
     }
   }, [
     autoApproveTools,
+    idle,
     pendingToolRequests,
     sessionApprovedTools,
     alwaysApprovedTools,
@@ -6867,11 +6990,20 @@ ${traceSummary}` : traceSummary;
     lastScrollTimeRef.current = now;
   }, []);
   const continueChat = useCallback2((promptText) => {
+    const promptToSend = String(promptText || "").trim();
+    if (!promptToSend) return;
+    if (turnLockRef.current || hasInFlightTurnWork()) {
+      queuePromptForLater(promptToSend);
+      return;
+    }
+    turnLockRef.current = true;
     handledToolCallSignaturesRef.current = /* @__PURE__ */ new Set();
     inFlightToolCallSignaturesRef.current = /* @__PURE__ */ new Set();
     toolContinuationCountRef.current = 0;
+    successfulMutationSignaturesRef.current = /* @__PURE__ */ new Set();
     activeStreamAppendBaseRef.current = null;
     toolReplaySummariesByKeyRef.current = {};
+    continuationDispatchPendingRef.current = false;
     setPendingToolRequests([]);
     setActiveToolCalls([]);
     setCollapsedBlocks((prev) => {
@@ -6891,29 +7023,17 @@ ${traceSummary}` : traceSummary;
     setUserHasScrolled(false);
     prevResponseLengthRef.current = 0;
     setResponse("");
-    if (!idle) {
-      stop(lastController);
-      setHistory((prevHistory) => __spreadProps(__spreadValues({}, prevHistory), {
-        [lastKey != null ? lastKey : ""]: {
-          content: processThinkingTags(response).cleanedText + "\n\n(response cancelled)",
-          callId: lastCallId || ""
-        }
-      }));
-      return;
-    }
     if (clearFollowOnQuestionsNextPrompt) {
       setFollowOnQuestionsState([]);
     }
-    const promptToSend = promptText;
-    if (!promptToSend || !promptToSend.trim()) return;
     setIsLoading(true);
     const timestamp = (/* @__PURE__ */ new Date()).toISOString();
-    const promptKey = `${timestamp}:${promptToSend.trim()}`;
+    const promptKey = `${timestamp}:${promptToSend}`;
     setHistory((prevHistory) => __spreadProps(__spreadValues({}, prevHistory), {
       [promptKey]: { content: "", callId: "" }
     }));
     toolReplaySummariesByKeyRef.current[promptKey] = [];
-    setLastPrompt(promptToSend.trim());
+    setLastPrompt(promptToSend);
     setLastKey(promptKey);
     setTimeout(() => {
       scrollToBottom(true);
@@ -6930,7 +7050,7 @@ ${traceSummary}` : traceSummary;
         messagesAndHistory.push({ role: "user", content: promptForHistory });
         messagesAndHistory.push({ role: "assistant", content: assistantContextContent });
       });
-      let fullPromptToSend = promptToSend.trim();
+      let fullPromptToSend = promptToSend;
       if (messagesAndHistory.length === 0 && promptTemplate) {
         fullPromptToSend = promptTemplate.replace("{{prompt}}", fullPromptToSend);
       }
@@ -6939,7 +7059,7 @@ ${traceSummary}` : traceSummary;
       if (onBeforeSend) {
         void Promise.resolve(
           onBeforeSend({
-            prompt: promptToSend.trim(),
+            prompt: promptToSend,
             conversationId: convId || null,
             agentId: agent,
             service,
@@ -6973,6 +7093,7 @@ ${traceSummary}` : traceSummary;
           if (isAbortError) {
             if (suppressAbortHistoryUpdateRef.current) {
               setIsLoading(false);
+              releaseTurnLockIfSettled();
               return;
             }
             console.log("[AIChatPanel] Request was aborted by user");
@@ -7037,6 +7158,7 @@ ${traceSummary}` : traceSummary;
             }
           }
           setIsLoading(false);
+          releaseTurnLockIfSettled();
         }
       );
       setLastMessages(messagesAndHistory);
@@ -7045,15 +7167,16 @@ ${traceSummary}` : traceSummary;
           onConversationCreated(convId);
         }, 100);
       }
+    }).catch((error2) => {
+      console.error("[AIChatPanel] Failed to send prompt:", error2);
+      setError({
+        message: error2 instanceof Error ? error2.message : "Failed to send prompt",
+        code: "UNKNOWN_ERROR"
+      });
+      setIsLoading(false);
+      releaseTurnLockIfSettled();
     });
   }, [
-    idle,
-    stop,
-    lastController,
-    lastKey,
-    response,
-    lastCallId,
-    processThinkingTags,
     clearFollowOnQuestionsNextPrompt,
     promptTemplate,
     send,
@@ -7067,17 +7190,16 @@ ${traceSummary}` : traceSummary;
     scrollToBottom,
     onConversationCreated,
     onBeforeSend,
+    hasInFlightTurnWork,
+    queuePromptForLater,
+    releaseTurnLockIfSettled,
     getThinkingBlockCollapseKey,
     getThinkingBlockRenderKey,
     setResponse
   ]);
   const handleQueuePrompt = useCallback2((promptText) => {
-    const normalizedPrompt = String(promptText || "").trim();
-    if (!normalizedPrompt) return;
-    const nextQueuedPrompts = [...queuedPromptsRef.current, normalizedPrompt];
-    queuedPromptsRef.current = nextQueuedPrompts;
-    setQueuedPrompts(nextQueuedPrompts);
-  }, []);
+    queuePromptForLater(promptText);
+  }, [queuePromptForLater]);
   const handleClearQueuedPrompt = useCallback2((index) => {
     const nextQueuedPrompts = queuedPromptsRef.current.filter((_, queueIndex) => queueIndex !== index);
     queuedPromptsRef.current = nextQueuedPrompts;
@@ -7089,18 +7211,6 @@ ${traceSummary}` : traceSummary;
   const handleStop = useCallback2(() => {
     stop(lastController);
   }, [stop, lastController]);
-  useEffect8(() => {
-    const nextQueuedPrompt = String(queuedPromptsRef.current[0] || "").trim();
-    if (!nextQueuedPrompt) return;
-    if (!idle || isLoading) return;
-    if (pendingToolRequests.length > 0 || activeToolCalls.length > 0) return;
-    if (toolRequestProcessingRef.current) return;
-    if ((queuedToolRequestsRef.current || []).length > 0) return;
-    const remainingQueuedPrompts = queuedPromptsRef.current.slice(1);
-    queuedPromptsRef.current = remainingQueuedPrompts;
-    setQueuedPrompts(remainingQueuedPrompts);
-    continueChat(nextQueuedPrompt);
-  }, [activeToolCalls.length, continueChat, idle, isLoading, pendingToolRequests.length, queuedPrompts.length]);
   const handleNewConversation = useCallback2(() => {
     if (!newConversationConfirm) {
       setNewConversationConfirm(true);
@@ -7141,6 +7251,7 @@ ${traceSummary}` : traceSummary;
     setUserHasScrolled(false);
     setError(null);
     setActiveToolCalls([]);
+    turnLockRef.current = false;
     setTimeout(() => {
       var _a2;
       setJustReset(false);
@@ -7160,6 +7271,7 @@ ${traceSummary}` : traceSummary;
       if (inFlightToolCallSignaturesRef.current.has(callSignature)) return false;
       return true;
     });
+    pendingToolRequestsRef.current = unseenToolRequests;
     setPendingToolRequests((prev) => {
       if (areToolRequestListsEqual(prev, unseenToolRequests)) {
         return prev;
@@ -7250,9 +7362,11 @@ ${traceSummary}` : traceSummary;
           responseCompleteCallbackRef.current(currentLastCallId, currentLastPrompt || "", entry.content);
         }
       }
+      releaseTurnLockIfSettled();
     }
     if (!isNowIdle && hasNotifiedCompletionRef.current) {
       hasNotifiedCompletionRef.current = false;
+      continuationDispatchPendingRef.current = false;
       prevResponseLengthRef.current = 0;
       const currentLastKey = lastKeyRef.current;
       const existingContent = currentLastKey ? ((_a2 = latestHistoryRef.current[currentLastKey]) == null ? void 0 : _a2.content) || "" : "";
@@ -7261,7 +7375,48 @@ ${traceSummary}` : traceSummary;
         base: existingContent
       } : null;
     }
-  }, [idle]);
+  }, [idle, releaseTurnLockIfSettled]);
+  useEffect8(() => {
+    if (!idle) return;
+    releaseTurnLockIfSettled();
+  }, [idle, response, pendingToolRequests.length, activeToolCalls.length, releaseTurnLockIfSettled]);
+  useEffect8(() => {
+    const nextQueuedPrompt = String(queuedPromptsRef.current[0] || "").trim();
+    if (!nextQueuedPrompt) return;
+    if (!idle || isLoading) return;
+    if (pendingToolRequests.length > 0 || activeToolCalls.length > 0) return;
+    if (turnLockRef.current || hasInFlightTurnWork()) return;
+    if (queuedDrainScheduledRef.current) return;
+    queuedDrainScheduledRef.current = true;
+    queuedDrainTimerRef.current = window.setTimeout(() => {
+      queuedDrainScheduledRef.current = false;
+      queuedDrainTimerRef.current = null;
+      const queuedPrompt = String(queuedPromptsRef.current[0] || "").trim();
+      if (!queuedPrompt) return;
+      if (turnLockRef.current || hasInFlightTurnWork()) return;
+      const remainingQueuedPrompts = queuedPromptsRef.current.slice(1);
+      queuedPromptsRef.current = remainingQueuedPrompts;
+      setQueuedPrompts(remainingQueuedPrompts);
+      continueChat(queuedPrompt);
+    }, 0);
+  }, [
+    continueChat,
+    hasInFlightTurnWork,
+    idle,
+    isLoading,
+    pendingToolRequests.length,
+    activeToolCalls.length,
+    queuedPrompts.length
+  ]);
+  useEffect8(() => {
+    return () => {
+      if (queuedDrainTimerRef.current !== null) {
+        window.clearTimeout(queuedDrainTimerRef.current);
+        queuedDrainTimerRef.current = null;
+      }
+      queuedDrainScheduledRef.current = false;
+    };
+  }, []);
   useEffect8(() => {
     scrollToEndRef.current = scrollToEnd || false;
   }, [scrollToEnd]);
@@ -8554,7 +8709,49 @@ var parseCallMessages = (rawMessages) => {
   return normalized;
 };
 var shouldSkipTranscriptMessage = (message) => {
-  return message.role === "system" || message.role === "user" && message.content.startsWith("__system__:");
+  const normalizedContent = message.content.trim().toLowerCase();
+  const isInternalContinuationPrompt = message.role === "user" && normalizedContent.startsWith("original request:") && normalizedContent.includes("tool execution summary for the previous request:");
+  return message.role === "system" || message.role === "user" && message.content.startsWith("__system__:") || isInternalContinuationPrompt;
+};
+var looksLikeToolStubResponse = (value) => {
+  const text = value.trim();
+  if (!text) return false;
+  return text.includes('"type":"tool_use"') || text.includes('"type": "tool_use"') || text.includes('"type":"function"') || text.includes('"type": "function"');
+};
+var pickMoreCompleteResponse = (primaryResponse, fallbackResponse) => {
+  const primary = primaryResponse.trim();
+  const fallback = fallbackResponse.trim();
+  if (!fallback) return primary;
+  if (!primary) return fallback;
+  if (primary === fallback) return primary;
+  if (looksLikeToolStubResponse(primary) && !looksLikeToolStubResponse(fallback)) {
+    return fallback;
+  }
+  if (fallback.startsWith(primary) || fallback.length > primary.length + 40) {
+    return fallback;
+  }
+  return primary;
+};
+var mergeContinuationResponse = (baseResponse, continuationResponse) => {
+  const base = baseResponse.trim();
+  const continuation = continuationResponse.trim();
+  if (!base) return continuation;
+  if (!continuation) return base;
+  if (looksLikeToolStubResponse(base) && !looksLikeToolStubResponse(continuation)) {
+    return continuation;
+  }
+  if (base.includes(continuation)) {
+    return base;
+  }
+  const maxOverlap = Math.min(base.length, continuation.length);
+  for (let overlap = maxOverlap; overlap > 0; overlap -= 1) {
+    if (base.slice(-overlap) === continuation.slice(0, overlap)) {
+      return `${base}${continuation.slice(overlap)}`;
+    }
+  }
+  return `${base}
+
+${continuation}`;
 };
 var parseTimestampMs = (value) => {
   const timestamp = toStringValue(value).trim();
@@ -8634,22 +8831,29 @@ var buildTurnsFromMessages = (messages, fallbackResponse, callId, timestampMs) =
     const prompt = message.content.trim();
     if (!prompt) continue;
     let response = "";
+    let lastAssistantIndex = -1;
     let reachedNextUser = false;
     for (let lookAhead = index + 1; lookAhead < messages.length; lookAhead += 1) {
       const nextMessage = messages[lookAhead];
       if (!nextMessage) continue;
       if (nextMessage.role === "assistant") {
-        response = nextMessage.content.trim();
-        index = lookAhead;
-        break;
+        const assistantResponse = nextMessage.content.trim();
+        if (assistantResponse) {
+          response = assistantResponse;
+          lastAssistantIndex = lookAhead;
+        }
+        continue;
       }
       if (nextMessage.role === "user") {
         reachedNextUser = true;
         break;
       }
     }
-    if (!response && !reachedNextUser) {
-      response = fallbackResponse;
+    if (lastAssistantIndex >= 0) {
+      index = lastAssistantIndex;
+    }
+    if (!reachedNextUser) {
+      response = pickMoreCompleteResponse(response, fallbackResponse);
     }
     if (!response) continue;
     turns.push({
@@ -8695,6 +8899,17 @@ var buildTranscriptTurnsFromCalls = (calls) => {
             timestampMs
           }
         ];
+      } else if (fallbackResponse && mergedTurns.length > 0) {
+        const previousTurn = mergedTurns[mergedTurns.length - 1];
+        mergedTurns = [
+          ...mergedTurns.slice(0, -1),
+          __spreadProps(__spreadValues({}, previousTurn), {
+            response: mergeContinuationResponse(previousTurn.response, fallbackResponse),
+            callId: callId || previousTurn.callId,
+            timestampMs: Math.max(previousTurn.timestampMs, timestampMs)
+          })
+        ];
+        continue;
       }
     }
     mergedTurns = mergeTranscriptTurns(mergedTurns, callTurns);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hef2024/llmasaservice-ui",
-  "version": "0.25.1",
+  "version": "0.25.2",
   "description": "Prebuilt UI components for LLMAsAService.io",
   "main": "dist/index.js",
   "module": "dist/index.mjs",