@mragentix/cli 4.2.37 → 4.2.38
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/cli.js +180 -9
- package/dist/cli.js.map +1 -1
- package/dist/config.d.ts.map +1 -1
- package/dist/config.js +64 -0
- package/dist/config.js.map +1 -1
- package/dist/core/agent-session.d.ts.map +1 -1
- package/dist/core/agent-session.js +13 -1
- package/dist/core/agent-session.js.map +1 -1
- package/dist/core/compaction/compactor.d.ts +1 -0
- package/dist/core/compaction/compactor.d.ts.map +1 -1
- package/dist/core/compaction/compactor.js +8 -1
- package/dist/core/compaction/compactor.js.map +1 -1
- package/dist/core/skills.d.ts +2 -1
- package/dist/core/skills.d.ts.map +1 -1
- package/dist/core/skills.js +8 -8
- package/dist/core/skills.js.map +1 -1
- package/dist/interactive.d.ts.map +1 -1
- package/dist/interactive.js +17 -5
- package/dist/interactive.js.map +1 -1
- package/dist/modes/json-mode.d.ts.map +1 -1
- package/dist/modes/json-mode.js +2 -1
- package/dist/modes/json-mode.js.map +1 -1
- package/dist/modes/print-mode.d.ts.map +1 -1
- package/dist/modes/print-mode.js +2 -1
- package/dist/modes/print-mode.js.map +1 -1
- package/dist/modes/rpc-mode.d.ts.map +1 -1
- package/dist/modes/rpc-mode.js +2 -1
- package/dist/modes/rpc-mode.js.map +1 -1
- package/dist/modes/serve-mode.d.ts.map +1 -1
- package/dist/modes/serve-mode.js +7 -6
- package/dist/modes/serve-mode.js.map +1 -1
- package/dist/system-prompt.d.ts +1 -1
- package/dist/system-prompt.d.ts.map +1 -1
- package/dist/system-prompt.js +40 -2
- package/dist/system-prompt.js.map +1 -1
- package/dist/tools/bash.d.ts +3 -1
- package/dist/tools/bash.d.ts.map +1 -1
- package/dist/tools/bash.js +4 -1
- package/dist/tools/bash.js.map +1 -1
- package/dist/tools/edit-diff.test.d.ts +2 -0
- package/dist/tools/edit-diff.test.d.ts.map +1 -0
- package/dist/tools/edit-diff.test.js +67 -0
- package/dist/tools/edit-diff.test.js.map +1 -0
- package/dist/tools/edit.d.ts +3 -1
- package/dist/tools/edit.d.ts.map +1 -1
- package/dist/tools/edit.js +4 -1
- package/dist/tools/edit.js.map +1 -1
- package/dist/tools/edit.test.d.ts +2 -0
- package/dist/tools/edit.test.d.ts.map +1 -0
- package/dist/tools/edit.test.js +93 -0
- package/dist/tools/edit.test.js.map +1 -0
- package/dist/tools/enter-plan.d.ts +8 -0
- package/dist/tools/enter-plan.d.ts.map +1 -0
- package/dist/tools/enter-plan.js +28 -0
- package/dist/tools/enter-plan.js.map +1 -0
- package/dist/tools/exit-plan.d.ts +8 -0
- package/dist/tools/exit-plan.d.ts.map +1 -0
- package/dist/tools/exit-plan.js +36 -0
- package/dist/tools/exit-plan.js.map +1 -0
- package/dist/tools/index.d.ts +10 -0
- package/dist/tools/index.d.ts.map +1 -1
- package/dist/tools/index.js +20 -4
- package/dist/tools/index.js.map +1 -1
- package/dist/tools/path-utils.test.d.ts +2 -0
- package/dist/tools/path-utils.test.d.ts.map +1 -0
- package/dist/tools/path-utils.test.js +49 -0
- package/dist/tools/path-utils.test.js.map +1 -0
- package/dist/tools/plan-mode.test.d.ts +2 -0
- package/dist/tools/plan-mode.test.d.ts.map +1 -0
- package/dist/tools/plan-mode.test.js +214 -0
- package/dist/tools/plan-mode.test.js.map +1 -0
- package/dist/tools/read.test.d.ts +2 -0
- package/dist/tools/read.test.d.ts.map +1 -0
- package/dist/tools/read.test.js +81 -0
- package/dist/tools/read.test.js.map +1 -0
- package/dist/tools/skill.d.ts +10 -0
- package/dist/tools/skill.d.ts.map +1 -0
- package/dist/tools/skill.js +36 -0
- package/dist/tools/skill.js.map +1 -0
- package/dist/tools/subagent.d.ts +3 -1
- package/dist/tools/subagent.d.ts.map +1 -1
- package/dist/tools/subagent.js +4 -1
- package/dist/tools/subagent.js.map +1 -1
- package/dist/tools/truncate-utils.test.d.ts +2 -0
- package/dist/tools/truncate-utils.test.d.ts.map +1 -0
- package/dist/tools/truncate-utils.test.js +93 -0
- package/dist/tools/truncate-utils.test.js.map +1 -0
- package/dist/tools/write.d.ts +3 -1
- package/dist/tools/write.d.ts.map +1 -1
- package/dist/tools/write.js +11 -1
- package/dist/tools/write.js.map +1 -1
- package/dist/tools/write.test.js +65 -52
- package/dist/tools/write.test.js.map +1 -1
- package/dist/ui/App.d.ts +18 -1
- package/dist/ui/App.d.ts.map +1 -1
- package/dist/ui/App.js +234 -31
- package/dist/ui/App.js.map +1 -1
- package/dist/ui/components/ActivityIndicator.d.ts +2 -1
- package/dist/ui/components/ActivityIndicator.d.ts.map +1 -1
- package/dist/ui/components/ActivityIndicator.js +25 -4
- package/dist/ui/components/ActivityIndicator.js.map +1 -1
- package/dist/ui/components/AssistantMessage.d.ts.map +1 -1
- package/dist/ui/components/AssistantMessage.js +6 -1
- package/dist/ui/components/AssistantMessage.js.map +1 -1
- package/dist/ui/components/Banner.d.ts.map +1 -1
- package/dist/ui/components/Banner.js +13 -4
- package/dist/ui/components/Banner.js.map +1 -1
- package/dist/ui/components/Footer.d.ts +2 -1
- package/dist/ui/components/Footer.d.ts.map +1 -1
- package/dist/ui/components/Footer.js +30 -16
- package/dist/ui/components/Footer.js.map +1 -1
- package/dist/ui/components/InputArea.d.ts +3 -1
- package/dist/ui/components/InputArea.d.ts.map +1 -1
- package/dist/ui/components/InputArea.js +27 -17
- package/dist/ui/components/InputArea.js.map +1 -1
- package/dist/ui/components/Markdown.d.ts +5 -1
- package/dist/ui/components/Markdown.d.ts.map +1 -1
- package/dist/ui/components/Markdown.js +14 -6
- package/dist/ui/components/Markdown.js.map +1 -1
- package/dist/ui/components/PlanApproval.d.ts +8 -0
- package/dist/ui/components/PlanApproval.d.ts.map +1 -0
- package/dist/ui/components/PlanApproval.js +58 -0
- package/dist/ui/components/PlanApproval.js.map +1 -0
- package/dist/ui/components/PlanBanner.d.ts +6 -0
- package/dist/ui/components/PlanBanner.d.ts.map +1 -0
- package/dist/ui/components/PlanBanner.js +28 -0
- package/dist/ui/components/PlanBanner.js.map +1 -0
- package/dist/ui/components/PlanOverlay.d.ts +11 -0
- package/dist/ui/components/PlanOverlay.d.ts.map +1 -0
- package/dist/ui/components/PlanOverlay.js +271 -0
- package/dist/ui/components/PlanOverlay.js.map +1 -0
- package/dist/ui/components/ServerToolExecution.d.ts.map +1 -1
- package/dist/ui/components/ServerToolExecution.js +11 -3
- package/dist/ui/components/ServerToolExecution.js.map +1 -1
- package/dist/ui/components/SkillsOverlay.d.ts +7 -0
- package/dist/ui/components/SkillsOverlay.d.ts.map +1 -0
- package/dist/ui/components/SkillsOverlay.js +162 -0
- package/dist/ui/components/SkillsOverlay.js.map +1 -0
- package/dist/ui/components/Spinner.js +1 -1
- package/dist/ui/components/Spinner.js.map +1 -1
- package/dist/ui/components/StreamingArea.d.ts +2 -1
- package/dist/ui/components/StreamingArea.d.ts.map +1 -1
- package/dist/ui/components/StreamingArea.js +7 -2
- package/dist/ui/components/StreamingArea.js.map +1 -1
- package/dist/ui/components/SubAgentPanel.d.ts.map +1 -1
- package/dist/ui/components/SubAgentPanel.js +21 -11
- package/dist/ui/components/SubAgentPanel.js.map +1 -1
- package/dist/ui/components/TaskOverlay.d.ts.map +1 -1
- package/dist/ui/components/TaskOverlay.js +10 -6
- package/dist/ui/components/TaskOverlay.js.map +1 -1
- package/dist/ui/components/ToolExecution.d.ts.map +1 -1
- package/dist/ui/components/ToolExecution.js +33 -13
- package/dist/ui/components/ToolExecution.js.map +1 -1
- package/dist/ui/components/ToolGroupExecution.js +1 -1
- package/dist/ui/components/ToolGroupExecution.js.map +1 -1
- package/dist/ui/hooks/useAgentLoop.d.ts +8 -0
- package/dist/ui/hooks/useAgentLoop.d.ts.map +1 -1
- package/dist/ui/hooks/useAgentLoop.js +318 -236
- package/dist/ui/hooks/useAgentLoop.js.map +1 -1
- package/dist/ui/hooks/useTerminalSize.d.ts +18 -16
- package/dist/ui/hooks/useTerminalSize.d.ts.map +1 -1
- package/dist/ui/hooks/useTerminalSize.js +31 -24
- package/dist/ui/hooks/useTerminalSize.js.map +1 -1
- package/dist/ui/login.d.ts.map +1 -1
- package/dist/ui/login.js +1 -5
- package/dist/ui/login.js.map +1 -1
- package/dist/ui/render.d.ts +11 -0
- package/dist/ui/render.d.ts.map +1 -1
- package/dist/ui/render.js +7 -2
- package/dist/ui/render.js.map +1 -1
- package/dist/ui/sessions.d.ts.map +1 -1
- package/dist/ui/sessions.js +1 -5
- package/dist/ui/sessions.js.map +1 -1
- package/dist/ui/theme/dark.json +3 -1
- package/dist/ui/theme/light.json +3 -1
- package/dist/ui/theme/theme.d.ts +2 -0
- package/dist/ui/theme/theme.d.ts.map +1 -1
- package/package.json +3 -3
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import fs from "node:fs/promises";
|
|
2
|
+
import path from "node:path";
|
|
3
|
+
import { z } from "zod";
|
|
4
|
+
import { resolvePath } from "./path-utils.js";
|
|
5
|
+
const ExitPlanParams = z.object({
|
|
6
|
+
plan_path: z
|
|
7
|
+
.string()
|
|
8
|
+
.describe("Path to the plan markdown file (must be under .mragentix/plans/)"),
|
|
9
|
+
});
|
|
10
|
+
export function createExitPlanTool(cwd, onExitPlan) {
|
|
11
|
+
return {
|
|
12
|
+
name: "exit_plan",
|
|
13
|
+
description: "Submit your plan for user review and exit plan mode. " +
|
|
14
|
+
"The plan file must be under .mragentix/plans/. The user will approve, reject with feedback, or cancel.",
|
|
15
|
+
parameters: ExitPlanParams,
|
|
16
|
+
async execute({ plan_path }) {
|
|
17
|
+
const resolved = resolvePath(cwd, plan_path);
|
|
18
|
+
const plansDir = path.join(cwd, ".mragentix", "plans");
|
|
19
|
+
if (!resolved.startsWith(plansDir)) {
|
|
20
|
+
return "Error: plan_path must be under .mragentix/plans/. Got: " + plan_path;
|
|
21
|
+
}
|
|
22
|
+
try {
|
|
23
|
+
const content = await fs.readFile(resolved, "utf-8");
|
|
24
|
+
if (!content.trim()) {
|
|
25
|
+
return "Error: Plan file is empty. Write your plan before calling exit_plan.";
|
|
26
|
+
}
|
|
27
|
+
}
|
|
28
|
+
catch {
|
|
29
|
+
return "Error: Could not read plan file at " + plan_path + ". Make sure the file exists.";
|
|
30
|
+
}
|
|
31
|
+
const result = await onExitPlan(resolved);
|
|
32
|
+
return result;
|
|
33
|
+
},
|
|
34
|
+
};
|
|
35
|
+
}
|
|
36
|
+
//# sourceMappingURL=exit-plan.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"exit-plan.js","sourceRoot":"","sources":["../../src/tools/exit-plan.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAE9C,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9B,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CAAC,kEAAkE,CAAC;CAChF,CAAC,CAAC;AAEH,MAAM,UAAU,kBAAkB,CAChC,GAAW,EACX,UAAiD;IAEjD,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,WAAW,EACT,uDAAuD;YACvD,wGAAwG;QAC1G,UAAU,EAAE,cAAc;QAC1B,KAAK,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE;YACzB,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;YAEvD,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACnC,OAAO,yDAAyD,GAAG,SAAS,CAAC;YAC/E,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACrD,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;oBACpB,OAAO,sEAAsE,CAAC;gBAChF,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,qCAAqC,GAAG,SAAS,GAAG,8BAA8B,CAAC;YAC5F,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC1C,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC"}
|
package/dist/tools/index.d.ts
CHANGED
|
@@ -2,12 +2,19 @@ import type { AgentTool } from "@mragentix/agent";
|
|
|
2
2
|
import { ProcessManager } from "../core/process-manager.js";
|
|
3
3
|
import { type ToolOperations } from "./operations.js";
|
|
4
4
|
import type { AgentDefinition } from "../core/agents.js";
|
|
5
|
+
import type { Skill } from "../core/skills.js";
|
|
5
6
|
export interface CreateToolsOptions {
|
|
6
7
|
agents?: AgentDefinition[];
|
|
7
8
|
provider?: string;
|
|
8
9
|
model?: string;
|
|
9
10
|
/** Custom I/O operations for remote execution (SSH, Docker, etc.). Defaults to local filesystem. */
|
|
10
11
|
operations?: ToolOperations;
|
|
12
|
+
skills?: Skill[];
|
|
13
|
+
planModeRef?: {
|
|
14
|
+
current: boolean;
|
|
15
|
+
};
|
|
16
|
+
onEnterPlan?: (reason?: string) => void;
|
|
17
|
+
onExitPlan?: (planPath: string) => Promise<string>;
|
|
11
18
|
}
|
|
12
19
|
export interface CreateToolsResult {
|
|
13
20
|
tools: AgentTool[];
|
|
@@ -25,6 +32,9 @@ export { createWebFetchTool } from "./web-fetch.js";
|
|
|
25
32
|
export { createTaskOutputTool } from "./task-output.js";
|
|
26
33
|
export { createTaskStopTool } from "./task-stop.js";
|
|
27
34
|
export { createTasksTool } from "./tasks.js";
|
|
35
|
+
export { createSkillTool } from "./skill.js";
|
|
36
|
+
export { createEnterPlanTool } from "./enter-plan.js";
|
|
37
|
+
export { createExitPlanTool } from "./exit-plan.js";
|
|
28
38
|
export { ProcessManager } from "../core/process-manager.js";
|
|
29
39
|
export { localOperations, type ToolOperations } from "./operations.js";
|
|
30
40
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAgB5D,OAAO,EAAmB,KAAK,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACvE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAE/C,MAAM,WAAW,kBAAkB;IACjC,MAAM,CAAC,EAAE,eAAe,EAAE,CAAC;IAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,oGAAoG;IACpG,UAAU,CAAC,EAAE,cAAc,CAAC;IAC5B,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC;IACjB,WAAW,CAAC,EAAE;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;IACnC,WAAW,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;IACxC,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;CACpD;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,cAAc,EAAE,cAAc,CAAC;CAChC;AAED,wBAAgB,WAAW,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,kBAAkB,GAAG,iBAAiB,CAqCrF;AAED,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,eAAe,EAAE,KAAK,cAAc,EAAE,MAAM,iBAAiB,CAAC"}
|
package/dist/tools/index.js
CHANGED
|
@@ -11,16 +11,20 @@ import { createWebFetchTool } from "./web-fetch.js";
|
|
|
11
11
|
import { createTaskOutputTool } from "./task-output.js";
|
|
12
12
|
import { createTaskStopTool } from "./task-stop.js";
|
|
13
13
|
import { createTasksTool } from "./tasks.js";
|
|
14
|
+
import { createSkillTool } from "./skill.js";
|
|
15
|
+
import { createEnterPlanTool } from "./enter-plan.js";
|
|
16
|
+
import { createExitPlanTool } from "./exit-plan.js";
|
|
14
17
|
import { localOperations } from "./operations.js";
|
|
15
18
|
export function createTools(cwd, opts) {
|
|
16
19
|
const readFiles = new Set();
|
|
17
20
|
const processManager = new ProcessManager();
|
|
18
21
|
const ops = opts?.operations ?? localOperations;
|
|
22
|
+
const planModeRef = opts?.planModeRef;
|
|
19
23
|
const tools = [
|
|
20
24
|
createReadTool(cwd, readFiles, ops),
|
|
21
|
-
createWriteTool(cwd, readFiles, ops),
|
|
22
|
-
createEditTool(cwd, readFiles, ops),
|
|
23
|
-
createBashTool(cwd, processManager, ops),
|
|
25
|
+
createWriteTool(cwd, readFiles, ops, planModeRef),
|
|
26
|
+
createEditTool(cwd, readFiles, ops, planModeRef),
|
|
27
|
+
createBashTool(cwd, processManager, ops, planModeRef),
|
|
24
28
|
createFindTool(cwd),
|
|
25
29
|
createGrepTool(cwd, ops),
|
|
26
30
|
createLsTool(cwd, ops),
|
|
@@ -30,7 +34,16 @@ export function createTools(cwd, opts) {
|
|
|
30
34
|
createTasksTool(cwd),
|
|
31
35
|
];
|
|
32
36
|
if (opts?.agents && opts.agents.length > 0 && opts.provider && opts.model) {
|
|
33
|
-
tools.push(createSubAgentTool(cwd, opts.agents, opts.provider, opts.model));
|
|
37
|
+
tools.push(createSubAgentTool(cwd, opts.agents, opts.provider, opts.model, planModeRef));
|
|
38
|
+
}
|
|
39
|
+
if (opts?.skills && opts.skills.length > 0) {
|
|
40
|
+
tools.push(createSkillTool(opts.skills));
|
|
41
|
+
}
|
|
42
|
+
if (opts?.onEnterPlan) {
|
|
43
|
+
tools.push(createEnterPlanTool(opts.onEnterPlan));
|
|
44
|
+
}
|
|
45
|
+
if (opts?.onExitPlan) {
|
|
46
|
+
tools.push(createExitPlanTool(cwd, opts.onExitPlan));
|
|
34
47
|
}
|
|
35
48
|
return { tools, processManager };
|
|
36
49
|
}
|
|
@@ -45,6 +58,9 @@ export { createWebFetchTool } from "./web-fetch.js";
|
|
|
45
58
|
export { createTaskOutputTool } from "./task-output.js";
|
|
46
59
|
export { createTaskStopTool } from "./task-stop.js";
|
|
47
60
|
export { createTasksTool } from "./tasks.js";
|
|
61
|
+
export { createSkillTool } from "./skill.js";
|
|
62
|
+
export { createEnterPlanTool } from "./enter-plan.js";
|
|
63
|
+
export { createExitPlanTool } from "./exit-plan.js";
|
|
48
64
|
export { ProcessManager } from "../core/process-manager.js";
|
|
49
65
|
export { localOperations } from "./operations.js";
|
|
50
66
|
//# sourceMappingURL=index.js.map
|
package/dist/tools/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAuB,MAAM,iBAAiB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAuB,MAAM,iBAAiB,CAAC;AAqBvE,MAAM,UAAU,WAAW,CAAC,GAAW,EAAE,IAAyB;IAChE,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;IAC5C,MAAM,GAAG,GAAG,IAAI,EAAE,UAAU,IAAI,eAAe,CAAC;IAChD,MAAM,WAAW,GAAG,IAAI,EAAE,WAAW,CAAC;IAEtC,MAAM,KAAK,GAAgB;QACzB,cAAc,CAAC,GAAG,EAAE,SAAS,EAAE,GAAG,CAAC;QACnC,eAAe,CAAC,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,WAAW,CAAC;QACjD,cAAc,CAAC,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,WAAW,CAAC;QAChD,cAAc,CAAC,GAAG,EAAE,cAAc,EAAE,GAAG,EAAE,WAAW,CAAC;QACrD,cAAc,CAAC,GAAG,CAAC;QACnB,cAAc,CAAC,GAAG,EAAE,GAAG,CAAC;QACxB,YAAY,CAAC,GAAG,EAAE,GAAG,CAAC;QACtB,kBAAkB,EAAE;QACpB,oBAAoB,CAAC,cAAc,CAAC;QACpC,kBAAkB,CAAC,cAAc,CAAC;QAClC,eAAe,CAAC,GAAG,CAAC;KACrB,CAAC;IAEF,IAAI,IAAI,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QAC1E,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC;IAC3F,CAAC;IAED,IAAI,IAAI,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3C,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,IAAI,EAAE,WAAW,EAAE,CAAC;QACtB,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;IACpD,CAAC;IAED,IAAI,IAAI,EAAE,UAAU,EAAE,CAAC;QACrB,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;AACnC,CAAC;AAED,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,eAAe,EAAuB,MAAM,iBAAiB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"path-utils.test.d.ts","sourceRoot":"","sources":["../../src/tools/path-utils.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach } from "vitest";
|
|
2
|
+
import fs from "node:fs/promises";
|
|
3
|
+
import path from "node:path";
|
|
4
|
+
import os from "node:os";
|
|
5
|
+
import { resolvePath, rejectSymlink } from "./path-utils.js";
|
|
6
|
+
describe("resolvePath", () => {
|
|
7
|
+
it("resolves relative path from cwd", () => {
|
|
8
|
+
const result = resolvePath("/home/user/project", "src/index.ts");
|
|
9
|
+
expect(result).toBe(path.resolve("/home/user/project", "src/index.ts"));
|
|
10
|
+
});
|
|
11
|
+
it("returns absolute path as-is", () => {
|
|
12
|
+
const result = resolvePath("/home/user/project", "/etc/hosts");
|
|
13
|
+
expect(result).toBe("/etc/hosts");
|
|
14
|
+
});
|
|
15
|
+
it("expands ~ to home directory", () => {
|
|
16
|
+
const result = resolvePath("/anywhere", "~");
|
|
17
|
+
expect(result).toBe(os.homedir());
|
|
18
|
+
});
|
|
19
|
+
it("expands ~/path to homedir/path", () => {
|
|
20
|
+
const result = resolvePath("/anywhere", "~/documents/file.txt");
|
|
21
|
+
expect(result).toBe(path.join(os.homedir(), "documents/file.txt"));
|
|
22
|
+
});
|
|
23
|
+
});
|
|
24
|
+
describe("rejectSymlink", () => {
|
|
25
|
+
let tmpDir;
|
|
26
|
+
beforeEach(async () => {
|
|
27
|
+
tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), "path-utils-test-"));
|
|
28
|
+
});
|
|
29
|
+
afterEach(async () => {
|
|
30
|
+
await fs.rm(tmpDir, { recursive: true, force: true });
|
|
31
|
+
});
|
|
32
|
+
it("does not throw for a regular file", async () => {
|
|
33
|
+
const filePath = path.join(tmpDir, "regular.txt");
|
|
34
|
+
await fs.writeFile(filePath, "hello");
|
|
35
|
+
await expect(rejectSymlink(filePath)).resolves.toBeUndefined();
|
|
36
|
+
});
|
|
37
|
+
it("throws for a symlink", async () => {
|
|
38
|
+
const target = path.join(tmpDir, "target.txt");
|
|
39
|
+
const link = path.join(tmpDir, "link.txt");
|
|
40
|
+
await fs.writeFile(target, "hello");
|
|
41
|
+
await fs.symlink(target, link);
|
|
42
|
+
await expect(rejectSymlink(link)).rejects.toThrow("Refusing to follow symlink");
|
|
43
|
+
});
|
|
44
|
+
it("does not throw for a non-existent file (ENOENT swallowed)", async () => {
|
|
45
|
+
const missing = path.join(tmpDir, "does-not-exist.txt");
|
|
46
|
+
await expect(rejectSymlink(missing)).resolves.toBeUndefined();
|
|
47
|
+
});
|
|
48
|
+
});
|
|
49
|
+
//# sourceMappingURL=path-utils.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"path-utils.test.js","sourceRoot":"","sources":["../../src/tools/path-utils.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAE7D,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,MAAM,GAAG,WAAW,CAAC,oBAAoB,EAAE,cAAc,CAAC,CAAC;QACjE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,oBAAoB,EAAE,cAAc,CAAC,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,MAAM,GAAG,WAAW,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,MAAM,GAAG,WAAW,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,MAAM,GAAG,WAAW,CAAC,WAAW,EAAE,sBAAsB,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,oBAAoB,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,MAAc,CAAC;IAEnB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,kBAAkB,CAAC,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAClD,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAC3C,MAAM,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACpC,MAAM,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC/B,MAAM,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,4BAA4B,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC;QACxD,MAAM,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;IAChE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"plan-mode.test.d.ts","sourceRoot":"","sources":["../../src/tools/plan-mode.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,214 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach } from "vitest";
|
|
2
|
+
import fs from "node:fs/promises";
|
|
3
|
+
import path from "node:path";
|
|
4
|
+
import os from "node:os";
|
|
5
|
+
import { createEnterPlanTool } from "./enter-plan.js";
|
|
6
|
+
import { createExitPlanTool } from "./exit-plan.js";
|
|
7
|
+
import { createWriteTool } from "./write.js";
|
|
8
|
+
import { createEditTool } from "./edit.js";
|
|
9
|
+
import { createBashTool } from "./bash.js";
|
|
10
|
+
import { ProcessManager } from "../core/process-manager.js";
|
|
11
|
+
import { buildSystemPrompt } from "../system-prompt.js";
|
|
12
|
+
function resultToString(result) {
|
|
13
|
+
return typeof result === "string" ? result : result.content;
|
|
14
|
+
}
|
|
15
|
+
const mockContext = {
|
|
16
|
+
signal: new AbortController().signal,
|
|
17
|
+
toolCallId: "test-1",
|
|
18
|
+
};
|
|
19
|
+
// ── enter_plan tool ──────────────────────────────────────
|
|
20
|
+
describe("createEnterPlanTool", () => {
|
|
21
|
+
it("calls onEnterPlan callback and returns instructions", async () => {
|
|
22
|
+
let calledWith;
|
|
23
|
+
const tool = createEnterPlanTool((reason) => {
|
|
24
|
+
calledWith = reason;
|
|
25
|
+
});
|
|
26
|
+
const result = resultToString(await tool.execute({ reason: "complex task" }, mockContext));
|
|
27
|
+
expect(calledWith).toBe("complex task");
|
|
28
|
+
expect(result).toContain("Plan mode activated");
|
|
29
|
+
expect(result).toContain(".mragentix/plans/");
|
|
30
|
+
expect(result).toContain("exit_plan");
|
|
31
|
+
});
|
|
32
|
+
it("works without a reason", async () => {
|
|
33
|
+
let called = false;
|
|
34
|
+
const tool = createEnterPlanTool(() => {
|
|
35
|
+
called = true;
|
|
36
|
+
});
|
|
37
|
+
await tool.execute({}, mockContext);
|
|
38
|
+
expect(called).toBe(true);
|
|
39
|
+
});
|
|
40
|
+
});
|
|
41
|
+
// ── exit_plan tool ───────────────────────────────────────
|
|
42
|
+
describe("createExitPlanTool", () => {
|
|
43
|
+
let tmpDir;
|
|
44
|
+
beforeEach(async () => {
|
|
45
|
+
tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), "exit-plan-test-"));
|
|
46
|
+
await fs.mkdir(path.join(tmpDir, ".gg", "plans"), { recursive: true });
|
|
47
|
+
});
|
|
48
|
+
afterEach(async () => {
|
|
49
|
+
await fs.rm(tmpDir, { recursive: true, force: true });
|
|
50
|
+
});
|
|
51
|
+
it("rejects paths outside .mragentix/plans/", async () => {
|
|
52
|
+
const tool = createExitPlanTool(tmpDir, async () => "ok");
|
|
53
|
+
const result = resultToString(await tool.execute({ plan_path: "src/hack.ts" }, mockContext));
|
|
54
|
+
expect(result).toContain("Error");
|
|
55
|
+
expect(result).toContain(".mragentix/plans/");
|
|
56
|
+
});
|
|
57
|
+
it("rejects empty plan files", async () => {
|
|
58
|
+
const planPath = path.join(tmpDir, ".gg", "plans", "empty.md");
|
|
59
|
+
await fs.writeFile(planPath, " \n \n");
|
|
60
|
+
const tool = createExitPlanTool(tmpDir, async () => "should not reach");
|
|
61
|
+
const result = resultToString(await tool.execute({ plan_path: ".mragentix/plans/empty.md" }, mockContext));
|
|
62
|
+
expect(result).toContain("Error");
|
|
63
|
+
expect(result).toContain("empty");
|
|
64
|
+
});
|
|
65
|
+
it("rejects non-existent plan files", async () => {
|
|
66
|
+
const tool = createExitPlanTool(tmpDir, async () => "should not reach");
|
|
67
|
+
const result = resultToString(await tool.execute({ plan_path: ".mragentix/plans/ghost.md" }, mockContext));
|
|
68
|
+
expect(result).toContain("Error");
|
|
69
|
+
expect(result).toContain("Could not read");
|
|
70
|
+
});
|
|
71
|
+
it("calls onExitPlan for valid plans and returns the result", async () => {
|
|
72
|
+
const planPath = path.join(tmpDir, ".gg", "plans", "my-plan.md");
|
|
73
|
+
await fs.writeFile(planPath, "# Plan\n\n1. Step one\n2. Step two\n");
|
|
74
|
+
let receivedPath = "";
|
|
75
|
+
const tool = createExitPlanTool(tmpDir, async (p) => {
|
|
76
|
+
receivedPath = p;
|
|
77
|
+
return "Plan submitted.";
|
|
78
|
+
});
|
|
79
|
+
const result = resultToString(await tool.execute({ plan_path: ".mragentix/plans/my-plan.md" }, mockContext));
|
|
80
|
+
expect(receivedPath).toBe(planPath);
|
|
81
|
+
expect(result).toBe("Plan submitted.");
|
|
82
|
+
});
|
|
83
|
+
});
|
|
84
|
+
// ── Tool restrictions in plan mode ───────────────────────
|
|
85
|
+
describe("plan mode tool restrictions", () => {
|
|
86
|
+
let tmpDir;
|
|
87
|
+
const planModeRef = { current: false };
|
|
88
|
+
beforeEach(async () => {
|
|
89
|
+
tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), "plan-restrict-test-"));
|
|
90
|
+
planModeRef.current = false;
|
|
91
|
+
});
|
|
92
|
+
afterEach(async () => {
|
|
93
|
+
await fs.rm(tmpDir, { recursive: true, force: true });
|
|
94
|
+
});
|
|
95
|
+
describe("bash", () => {
|
|
96
|
+
it("allows execution when plan mode is off", async () => {
|
|
97
|
+
const pm = new ProcessManager();
|
|
98
|
+
const tool = createBashTool(tmpDir, pm, undefined, planModeRef);
|
|
99
|
+
const result = resultToString(await tool.execute({ command: "echo hello" }, mockContext));
|
|
100
|
+
expect(result).toContain("Exit code: 0");
|
|
101
|
+
expect(result).toContain("hello");
|
|
102
|
+
pm.shutdownAll();
|
|
103
|
+
});
|
|
104
|
+
it("blocks execution when plan mode is on", async () => {
|
|
105
|
+
planModeRef.current = true;
|
|
106
|
+
const pm = new ProcessManager();
|
|
107
|
+
const tool = createBashTool(tmpDir, pm, undefined, planModeRef);
|
|
108
|
+
const result = resultToString(await tool.execute({ command: "echo hello" }, mockContext));
|
|
109
|
+
expect(result).toContain("Error");
|
|
110
|
+
expect(result).toContain("restricted in plan mode");
|
|
111
|
+
pm.shutdownAll();
|
|
112
|
+
});
|
|
113
|
+
});
|
|
114
|
+
describe("edit", () => {
|
|
115
|
+
it("blocks execution when plan mode is on", async () => {
|
|
116
|
+
planModeRef.current = true;
|
|
117
|
+
const readFiles = new Set();
|
|
118
|
+
const tool = createEditTool(tmpDir, readFiles, undefined, planModeRef);
|
|
119
|
+
const result = resultToString(await tool.execute({ file_path: "test.ts", old_text: "foo", new_text: "bar" }, mockContext));
|
|
120
|
+
expect(result).toContain("Error");
|
|
121
|
+
expect(result).toContain("restricted in plan mode");
|
|
122
|
+
});
|
|
123
|
+
});
|
|
124
|
+
describe("write", () => {
|
|
125
|
+
it("blocks writing to non-plan paths when plan mode is on", async () => {
|
|
126
|
+
planModeRef.current = true;
|
|
127
|
+
const tool = createWriteTool(tmpDir, undefined, undefined, planModeRef);
|
|
128
|
+
const result = resultToString(await tool.execute({ file_path: "src/hack.ts", content: "malicious" }, mockContext));
|
|
129
|
+
expect(result).toContain("Error");
|
|
130
|
+
expect(result).toContain("restricted in plan mode");
|
|
131
|
+
expect(result).toContain(".mragentix/plans/");
|
|
132
|
+
});
|
|
133
|
+
it("allows writing to .mragentix/plans/ when plan mode is on", async () => {
|
|
134
|
+
planModeRef.current = true;
|
|
135
|
+
const tool = createWriteTool(tmpDir, undefined, undefined, planModeRef);
|
|
136
|
+
const result = resultToString(await tool.execute({ file_path: ".mragentix/plans/my-plan.md", content: "# Plan\n\nDo stuff." }, mockContext));
|
|
137
|
+
expect(result).toContain("Wrote");
|
|
138
|
+
expect(result).toContain("bytes");
|
|
139
|
+
// Verify file was actually created
|
|
140
|
+
const written = await fs.readFile(path.join(tmpDir, ".gg", "plans", "my-plan.md"), "utf-8");
|
|
141
|
+
expect(written).toBe("# Plan\n\nDo stuff.");
|
|
142
|
+
});
|
|
143
|
+
it("creates .mragentix/plans/ directory if it doesn't exist", async () => {
|
|
144
|
+
planModeRef.current = true;
|
|
145
|
+
const tool = createWriteTool(tmpDir, undefined, undefined, planModeRef);
|
|
146
|
+
await tool.execute({ file_path: ".mragentix/plans/auto-created.md", content: "# Plan" }, mockContext);
|
|
147
|
+
const stat = await fs.stat(path.join(tmpDir, ".gg", "plans"));
|
|
148
|
+
expect(stat.isDirectory()).toBe(true);
|
|
149
|
+
});
|
|
150
|
+
it("allows writing anywhere when plan mode is off", async () => {
|
|
151
|
+
planModeRef.current = false;
|
|
152
|
+
const tool = createWriteTool(tmpDir, undefined, undefined, planModeRef);
|
|
153
|
+
const result = resultToString(await tool.execute({ file_path: "src/normal.ts", content: "export const x = 1;" }, mockContext));
|
|
154
|
+
expect(result).toContain("Wrote");
|
|
155
|
+
});
|
|
156
|
+
});
|
|
157
|
+
});
|
|
158
|
+
// ── System prompt generation ─────────────────────────────
|
|
159
|
+
describe("buildSystemPrompt with plan mode", () => {
|
|
160
|
+
let tmpDir;
|
|
161
|
+
beforeEach(async () => {
|
|
162
|
+
tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), "prompt-test-"));
|
|
163
|
+
});
|
|
164
|
+
afterEach(async () => {
|
|
165
|
+
await fs.rm(tmpDir, { recursive: true, force: true });
|
|
166
|
+
});
|
|
167
|
+
it("includes plan mode section when planMode is true", async () => {
|
|
168
|
+
const prompt = await buildSystemPrompt(tmpDir, [], true);
|
|
169
|
+
expect(prompt).toContain("Plan Mode (ACTIVE)");
|
|
170
|
+
expect(prompt).toContain("read-only");
|
|
171
|
+
expect(prompt).toContain("exit_plan");
|
|
172
|
+
expect(prompt).toContain(".mragentix/plans/");
|
|
173
|
+
});
|
|
174
|
+
it("does not include plan mode section when planMode is false", async () => {
|
|
175
|
+
const prompt = await buildSystemPrompt(tmpDir, [], false);
|
|
176
|
+
expect(prompt).not.toContain("Plan Mode (ACTIVE)");
|
|
177
|
+
});
|
|
178
|
+
it("includes enter_plan and exit_plan tool descriptions", async () => {
|
|
179
|
+
const prompt = await buildSystemPrompt(tmpDir, [], false);
|
|
180
|
+
expect(prompt).toContain("enter_plan");
|
|
181
|
+
expect(prompt).toContain("exit_plan");
|
|
182
|
+
});
|
|
183
|
+
it("includes approved plan when approvedPlanPath is set", async () => {
|
|
184
|
+
const plansDir = path.join(tmpDir, ".gg", "plans");
|
|
185
|
+
await fs.mkdir(plansDir, { recursive: true });
|
|
186
|
+
const planPath = path.join(plansDir, "approved.md");
|
|
187
|
+
await fs.writeFile(planPath, "# My Plan\n\n1. Do this\n2. Do that\n");
|
|
188
|
+
const prompt = await buildSystemPrompt(tmpDir, [], false, planPath);
|
|
189
|
+
expect(prompt).toContain("Approved Plan");
|
|
190
|
+
expect(prompt).toContain("approved.md");
|
|
191
|
+
expect(prompt).toContain("# My Plan");
|
|
192
|
+
expect(prompt).toContain("Do this");
|
|
193
|
+
expect(prompt).toContain("Follow the plan");
|
|
194
|
+
});
|
|
195
|
+
it("does not include approved plan section when no path is set", async () => {
|
|
196
|
+
const prompt = await buildSystemPrompt(tmpDir, [], false);
|
|
197
|
+
expect(prompt).not.toContain("Approved Plan");
|
|
198
|
+
expect(prompt).not.toContain("<approved_plan>");
|
|
199
|
+
});
|
|
200
|
+
it("does not include approved plan when in plan mode (even if path is set)", async () => {
|
|
201
|
+
const planPath = path.join(tmpDir, "fake-plan.md");
|
|
202
|
+
const prompt = await buildSystemPrompt(tmpDir, [], true, planPath);
|
|
203
|
+
// Plan mode section should be present, but not the approved plan
|
|
204
|
+
expect(prompt).toContain("Plan Mode (ACTIVE)");
|
|
205
|
+
expect(prompt).not.toContain("Approved Plan");
|
|
206
|
+
});
|
|
207
|
+
it("gracefully handles missing approved plan file", async () => {
|
|
208
|
+
const prompt = await buildSystemPrompt(tmpDir, [], false, "/nonexistent/plan.md");
|
|
209
|
+
// Should not crash, and should not include the section
|
|
210
|
+
expect(prompt).not.toContain("Approved Plan");
|
|
211
|
+
expect(prompt).not.toContain("<approved_plan>");
|
|
212
|
+
});
|
|
213
|
+
});
|
|
214
|
+
//# sourceMappingURL=plan-mode.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"plan-mode.test.js","sourceRoot":"","sources":["../../src/tools/plan-mode.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAExD,SAAS,cAAc,CAAC,MAAoC;IAC1D,OAAO,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;AAC9D,CAAC;AAED,MAAM,WAAW,GAAG;IAClB,MAAM,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;IACpC,UAAU,EAAE,QAAQ;CACrB,CAAC;AAEF,4DAA4D;AAE5D,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,IAAI,UAA8B,CAAC;QACnC,MAAM,IAAI,GAAG,mBAAmB,CAAC,CAAC,MAAM,EAAE,EAAE;YAC1C,UAAU,GAAG,MAAM,CAAC;QACtB,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;QAE3F,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,IAAI,MAAM,GAAG,KAAK,CAAC;QACnB,MAAM,IAAI,GAAG,mBAAmB,CAAC,GAAG,EAAE;YACpC,MAAM,GAAG,IAAI,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,4DAA4D;AAE5D,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,MAAc,CAAC;IAEnB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;QACrE,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,IAAI,GAAG,kBAAkB,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC;QAE1D,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;QAE7F,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAC/D,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAE1C,MAAM,IAAI,GAAG,kBAAkB,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,kBAAkB,CAAC,CAAC;QAExE,MAAM,MAAM,GAAG,cAAc,CAC3B,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,2BAA2B,EAAE,EAAE,WAAW,CAAC,CAC5E,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,IAAI,GAAG,kBAAkB,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,kBAAkB,CAAC,CAAC;QAExE,MAAM,MAAM,GAAG,cAAc,CAC3B,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,2BAA2B,EAAE,EAAE,WAAW,CAAC,CAC5E,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;QACjE,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,sCAAsC,CAAC,CAAC;QAErE,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,MAAM,IAAI,GAAG,kBAAkB,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;YAClD,YAAY,GAAG,CAAC,CAAC;YACjB,OAAO,iBAAiB,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAC3B,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,6BAA6B,EAAE,EAAE,WAAW,CAAC,CAC9E,CAAC;QAEF,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,4DAA4D;AAE5D,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,IAAI,MAAc,CAAC;IACnB,MAAM,WAAW,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAEvC,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,qBAAqB,CAAC,CAAC,CAAC;QACzE,WAAW,CAAC,OAAO,GAAG,KAAK,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,EAAE,GAAG,IAAI,cAAc,EAAE,CAAC;YAChC,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;YAE1F,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAClC,EAAE,CAAC,WAAW,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;YAC3B,MAAM,EAAE,GAAG,IAAI,cAAc,EAAE,CAAC;YAChC,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;YAE1F,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;YACpD,EAAE,CAAC,WAAW,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;YAC3B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;YACpC,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,cAAc,CAC3B,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,WAAW,CAAC,CAC5F,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;YAC3B,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAExE,MAAM,MAAM,GAAG,cAAc,CAC3B,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,WAAW,CAAC,CACpF,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;YAC3B,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAExE,MAAM,MAAM,GAAG,cAAc,CAC3B,MAAM,IAAI,CAAC,OAAO,CAChB,EAAE,SAAS,EAAE,6BAA6B,EAAE,OAAO,EAAE,qBAAqB,EAAE,EAC5E,WAAW,CACZ,CACF,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAElC,mCAAmC;YACnC,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC;YAC5F,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;YAC3B,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAExE,MAAM,IAAI,CAAC,OAAO,CAChB,EAAE,SAAS,EAAE,kCAAkC,EAAE,OAAO,EAAE,QAAQ,EAAE,EACpE,WAAW,CACZ,CAAC;YAEF,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,WAAW,CAAC,OAAO,GAAG,KAAK,CAAC;YAC5B,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAExE,MAAM,MAAM,GAAG,cAAc,CAC3B,MAAM,IAAI,CAAC,OAAO,CAChB,EAAE,SAAS,EAAE,eAAe,EAAE,OAAO,EAAE,qBAAqB,EAAE,EAC9D,WAAW,CACZ,CACF,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,4DAA4D;AAE5D,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;IAChD,IAAI,MAAc,CAAC;IAEnB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,cAAc,CAAC,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;QAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACnD,MAAM,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QACpD,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,uCAAuC,CAAC,CAAC;QAEtE,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QAEpE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;QAEnE,iEAAiE;QACjE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,sBAAsB,CAAC,CAAC;QAElF,uDAAuD;QACvD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"read.test.d.ts","sourceRoot":"","sources":["../../src/tools/read.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,81 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach } from "vitest";
|
|
2
|
+
import fs from "node:fs/promises";
|
|
3
|
+
import path from "node:path";
|
|
4
|
+
import os from "node:os";
|
|
5
|
+
import { createReadTool, BINARY_EXTENSIONS } from "./read.js";
|
|
6
|
+
describe("createReadTool", () => {
|
|
7
|
+
let tmpDir;
|
|
8
|
+
beforeEach(async () => {
|
|
9
|
+
tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), "read-tool-test-"));
|
|
10
|
+
});
|
|
11
|
+
afterEach(async () => {
|
|
12
|
+
await fs.rm(tmpDir, { recursive: true, force: true });
|
|
13
|
+
});
|
|
14
|
+
function ctx(id) {
|
|
15
|
+
return { signal: new AbortController().signal, toolCallId: id };
|
|
16
|
+
}
|
|
17
|
+
it("reads a file and returns numbered lines in cat -n format", async () => {
|
|
18
|
+
const filePath = path.join(tmpDir, "hello.txt");
|
|
19
|
+
await fs.writeFile(filePath, "alpha\nbeta\ngamma");
|
|
20
|
+
const tool = createReadTool(tmpDir);
|
|
21
|
+
const result = await tool.execute({ file_path: filePath }, ctx("test-1"));
|
|
22
|
+
expect(result).toBe(" 1\talpha\n 2\tbeta\n 3\tgamma");
|
|
23
|
+
});
|
|
24
|
+
it("returns all lines when no offset/limit provided", async () => {
|
|
25
|
+
const filePath = path.join(tmpDir, "five.txt");
|
|
26
|
+
await fs.writeFile(filePath, "a\nb\nc\nd\ne");
|
|
27
|
+
const tool = createReadTool(tmpDir);
|
|
28
|
+
const result = await tool.execute({ file_path: filePath }, ctx("test-2"));
|
|
29
|
+
const lines = result.split("\n");
|
|
30
|
+
expect(lines).toHaveLength(5);
|
|
31
|
+
expect(lines[0]).toBe(" 1\ta");
|
|
32
|
+
expect(lines[4]).toBe(" 5\te");
|
|
33
|
+
});
|
|
34
|
+
it("supports offset parameter (1-based, skips first N-1 lines)", async () => {
|
|
35
|
+
const filePath = path.join(tmpDir, "offset.txt");
|
|
36
|
+
await fs.writeFile(filePath, "line1\nline2\nline3\nline4\nline5");
|
|
37
|
+
const tool = createReadTool(tmpDir);
|
|
38
|
+
const result = await tool.execute({ file_path: filePath, offset: 3 }, ctx("test-3"));
|
|
39
|
+
expect(result).toBe(" 3\tline3\n 4\tline4\n 5\tline5");
|
|
40
|
+
});
|
|
41
|
+
it("supports limit parameter (returns only N lines)", async () => {
|
|
42
|
+
const filePath = path.join(tmpDir, "limit.txt");
|
|
43
|
+
await fs.writeFile(filePath, "a\nb\nc\nd\ne");
|
|
44
|
+
const tool = createReadTool(tmpDir);
|
|
45
|
+
const result = await tool.execute({ file_path: filePath, limit: 2 }, ctx("test-4"));
|
|
46
|
+
expect(result).toBe(" 1\ta\n 2\tb");
|
|
47
|
+
});
|
|
48
|
+
it("supports offset + limit together", async () => {
|
|
49
|
+
const filePath = path.join(tmpDir, "both.txt");
|
|
50
|
+
await fs.writeFile(filePath, "a\nb\nc\nd\ne");
|
|
51
|
+
const tool = createReadTool(tmpDir);
|
|
52
|
+
const result = await tool.execute({ file_path: filePath, offset: 2, limit: 2 }, ctx("test-5"));
|
|
53
|
+
expect(result).toBe(" 2\tb\n 3\tc");
|
|
54
|
+
});
|
|
55
|
+
it("detects binary files and returns size notice", async () => {
|
|
56
|
+
const filePath = path.join(tmpDir, "image.png");
|
|
57
|
+
const buf = Buffer.alloc(128, 0xff);
|
|
58
|
+
await fs.writeFile(filePath, buf);
|
|
59
|
+
const tool = createReadTool(tmpDir);
|
|
60
|
+
const result = await tool.execute({ file_path: filePath }, ctx("test-6"));
|
|
61
|
+
expect(result).toBe(`Binary file: ${filePath} (.png, 128 bytes)`);
|
|
62
|
+
});
|
|
63
|
+
it("adds resolved path to readFiles set", async () => {
|
|
64
|
+
const filePath = path.join(tmpDir, "tracked.txt");
|
|
65
|
+
await fs.writeFile(filePath, "content");
|
|
66
|
+
const readFiles = new Set();
|
|
67
|
+
const tool = createReadTool(tmpDir, readFiles);
|
|
68
|
+
await tool.execute({ file_path: filePath }, ctx("test-7"));
|
|
69
|
+
expect(readFiles.has(filePath)).toBe(true);
|
|
70
|
+
expect(readFiles.size).toBe(1);
|
|
71
|
+
});
|
|
72
|
+
});
|
|
73
|
+
describe("BINARY_EXTENSIONS", () => {
|
|
74
|
+
it("contains expected entries", () => {
|
|
75
|
+
expect(BINARY_EXTENSIONS.has(".png")).toBe(true);
|
|
76
|
+
expect(BINARY_EXTENSIONS.has(".pdf")).toBe(true);
|
|
77
|
+
expect(BINARY_EXTENSIONS.has(".exe")).toBe(true);
|
|
78
|
+
expect(BINARY_EXTENSIONS.has(".wasm")).toBe(true);
|
|
79
|
+
});
|
|
80
|
+
});
|
|
81
|
+
//# sourceMappingURL=read.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"read.test.js","sourceRoot":"","sources":["../../src/tools/read.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAE9D,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,MAAc,CAAC;IAEnB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,SAAS,GAAG,CAAC,EAAU;QACrB,OAAO,EAAE,MAAM,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAClE,CAAC;IAED,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAChD,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,oBAAoB,CAAC,CAAC;QAEnD,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE1E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAC/C,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;QAE9C,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE1E,MAAM,KAAK,GAAI,MAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QACjD,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,mCAAmC,CAAC,CAAC;QAElE,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAErF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAChD,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;QAE9C,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEpF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAC/C,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;QAE9C,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE/F,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAChD,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACpC,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;QAElC,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE1E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,gBAAgB,QAAQ,oBAAoB,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAClD,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAExC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;QACpC,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC/C,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE3D,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
import type { AgentTool } from "@mragentix/agent";
|
|
3
|
+
import type { Skill } from "../core/skills.js";
|
|
4
|
+
declare const parameters: z.ZodObject<{
|
|
5
|
+
skill: z.ZodString;
|
|
6
|
+
args: z.ZodOptional<z.ZodString>;
|
|
7
|
+
}, z.core.$strip>;
|
|
8
|
+
export declare function createSkillTool(skills: Skill[]): AgentTool<typeof parameters>;
|
|
9
|
+
export {};
|
|
10
|
+
//# sourceMappingURL=skill.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"skill.d.ts","sourceRoot":"","sources":["../../src/tools/skill.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAE/C,QAAA,MAAM,UAAU;;;iBAGd,CAAC;AAEH,wBAAgB,eAAe,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC,OAAO,UAAU,CAAC,CAwB7E"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
const parameters = z.object({
|
|
3
|
+
skill: z.string().describe("The name of the skill to invoke"),
|
|
4
|
+
args: z.string().optional().describe("Optional arguments or context for the skill"),
|
|
5
|
+
});
|
|
6
|
+
export function createSkillTool(skills) {
|
|
7
|
+
const skillMap = new Map(skills.map((s) => [s.name, s]));
|
|
8
|
+
return {
|
|
9
|
+
name: "skill",
|
|
10
|
+
description: generateSkillDescription(skills),
|
|
11
|
+
parameters,
|
|
12
|
+
async execute(input) {
|
|
13
|
+
const skill = skillMap.get(input.skill);
|
|
14
|
+
if (!skill) {
|
|
15
|
+
const available = skills.map((s) => s.name).join(", ");
|
|
16
|
+
return `Error: Skill "${input.skill}" not found. Available skills: ${available || "none"}`;
|
|
17
|
+
}
|
|
18
|
+
const parts = [`<skill_content name="${skill.name}">`, skill.content, `</skill_content>`];
|
|
19
|
+
if (input.args) {
|
|
20
|
+
parts.push(`\nUser context: ${input.args}`);
|
|
21
|
+
}
|
|
22
|
+
parts.push("\nTreat the above skill instructions as authoritative. Follow them to complete the task.");
|
|
23
|
+
return parts.join("\n");
|
|
24
|
+
},
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
function generateSkillDescription(skills) {
|
|
28
|
+
if (skills.length === 0) {
|
|
29
|
+
return "Invoke a skill by name. No skills are currently available.";
|
|
30
|
+
}
|
|
31
|
+
const list = skills
|
|
32
|
+
.map((s) => `- **${s.name}**: ${s.description || "No description"}`)
|
|
33
|
+
.join("\n");
|
|
34
|
+
return `Invoke a skill by name to get specialized instructions for a task.\n\nAvailable skills:\n${list}`;
|
|
35
|
+
}
|
|
36
|
+
//# sourceMappingURL=skill.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"skill.js","sourceRoot":"","sources":["../../src/tools/skill.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;IAC7D,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6CAA6C,CAAC;CACpF,CAAC,CAAC;AAEH,MAAM,UAAU,eAAe,CAAC,MAAe;IAC7C,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAEzD,OAAO;QACL,IAAI,EAAE,OAAO;QACb,WAAW,EAAE,wBAAwB,CAAC,MAAM,CAAC;QAC7C,UAAU;QACV,KAAK,CAAC,OAAO,CAAC,KAAK;YACjB,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACxC,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACvD,OAAO,iBAAiB,KAAK,CAAC,KAAK,kCAAkC,SAAS,IAAI,MAAM,EAAE,CAAC;YAC7F,CAAC;YAED,MAAM,KAAK,GAAG,CAAC,wBAAwB,KAAK,CAAC,IAAI,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC;YAC1F,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;gBACf,KAAK,CAAC,IAAI,CAAC,mBAAmB,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YAC9C,CAAC;YACD,KAAK,CAAC,IAAI,CACR,0FAA0F,CAC3F,CAAC;YACF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,wBAAwB,CAAC,MAAe;IAC/C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,4DAA4D,CAAC;IACtE,CAAC;IAED,MAAM,IAAI,GAAG,MAAM;SAChB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,WAAW,IAAI,gBAAgB,EAAE,CAAC;SACnE,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,OAAO,4FAA4F,IAAI,EAAE,CAAC;AAC5G,CAAC"}
|
package/dist/tools/subagent.d.ts
CHANGED
|
@@ -21,6 +21,8 @@ export interface SubAgentDetails {
|
|
|
21
21
|
};
|
|
22
22
|
durationMs: number;
|
|
23
23
|
}
|
|
24
|
-
export declare function createSubAgentTool(cwd: string, agents: AgentDefinition[], parentProvider: string, parentModel: string
|
|
24
|
+
export declare function createSubAgentTool(cwd: string, agents: AgentDefinition[], parentProvider: string, parentModel: string, planModeRef?: {
|
|
25
|
+
current: boolean;
|
|
26
|
+
}): AgentTool<typeof SubAgentParams>;
|
|
25
27
|
export {};
|
|
26
28
|
//# sourceMappingURL=subagent.d.ts.map
|