@ash-cloud/ash-ui 0.0.4 → 0.0.5

package/dist/index.cjs

@@ -1742,6 +1742,9 @@ function isToolCallEntry(entry) {
 function isErrorEntry(entry) {
   return entry.type === "error";
 }
+function isWidgetEntry(entry) {
+  return entry.type === "widget";
+}
 var DEFAULT_DISPLAY_CONFIG = {
   mode: "inline",
   breakEveryNToolCalls: 0,
@@ -1797,10 +1800,25 @@ function MessageList({
   streamingContent,
   displayConfig: displayConfigProp,
   onOptionSelect,
+  renderWidget,
+  onWidgetAction,
   className
 }) {
   const contextConfig = useDisplayConfig();
   const config = displayConfigProp || contextConfig;
+  const createWidgetActionHandler = react.useCallback(
+    (entryId, widgetType) => {
+      if (!onWidgetAction) return void 0;
+      return (action) => {
+        onWidgetAction({
+          ...action,
+          entryId,
+          widgetType
+        });
+      };
+    },
+    [onWidgetAction]
+  );
   const groupedEntries = react.useMemo(() => {
     if (config.mode === "inline") {
       return entries.map((entry) => ({
@@ -1814,7 +1832,20 @@ function MessageList({
   return /* @__PURE__ */ jsxRuntime.jsxs("div", { className: cn("flex-1 overflow-y-auto p-4 space-y-4 ash-scrollbar", className), children: [
     groupedEntries.map((groupedEntry) => {
       if (groupedEntry.type === "single") {
-        return /* @__PURE__ */ jsxRuntime.jsx(MessageEntry, { entry: groupedEntry.entry, onOptionSelect }, groupedEntry.entry.id);
+        const entry = groupedEntry.entry;
+        if (entry.entryType.type === "widget" && renderWidget) {
+          const widgetEntry = entry.entryType;
+          const widgetContent = renderWidget({
+            entry,
+            widgetType: widgetEntry.widgetType,
+            widgetData: widgetEntry.widgetData,
+            onAction: createWidgetActionHandler(entry.id, widgetEntry.widgetType)
+          });
+          if (widgetContent !== null) {
+            return /* @__PURE__ */ jsxRuntime.jsx("div", { className: "ash-animate-fade-in", children: widgetContent }, entry.id);
+          }
+        }
+        return /* @__PURE__ */ jsxRuntime.jsx(MessageEntry, { entry, onOptionSelect }, entry.id);
       }
       const toolCalls = extractToolCallsFromGroup(groupedEntry.entries);
       return /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex gap-3 ash-animate-fade-in", children: [
@@ -3036,6 +3067,7 @@ exports.isTodoWriteAction = isTodoWriteAction;
 exports.isToolCallEntry = isToolCallEntry;
 exports.isWebFetchAction = isWebFetchAction;
 exports.isWebSearchAction = isWebSearchAction;
+exports.isWidgetEntry = isWidgetEntry;
 exports.keyframes = keyframes;
 exports.keyframesCss = keyframesCss;
 exports.mapToolToActionType = mapToolToActionType;