feat: useAssistantTool allow disabling tools (#1277)

assistant-ui · Dec 28, 2024 · 2c7dec0 · 2c7dec0
1 parent 267c521
commit 2c7dec0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/.changeset/dull-owls-work.md b/.changeset/dull-owls-work.md
@@ -0,0 +1,5 @@
+---
+"@assistant-ui/react": patch
+---
+
+feat: useAssistantTool allow disabling tools
diff --git a/packages/react/src/model-config/useAssistantTool.tsx b/packages/react/src/model-config/useAssistantTool.tsx
@@ -14,6 +14,7 @@ export type AssistantToolProps<
 > = Tool<TArgs, TResult> & {
   toolName: string;
   render?: ToolCallContentPartComponent<TArgs, TResult> | undefined;
+  disabled?: boolean | undefined;
 };
 
 export const useAssistantTool = <
@@ -32,7 +33,9 @@ export const useAssistantTool = <
   }, [toolUIsStore, tool.toolName, tool.render]);
 
   useEffect(() => {
-    const { toolName, render, ...rest } = tool;
+    const { toolName, render, disabled, ...rest } = tool;
+    if (disabled) return;
+
     const config = {
       tools: {
         [toolName]: rest,