@arbidocs/sdk 0.3.79 → 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: "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 "";