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