@arbidocs/tui 0.3.131 → 0.3.133

package/dist/index.js

@@ -1953,16 +1953,6 @@ function registerAllCommands() {
   registerCommands(historyCommands);
   registerCommands(skillCommands);
 }
-function extractStepDetails(data) {
-  const detail = data.detail;
-  if (!detail || detail.length === 0) return void 0;
-  const d = detail[0];
-  if (!d) return void 0;
-  const result = {};
-  if (d.tool) result.tool = d.tool;
-  if (d.message) result.args = d.message;
-  return result.tool || result.args || result.result ? result : void 0;
-}
 async function streamResponse(tui, response, docNames) {
   tui.chatLog.startAssistant();
   tui.state.activityStatus = "streaming";
@@ -1970,6 +1960,16 @@ async function streamResponse(tui, response, docNames) {
   let accumulated = "";
   let elapsedTime = null;
   let firstToken = true;
+  const updateAnswer = (content) => {
+    if (!content) return;
+    if (firstToken) {
+      firstToken = false;
+      tui.chatLog.clearActiveSteps();
+    }
+    accumulated = content;
+    tui.chatLog.updateAssistant(accumulated);
+    tui.requestRender();
+  };
   let lastStepIndex = null;
   let lastStepLabel = "";
   const callbacks = {
@@ -1979,21 +1979,16 @@ async function streamResponse(tui, response, docNames) {
       }
     },
     onToken: (content) => {
-      if (firstToken) {
-        firstToken = false;
-        tui.chatLog.clearActiveSteps();
-      }
-      accumulated += content;
-      tui.chatLog.updateAssistant(accumulated);
-      tui.requestRender();
+      updateAnswer(accumulated + content);
     },
+    onTextDone: updateAnswer,
     onAgentStep: (data) => {
       const label = sdk.formatAgentStepLabel(data, docNames);
       if (label) {
         if (data.index === lastStepIndex && label === lastStepLabel) return;
         lastStepIndex = data.index;
         lastStepLabel = label;
-        tui.chatLog.addAgentStep(label, extractStepDetails(data));
+        tui.chatLog.addAgentStep(label);
         tui.requestRender();
       }
     },
@@ -2007,6 +2002,7 @@ async function streamResponse(tui, response, docNames) {
   };
   try {
     const result = await sdk.streamSSE(response, callbacks);
+    if (result.text && result.text !== accumulated) updateAnswer(result.text);
     tui.chatLog.finalizeAssistant();
     tui.lastMetadata = result.metadata ?? null;
     tui.sourcesIndex.recordCitations(result.metadata);