@apteva/apteva-kit 0.1.33 → 0.1.35

package/dist/index.mjs

@@ -1325,23 +1325,48 @@ function Message({ message, onAction, enableWidgets, onWidgetRender }) {
       ] });
     }
     if (contentSegments && contentSegments.length > 0) {
-      return /* @__PURE__ */ jsx11("div", { children: contentSegments.map((segment, index) => {
-        if (segment.type === "text") {
-          return segment.content ? /* @__PURE__ */ jsx11("div", { children: renderTextContent(segment.content) }, `text-${index}`) : null;
-        } else if (segment.type === "tool") {
-          return /* @__PURE__ */ jsx11("div", { className: "my-2", children: /* @__PURE__ */ jsx11(
+      return null;
+    }
+    return renderTextContent(message.content);
+  };
+  const renderSegmentedContent = () => {
+    if (!contentSegments || contentSegments.length === 0) {
+      return null;
+    }
+    const elements = [];
+    contentSegments.forEach((segment, index) => {
+      if (segment.type === "text" && segment.content) {
+        elements.push(
+          /* @__PURE__ */ jsx11(
+            "div",
+            {
+              className: "apteva-message-bubble apteva-message-assistant",
+              children: /* @__PURE__ */ jsx11("div", { className: "apteva-message-content-assistant", children: renderTextContent(segment.content) })
+            },
+            `text-${index}`
+          )
+        );
+      } else if (segment.type === "tool") {
+        elements.push(
+          /* @__PURE__ */ jsx11("div", { className: "apteva-tool-call-standalone", children: /* @__PURE__ */ jsx11(
             ToolCall,
             {
               name: segment.name,
               status: segment.result !== void 0 ? "completed" : "running"
             }
-          ) }, segment.id);
-        }
-        return null;
-      }) });
-    }
-    return renderTextContent(message.content);
+          ) }, segment.id)
+        );
+      }
+    });
+    return elements;
   };
+  if (!isUser && contentSegments && contentSegments.length > 0) {
+    return /* @__PURE__ */ jsxs8("div", { className: "apteva-message-segmented", children: [
+      renderSegmentedContent(),
+      message.widgets && message.widgets.length > 0 && /* @__PURE__ */ jsx11("div", { className: "apteva-message-widgets", children: /* @__PURE__ */ jsx11(Widgets, { widgets: message.widgets, onAction, layout: "stack" }) }),
+      /* @__PURE__ */ jsx11("div", { className: "apteva-message-timestamp apteva-message-timestamp-assistant", suppressHydrationWarning: true, children: message.timestamp.toLocaleTimeString([], { hour: "2-digit", minute: "2-digit" }) })
+    ] });
+  }
   return /* @__PURE__ */ jsxs8(
     "div",
     {
@@ -2288,6 +2313,7 @@ var Chat = forwardRef(function Chat2({
   onFileUpload,
   onComplete,
   onError,
+  onToolCall,
   onToolResult,
   // UI
   placeholder,
@@ -2480,6 +2506,7 @@ ${widgetContext}` : widgetContext;
                   contentSegments.push({ type: "tool", id: chunk.tool_id, name: chunk.tool_name });
                   toolInputBuffer = "";
                   setChatToolName(chunk.tool_name);
+                  onToolCall?.(chunk.tool_name, chunk.tool_id);
                   updateMessage();
                 }
                 break;
@@ -2699,6 +2726,7 @@ ${commandInstruction}` : commandInstruction;
               } else if (chunk.type === "tool_call" && chunk.tool_name) {
                 lastToolName = chunk.tool_name;
                 setCurrentToolName(chunk.tool_name);
+                onToolCall?.(chunk.tool_name, chunk.tool_id || "");
                 accumulatedContent = "";
                 setStreamedContent("");
               } else if (chunk.type === "tool_result") {