Restrict tool name

daohoangson · Jul 19, 2024 · ec5005a · ec5005a
1 parent 7be3625
commit ec5005a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/core/src/interfaces/ai/tool.ts b/packages/core/src/interfaces/ai/tool.ts
@@ -10,6 +10,6 @@ type ToolHandlerInput<T> = {
 export type Tool<T> = {
   description: string;
   handler: (input: ToolHandlerInput<T>) => Promise<any>;
-  name: string;
+  name: "generate_image" | "overwrite_memory" | "new_thread";
   parametersSchema: z.ZodType<T>;
 };