@arbidocs/sdk 0.3.132 → 0.3.134

package/dist/index.cjs

@@ -4187,14 +4187,10 @@ async function resolveWorkspace(store, workspaceOpt) {
 }
 
 // src/sse.ts
-var LIFECYCLE_LABELS_FALLBACK = {
-  goal: "Thinking\u2026",
-  evaluation: "Evaluating results",
-  answering: "Answered",
-  interjection: "Your message",
-  planning: "Planning",
-  tool_progress: "Working"
-};
+function humanizeStep(identifier) {
+  const spaced = identifier.replace(/[_-]+/g, " ").trim();
+  return spaced.charAt(0).toUpperCase() + spaced.slice(1);
+}
 function buildDocNameMap(docs) {
   const map = {};
   for (const doc of docs) {
@@ -4247,13 +4243,13 @@ function formatAgentStepLabel(step, docNames) {
   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;
+    const label = detail[0].label || detail[0].tool && humanizeStep(detail[0].tool) || humanizeStep(step.step);
     if (suffix) return `${label}${suffix}`;
     const message = detail[0].message;
     return message ? `${label}: ${resolveDocRefs(message, docNames)}` : label;
   }
   if (step.step) {
-    return LIFECYCLE_LABELS_FALLBACK[step.step] || step.step;
+    return humanizeStep(step.step);
   }
   if (detail && detail.length > 0) {
     if (detail[0].label) return `${detail[0].label}${suffix}`;
@@ -4376,6 +4372,11 @@ async function streamSSE(response, callbacks = {}) {
       callbacks.onAgentStep?.(data);
       if (data.t != null) callbacks.onElapsedTime?.(data.t);
     },
+    "arbi.generating": (raw) => {
+      const data = JSON.parse(raw);
+      callbacks.onGenerating?.(data);
+      if (data.t != null) callbacks.onElapsedTime?.(data.t);
+    },
     "arbi.agent_step_delta": (raw) => {
       const data = JSON.parse(raw);
       callbacks.onAgentStepDelta?.(data);