@assistant-ui/react 0.5.45 → 0.5.46

package/dist/index.mjs

@@ -2156,6 +2156,7 @@ var makeAssistantTool = (tool) => {
     useAssistantTool(tool);
     return null;
   };
+  Tool.unstable_tool = tool;
   return Tool;
 };
 
@@ -2177,6 +2178,7 @@ var makeAssistantToolUI = (tool) => {
     useAssistantToolUI(tool);
     return null;
   };
+  ToolUI.unstable_tool = tool;
   return ToolUI;
 };
 
@@ -3033,21 +3035,22 @@ ContentPartPrimitiveInProgress.displayName = "ContentPartPrimitive.InProgress";
 
 // src/primitives/message/MessageContent.tsx
 import { jsx as jsx22, jsxs as jsxs3 } from "react/jsx-runtime";
+var ToolUIDisplay = ({
+  UI,
+  ...props
+}) => {
+  const { useToolUIs } = useAssistantContext();
+  const Render = useToolUIs((s) => s.getToolUI(props.part.toolName)) ?? UI;
+  if (!Render) return null;
+  return /* @__PURE__ */ jsx22(Render, { ...props });
+};
 var defaultComponents = {
   Text: () => /* @__PURE__ */ jsxs3("p", { style: { whiteSpace: "pre-line" }, children: [
     /* @__PURE__ */ jsx22(ContentPartPrimitiveText, {}),
     /* @__PURE__ */ jsx22(ContentPartPrimitiveInProgress, { children: /* @__PURE__ */ jsx22("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })
   ] }),
   Image: () => /* @__PURE__ */ jsx22(ContentPartPrimitiveImage, {}),
-  UI: () => /* @__PURE__ */ jsx22(ContentPartPrimitiveDisplay, {}),
-  tools: {
-    Fallback: (props) => {
-      const { useToolUIs } = useAssistantContext();
-      const Render = useToolUIs((s) => s.getToolUI(props.part.toolName));
-      if (!Render) return null;
-      return /* @__PURE__ */ jsx22(Render, { ...props });
-    }
-  }
+  UI: () => /* @__PURE__ */ jsx22(ContentPartPrimitiveDisplay, {})
 };
 var MessageContentPartComponent = ({
   components: {
@@ -3055,7 +3058,7 @@ var MessageContentPartComponent = ({
     Text: Text2 = defaultComponents.Text,
     Image: Image2 = defaultComponents.Image,
     UI = defaultComponents.UI,
-    tools: { by_name = {}, Fallback: Fallback2 = defaultComponents.tools.Fallback } = {}
+    tools: { by_name = {}, Fallback: Fallback2 = void 0 } = {}
   } = {}
 }) => {
   const { useThreadActions } = useThreadContext();
@@ -3086,7 +3089,15 @@ var MessageContentPartComponent = ({
         toolCallId: part.toolCallId,
         result
       });
-      return /* @__PURE__ */ jsx22(Tool, { part, status, addResult });
+      return /* @__PURE__ */ jsx22(
+        ToolUIDisplay,
+        {
+          UI: Tool,
+          part,
+          status,
+          addResult
+        }
+      );
     }
     default:
       const unhandledType = type;
@@ -3955,13 +3966,22 @@ var AssistantMessageContentWrapper = withDefaults("div", {
   className: "aui-assistant-message-content"
 });
 var AssistantMessageContent = forwardRef22(({ components: componentsProp, ...rest }, ref) => {
-  const { assistantMessage: { components = {} } = {} } = useThreadConfig();
+  const { tools, assistantMessage: { components = {} } = {} } = useThreadConfig();
   return /* @__PURE__ */ jsx36(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ jsx36(
     message_exports.Content,
     {
       components: {
         ...componentsProp,
-        Text: componentsProp?.Text ?? components.Text ?? content_part_default.Text
+        Text: componentsProp?.Text ?? components.Text ?? content_part_default.Text,
+        tools: {
+          by_name: !tools ? void 0 : Object.fromEntries(
+            tools.map((t) => [
+              t.unstable_tool.toolName,
+              t.unstable_tool.render
+            ])
+          ),
+          Fallback: components.ToolFallback
+        }
       }
     }
   ) });