Tool factories

mcp/selectMode.ts

@@ -1,5 +1,6 @@
 import { type } from "arktype";
-import { contextualize, tool } from "./shared.ts";
+import type { Context } from "../main.ts";
+import { execute, tool } from "./shared.ts";
 
 export const SelectMode = type({
   modeName: type.string.describe(
@@ -7,26 +8,28 @@ export const SelectMode = type({
   ),
 });
 
-export const SelectModeTool = tool({
-  name: "select_mode",
-  description:
-    "Select a mode and get its detailed prompt instructions. Call this first to determine which mode to use based on the request.",
-  parameters: SelectMode,
-  execute: contextualize(async ({ modeName }, ctx) => {
-    const selectedMode = ctx.modes.find((m) => m.name.toLowerCase() === modeName.toLowerCase());
+export function SelectModeTool(ctx: Context) {
+  return tool({
+    name: "select_mode",
+    description:
+      "Select a mode and get its detailed prompt instructions. Call this first to determine which mode to use based on the request.",
+    parameters: SelectMode,
+    execute: execute(ctx, async ({ modeName }) => {
+      const selectedMode = ctx.modes.find((m) => m.name.toLowerCase() === modeName.toLowerCase());
+
+      if (!selectedMode) {
+        const availableModes = ctx.modes.map((m) => m.name).join(", ");
+        return {
+          error: `Mode "${modeName}" not found. Available modes: ${availableModes}`,
+          availableModes: ctx.modes.map((m) => ({ name: m.name, description: m.description })),
+        };
+      }
 
-    if (!selectedMode) {
-      const availableModes = ctx.modes.map((m) => m.name).join(", ");
       return {
-        error: `Mode "${modeName}" not found. Available modes: ${availableModes}`,
-        availableModes: ctx.modes.map((m) => ({ name: m.name, description: m.description })),
+        modeName: selectedMode.name,
+        description: selectedMode.description,
+        prompt: selectedMode.prompt,
       };
-    }
-
-    return {
-      modeName: selectedMode.name,
-      description: selectedMode.description,
-      prompt: selectedMode.prompt,
-    };
-  }),
-});
+    }),
+  });
+}