feat: useAssistantTool (#219)

packages/react/src/experimental.ts

@@ -8,3 +8,4 @@ export type { ModelConfigProvider } from "./utils/ModelConfigTypes";
 
 export * from "./context";
 export { useAssistantInstructions } from "./model-config/useAssistantInstructions";
+export { useAssistantTool } from "./model-config/useAssistantTool";

packages/react/src/model-config/useAssistantInstructions.tsx

@@ -5,16 +5,16 @@ import { useAssistantContext } from "../context/AssistantContext";
 
 export const useAssistantInstructions = (instruction: string) => {
   const { useModelConfig } = useAssistantContext();
-  const addContextProvider = useModelConfig(
+  const registerModelConfigProvider = useModelConfig(
     (s) => s.registerModelConfigProvider,
   );
   useEffect(
     () =>
-      addContextProvider(() => {
+      registerModelConfigProvider(() => {
         return {
           system: instruction,
         };
       }),
-    [addContextProvider, instruction],
+    [registerModelConfigProvider, instruction],
   );
 };

packages/react/src/model-config/useAssistantTool.tsx

@@ -0,0 +1,23 @@
+"use client";
+
+import { useEffect } from "react";
+import { useAssistantContext } from "../context/AssistantContext";
+import type { Tool } from "../utils/ModelConfigTypes";
+
+export const useAssistantTool = <T,>(tool: Tool<T>) => {
+  const { useModelConfig } = useAssistantContext();
+  const registerModelConfigProvider = useModelConfig(
+    (s) => s.registerModelConfigProvider,
+  );
+  useEffect(
+    () =>
+      registerModelConfigProvider(() => {
+        return {
+          tools: {
+            [tool.name]: tool,
+          },
+        };
+      }),
+    [registerModelConfigProvider, tool],
+  );
+};

packages/react/src/utils/ModelConfigTypes.ts

@@ -2,6 +2,7 @@
 import type { z } from "zod";
 
 export type Tool<TArgs> = {
+  name: string;
   description: string;
   parameters: z.ZodSchema<TArgs>;
   execute: (args: TArgs) => Promise<unknown>; // TODO return type