cray-code 1.0.0
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/README.md +316 -0
- package/dist/Tool.d.ts +217 -0
- package/dist/Tool.d.ts.map +1 -0
- package/dist/Tool.js +89 -0
- package/dist/Tool.js.map +1 -0
- package/dist/branding/logo.d.ts +8 -0
- package/dist/branding/logo.d.ts.map +1 -0
- package/dist/branding/logo.js +26 -0
- package/dist/branding/logo.js.map +1 -0
- package/dist/branding/theme.d.ts +27 -0
- package/dist/branding/theme.d.ts.map +1 -0
- package/dist/branding/theme.js +28 -0
- package/dist/branding/theme.js.map +1 -0
- package/dist/commands/registry.d.ts +32 -0
- package/dist/commands/registry.d.ts.map +1 -0
- package/dist/commands/registry.js +759 -0
- package/dist/commands/registry.js.map +1 -0
- package/dist/components/MessageView.d.ts +12 -0
- package/dist/components/MessageView.d.ts.map +1 -0
- package/dist/components/MessageView.js +35 -0
- package/dist/components/MessageView.js.map +1 -0
- package/dist/components/PermissionPrompt.d.ts +11 -0
- package/dist/components/PermissionPrompt.d.ts.map +1 -0
- package/dist/components/PermissionPrompt.js +6 -0
- package/dist/components/PermissionPrompt.js.map +1 -0
- package/dist/components/PluginManager.d.ts +27 -0
- package/dist/components/PluginManager.d.ts.map +1 -0
- package/dist/components/PluginManager.js +391 -0
- package/dist/components/PluginManager.js.map +1 -0
- package/dist/components/ThinkingBlock.d.ts +27 -0
- package/dist/components/ThinkingBlock.d.ts.map +1 -0
- package/dist/components/ThinkingBlock.js +29 -0
- package/dist/components/ThinkingBlock.js.map +1 -0
- package/dist/components/ToolCallBlock.d.ts +14 -0
- package/dist/components/ToolCallBlock.d.ts.map +1 -0
- package/dist/components/ToolCallBlock.js +83 -0
- package/dist/components/ToolCallBlock.js.map +1 -0
- package/dist/components/TrustDialog.d.ts +20 -0
- package/dist/components/TrustDialog.d.ts.map +1 -0
- package/dist/components/TrustDialog.js +80 -0
- package/dist/components/TrustDialog.js.map +1 -0
- package/dist/context.d.ts +25 -0
- package/dist/context.d.ts.map +1 -0
- package/dist/context.js +268 -0
- package/dist/context.js.map +1 -0
- package/dist/cray.d.ts +114 -0
- package/dist/cray.d.ts.map +1 -0
- package/dist/cray.js +338 -0
- package/dist/cray.js.map +1 -0
- package/dist/index.d.ts +16 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +122 -0
- package/dist/index.js.map +1 -0
- package/dist/plugins/registry.d.ts +106 -0
- package/dist/plugins/registry.d.ts.map +1 -0
- package/dist/plugins/registry.js +695 -0
- package/dist/plugins/registry.js.map +1 -0
- package/dist/query.d.ts +31 -0
- package/dist/query.d.ts.map +1 -0
- package/dist/query.js +637 -0
- package/dist/query.js.map +1 -0
- package/dist/queryStream.d.ts +36 -0
- package/dist/queryStream.d.ts.map +1 -0
- package/dist/queryStream.js +704 -0
- package/dist/queryStream.js.map +1 -0
- package/dist/screens/ReplScreen.d.ts +22 -0
- package/dist/screens/ReplScreen.d.ts.map +1 -0
- package/dist/screens/ReplScreen.js +763 -0
- package/dist/screens/ReplScreen.js.map +1 -0
- package/dist/services/agentRunner.d.ts +39 -0
- package/dist/services/agentRunner.d.ts.map +1 -0
- package/dist/services/agentRunner.js +115 -0
- package/dist/services/agentRunner.js.map +1 -0
- package/dist/services/compact.d.ts +34 -0
- package/dist/services/compact.d.ts.map +1 -0
- package/dist/services/compact.js +179 -0
- package/dist/services/compact.js.map +1 -0
- package/dist/services/loadPluginCommands.d.ts +55 -0
- package/dist/services/loadPluginCommands.d.ts.map +1 -0
- package/dist/services/loadPluginCommands.js +219 -0
- package/dist/services/loadPluginCommands.js.map +1 -0
- package/dist/services/mcp/index.d.ts +3 -0
- package/dist/services/mcp/index.d.ts.map +1 -0
- package/dist/services/mcp/index.js +3 -0
- package/dist/services/mcp/index.js.map +1 -0
- package/dist/services/mcp/manager.d.ts +24 -0
- package/dist/services/mcp/manager.d.ts.map +1 -0
- package/dist/services/mcp/manager.js +138 -0
- package/dist/services/mcp/manager.js.map +1 -0
- package/dist/services/mcp/types.d.ts +52 -0
- package/dist/services/mcp/types.d.ts.map +1 -0
- package/dist/services/mcp/types.js +5 -0
- package/dist/services/mcp/types.js.map +1 -0
- package/dist/services/memory.d.ts +38 -0
- package/dist/services/memory.d.ts.map +1 -0
- package/dist/services/memory.js +181 -0
- package/dist/services/memory.js.map +1 -0
- package/dist/services/permissionPrompt.d.ts +38 -0
- package/dist/services/permissionPrompt.d.ts.map +1 -0
- package/dist/services/permissionPrompt.js +83 -0
- package/dist/services/permissionPrompt.js.map +1 -0
- package/dist/services/permissions.d.ts +15 -0
- package/dist/services/permissions.d.ts.map +1 -0
- package/dist/services/permissions.js +237 -0
- package/dist/services/permissions.js.map +1 -0
- package/dist/services/sessionStorage.d.ts +51 -0
- package/dist/services/sessionStorage.d.ts.map +1 -0
- package/dist/services/sessionStorage.js +266 -0
- package/dist/services/sessionStorage.js.map +1 -0
- package/dist/setup.d.ts +22 -0
- package/dist/setup.d.ts.map +1 -0
- package/dist/setup.js +160 -0
- package/dist/setup.js.map +1 -0
- package/dist/skills/bundledSkills.d.ts +18 -0
- package/dist/skills/bundledSkills.d.ts.map +1 -0
- package/dist/skills/bundledSkills.js +277 -0
- package/dist/skills/bundledSkills.js.map +1 -0
- package/dist/skills/index.d.ts +4 -0
- package/dist/skills/index.d.ts.map +1 -0
- package/dist/skills/index.js +3 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/loadSkillsDir.d.ts +45 -0
- package/dist/skills/loadSkillsDir.d.ts.map +1 -0
- package/dist/skills/loadSkillsDir.js +233 -0
- package/dist/skills/loadSkillsDir.js.map +1 -0
- package/dist/state/AppState.d.ts +70 -0
- package/dist/state/AppState.d.ts.map +1 -0
- package/dist/state/AppState.js +106 -0
- package/dist/state/AppState.js.map +1 -0
- package/dist/tools/AgentTool.d.ts +62 -0
- package/dist/tools/AgentTool.d.ts.map +1 -0
- package/dist/tools/AgentTool.js +133 -0
- package/dist/tools/AgentTool.js.map +1 -0
- package/dist/tools/AskUserQuestionTool.d.ts +60 -0
- package/dist/tools/AskUserQuestionTool.d.ts.map +1 -0
- package/dist/tools/AskUserQuestionTool.js +52 -0
- package/dist/tools/AskUserQuestionTool.js.map +1 -0
- package/dist/tools/BashTool.d.ts +33 -0
- package/dist/tools/BashTool.d.ts.map +1 -0
- package/dist/tools/BashTool.js +211 -0
- package/dist/tools/BashTool.js.map +1 -0
- package/dist/tools/EditTool.d.ts +24 -0
- package/dist/tools/EditTool.d.ts.map +1 -0
- package/dist/tools/EditTool.js +102 -0
- package/dist/tools/EditTool.js.map +1 -0
- package/dist/tools/GlobTool.d.ts +17 -0
- package/dist/tools/GlobTool.d.ts.map +1 -0
- package/dist/tools/GlobTool.js +65 -0
- package/dist/tools/GlobTool.js.map +1 -0
- package/dist/tools/GrepTool.d.ts +30 -0
- package/dist/tools/GrepTool.d.ts.map +1 -0
- package/dist/tools/GrepTool.js +140 -0
- package/dist/tools/GrepTool.js.map +1 -0
- package/dist/tools/MCPTool.d.ts +24 -0
- package/dist/tools/MCPTool.d.ts.map +1 -0
- package/dist/tools/MCPTool.js +67 -0
- package/dist/tools/MCPTool.js.map +1 -0
- package/dist/tools/NotebookEditTool.d.ts +28 -0
- package/dist/tools/NotebookEditTool.d.ts.map +1 -0
- package/dist/tools/NotebookEditTool.js +213 -0
- package/dist/tools/NotebookEditTool.js.map +1 -0
- package/dist/tools/NotebookReadTool.d.ts +19 -0
- package/dist/tools/NotebookReadTool.d.ts.map +1 -0
- package/dist/tools/NotebookReadTool.js +191 -0
- package/dist/tools/NotebookReadTool.js.map +1 -0
- package/dist/tools/PlanTools.d.ts +17 -0
- package/dist/tools/PlanTools.d.ts.map +1 -0
- package/dist/tools/PlanTools.js +65 -0
- package/dist/tools/PlanTools.js.map +1 -0
- package/dist/tools/ReadTool.d.ts +21 -0
- package/dist/tools/ReadTool.d.ts.map +1 -0
- package/dist/tools/ReadTool.js +202 -0
- package/dist/tools/ReadTool.js.map +1 -0
- package/dist/tools/SkillTool.d.ts +32 -0
- package/dist/tools/SkillTool.d.ts.map +1 -0
- package/dist/tools/SkillTool.js +217 -0
- package/dist/tools/SkillTool.js.map +1 -0
- package/dist/tools/TodoWriteTool.d.ts +35 -0
- package/dist/tools/TodoWriteTool.d.ts.map +1 -0
- package/dist/tools/TodoWriteTool.js +58 -0
- package/dist/tools/TodoWriteTool.js.map +1 -0
- package/dist/tools/WebFetchTool.d.ts +17 -0
- package/dist/tools/WebFetchTool.d.ts.map +1 -0
- package/dist/tools/WebFetchTool.js +97 -0
- package/dist/tools/WebFetchTool.js.map +1 -0
- package/dist/tools/WebSearchTool.d.ts +18 -0
- package/dist/tools/WebSearchTool.d.ts.map +1 -0
- package/dist/tools/WebSearchTool.js +76 -0
- package/dist/tools/WebSearchTool.js.map +1 -0
- package/dist/tools/WriteTool.d.ts +17 -0
- package/dist/tools/WriteTool.d.ts.map +1 -0
- package/dist/tools/WriteTool.js +84 -0
- package/dist/tools/WriteTool.js.map +1 -0
- package/dist/tools/index.d.ts +21 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +20 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools.d.ts +34 -0
- package/dist/tools.d.ts.map +1 -0
- package/dist/tools.js +102 -0
- package/dist/tools.js.map +1 -0
- package/dist/types/events.d.ts +85 -0
- package/dist/types/events.d.ts.map +1 -0
- package/dist/types/events.js +12 -0
- package/dist/types/events.js.map +1 -0
- package/dist/types/index.d.ts +4 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +4 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/message.d.ts +71 -0
- package/dist/types/message.d.ts.map +1 -0
- package/dist/types/message.js +5 -0
- package/dist/types/message.js.map +1 -0
- package/dist/types/permission.d.ts +56 -0
- package/dist/types/permission.d.ts.map +1 -0
- package/dist/types/permission.js +46 -0
- package/dist/types/permission.js.map +1 -0
- package/dist/types/processUserInput.d.ts +18 -0
- package/dist/types/processUserInput.d.ts.map +1 -0
- package/dist/types/processUserInput.js +8 -0
- package/dist/types/processUserInput.js.map +1 -0
- package/dist/types/tool.d.ts +32 -0
- package/dist/types/tool.d.ts.map +1 -0
- package/dist/types/tool.js +5 -0
- package/dist/types/tool.js.map +1 -0
- package/dist/utils/compactBoundary.d.ts +11 -0
- package/dist/utils/compactBoundary.d.ts.map +1 -0
- package/dist/utils/compactBoundary.js +26 -0
- package/dist/utils/compactBoundary.js.map +1 -0
- package/dist/utils/configStore.d.ts +41 -0
- package/dist/utils/configStore.d.ts.map +1 -0
- package/dist/utils/configStore.js +111 -0
- package/dist/utils/configStore.js.map +1 -0
- package/dist/utils/forkedAgent.d.ts +40 -0
- package/dist/utils/forkedAgent.d.ts.map +1 -0
- package/dist/utils/forkedAgent.js +231 -0
- package/dist/utils/forkedAgent.js.map +1 -0
- package/dist/utils/messages.d.ts +14 -0
- package/dist/utils/messages.d.ts.map +1 -0
- package/dist/utils/messages.js +29 -0
- package/dist/utils/messages.js.map +1 -0
- package/dist/utils/sandbox.d.ts +22 -0
- package/dist/utils/sandbox.d.ts.map +1 -0
- package/dist/utils/sandbox.js +59 -0
- package/dist/utils/sandbox.js.map +1 -0
- package/dist/utils/sideQuestion.d.ts +29 -0
- package/dist/utils/sideQuestion.d.ts.map +1 -0
- package/dist/utils/sideQuestion.js +81 -0
- package/dist/utils/sideQuestion.js.map +1 -0
- package/install.ps1 +86 -0
- package/install.sh +92 -0
- package/package.json +68 -0
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* MCPTool — Model Context Protocol integration for Cray Code.
|
|
3
|
+
* Dynamically wraps external MCP server tools for use by the LLM agent.
|
|
4
|
+
*/
|
|
5
|
+
import { z } from 'zod';
|
|
6
|
+
import { type ToolDef } from '../Tool.js';
|
|
7
|
+
export declare const MCP_TOOL_NAME = "mcp";
|
|
8
|
+
declare const inputSchema: z.ZodObject<{
|
|
9
|
+
server: z.ZodString;
|
|
10
|
+
tool: z.ZodString;
|
|
11
|
+
arguments: z.ZodRecord<z.ZodString, z.ZodUnknown>;
|
|
12
|
+
}, "passthrough", z.ZodTypeAny, z.objectOutputType<{
|
|
13
|
+
server: z.ZodString;
|
|
14
|
+
tool: z.ZodString;
|
|
15
|
+
arguments: z.ZodRecord<z.ZodString, z.ZodUnknown>;
|
|
16
|
+
}, z.ZodTypeAny, "passthrough">, z.objectInputType<{
|
|
17
|
+
server: z.ZodString;
|
|
18
|
+
tool: z.ZodString;
|
|
19
|
+
arguments: z.ZodRecord<z.ZodString, z.ZodUnknown>;
|
|
20
|
+
}, z.ZodTypeAny, "passthrough">>;
|
|
21
|
+
/** Create a wrapped MCP tool instance for a specific MCP server tool */
|
|
22
|
+
export declare function createMCPTool(serverName: string, toolName: string, toolDescription: string, toolInputSchema: Record<string, unknown>): ToolDef<typeof inputSchema, string>;
|
|
23
|
+
export {};
|
|
24
|
+
//# sourceMappingURL=MCPTool.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"MCPTool.d.ts","sourceRoot":"","sources":["../../src/tools/MCPTool.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAa,KAAK,OAAO,EAA4D,MAAM,YAAY,CAAC;AAI/G,eAAO,MAAM,aAAa,QAAQ,CAAC;AAEnC,QAAA,MAAM,WAAW;;;;;;;;;;;;gCAID,CAAC;AAIjB,wEAAwE;AACxE,wBAAgB,aAAa,CAC3B,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,eAAe,EAAE,MAAM,EACvB,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACvC,OAAO,CAAC,OAAO,WAAW,EAAE,MAAM,CAAC,CA2DrC"}
|
|
@@ -0,0 +1,67 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* MCPTool — Model Context Protocol integration for Cray Code.
|
|
3
|
+
* Dynamically wraps external MCP server tools for use by the LLM agent.
|
|
4
|
+
*/
|
|
5
|
+
import { z } from 'zod';
|
|
6
|
+
import { buildTool } from '../Tool.js';
|
|
7
|
+
import { getAllMCPServers } from '../services/mcp/manager.js';
|
|
8
|
+
export const MCP_TOOL_NAME = 'mcp';
|
|
9
|
+
const inputSchema = z.object({
|
|
10
|
+
server: z.string().describe('MCP server name'),
|
|
11
|
+
tool: z.string().describe('Tool name on the MCP server'),
|
|
12
|
+
arguments: z.record(z.unknown()).describe('Tool arguments'),
|
|
13
|
+
}).passthrough();
|
|
14
|
+
/** Create a wrapped MCP tool instance for a specific MCP server tool */
|
|
15
|
+
export function createMCPTool(serverName, toolName, toolDescription, toolInputSchema) {
|
|
16
|
+
return buildTool({
|
|
17
|
+
name: `mcp__${serverName}__${toolName}`,
|
|
18
|
+
mcpInfo: { serverName, toolName },
|
|
19
|
+
searchHint: `MCP ${serverName} ${toolName}`,
|
|
20
|
+
inputSchema,
|
|
21
|
+
description() {
|
|
22
|
+
return toolDescription;
|
|
23
|
+
},
|
|
24
|
+
userFacingName() {
|
|
25
|
+
return `${serverName}:${toolName}`;
|
|
26
|
+
},
|
|
27
|
+
isConcurrencySafe() {
|
|
28
|
+
return true;
|
|
29
|
+
},
|
|
30
|
+
async checkPermissions(_input, _context) {
|
|
31
|
+
return { behavior: 'passthrough' };
|
|
32
|
+
},
|
|
33
|
+
async call(input, _context) {
|
|
34
|
+
try {
|
|
35
|
+
const servers = getAllMCPServers();
|
|
36
|
+
const server = servers.get(serverName);
|
|
37
|
+
if (!server || !server.connected) {
|
|
38
|
+
return {
|
|
39
|
+
data: '',
|
|
40
|
+
content: `MCP server "${serverName}" is not connected`,
|
|
41
|
+
isError: true,
|
|
42
|
+
title: `MCP Error: ${serverName}`,
|
|
43
|
+
};
|
|
44
|
+
}
|
|
45
|
+
// Dispatch to the MCP server
|
|
46
|
+
const result = await server.callTool(toolName, input.arguments);
|
|
47
|
+
const output = typeof result === 'string' ? result : JSON.stringify(result, null, 2);
|
|
48
|
+
const truncated = output.length > 100_000;
|
|
49
|
+
return {
|
|
50
|
+
data: truncated ? output.slice(0, 100_000) + '\n... [truncated]' : output,
|
|
51
|
+
content: output,
|
|
52
|
+
truncated,
|
|
53
|
+
title: `MCP ${serverName}/${toolName}`,
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
catch (err) {
|
|
57
|
+
return {
|
|
58
|
+
data: '',
|
|
59
|
+
content: `MCP tool error: ${err.message}`,
|
|
60
|
+
isError: true,
|
|
61
|
+
title: `MCP Error: ${serverName}/${toolName}`,
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
},
|
|
65
|
+
});
|
|
66
|
+
}
|
|
67
|
+
//# sourceMappingURL=MCPTool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"MCPTool.js","sourceRoot":"","sources":["../../src/tools/MCPTool.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAA0E,MAAM,YAAY,CAAC;AAE/G,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAE9D,MAAM,CAAC,MAAM,aAAa,GAAG,KAAK,CAAC;AAEnC,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;IAC9C,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;IACxD,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,gBAAgB,CAAC;CAC5D,CAAC,CAAC,WAAW,EAAE,CAAC;AAIjB,wEAAwE;AACxE,MAAM,UAAU,aAAa,CAC3B,UAAkB,EAClB,QAAgB,EAChB,eAAuB,EACvB,eAAwC;IAExC,OAAO,SAAS,CAAC;QACf,IAAI,EAAE,QAAQ,UAAU,KAAK,QAAQ,EAAE;QACvC,OAAO,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE;QACjC,UAAU,EAAE,OAAO,UAAU,IAAI,QAAQ,EAAE;QAC3C,WAAW;QAEX,WAAW;YACT,OAAO,eAAe,CAAC;QACzB,CAAC;QAED,cAAc;YACZ,OAAO,GAAG,UAAU,IAAI,QAAQ,EAAE,CAAC;QACrC,CAAC;QAED,iBAAiB;YACf,OAAO,IAAI,CAAC;QACd,CAAC;QAED,KAAK,CAAC,gBAAgB,CAAC,MAAa,EAAE,QAAa;YACjD,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC;QACrC,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,KAAY,EAAE,QAAwB;YAC/C,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,gBAAgB,EAAE,CAAC;gBACnC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBAEvC,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;oBACjC,OAAO;wBACL,IAAI,EAAE,EAAE;wBACR,OAAO,EAAE,eAAe,UAAU,oBAAoB;wBACtD,OAAO,EAAE,IAAI;wBACb,KAAK,EAAE,cAAc,UAAU,EAAE;qBAClC,CAAC;gBACJ,CAAC;gBAED,6BAA6B;gBAC7B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,KAAK,CAAC,SAAoC,CAAC,CAAC;gBAE3F,MAAM,MAAM,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;gBACrF,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,GAAG,OAAO,CAAC;gBAE1C,OAAO;oBACL,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,mBAAmB,CAAC,CAAC,CAAC,MAAM;oBACzE,OAAO,EAAE,MAAM;oBACf,SAAS;oBACT,KAAK,EAAE,OAAO,UAAU,IAAI,QAAQ,EAAE;iBACvC,CAAC;YACJ,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,OAAO;oBACL,IAAI,EAAE,EAAE;oBACR,OAAO,EAAE,mBAAmB,GAAG,CAAC,OAAO,EAAE;oBACzC,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE,cAAc,UAAU,IAAI,QAAQ,EAAE;iBAC9C,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* NotebookEditTool — Jupyter notebook (.ipynb) cell editor for Cray Code.
|
|
3
|
+
* Supports replace, insert, and delete operations on individual cells.
|
|
4
|
+
* Matches Claude Code's NotebookEdit capabilities.
|
|
5
|
+
*/
|
|
6
|
+
import { z } from 'zod';
|
|
7
|
+
import { type ToolDef } from '../Tool.js';
|
|
8
|
+
export declare const NOTEBOOK_EDIT_TOOL_NAME = "notebook_edit";
|
|
9
|
+
export declare const NotebookEditTool: ToolDef<z.ZodObject<{
|
|
10
|
+
notebook_path: z.ZodString;
|
|
11
|
+
cell_id: z.ZodOptional<z.ZodString>;
|
|
12
|
+
new_source: z.ZodString;
|
|
13
|
+
cell_type: z.ZodOptional<z.ZodEnum<["code", "markdown"]>>;
|
|
14
|
+
edit_mode: z.ZodDefault<z.ZodOptional<z.ZodEnum<["replace", "insert", "delete"]>>>;
|
|
15
|
+
}, "strip", z.ZodTypeAny, {
|
|
16
|
+
notebook_path: string;
|
|
17
|
+
new_source: string;
|
|
18
|
+
edit_mode: "replace" | "insert" | "delete";
|
|
19
|
+
cell_id?: string | undefined;
|
|
20
|
+
cell_type?: "code" | "markdown" | undefined;
|
|
21
|
+
}, {
|
|
22
|
+
notebook_path: string;
|
|
23
|
+
new_source: string;
|
|
24
|
+
cell_id?: string | undefined;
|
|
25
|
+
cell_type?: "code" | "markdown" | undefined;
|
|
26
|
+
edit_mode?: "replace" | "insert" | "delete" | undefined;
|
|
27
|
+
}>, string>;
|
|
28
|
+
//# sourceMappingURL=NotebookEditTool.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"NotebookEditTool.d.ts","sourceRoot":"","sources":["../../src/tools/NotebookEditTool.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,OAAO,EAAa,KAAK,OAAO,EAAwC,MAAM,YAAY,CAAC;AAG3F,eAAO,MAAM,uBAAuB,kBAAkB,CAAC;AA+BvD,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;WA4N3B,CAAC"}
|
|
@@ -0,0 +1,213 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* NotebookEditTool — Jupyter notebook (.ipynb) cell editor for Cray Code.
|
|
3
|
+
* Supports replace, insert, and delete operations on individual cells.
|
|
4
|
+
* Matches Claude Code's NotebookEdit capabilities.
|
|
5
|
+
*/
|
|
6
|
+
import { z } from 'zod';
|
|
7
|
+
import { readFile, writeFile } from 'fs/promises';
|
|
8
|
+
import { existsSync, mkdirSync } from 'fs';
|
|
9
|
+
import * as path from 'path';
|
|
10
|
+
import { buildTool } from '../Tool.js';
|
|
11
|
+
export const NOTEBOOK_EDIT_TOOL_NAME = 'notebook_edit';
|
|
12
|
+
const inputSchema = z.object({
|
|
13
|
+
notebook_path: z.string().describe('The absolute path to the Jupyter notebook file'),
|
|
14
|
+
cell_id: z.string().optional().describe('The ID of the cell to edit. When inserting, the new cell is inserted after this cell, or at the beginning if not specified.'),
|
|
15
|
+
new_source: z.string().describe('The new source for the cell'),
|
|
16
|
+
cell_type: z.enum(['code', 'markdown']).optional().describe('The type of the cell. Defaults to current cell type. Required for insert.'),
|
|
17
|
+
edit_mode: z.enum(['replace', 'insert', 'delete']).optional().default('replace'),
|
|
18
|
+
});
|
|
19
|
+
function generateCellId() {
|
|
20
|
+
return `cell-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
|
21
|
+
}
|
|
22
|
+
export const NotebookEditTool = buildTool({
|
|
23
|
+
name: NOTEBOOK_EDIT_TOOL_NAME,
|
|
24
|
+
aliases: ['nb_edit', 'edit_notebook'],
|
|
25
|
+
searchHint: 'jupyter notebook ipynb edit cell modify',
|
|
26
|
+
inputSchema,
|
|
27
|
+
async description(input) {
|
|
28
|
+
const op = input.edit_mode === 'delete' ? 'Delete' : input.edit_mode === 'insert' ? 'Insert' : 'Edit';
|
|
29
|
+
return `${op} cell in ${path.basename(input.notebook_path)}`;
|
|
30
|
+
},
|
|
31
|
+
userFacingName() {
|
|
32
|
+
return 'NotebookEdit';
|
|
33
|
+
},
|
|
34
|
+
isReadOnly() {
|
|
35
|
+
return false;
|
|
36
|
+
},
|
|
37
|
+
isConcurrencySafe() {
|
|
38
|
+
return false;
|
|
39
|
+
},
|
|
40
|
+
async checkPermissions(input, context) {
|
|
41
|
+
const mode = (context?.permissionContext?.mode ?? context?.mode) ?? 'default';
|
|
42
|
+
if (mode === 'bypass' || mode === 'acceptEdits')
|
|
43
|
+
return { behavior: 'allow' };
|
|
44
|
+
const op = input.edit_mode === 'delete' ? 'Delete cell from' : input.edit_mode === 'insert' ? 'Insert cell into' : 'Edit cell in';
|
|
45
|
+
return {
|
|
46
|
+
behavior: 'ask',
|
|
47
|
+
message: `${op} ${path.basename(input.notebook_path)}?`,
|
|
48
|
+
};
|
|
49
|
+
},
|
|
50
|
+
async call(input, context) {
|
|
51
|
+
const notebookPath = path.resolve(input.notebook_path);
|
|
52
|
+
try {
|
|
53
|
+
if (!existsSync(notebookPath)) {
|
|
54
|
+
if (input.edit_mode === 'insert') {
|
|
55
|
+
// Create a new empty notebook
|
|
56
|
+
const newNb = {
|
|
57
|
+
cells: [],
|
|
58
|
+
metadata: {},
|
|
59
|
+
nbformat: 4,
|
|
60
|
+
nbformat_minor: 5,
|
|
61
|
+
};
|
|
62
|
+
const dir = path.dirname(notebookPath);
|
|
63
|
+
if (!existsSync(dir))
|
|
64
|
+
mkdirSync(dir, { recursive: true });
|
|
65
|
+
await writeFile(notebookPath, JSON.stringify(newNb, null, 2), 'utf-8');
|
|
66
|
+
return {
|
|
67
|
+
data: `Created new notebook: ${path.basename(notebookPath)}`,
|
|
68
|
+
content: `Created new notebook: ${path.basename(notebookPath)}. Use notebook_edit with edit_mode="insert" to add cells.`,
|
|
69
|
+
title: `NotebookEdit: ${path.basename(notebookPath)}`,
|
|
70
|
+
};
|
|
71
|
+
}
|
|
72
|
+
return {
|
|
73
|
+
data: '',
|
|
74
|
+
content: `Notebook not found: ${notebookPath}`,
|
|
75
|
+
isError: true,
|
|
76
|
+
title: `NotebookEdit: ${path.basename(notebookPath)}`,
|
|
77
|
+
};
|
|
78
|
+
}
|
|
79
|
+
const raw = await readFile(notebookPath, 'utf-8');
|
|
80
|
+
const nb = JSON.parse(raw);
|
|
81
|
+
if (!nb.cells || !Array.isArray(nb.cells)) {
|
|
82
|
+
return {
|
|
83
|
+
data: '',
|
|
84
|
+
content: `Invalid notebook format in ${path.basename(notebookPath)}`,
|
|
85
|
+
isError: true,
|
|
86
|
+
title: `NotebookEdit: ${path.basename(notebookPath)}`,
|
|
87
|
+
};
|
|
88
|
+
}
|
|
89
|
+
if (input.edit_mode === 'delete') {
|
|
90
|
+
if (!input.cell_id) {
|
|
91
|
+
return {
|
|
92
|
+
data: '',
|
|
93
|
+
content: 'cell_id is required for delete mode.',
|
|
94
|
+
isError: true,
|
|
95
|
+
title: 'NotebookEdit Error',
|
|
96
|
+
};
|
|
97
|
+
}
|
|
98
|
+
const idx = nb.cells.findIndex((c) => c.id === input.cell_id);
|
|
99
|
+
if (idx === -1) {
|
|
100
|
+
return {
|
|
101
|
+
data: '',
|
|
102
|
+
content: `Cell not found: ${input.cell_id}`,
|
|
103
|
+
isError: true,
|
|
104
|
+
title: 'NotebookEdit: cell not found',
|
|
105
|
+
};
|
|
106
|
+
}
|
|
107
|
+
const removed = nb.cells[idx];
|
|
108
|
+
nb.cells.splice(idx, 1);
|
|
109
|
+
await writeFile(notebookPath, JSON.stringify(nb, null, 2), 'utf-8');
|
|
110
|
+
const cellLabel = removed.cell_type === 'code'
|
|
111
|
+
? `[${removed.execution_count ?? ' '}] ${(removed.source.join('') || '').slice(0, 60)}`
|
|
112
|
+
: `[markdown] ${(removed.source.join('') || '').slice(0, 60)}`;
|
|
113
|
+
return {
|
|
114
|
+
data: `Deleted cell ${idx + 1}: ${cellLabel}`,
|
|
115
|
+
content: `Deleted cell ${idx + 1} (id: ${input.cell_id}) from ${path.basename(notebookPath)}`,
|
|
116
|
+
title: `NotebookEdit: deleted cell`,
|
|
117
|
+
};
|
|
118
|
+
}
|
|
119
|
+
if (input.edit_mode === 'insert') {
|
|
120
|
+
const cellType = input.cell_type || 'code';
|
|
121
|
+
const newSource = input.new_source.split('\n').map((l, i, arr) => i < arr.length - 1 ? l + '\n' : l);
|
|
122
|
+
const newCell = {
|
|
123
|
+
cell_type: cellType,
|
|
124
|
+
source: newSource.length === 0 ? [''] : newSource,
|
|
125
|
+
id: generateCellId(),
|
|
126
|
+
execution_count: null,
|
|
127
|
+
};
|
|
128
|
+
if (cellType === 'code') {
|
|
129
|
+
newCell.outputs = [];
|
|
130
|
+
}
|
|
131
|
+
if (input.cell_id) {
|
|
132
|
+
const idx = nb.cells.findIndex((c) => c.id === input.cell_id);
|
|
133
|
+
if (idx === -1) {
|
|
134
|
+
return {
|
|
135
|
+
data: '',
|
|
136
|
+
content: `Cell not found: ${input.cell_id}. Insert at beginning with no cell_id.`,
|
|
137
|
+
isError: true,
|
|
138
|
+
title: 'NotebookEdit: cell not found',
|
|
139
|
+
};
|
|
140
|
+
}
|
|
141
|
+
nb.cells.splice(idx + 1, 0, newCell);
|
|
142
|
+
}
|
|
143
|
+
else {
|
|
144
|
+
nb.cells.unshift(newCell);
|
|
145
|
+
}
|
|
146
|
+
await writeFile(notebookPath, JSON.stringify(nb, null, 2), 'utf-8');
|
|
147
|
+
const pos = input.cell_id
|
|
148
|
+
? `after cell ${nb.cells.findIndex((c) => c.id === input.cell_id)}`
|
|
149
|
+
: 'at beginning';
|
|
150
|
+
return {
|
|
151
|
+
data: `Inserted ${cellType} cell ${pos}`,
|
|
152
|
+
content: `Inserted new ${cellType} cell (id: ${newCell.id}) ${pos} in ${path.basename(notebookPath)}`,
|
|
153
|
+
title: `NotebookEdit: inserted ${cellType} cell`,
|
|
154
|
+
};
|
|
155
|
+
}
|
|
156
|
+
// edit_mode === 'replace'
|
|
157
|
+
if (!input.cell_id) {
|
|
158
|
+
return {
|
|
159
|
+
data: '',
|
|
160
|
+
content: 'cell_id is required for replace mode.',
|
|
161
|
+
isError: true,
|
|
162
|
+
title: 'NotebookEdit Error',
|
|
163
|
+
};
|
|
164
|
+
}
|
|
165
|
+
const cellIdx = nb.cells.findIndex((c) => c.id === input.cell_id);
|
|
166
|
+
if (cellIdx === -1) {
|
|
167
|
+
return {
|
|
168
|
+
data: '',
|
|
169
|
+
content: `Cell not found: ${input.cell_id}`,
|
|
170
|
+
isError: true,
|
|
171
|
+
title: 'NotebookEdit: cell not found',
|
|
172
|
+
};
|
|
173
|
+
}
|
|
174
|
+
const oldCell = nb.cells[cellIdx];
|
|
175
|
+
const newSource = input.new_source.split('\n').map((l, i, arr) => i < arr.length - 1 ? l + '\n' : l);
|
|
176
|
+
nb.cells[cellIdx] = {
|
|
177
|
+
...oldCell,
|
|
178
|
+
source: newSource.length === 0 ? [''] : newSource,
|
|
179
|
+
cell_type: input.cell_type || oldCell.cell_type,
|
|
180
|
+
};
|
|
181
|
+
// If type changed to/from code, adjust outputs
|
|
182
|
+
if (input.cell_type && input.cell_type !== oldCell.cell_type) {
|
|
183
|
+
if (input.cell_type === 'code' && !nb.cells[cellIdx].outputs) {
|
|
184
|
+
nb.cells[cellIdx].outputs = [];
|
|
185
|
+
}
|
|
186
|
+
else if (input.cell_type === 'markdown') {
|
|
187
|
+
delete nb.cells[cellIdx].outputs;
|
|
188
|
+
delete nb.cells[cellIdx].execution_count;
|
|
189
|
+
}
|
|
190
|
+
}
|
|
191
|
+
await writeFile(notebookPath, JSON.stringify(nb, null, 2), 'utf-8');
|
|
192
|
+
const preview = input.new_source.slice(0, 60);
|
|
193
|
+
return {
|
|
194
|
+
data: `Replaced cell ${cellIdx + 1} (id: ${input.cell_id})`,
|
|
195
|
+
content: `Replaced cell ${cellIdx + 1} (id: ${input.cell_id}) in ${path.basename(notebookPath)}: "${preview}"`,
|
|
196
|
+
title: `NotebookEdit: replaced cell`,
|
|
197
|
+
};
|
|
198
|
+
}
|
|
199
|
+
catch (err) {
|
|
200
|
+
return {
|
|
201
|
+
data: '',
|
|
202
|
+
content: `Failed to edit notebook ${notebookPath}: ${err.message}`,
|
|
203
|
+
isError: true,
|
|
204
|
+
title: `NotebookEdit Error: ${path.basename(notebookPath)}`,
|
|
205
|
+
};
|
|
206
|
+
}
|
|
207
|
+
},
|
|
208
|
+
renderToolUseMessage(input) {
|
|
209
|
+
const op = input.edit_mode === 'delete' ? 'Deleting cell from' : input.edit_mode === 'insert' ? 'Inserting cell into' : 'Editing cell in';
|
|
210
|
+
return `${op} ${path.basename(input.notebook_path)}...`;
|
|
211
|
+
},
|
|
212
|
+
});
|
|
213
|
+
//# sourceMappingURL=NotebookEditTool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"NotebookEditTool.js","sourceRoot":"","sources":["../../src/tools/NotebookEditTool.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC3C,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAsD,MAAM,YAAY,CAAC;AAG3F,MAAM,CAAC,MAAM,uBAAuB,GAAG,eAAe,CAAC;AAEvD,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gDAAgD,CAAC;IACpF,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6HAA6H,CAAC;IACtK,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;IAC9D,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2EAA2E,CAAC;IACxI,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC;CACjF,CAAC,CAAC;AAmBH,SAAS,cAAc;IACrB,OAAO,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AACxE,CAAC;AAED,MAAM,CAAC,MAAM,gBAAgB,GAAG,SAAS,CAAC;IACxC,IAAI,EAAE,uBAAuB;IAC7B,OAAO,EAAE,CAAC,SAAS,EAAE,eAAe,CAAC;IACrC,UAAU,EAAE,yCAAyC;IACrD,WAAW;IAEX,KAAK,CAAC,WAAW,CAAC,KAAY;QAC5B,MAAM,EAAE,GAAG,KAAK,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC;QACtG,OAAO,GAAG,EAAE,YAAY,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;IAC/D,CAAC;IAED,cAAc;QACZ,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,UAAU;QACR,OAAO,KAAK,CAAC;IACf,CAAC;IAED,iBAAiB;QACf,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAAY,EAAE,OAAY;QAC/C,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,iBAAiB,EAAE,IAAI,IAAI,OAAO,EAAE,IAAI,CAAC,IAAI,SAAS,CAAC;QAC9E,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,aAAa;YAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;QAE9E,MAAM,EAAE,GAAG,KAAK,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,cAAc,CAAC;QAClI,OAAO;YACL,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE,GAAG,EAAE,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG;SACxD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAY,EAAE,OAAuB;QAC9C,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAEvD,IAAI,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;gBAC9B,IAAI,KAAK,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBACjC,8BAA8B;oBAC9B,MAAM,KAAK,GAAa;wBACtB,KAAK,EAAE,EAAE;wBACT,QAAQ,EAAE,EAAE;wBACZ,QAAQ,EAAE,CAAC;wBACX,cAAc,EAAE,CAAC;qBAClB,CAAC;oBACF,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;oBACvC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;wBAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;oBAC1D,MAAM,SAAS,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;oBACvE,OAAO;wBACL,IAAI,EAAE,yBAAyB,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;wBAC5D,OAAO,EAAE,yBAAyB,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,2DAA2D;wBACxH,KAAK,EAAE,iBAAiB,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;qBACtD,CAAC;gBACJ,CAAC;gBACD,OAAO;oBACL,IAAI,EAAE,EAAE;oBACR,OAAO,EAAE,uBAAuB,YAAY,EAAE;oBAC9C,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE,iBAAiB,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;iBACtD,CAAC;YACJ,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAClD,MAAM,EAAE,GAAa,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAErC,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1C,OAAO;oBACL,IAAI,EAAE,EAAE;oBACR,OAAO,EAAE,8BAA8B,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;oBACpE,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE,iBAAiB,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;iBACtD,CAAC;YACJ,CAAC;YAED,IAAI,KAAK,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBACjC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;oBACnB,OAAO;wBACL,IAAI,EAAE,EAAE;wBACR,OAAO,EAAE,sCAAsC;wBAC/C,OAAO,EAAE,IAAI;wBACb,KAAK,EAAE,oBAAoB;qBAC5B,CAAC;gBACJ,CAAC;gBAED,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC9D,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;oBACf,OAAO;wBACL,IAAI,EAAE,EAAE;wBACR,OAAO,EAAE,mBAAmB,KAAK,CAAC,OAAO,EAAE;wBAC3C,OAAO,EAAE,IAAI;wBACb,KAAK,EAAE,8BAA8B;qBACtC,CAAC;gBACJ,CAAC;gBAED,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBAC9B,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;gBAExB,MAAM,SAAS,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;gBAEpE,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,KAAK,MAAM;oBAC5C,CAAC,CAAC,IAAI,OAAO,CAAC,eAAe,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE;oBACvF,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;gBAEjE,OAAO;oBACL,IAAI,EAAE,gBAAgB,GAAG,GAAG,CAAC,KAAK,SAAS,EAAE;oBAC7C,OAAO,EAAE,gBAAgB,GAAG,GAAG,CAAC,SAAS,KAAK,CAAC,OAAO,UAAU,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;oBAC7F,KAAK,EAAE,4BAA4B;iBACpC,CAAC;YACJ,CAAC;YAED,IAAI,KAAK,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBACjC,MAAM,QAAQ,GAAG,KAAK,CAAC,SAAS,IAAI,MAAM,CAAC;gBAC3C,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,CAC/D,CAAC,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAClC,CAAC;gBAEF,MAAM,OAAO,GAAiB;oBAC5B,SAAS,EAAE,QAAQ;oBACnB,MAAM,EAAE,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;oBACjD,EAAE,EAAE,cAAc,EAAE;oBACpB,eAAe,EAAE,IAAI;iBACtB,CAAC;gBAEF,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;oBACxB,OAAO,CAAC,OAAO,GAAG,EAAE,CAAC;gBACvB,CAAC;gBAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBAClB,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,OAAO,CAAC,CAAC;oBAC9D,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;wBACf,OAAO;4BACL,IAAI,EAAE,EAAE;4BACR,OAAO,EAAE,mBAAmB,KAAK,CAAC,OAAO,wCAAwC;4BACjF,OAAO,EAAE,IAAI;4BACb,KAAK,EAAE,8BAA8B;yBACtC,CAAC;oBACJ,CAAC;oBACD,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;gBACvC,CAAC;qBAAM,CAAC;oBACN,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC5B,CAAC;gBAED,MAAM,SAAS,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;gBAEpE,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO;oBACvB,CAAC,CAAC,cAAc,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,OAAO,CAAC,EAAE;oBACnE,CAAC,CAAC,cAAc,CAAC;gBAEnB,OAAO;oBACL,IAAI,EAAE,YAAY,QAAQ,SAAS,GAAG,EAAE;oBACxC,OAAO,EAAE,gBAAgB,QAAQ,cAAc,OAAO,CAAC,EAAE,KAAK,GAAG,OAAO,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;oBACrG,KAAK,EAAE,0BAA0B,QAAQ,OAAO;iBACjD,CAAC;YACJ,CAAC;YAED,0BAA0B;YAC1B,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBACnB,OAAO;oBACL,IAAI,EAAE,EAAE;oBACR,OAAO,EAAE,uCAAuC;oBAChD,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE,oBAAoB;iBAC5B,CAAC;YACJ,CAAC;YAED,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,OAAO,CAAC,CAAC;YAClE,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;gBACnB,OAAO;oBACL,IAAI,EAAE,EAAE;oBACR,OAAO,EAAE,mBAAmB,KAAK,CAAC,OAAO,EAAE;oBAC3C,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE,8BAA8B;iBACtC,CAAC;YACJ,CAAC;YAED,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAClC,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,CAC/D,CAAC,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAClC,CAAC;YAEF,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG;gBAClB,GAAG,OAAO;gBACV,MAAM,EAAE,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;gBACjD,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS;aAChD,CAAC;YAEF,+CAA+C;YAC/C,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,KAAK,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC7D,IAAI,KAAK,CAAC,SAAS,KAAK,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;oBAC7D,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,OAAO,GAAG,EAAE,CAAC;gBACjC,CAAC;qBAAM,IAAI,KAAK,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;oBAC1C,OAAO,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;oBACjC,OAAO,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC;gBAC3C,CAAC;YACH,CAAC;YAED,MAAM,SAAS,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YAEpE,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9C,OAAO;gBACL,IAAI,EAAE,iBAAiB,OAAO,GAAG,CAAC,SAAS,KAAK,CAAC,OAAO,GAAG;gBAC3D,OAAO,EAAE,iBAAiB,OAAO,GAAG,CAAC,SAAS,KAAK,CAAC,OAAO,QAAQ,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,MAAM,OAAO,GAAG;gBAC9G,KAAK,EAAE,6BAA6B;aACrC,CAAC;QACJ,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO;gBACL,IAAI,EAAE,EAAE;gBACR,OAAO,EAAE,2BAA2B,YAAY,KAAK,GAAG,CAAC,OAAO,EAAE;gBAClE,OAAO,EAAE,IAAI;gBACb,KAAK,EAAE,uBAAuB,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;aAC5D,CAAC;QACJ,CAAC;IACH,CAAC;IAED,oBAAoB,CAAC,KAAY;QAC/B,MAAM,EAAE,GAAG,KAAK,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,iBAAiB,CAAC;QAC1I,OAAO,GAAG,EAAE,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC;IAC1D,CAAC;CACF,CAAC,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* NotebookReadTool — Jupyter notebook (.ipynb) reader for Cray Code.
|
|
3
|
+
* Displays cells with their outputs (code, text, visualizations).
|
|
4
|
+
* Matches Claude Code's NotebookRead capabilities.
|
|
5
|
+
*/
|
|
6
|
+
import { z } from 'zod';
|
|
7
|
+
import { type ToolDef } from '../Tool.js';
|
|
8
|
+
export declare const NOTEBOOK_READ_TOOL_NAME = "notebook_read";
|
|
9
|
+
export declare const NotebookReadTool: ToolDef<z.ZodObject<{
|
|
10
|
+
file_path: z.ZodString;
|
|
11
|
+
pages: z.ZodOptional<z.ZodString>;
|
|
12
|
+
}, "strip", z.ZodTypeAny, {
|
|
13
|
+
file_path: string;
|
|
14
|
+
pages?: string | undefined;
|
|
15
|
+
}, {
|
|
16
|
+
file_path: string;
|
|
17
|
+
pages?: string | undefined;
|
|
18
|
+
}>, string>;
|
|
19
|
+
//# sourceMappingURL=NotebookReadTool.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"NotebookReadTool.d.ts","sourceRoot":"","sources":["../../src/tools/NotebookReadTool.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,OAAO,EAAa,KAAK,OAAO,EAAwC,MAAM,YAAY,CAAC;AAG3F,eAAO,MAAM,uBAAuB,kBAAkB,CAAC;AAmFvD,eAAO,MAAM,gBAAgB;;;;;;;;;WAyJ3B,CAAC"}
|
|
@@ -0,0 +1,191 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* NotebookReadTool — Jupyter notebook (.ipynb) reader for Cray Code.
|
|
3
|
+
* Displays cells with their outputs (code, text, visualizations).
|
|
4
|
+
* Matches Claude Code's NotebookRead capabilities.
|
|
5
|
+
*/
|
|
6
|
+
import { z } from 'zod';
|
|
7
|
+
import { readFile } from 'fs/promises';
|
|
8
|
+
import { existsSync, statSync } from 'fs';
|
|
9
|
+
import * as path from 'path';
|
|
10
|
+
import { buildTool } from '../Tool.js';
|
|
11
|
+
export const NOTEBOOK_READ_TOOL_NAME = 'notebook_read';
|
|
12
|
+
const inputSchema = z.object({
|
|
13
|
+
file_path: z.string().describe('The absolute path to the .ipynb notebook file to read'),
|
|
14
|
+
pages: z.string().optional().describe('Cell range to read (e.g., "1-5", "3", "10-20"). 1-indexed.'),
|
|
15
|
+
});
|
|
16
|
+
function parsePageRange(pages, maxCells) {
|
|
17
|
+
if (pages.includes('-')) {
|
|
18
|
+
const [a, b] = pages.split('-').map(Number);
|
|
19
|
+
return { start: Math.max(1, a) - 1, end: Math.min(maxCells, b) };
|
|
20
|
+
}
|
|
21
|
+
const n = Number(pages);
|
|
22
|
+
return { start: n - 1, end: n };
|
|
23
|
+
}
|
|
24
|
+
function formatCellSource(source) {
|
|
25
|
+
return Array.isArray(source) ? source.join('') : source;
|
|
26
|
+
}
|
|
27
|
+
function formatOutput(output) {
|
|
28
|
+
const lines = [];
|
|
29
|
+
lines.push(` [${output.output_type}]`);
|
|
30
|
+
if (output.name)
|
|
31
|
+
lines.push(` name: ${output.name}`);
|
|
32
|
+
if (output.text) {
|
|
33
|
+
const text = Array.isArray(output.text) ? output.text.join('') : output.text;
|
|
34
|
+
for (const l of String(text).split('\n')) {
|
|
35
|
+
lines.push(` ${l}`);
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
if (output.data) {
|
|
39
|
+
for (const [mime, rawVal] of Object.entries(output.data)) {
|
|
40
|
+
const val = Array.isArray(rawVal) ? rawVal.join('') : String(rawVal);
|
|
41
|
+
if (mime.startsWith('image/')) {
|
|
42
|
+
lines.push(` [${mime}: ${Math.ceil(val.length * 0.75)} bytes]`);
|
|
43
|
+
}
|
|
44
|
+
else if (val.length < 5000) {
|
|
45
|
+
lines.push(` [${mime}]:`);
|
|
46
|
+
for (const l of val.split('\n').slice(0, 40)) {
|
|
47
|
+
lines.push(` ${l}`);
|
|
48
|
+
}
|
|
49
|
+
}
|
|
50
|
+
else {
|
|
51
|
+
lines.push(` [${mime}: ${val.length} chars]`);
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
return lines.join('\n');
|
|
56
|
+
}
|
|
57
|
+
export const NotebookReadTool = buildTool({
|
|
58
|
+
name: NOTEBOOK_READ_TOOL_NAME,
|
|
59
|
+
aliases: ['nb_read', 'read_notebook'],
|
|
60
|
+
searchHint: 'jupyter notebook ipynb read cells',
|
|
61
|
+
inputSchema,
|
|
62
|
+
async description(input) {
|
|
63
|
+
return `Reads notebook: ${path.basename(input.file_path)}`;
|
|
64
|
+
},
|
|
65
|
+
userFacingName() {
|
|
66
|
+
return 'NotebookRead';
|
|
67
|
+
},
|
|
68
|
+
isReadOnly() {
|
|
69
|
+
return true;
|
|
70
|
+
},
|
|
71
|
+
isConcurrencySafe() {
|
|
72
|
+
return true;
|
|
73
|
+
},
|
|
74
|
+
async checkPermissions(_input, _context) {
|
|
75
|
+
return { behavior: 'passthrough' };
|
|
76
|
+
},
|
|
77
|
+
async call(input, context) {
|
|
78
|
+
const filePath = path.resolve(input.file_path);
|
|
79
|
+
try {
|
|
80
|
+
if (!existsSync(filePath)) {
|
|
81
|
+
return {
|
|
82
|
+
data: '',
|
|
83
|
+
content: `Notebook not found: ${filePath}`,
|
|
84
|
+
isError: true,
|
|
85
|
+
title: `NotebookRead: ${path.basename(filePath)}`,
|
|
86
|
+
};
|
|
87
|
+
}
|
|
88
|
+
const stats = statSync(filePath);
|
|
89
|
+
if (stats.size > 50 * 1024 * 1024) {
|
|
90
|
+
return {
|
|
91
|
+
data: '',
|
|
92
|
+
content: `Notebook too large (${(stats.size / 1024 / 1024).toFixed(1)}MB). Max 50MB.`,
|
|
93
|
+
isError: true,
|
|
94
|
+
title: `NotebookRead: ${path.basename(filePath)} (too large)`,
|
|
95
|
+
};
|
|
96
|
+
}
|
|
97
|
+
const raw = await readFile(filePath, 'utf-8');
|
|
98
|
+
const nb = JSON.parse(raw);
|
|
99
|
+
if (!nb.cells || !Array.isArray(nb.cells)) {
|
|
100
|
+
return {
|
|
101
|
+
data: '',
|
|
102
|
+
content: `Invalid notebook format in ${path.basename(filePath)}`,
|
|
103
|
+
isError: true,
|
|
104
|
+
title: `NotebookRead: ${path.basename(filePath)} (invalid)`,
|
|
105
|
+
};
|
|
106
|
+
}
|
|
107
|
+
const totalCells = nb.cells.length;
|
|
108
|
+
let startIdx = 0;
|
|
109
|
+
let endIdx = totalCells;
|
|
110
|
+
if (input.pages) {
|
|
111
|
+
const range = parsePageRange(input.pages, totalCells);
|
|
112
|
+
startIdx = range.start;
|
|
113
|
+
endIdx = range.end;
|
|
114
|
+
}
|
|
115
|
+
// Limit: max 100 cells per read
|
|
116
|
+
endIdx = Math.min(endIdx, startIdx + 100);
|
|
117
|
+
const parts = [];
|
|
118
|
+
parts.push(`Notebook: ${path.basename(filePath)}`);
|
|
119
|
+
parts.push(`Format: nbformat v${nb.nbformat}.${nb.nbformat_minor}`);
|
|
120
|
+
parts.push(`Total cells: ${totalCells}`);
|
|
121
|
+
if (input.pages)
|
|
122
|
+
parts.push(`Showing cells: ${startIdx + 1}-${endIdx}`);
|
|
123
|
+
parts.push('');
|
|
124
|
+
let charCount = 0;
|
|
125
|
+
const MAX_OUTPUT = 80_000;
|
|
126
|
+
for (let i = startIdx; i < endIdx; i++) {
|
|
127
|
+
const cell = nb.cells[i];
|
|
128
|
+
const cellNum = i + 1;
|
|
129
|
+
const source = formatCellSource(cell.source);
|
|
130
|
+
const execCount = cell.execution_count;
|
|
131
|
+
const execLabel = execCount != null ? ` [${execCount}]` : '';
|
|
132
|
+
if (cell.cell_type === 'markdown') {
|
|
133
|
+
parts.push(`[${cellNum}] Markdown:${execLabel}`);
|
|
134
|
+
for (const l of source.split('\n').slice(0, 30)) {
|
|
135
|
+
parts.push(` ${l}`);
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
else if (cell.cell_type === 'code') {
|
|
139
|
+
parts.push(`[${cellNum}] Code:${execLabel}`);
|
|
140
|
+
for (const l of source.split('\n').slice(0, 50)) {
|
|
141
|
+
parts.push(` ${l}`);
|
|
142
|
+
}
|
|
143
|
+
if (cell.outputs && cell.outputs.length > 0) {
|
|
144
|
+
parts.push(` --- outputs (${cell.outputs.length}) ---`);
|
|
145
|
+
for (const out of cell.outputs.slice(0, 5)) {
|
|
146
|
+
parts.push(formatOutput(out));
|
|
147
|
+
}
|
|
148
|
+
if (cell.outputs.length > 5) {
|
|
149
|
+
parts.push(` ... ${cell.outputs.length - 5} more outputs`);
|
|
150
|
+
}
|
|
151
|
+
}
|
|
152
|
+
}
|
|
153
|
+
else {
|
|
154
|
+
parts.push(`[${cellNum}] Raw:`);
|
|
155
|
+
for (const l of source.split('\n').slice(0, 20)) {
|
|
156
|
+
parts.push(` ${l}`);
|
|
157
|
+
}
|
|
158
|
+
}
|
|
159
|
+
parts.push('');
|
|
160
|
+
// Truncation check
|
|
161
|
+
const current = parts.join('\n').length;
|
|
162
|
+
if (current > MAX_OUTPUT) {
|
|
163
|
+
parts.push(`... [truncated at ${MAX_OUTPUT / 1000}KB, ${endIdx - i - 1} cells remaining]`);
|
|
164
|
+
break;
|
|
165
|
+
}
|
|
166
|
+
}
|
|
167
|
+
if (endIdx < totalCells) {
|
|
168
|
+
parts.push(`... [${totalCells - endIdx} more cells available, use pages="${endIdx + 1}-${Math.min(endIdx + 100, totalCells)}" to continue]`);
|
|
169
|
+
}
|
|
170
|
+
const result = parts.join('\n');
|
|
171
|
+
return {
|
|
172
|
+
data: result,
|
|
173
|
+
content: result,
|
|
174
|
+
truncated: endIdx < totalCells || result.length >= MAX_OUTPUT,
|
|
175
|
+
title: `NotebookRead: ${path.basename(filePath)} (cells ${startIdx + 1}-${endIdx} of ${totalCells})`,
|
|
176
|
+
};
|
|
177
|
+
}
|
|
178
|
+
catch (err) {
|
|
179
|
+
return {
|
|
180
|
+
data: '',
|
|
181
|
+
content: `Error reading notebook ${filePath}: ${err.message}`,
|
|
182
|
+
isError: true,
|
|
183
|
+
title: `NotebookRead Error: ${path.basename(filePath)}`,
|
|
184
|
+
};
|
|
185
|
+
}
|
|
186
|
+
},
|
|
187
|
+
renderToolUseMessage(input) {
|
|
188
|
+
return `Reading notebook ${path.basename(input.file_path)}...`;
|
|
189
|
+
},
|
|
190
|
+
});
|
|
191
|
+
//# sourceMappingURL=NotebookReadTool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"NotebookReadTool.js","sourceRoot":"","sources":["../../src/tools/NotebookReadTool.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAsD,MAAM,YAAY,CAAC;AAG3F,MAAM,CAAC,MAAM,uBAAuB,GAAG,eAAe,CAAC;AAEvD,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uDAAuD,CAAC;IACvF,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4DAA4D,CAAC;CACpG,CAAC,CAAC;AAyBH,SAAS,cAAc,CAAC,KAAa,EAAE,QAAgB;IACrD,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC5C,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC;IACnE,CAAC;IACD,MAAM,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IACxB,OAAO,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;AAClC,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAyB;IACjD,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;AAC1D,CAAC;AAUD,SAAS,YAAY,CAAC,MAAkB;IACtC,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,CAAC,IAAI,CAAC,MAAM,MAAM,CAAC,WAAW,GAAG,CAAC,CAAC;IAExC,IAAI,MAAM,CAAC,IAAI;QAAE,KAAK,CAAC,IAAI,CAAC,WAAW,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IAEtD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC;QAC7E,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACzD,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACrE,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC9B,KAAK,CAAC,IAAI,CAAC,MAAM,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;YACnE,CAAC;iBAAM,IAAI,GAAG,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC;gBAC7B,KAAK,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;gBAC3B,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;oBAC7C,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,IAAI,CAAC,MAAM,IAAI,KAAK,GAAG,CAAC,MAAM,SAAS,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,CAAC,MAAM,gBAAgB,GAAG,SAAS,CAAC;IACxC,IAAI,EAAE,uBAAuB;IAC7B,OAAO,EAAE,CAAC,SAAS,EAAE,eAAe,CAAC;IACrC,UAAU,EAAE,mCAAmC;IAC/C,WAAW;IAEX,KAAK,CAAC,WAAW,CAAC,KAAY;QAC5B,OAAO,mBAAmB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;IAC7D,CAAC;IAED,cAAc;QACZ,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC;IACd,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,MAAa,EAAE,QAAa;QACjD,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAY,EAAE,OAAuB;QAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAE/C,IAAI,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC1B,OAAO;oBACL,IAAI,EAAE,EAAE;oBACR,OAAO,EAAE,uBAAuB,QAAQ,EAAE;oBAC1C,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE,iBAAiB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE;iBAClD,CAAC;YACJ,CAAC;YAED,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAEjC,IAAI,KAAK,CAAC,IAAI,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC;gBAClC,OAAO;oBACL,IAAI,EAAE,EAAE;oBACR,OAAO,EAAE,uBAAuB,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,gBAAgB;oBACrF,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE,iBAAiB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,cAAc;iBAC9D,CAAC;YACJ,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC9C,MAAM,EAAE,GAAa,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAErC,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1C,OAAO;oBACL,IAAI,EAAE,EAAE;oBACR,OAAO,EAAE,8BAA8B,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE;oBAChE,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE,iBAAiB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY;iBAC5D,CAAC;YACJ,CAAC;YAED,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC;YACnC,IAAI,QAAQ,GAAG,CAAC,CAAC;YACjB,IAAI,MAAM,GAAG,UAAU,CAAC;YAExB,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;gBAChB,MAAM,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;gBACtD,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC;gBACvB,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC;YACrB,CAAC;YAED,gCAAgC;YAChC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,QAAQ,GAAG,GAAG,CAAC,CAAC;YAE1C,MAAM,KAAK,GAAa,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACnD,KAAK,CAAC,IAAI,CAAC,qBAAqB,EAAE,CAAC,QAAQ,IAAI,EAAE,CAAC,cAAc,EAAE,CAAC,CAAC;YACpE,KAAK,CAAC,IAAI,CAAC,gBAAgB,UAAU,EAAE,CAAC,CAAC;YACzC,IAAI,KAAK,CAAC,KAAK;gBAAE,KAAK,CAAC,IAAI,CAAC,kBAAkB,QAAQ,GAAG,CAAC,IAAI,MAAM,EAAE,CAAC,CAAC;YACxE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAEf,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,MAAM,UAAU,GAAG,MAAM,CAAC;YAE1B,KAAK,IAAI,CAAC,GAAG,QAAQ,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,IAAI,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACzB,MAAM,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC;gBACtB,MAAM,MAAM,GAAG,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC7C,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC;gBACvC,MAAM,SAAS,GAAG,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,SAAS,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBAE7D,IAAI,IAAI,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;oBAClC,KAAK,CAAC,IAAI,CAAC,IAAI,OAAO,cAAc,SAAS,EAAE,CAAC,CAAC;oBACjD,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;wBAChD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;oBACvB,CAAC;gBACH,CAAC;qBAAM,IAAI,IAAI,CAAC,SAAS,KAAK,MAAM,EAAE,CAAC;oBACrC,KAAK,CAAC,IAAI,CAAC,IAAI,OAAO,UAAU,SAAS,EAAE,CAAC,CAAC;oBAC7C,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;wBAChD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;oBACvB,CAAC;oBACD,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC5C,KAAK,CAAC,IAAI,CAAC,kBAAkB,IAAI,CAAC,OAAO,CAAC,MAAM,OAAO,CAAC,CAAC;wBACzD,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;4BAC3C,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC;wBAChC,CAAC;wBACD,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BAC5B,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,eAAe,CAAC,CAAC;wBAC9D,CAAC;oBACH,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,KAAK,CAAC,IAAI,CAAC,IAAI,OAAO,QAAQ,CAAC,CAAC;oBAChC,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;wBAChD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;oBACvB,CAAC;gBACH,CAAC;gBAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAEf,mBAAmB;gBACnB,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;gBACxC,IAAI,OAAO,GAAG,UAAU,EAAE,CAAC;oBACzB,KAAK,CAAC,IAAI,CAAC,qBAAqB,UAAU,GAAG,IAAI,OAAO,MAAM,GAAG,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;oBAC3F,MAAM;gBACR,CAAC;YACH,CAAC;YAED,IAAI,MAAM,GAAG,UAAU,EAAE,CAAC;gBACxB,KAAK,CAAC,IAAI,CAAC,QAAQ,UAAU,GAAG,MAAM,qCAAqC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,GAAG,EAAE,UAAU,CAAC,gBAAgB,CAAC,CAAC;YAC/I,CAAC;YAED,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEhC,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,MAAM;gBACf,SAAS,EAAE,MAAM,GAAG,UAAU,IAAI,MAAM,CAAC,MAAM,IAAI,UAAU;gBAC7D,KAAK,EAAE,iBAAiB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,QAAQ,GAAG,CAAC,IAAI,MAAM,OAAO,UAAU,GAAG;aACrG,CAAC;QACJ,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO;gBACL,IAAI,EAAE,EAAE;gBACR,OAAO,EAAE,0BAA0B,QAAQ,KAAK,GAAG,CAAC,OAAO,EAAE;gBAC7D,OAAO,EAAE,IAAI;gBACb,KAAK,EAAE,uBAAuB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE;aACxD,CAAC;QACJ,CAAC;IACH,CAAC;IAED,oBAAoB,CAAC,KAAY;QAC/B,OAAO,oBAAoB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC;IACjE,CAAC;CACF,CAAC,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* PlanTools — Enter/Exit plan mode tools for Cray Code.
|
|
3
|
+
* Plan mode allows the agent to design an approach before implementation.
|
|
4
|
+
*/
|
|
5
|
+
import { z } from 'zod';
|
|
6
|
+
import { type ToolDef } from '../Tool.js';
|
|
7
|
+
export declare const ENTER_PLAN_TOOL_NAME = "enter_plan_mode";
|
|
8
|
+
export declare const EnterPlanModeTool: ToolDef<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, string>;
|
|
9
|
+
export declare const EXIT_PLAN_TOOL_NAME = "exit_plan_mode";
|
|
10
|
+
export declare const ExitPlanModeTool: ToolDef<z.ZodObject<{
|
|
11
|
+
plan: z.ZodString;
|
|
12
|
+
}, "strip", z.ZodTypeAny, {
|
|
13
|
+
plan: string;
|
|
14
|
+
}, {
|
|
15
|
+
plan: string;
|
|
16
|
+
}>, string>;
|
|
17
|
+
//# sourceMappingURL=PlanTools.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"PlanTools.d.ts","sourceRoot":"","sources":["../../src/tools/PlanTools.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAa,KAAK,OAAO,EAAwC,MAAM,YAAY,CAAC;AAE3F,eAAO,MAAM,oBAAoB,oBAAoB,CAAC;AAItD,eAAO,MAAM,iBAAiB,iEA8B5B,CAAC;AAEH,eAAO,MAAM,mBAAmB,mBAAmB,CAAC;AAMpD,eAAO,MAAM,gBAAgB;;;;;;WA8B3B,CAAC"}
|