@apteva/apteva-kit 0.1.72 → 0.1.77

package/dist/index.mjs

@@ -606,6 +606,14 @@ function findMatchingBracket(text, startIndex) {
   }
   return -1;
 }
+function findOpeningBracket(text) {
+  const squareIndex = text.indexOf("[");
+  const curlyIndex = text.indexOf("{");
+  if (squareIndex === -1 && curlyIndex === -1) return null;
+  if (squareIndex === -1) return { index: curlyIndex, char: "{" };
+  if (curlyIndex === -1) return { index: squareIndex, char: "[" };
+  return squareIndex < curlyIndex ? { index: squareIndex, char: "[" } : { index: curlyIndex, char: "{" };
+}
 function parseWidgetsFromText(text) {
   const segments = [];
   let hasWidgets = false;
@@ -619,13 +627,14 @@ function parseWidgetsFromText(text) {
     const typeMatch = afterStart.match(/^@ui:(\w+)/);
     if (typeMatch) {
       const widgetType = typeMatch[1];
-      const bracketOpenIndex = afterStart.indexOf("[");
-      if (bracketOpenIndex === -1) {
+      const afterType = afterStart.slice(typeMatch[0].length);
+      const bracketInfo = findOpeningBracket(afterType);
+      if (!bracketInfo) {
         processText = text.slice(0, lastWidgetStart);
         pendingWidgetType = widgetType;
         hasPendingWidget = true;
       } else {
-        const fullBracketStart = lastWidgetStart + bracketOpenIndex;
+        const fullBracketStart = lastWidgetStart + typeMatch[0].length + bracketInfo.index;
         const bracketEnd = findMatchingBracket(text, fullBracketStart);
         if (bracketEnd === -1) {
           if (STREAMABLE_WIDGET_TYPES.includes(widgetType)) {
@@ -667,16 +676,22 @@ function parseWidgetsFromText(text) {
   if (hasPendingWidget) {
     processText = processText.replace(/[\s:;\-–—\.]+$/g, "");
   }
-  const startPattern = /@ui:(\w+)\[/g;
+  const startPattern = /@ui:(\w+)([\[{])/g;
   let match;
   while ((match = startPattern.exec(processText)) !== null) {
     const widgetType = match[1];
+    const openingBracket = match[2];
     const bracketStart = match.index + match[0].length - 1;
     const bracketEnd = findMatchingBracket(processText, bracketStart);
     if (bracketEnd === -1) {
       continue;
     }
-    const jsonContent = processText.slice(bracketStart + 1, bracketEnd);
+    let jsonContent;
+    if (openingBracket === "[") {
+      jsonContent = processText.slice(bracketStart + 1, bracketEnd);
+    } else {
+      jsonContent = processText.slice(bracketStart, bracketEnd + 1);
+    }
     if (match.index > currentIndex) {
       const textContent = processText.slice(currentIndex, match.index).trim();
       if (textContent) {
@@ -1520,19 +1535,14 @@ function MarkdownContent({ content, className = "" }) {
 import { jsx as jsx11, jsxs as jsxs8 } from "react/jsx-runtime";
 function ToolCall({ name, status, isReceiving = false, inputLength = 0 }) {
   if (status === "preparing") {
-    return /* @__PURE__ */ jsxs8("div", { className: `apteva-tool-card apteva-tool-card-preparing ${isReceiving ? "apteva-tool-receiving" : ""}`, children: [
+    return /* @__PURE__ */ jsxs8("div", { className: "apteva-tool-card apteva-tool-card-preparing", children: [
       /* @__PURE__ */ jsxs8("svg", { className: "apteva-tool-icon apteva-tool-icon-spin", fill: "none", viewBox: "0 0 24 24", children: [
         /* @__PURE__ */ jsx11("circle", { className: "apteva-tool-spinner-track", cx: "12", cy: "12", r: "10", stroke: "currentColor", strokeWidth: "4" }),
         /* @__PURE__ */ jsx11("path", { className: "apteva-tool-spinner-fill", fill: "currentColor", d: "M4 12a8 8 0 018-8V0C5.373 0 0 5.373 0 12h4z" })
       ] }),
       /* @__PURE__ */ jsxs8("span", { className: "apteva-tool-label", children: [
         /* @__PURE__ */ jsx11("strong", { children: name }),
-        /* @__PURE__ */ jsx11("span", { className: "apteva-tool-status-text", children: " preparing" }),
-        inputLength > 0 && /* @__PURE__ */ jsxs8("span", { className: "apteva-tool-char-count", children: [
-          " (",
-          inputLength,
-          " chars)"
-        ] })
+        /* @__PURE__ */ jsx11("span", { className: "apteva-tool-status-text", children: " preparing..." })
       ] })
     ] });
   }
@@ -2871,8 +2881,8 @@ ${widgetContext}` : widgetContext;
         let currentTextBuffer = "";
         let accumulatedWidgets = [];
         let responseThreadId = currentThreadId;
-        let toolInputBuffer = "";
-        let receivingTimeout = null;
+        const toolInputBuffers = {};
+        const receivingTimeouts = {};
         const streamingMessageId = `msg-${Date.now()}`;
         const updateMessage = () => {
           const segments = [...contentSegments];
@@ -2950,7 +2960,7 @@ ${widgetContext}` : widgetContext;
                     currentTextBuffer = "";
                   }
                   contentSegments.push({ type: "tool", id: chunk.tool_id, name: displayName, status: "preparing" });
-                  toolInputBuffer = "";
+                  toolInputBuffers[chunk.tool_id] = "";
                   setChatToolName(displayName);
                   onToolCall?.(chunk.tool_name, chunk.tool_id);
                   updateMessage();
@@ -2958,14 +2968,18 @@ ${widgetContext}` : widgetContext;
                 break;
               case "tool_input_delta":
                 if (chunk.tool_id && chunk.content) {
-                  toolInputBuffer += chunk.content;
-                  const toolSegment = contentSegments.find((s) => s.type === "tool" && s.id === chunk.tool_id);
+                  const toolId = chunk.tool_id;
+                  if (toolInputBuffers[toolId] === void 0) {
+                    toolInputBuffers[toolId] = "";
+                  }
+                  toolInputBuffers[toolId] += chunk.content;
+                  const toolSegment = contentSegments.find((s) => s.type === "tool" && s.id === toolId);
                   if (toolSegment) {
                     toolSegment.isReceiving = true;
-                    toolSegment.inputLength = toolInputBuffer.length;
+                    toolSegment.inputLength = toolInputBuffers[toolId].length;
                     updateMessage();
-                    if (receivingTimeout) clearTimeout(receivingTimeout);
-                    receivingTimeout = setTimeout(() => {
+                    if (receivingTimeouts[toolId]) clearTimeout(receivingTimeouts[toolId]);
+                    receivingTimeouts[toolId] = setTimeout(() => {
                       if (toolSegment.status === "preparing") {
                         toolSegment.isReceiving = false;
                         updateMessage();