30 lines
776 B
TypeScript
30 lines
776 B
TypeScript
|
import { BLOCKED_TOOLS, REPLACEMENT_MESSAGE } from "./constants";
|
||
|
|
|
||
|
|
export interface TasksTodowriteDisablerConfig {
|
||
|
|
experimental?: {
|
||
|
|
task_system?: boolean;
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
export function createTasksTodowriteDisablerHook(
|
||
|
|
config: TasksTodowriteDisablerConfig,
|
||
|
|
) {
|
||
|
|
const isTaskSystemEnabled = config.experimental?.task_system ?? false;
|
||
|
|
|
||
|
|
return {
|
||
|
|
"tool.execute.before": async (
|
||
|
|
input: { tool: string; sessionID: string; callID: string },
|
||
|
|
output: { args: Record<string, unknown> },
|
||
|
|
) => {
|
||
|
|
if (!isTaskSystemEnabled) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
const toolName = input.tool as string;
|
||
|
|
if (BLOCKED_TOOLS.some((blocked) => blocked.toLowerCase() === toolName.toLowerCase())) {
|
||
|
|
throw new Error(REPLACEMENT_MESSAGE);
|
||
|
|
}
|
||
|
|
},
|
||
|
|
};
|
||
|
|
}
|