@assistant-ui/react 0.1.7 → 0.1.8

package/dist/index.mjs

@@ -9,7 +9,7 @@ import {
   useContentPartContext,
   useMessageContext,
   useThreadContext
-} from "./chunk-XVZ2GVQM.mjs";
+} from "./chunk-RITM2IUH.mjs";
 
 // src/actions/useCopyMessage.tsx
 import { useCallback } from "react";
@@ -841,10 +841,8 @@ var defaultComponents = {
   UI: () => /* @__PURE__ */ jsx15(ContentPartDisplay, {}),
   tools: {
     Fallback: (props) => {
-      const { useToolRenderers } = useAssistantContext();
-      const Render = useToolRenderers(
-        (s) => s.getToolRenderer(props.part.toolName)
-      );
+      const { useToolUIs } = useAssistantContext();
+      const Render = useToolUIs((s) => s.getToolUI(props.part.toolName));
       if (!Render) return null;
       return /* @__PURE__ */ jsx15(Render, { ...props });
     }
@@ -858,6 +856,8 @@ var MessageContentPartComponent = ({
     tools: { by_name = {}, Fallback = defaultComponents.tools.Fallback } = {}
   } = {}
 }) => {
+  const { useThread } = useThreadContext();
+  const addToolResult = useThread((t) => t.addToolResult);
   const { useContentPart } = useContentPartContext();
   const { part, status } = useContentPart();
   const type = part.type;
@@ -870,7 +870,8 @@ var MessageContentPartComponent = ({
       return /* @__PURE__ */ jsx15(UI, { part, status });
     case "tool-call": {
       const Tool = by_name[part.toolName] || Fallback;
-      return /* @__PURE__ */ jsx15(Tool, { part, status });
+      const addResult = (result) => addToolResult(part.toolCallId, result);
+      return /* @__PURE__ */ jsx15(Tool, { part, status, addResult });
     }
     default:
       throw new Error(`Unknown content part type: ${type}`);
@@ -1337,6 +1338,9 @@ var LocalRuntime = class {
     this._configProviders.add(provider);
     return () => this._configProviders.delete(provider);
   }
+  addToolResult() {
+    throw new Error("LocalRuntime does not yet support tool results");
+  }
 };
 
 // src/runtime/local/useLocalRuntime.tsx
@@ -1384,18 +1388,18 @@ var makeAssistantModelConfigStore = () => create4(() => {
   };
 });
 
-// src/context/stores/AssistantToolRenderers.ts
+// src/context/stores/AssistantToolUIs.ts
 import { create as create5 } from "zustand";
-var makeAssistantToolRenderersStore = () => create5((set) => {
+var makeAssistantToolUIsStore = () => create5((set) => {
   const renderers = /* @__PURE__ */ new Map();
   return {
-    getToolRenderer: (name) => {
+    getToolUI: (name) => {
       const arr = renderers.get(name);
       const last = arr?.at(-1);
       if (last) return last;
       return null;
     },
-    setToolRenderer: (name, render) => {
+    setToolUI: (name, render) => {
       let arr = renderers.get(name);
       if (!arr) {
         arr = [];
@@ -1450,7 +1454,8 @@ var makeThreadStore = (runtimeRef) => {
     switchToBranch: (branchId) => runtimeRef.current.switchToBranch(branchId),
     startRun: (parentId) => runtimeRef.current.startRun(parentId),
     append: (message) => runtimeRef.current.append(message),
-    cancelRun: () => runtimeRef.current.cancelRun()
+    cancelRun: () => runtimeRef.current.cancelRun(),
+    addToolResult: (toolCallId, result) => runtimeRef.current.addToolResult(toolCallId, result)
   }));
   const onRuntimeUpdate = () => {
     useThread.setState({
@@ -1527,8 +1532,8 @@ var AssistantProvider = ({ children, runtime }) => {
   });
   const [context] = useState5(() => {
     const useModelConfig = makeAssistantModelConfigStore();
-    const useToolRenderers = makeAssistantToolRenderersStore();
-    return { useModelConfig, useToolRenderers };
+    const useToolUIs = makeAssistantToolUIsStore();
+    return { useModelConfig, useToolUIs };
   });
   const getModelCOnfig = context.useModelConfig((c) => c.getModelConfig);
   useEffect7(() => {