David Blass
56 lines
1.5 KiB
TypeScript
56 lines
1.5 KiB
TypeScript
import { type } from "arktype";
|
|
import type { Mode } from "../modes.ts";
|
|
import { contextualize, tool } from "./shared.ts";
|
|
|
|
// Get modes from environment variable (set by createMcpConfigs)
|
|
function getModes(): Mode[] {
|
|
const modesJson = process.env.PULLFROG_MODES;
|
|
if (modesJson) {
|
|
try {
|
|
return JSON.parse(modesJson);
|
|
} catch {
|
|
return [];
|
|
}
|
|
}
|
|
return [];
|
|
}
|
|
|
|
export const SelectMode = type({
|
|
modeName: type.string.describe(
|
|
"the name of the mode to select (e.g., 'Plan', 'Build', 'Review', 'Prompt')"
|
|
),
|
|
});
|
|
|
|
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 }) => {
|
|
const allModes = getModes();
|
|
|
|
if (allModes.length === 0) {
|
|
return {
|
|
error:
|
|
"No modes available. Modes must be provided via PULLFROG_MODES environment variable.",
|
|
};
|
|
}
|
|
|
|
const selectedMode = allModes.find((m) => m.name.toLowerCase() === modeName.toLowerCase());
|
|
|
|
if (!selectedMode) {
|
|
const availableModes = allModes.map((m) => m.name).join(", ");
|
|
return {
|
|
error: `Mode "${modeName}" not found. Available modes: ${availableModes}`,
|
|
availableModes: allModes.map((m) => ({ name: m.name, description: m.description })),
|
|
};
|
|
}
|
|
|
|
return {
|
|
modeName: selectedMode.name,
|
|
description: selectedMode.description,
|
|
prompt: selectedMode.prompt,
|
|
};
|
|
}),
|
|
});
|