@assistant-ui/react 0.1.7 → 0.1.8

package/dist/index.js

@@ -934,10 +934,8 @@ var defaultComponents = {
   UI: () => /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(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__ */ (0, import_jsx_runtime15.jsx)(Render, { ...props });
     }
@@ -951,6 +949,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;
@@ -963,7 +963,8 @@ var MessageContentPartComponent = ({
       return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(UI, { part, status });
     case "tool-call": {
       const Tool = by_name[part.toolName] || Fallback;
-      return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(Tool, { part, status });
+      const addResult = (result) => addToolResult(part.toolCallId, result);
+      return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(Tool, { part, status, addResult });
     }
     default:
       throw new Error(`Unknown content part type: ${type}`);
@@ -1427,6 +1428,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
@@ -1474,18 +1478,18 @@ var makeAssistantModelConfigStore = () => (0, import_zustand4.create)(() => {
   };
 });
 
-// src/context/stores/AssistantToolRenderers.ts
+// src/context/stores/AssistantToolUIs.ts
 var import_zustand5 = require("zustand");
-var makeAssistantToolRenderersStore = () => (0, import_zustand5.create)((set) => {
+var makeAssistantToolUIsStore = () => (0, import_zustand5.create)((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 = [];
@@ -1540,7 +1544,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({
@@ -1617,8 +1622,8 @@ var AssistantProvider = ({ children, runtime }) => {
   });
   const [context] = (0, import_react36.useState)(() => {
     const useModelConfig = makeAssistantModelConfigStore();
-    const useToolRenderers = makeAssistantToolRenderersStore();
-    return { useModelConfig, useToolRenderers };
+    const useToolUIs = makeAssistantToolUIsStore();
+    return { useModelConfig, useToolUIs };
   });
   const getModelCOnfig = context.useModelConfig((c) => c.getModelConfig);
   (0, import_react36.useEffect)(() => {