toolz

1 files changed, 28 insertions, 0 deletions

diff --git a/js/src/tool/tool.ts b/js/src/tool/tool.ts
new file mode 100644
index 000000000..08ab935b3
--- /dev/null
+++ b/js/src/tool/tool.ts
@@ -0,0 +1,28 @@
+import { type Tool, tool as AITool } from "ai";
+import { Log } from "../util/log";
+
+const log = Log.create({ service: "tool" });
+
+export function tool<Params, Result>(
+  tool: Tool<Params, Result> & {
+    name: string;
+  },
+) {
+  return {
+    [tool.name]: AITool({
+      ...tool,
+      execute: async (params, opts) => {
+        log.info("invoking", {
+          id: opts.toolCallId,
+          name: tool.name,
+          ...params,
+        });
+        try {
+          return tool.execute!(params, opts);
+        } catch (e: any) {
+          return "An error occurred: " + e.toString();
+        }
+      },
+    }),
+  };
+}