@arbidocs/sdk 0.3.78 → 0.3.80

package/dist/index.cjs

@@ -4188,26 +4188,66 @@ async function resolveWorkspace(store, workspaceOpt) {
 
 // src/sse.ts
 var LIFECYCLE_LABELS_FALLBACK = {
+  goal: "Thinking\u2026",
   evaluation: "Evaluating results",
-  answering: "Writing answer",
+  answering: "Answered",
   reviewing: "Reviewing answer",
   planning: "Planning",
   tool_progress: "Working"
 };
-function formatAgentStepLabel(step) {
-  if (step.focus) return step.focus;
-  if (step.label) return step.label;
+function resolveDocRefs(text, docNames) {
+  if (!docNames) return text;
+  return text.replace(/\{(doc-[^}]+)\}/g, (_match, id) => docNames[id] ?? id);
+}
+function resolveStepDocName(step, docNames) {
+  if (!docNames) return void 0;
+  const detail = step.detail;
+  if (!detail || detail.length === 0) return void 0;
+  for (const d2 of detail) {
+    const id = d2.doc_ext_id ?? d2.document_ext_id;
+    if (typeof id === "string") return docNames[id] ?? id;
+  }
+  return void 0;
+}
+function extractPageRange(step) {
+  const detail = step.detail;
+  if (!detail || detail.length === 0) return void 0;
+  for (const d2 of detail) {
+    const summary = typeof d2.summary === "string" ? d2.summary : void 0;
+    if (!summary) continue;
+    const match = summary.match(/\bpp?\.\s*(\d[\d.,\s–—-]*)/);
+    if (!match) continue;
+    const floored = match[1].replace(/\d+(?:\.\d+)?/g, (n2) => String(Math.floor(parseFloat(n2)))).replace(/(\d+)\s*[–—-]\s*\1(?!\d)/g, "$1");
+    const items = [];
+    for (const raw of floored.split(",")) {
+      const item = raw.trim();
+      if (item && !items.includes(item)) items.push(item);
+    }
+    if (items.length > 0) return `pp. ${items.join(", ")}`;
+  }
+  return void 0;
+}
+function formatAgentStepLabel(step, docNames) {
+  if (step.step === "goal") {
+    return step.label && step.label !== "Goal" ? step.label : "Thinking\u2026";
+  }
+  if (step.focus) return resolveDocRefs(step.focus, docNames);
+  const docName = resolveStepDocName(step, docNames);
+  const pages = docName ? extractPageRange(step) : void 0;
+  const suffix = docName ? ` \u2014 ${docName}${pages ? ` \xB7 ${pages}` : ""}` : "";
+  if (step.label) return `${step.label}${suffix}`;
   const detail = step.detail;
   if (step.step === "tool_progress" && detail && detail.length > 0) {
     const label = detail[0].label || LIFECYCLE_LABELS_FALLBACK.tool_progress;
+    if (suffix) return `${label}${suffix}`;
     const message = detail[0].message;
-    return message ? `${label}: ${message}` : label;
+    return message ? `${label}: ${resolveDocRefs(message, docNames)}` : label;
   }
   if (step.step) {
     return LIFECYCLE_LABELS_FALLBACK[step.step] || step.step;
   }
   if (detail && detail.length > 0) {
-    if (detail[0].label) return detail[0].label;
+    if (detail[0].label) return `${detail[0].label}${suffix}`;
     if (detail[0].tool) return detail[0].tool;
   }
   return "";
@@ -4321,6 +4361,11 @@ async function streamSSE(response, callbacks = {}) {
       callbacks.onAgentStep?.(data);
       if (data.t != null) callbacks.onElapsedTime?.(data.t);
     },
+    "arbi.agent_step_delta": (raw) => {
+      const data = JSON.parse(raw);
+      callbacks.onAgentStepDelta?.(data);
+      if (data.t != null) callbacks.onElapsedTime?.(data.t);
+    },
     "arbi.user_message": (raw) => {
       const data = JSON.parse(raw);
       userMessage = data;
@@ -4346,16 +4391,21 @@ async function streamSSE(response, callbacks = {}) {
       errors.push(message);
       callbacks.onArbiError?.(data);
       callbacks.onError?.(message);
-    },
-    "response.complete": () => {
-      callbacks.onComplete?.();
     }
   };
   const processEvents = (events) => {
     for (const { event, data } of events) {
       try {
+        if (event === "reasoning") {
+          callbacks.onReasoningDelta?.(data);
+          continue;
+        }
         const handler = eventHandlers[event];
-        if (handler) handler(data);
+        if (handler) {
+          handler(data);
+        } else {
+          console.warn(`[arbi-sse] unhandled SSE event: ${event}`);
+        }
       } catch {
       }
     }