@arbidocs/sdk 0.3.134 → 0.3.135

package/dist/index.cjs

@@ -4232,6 +4232,12 @@ function extractPageRange(step) {
   }
   return void 0;
 }
+function formatItemLabel(item, docNames) {
+  const arbi = item.arbi;
+  if (arbi?.focus) return resolveDocRefs(arbi.focus, docNames);
+  if (arbi?.label) return arbi.label;
+  return item.name ?? "";
+}
 function formatAgentStepLabel(step, docNames) {
   if (step.step === "goal") {
     return step.label && step.label !== "Goal" ? step.label : "Thinking\u2026";
@@ -4312,7 +4318,11 @@ async function streamSSE(response, callbacks = {}) {
       const data = JSON.parse(raw);
       if (data.delta) {
         text += data.delta;
-        callbacks.onToken?.(data.delta);
+        if (callbacks.onOutputTextDelta) {
+          callbacks.onOutputTextDelta({ item_id: data.item_id ?? "", delta: data.delta });
+        } else {
+          callbacks.onToken?.(data.delta);
+        }
       }
     },
     "response.output_text.done": (raw) => {
@@ -4320,9 +4330,20 @@ async function streamSSE(response, callbacks = {}) {
       text = data.text;
       callbacks.onTextDone?.(data.text);
     },
+    "response.function_call_arguments.delta": (raw) => {
+      const data = JSON.parse(raw);
+      callbacks.onFunctionCallArgumentsDelta?.(data);
+    },
     "response.output_item.done": (raw) => {
       const data = JSON.parse(raw);
+      const item = data.item;
+      if (item?.type === "function_call" || item?.type === "reasoning") {
+        const label = formatItemLabel(item);
+        if (label) agentSteps.push(label);
+        if (item.type === "function_call") toolCallCount += 1;
+      }
       callbacks.onOutputItemDone?.(data);
+      if (data.t != null) callbacks.onElapsedTime?.(data.t);
     },
     "response.completed": (raw) => {
       const data = JSON.parse(raw);
@@ -4361,27 +4382,6 @@ async function streamSSE(response, callbacks = {}) {
       }
       callbacks.onMessageQueued?.(data);
     },
-    "arbi.agent_step": (raw) => {
-      const data = JSON.parse(raw);
-      const label = formatAgentStepLabel(data);
-      if (label) agentSteps.push(label);
-      const detail = data.detail;
-      if (detail && Array.isArray(detail)) {
-        toolCallCount += detail.filter((d2) => d2.tool).length;
-      }
-      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);
-      if (data.t != null) callbacks.onElapsedTime?.(data.t);
-    },
     "arbi.user_message": (raw) => {
       const data = JSON.parse(raw);
       userMessage = data;
@@ -6456,7 +6456,7 @@ var Arbi = class {
       const callbacks = {
         onToken: options.onToken,
         onStreamStart: options.onStreamStart,
-        onAgentStep: options.onAgentStep,
+        onOutputItemDone: options.onOutputItemDone,
         onError: options.onError,
         onUserMessage: options.onUserMessage,
         onMetadata: options.onMetadata,
@@ -7824,6 +7824,7 @@ exports.files = files_exports;
 exports.filterSkills = filterSkills;
 exports.formatAgentStepLabel = formatAgentStepLabel;
 exports.formatFileSize = formatFileSize;
+exports.formatItemLabel = formatItemLabel;
 exports.formatStreamSummary = formatStreamSummary;
 exports.formatUserName = formatUserName;
 exports.formatWorkspaceChoices = formatWorkspaceChoices;