@cjhyy/code-shell-core 0.6.0-rc.13 → 0.6.0-rc.15
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/capability-control/service.d.ts +2 -0
- package/dist/capability-control/service.js +4 -2
- package/dist/cc-orchestrator/codex-session-history.js +1 -1
- package/dist/cc-orchestrator/external-agent-bindings.js +1 -1
- package/dist/cc-orchestrator/external-agent-session-store.js +3 -1
- package/dist/cc-orchestrator/session-history.js +2 -2
- package/dist/cli/agent-server-stdio.js +7 -2
- package/dist/context/compaction.d.ts +8 -1
- package/dist/context/compaction.js +49 -4
- package/dist/context/manager.d.ts +11 -2
- package/dist/context/manager.js +64 -3
- package/dist/credentials/inject-credential-tool.d.ts +3 -1
- package/dist/credentials/inject-credential-tool.js +29 -10
- package/dist/credentials/use-credential-tool.d.ts +1 -0
- package/dist/credentials/use-credential-tool.js +3 -0
- package/dist/engine/engine.d.ts +9 -0
- package/dist/engine/engine.js +96 -17
- package/dist/engine/turn-loop.d.ts +3 -1
- package/dist/engine/turn-loop.js +4 -2
- package/dist/engine/types.d.ts +8 -0
- package/dist/git/worktree/crud.d.ts +69 -0
- package/dist/git/worktree/crud.js +206 -0
- package/dist/git/worktree/diff.d.ts +14 -0
- package/dist/git/worktree/diff.js +82 -0
- package/dist/git/worktree/git-exec.d.ts +7 -0
- package/dist/git/worktree/git-exec.js +51 -0
- package/dist/git/worktree/index.d.ts +5 -0
- package/dist/git/worktree/index.js +5 -0
- package/dist/git/worktree/query.d.ts +42 -0
- package/dist/git/worktree/query.js +121 -0
- package/dist/git/worktree/slug.d.ts +11 -0
- package/dist/git/worktree/slug.js +58 -0
- package/dist/git/worktree.d.ts +1 -127
- package/dist/git/worktree.js +5 -464
- package/dist/index.d.ts +28 -27
- package/dist/index.js +24 -24
- package/dist/plugins/installer/checkUpdate.d.ts +4 -1
- package/dist/plugins/installer/checkUpdate.js +4 -2
- package/dist/plugins/installer/install.js +2 -0
- package/dist/plugins/installer/parseSource.d.ts +4 -1
- package/dist/plugins/installer/parseSource.js +28 -10
- package/dist/plugins/installer/update.d.ts +4 -1
- package/dist/plugins/installer/update.js +5 -3
- package/dist/plugins/parseMarketplaceInput.d.ts +4 -1
- package/dist/plugins/parseMarketplaceInput.js +4 -3
- package/dist/preset/index.d.ts +1 -0
- package/dist/preset/index.js +6 -0
- package/dist/prompt/sections/base.md +1 -1
- package/dist/protocol/chat-session-manager.js +7 -0
- package/dist/protocol/server.d.ts +10 -0
- package/dist/protocol/server.js +88 -4
- package/dist/protocol/types.d.ts +6 -0
- package/dist/protocol/types.js +2 -0
- package/dist/runtime/background-shell.js +2 -3
- package/dist/services/auto-dream.d.ts +15 -4
- package/dist/services/auto-dream.js +20 -20
- package/dist/services/dream-consolidation.d.ts +2 -3
- package/dist/services/dream-consolidation.js +65 -15
- package/dist/services/extract-memories.d.ts +14 -5
- package/dist/services/extract-memories.js +20 -3
- package/dist/services/global-dream-promotion.d.ts +23 -0
- package/dist/services/global-dream-promotion.js +112 -0
- package/dist/services/memory-orchestrator.js +347 -34
- package/dist/session/memory.d.ts +56 -17
- package/dist/session/memory.js +337 -78
- package/dist/session/session-manager.d.ts +1 -1
- package/dist/session/session-manager.js +6 -6
- package/dist/session/transcript.d.ts +1 -1
- package/dist/session/transcript.js +17 -5
- package/dist/settings/manager.js +43 -12
- package/dist/settings/schema.d.ts +21 -0
- package/dist/settings/schema.js +12 -0
- package/dist/tool-system/builtin/index.js +18 -8
- package/dist/tool-system/builtin/memory.js +40 -8
- package/dist/tool-system/builtin/view-image.d.ts +2 -2
- package/dist/tool-system/builtin/view-image.js +100 -19
- package/dist/tool-system/builtin/worktree.d.ts +2 -0
- package/dist/tool-system/builtin/worktree.js +59 -11
- package/dist/tool-system/context.d.ts +11 -1
- package/dist/tool-system/mcp-manager.d.ts +8 -5
- package/dist/tool-system/mcp-manager.js +85 -55
- package/dist/tool-system/path-policy.d.ts +2 -0
- package/dist/tool-system/path-policy.js +28 -12
- package/dist/tool-system/workspace-bridge.d.ts +11 -0
- package/dist/tool-system/workspace-bridge.js +1 -0
- package/dist/types.d.ts +14 -0
- package/package.json +1 -1
package/dist/settings/manager.js
CHANGED
|
@@ -51,14 +51,35 @@ export function isProtectedSettingKey(key) {
|
|
|
51
51
|
return PROTECTED_SETTING_ROOTS.has(root);
|
|
52
52
|
}
|
|
53
53
|
const FORBIDDEN_SETTING_KEY_SEGMENTS = new Set(["__proto__", "prototype", "constructor"]);
|
|
54
|
+
function isForbiddenSettingKeySegment(key) {
|
|
55
|
+
return FORBIDDEN_SETTING_KEY_SEGMENTS.has(key);
|
|
56
|
+
}
|
|
54
57
|
function parseDottedSettingKey(key) {
|
|
55
58
|
const parts = key.split(".");
|
|
56
59
|
if (parts.length === 0 ||
|
|
57
|
-
parts.some((seg) => seg.length === 0 ||
|
|
60
|
+
parts.some((seg) => seg.length === 0 || isForbiddenSettingKeySegment(seg))) {
|
|
58
61
|
throw new Error(`invalid setting key: ${key}`);
|
|
59
62
|
}
|
|
60
63
|
return parts;
|
|
61
64
|
}
|
|
65
|
+
function sanitizeSettingsValue(value) {
|
|
66
|
+
if (Array.isArray(value)) {
|
|
67
|
+
return value.map((entry) => sanitizeSettingsValue(entry));
|
|
68
|
+
}
|
|
69
|
+
if (!value || typeof value !== "object") {
|
|
70
|
+
return value;
|
|
71
|
+
}
|
|
72
|
+
const out = {};
|
|
73
|
+
for (const [key, child] of Object.entries(value)) {
|
|
74
|
+
if (isForbiddenSettingKeySegment(key))
|
|
75
|
+
continue;
|
|
76
|
+
out[key] = sanitizeSettingsValue(child);
|
|
77
|
+
}
|
|
78
|
+
return out;
|
|
79
|
+
}
|
|
80
|
+
function sanitizeSettingsObject(data) {
|
|
81
|
+
return sanitizeSettingsValue(data);
|
|
82
|
+
}
|
|
62
83
|
function isOwnPlainObject(parent, key) {
|
|
63
84
|
if (!Object.prototype.hasOwnProperty.call(parent, key))
|
|
64
85
|
return false;
|
|
@@ -174,7 +195,7 @@ export class SettingsManager {
|
|
|
174
195
|
}
|
|
175
196
|
// 5. CLI flags (highest priority)
|
|
176
197
|
if (flagOverrides && Object.keys(flagOverrides).length > 0) {
|
|
177
|
-
this.sources.push({ name: "flag", priority: 4, data: flagOverrides });
|
|
198
|
+
this.sources.push({ name: "flag", priority: 4, data: sanitizeSettingsObject(flagOverrides) });
|
|
178
199
|
}
|
|
179
200
|
// Sort by priority ascending (merge in order, later wins)
|
|
180
201
|
this.sources.sort((a, b) => a.priority - b.priority);
|
|
@@ -215,7 +236,7 @@ export class SettingsManager {
|
|
|
215
236
|
const userPath = join(userHome(), ".code-shell", "settings.json");
|
|
216
237
|
if (readUser && existsSync(userPath)) {
|
|
217
238
|
try {
|
|
218
|
-
const userRaw = JSON.parse(readFileSync(userPath, "utf-8"));
|
|
239
|
+
const userRaw = sanitizeSettingsObject(JSON.parse(readFileSync(userPath, "utf-8")));
|
|
219
240
|
const result = migrateModels({
|
|
220
241
|
providers: userRaw.providers ?? [],
|
|
221
242
|
models: userRaw.models ?? [],
|
|
@@ -227,14 +248,15 @@ export class SettingsManager {
|
|
|
227
248
|
providers: result.providers,
|
|
228
249
|
models: result.models,
|
|
229
250
|
};
|
|
251
|
+
const sanitized = sanitizeSettingsObject(migrated);
|
|
230
252
|
// Atomic write (tmp+rename) — a concurrent load must not see a
|
|
231
253
|
// half-written file. File exists here (existsSync guard above).
|
|
232
|
-
this.atomicWriteJson(userPath,
|
|
254
|
+
this.atomicWriteJson(userPath, sanitized);
|
|
233
255
|
// Re-deep-merge with the migrated user data so the validate
|
|
234
256
|
// call sees the new shape rather than the legacy one.
|
|
235
257
|
const userSource = this.sources.find((s) => s.name === "user");
|
|
236
258
|
if (userSource)
|
|
237
|
-
userSource.data =
|
|
259
|
+
userSource.data = sanitized;
|
|
238
260
|
const remerged = this.deepMerge();
|
|
239
261
|
this.merged = validateSettings(remerged);
|
|
240
262
|
return this.merged;
|
|
@@ -263,7 +285,7 @@ export class SettingsManager {
|
|
|
263
285
|
const parsed = JSON.parse(readFileSync(path, "utf-8"));
|
|
264
286
|
if (!parsed || typeof parsed !== "object" || Array.isArray(parsed))
|
|
265
287
|
return;
|
|
266
|
-
const raw = parsed;
|
|
288
|
+
const raw = sanitizeSettingsObject(parsed);
|
|
267
289
|
const result = migrateConfig(raw);
|
|
268
290
|
if (!result.changed)
|
|
269
291
|
return;
|
|
@@ -279,10 +301,11 @@ export class SettingsManager {
|
|
|
279
301
|
// Atomic write (tmp+rename) so a concurrent load can't read a half-written
|
|
280
302
|
// migrated file — matches the normal save path (atomicWriteJson). The file
|
|
281
303
|
// exists here (existsSync guard above), so the recursive mkdir is a no-op.
|
|
282
|
-
|
|
304
|
+
const sanitized = sanitizeSettingsObject(result.config);
|
|
305
|
+
this.atomicWriteJson(path, sanitized);
|
|
283
306
|
const source = this.sources.find((s) => s.name === sourceName);
|
|
284
307
|
if (source)
|
|
285
|
-
source.data =
|
|
308
|
+
source.data = sanitized;
|
|
286
309
|
}
|
|
287
310
|
catch {
|
|
288
311
|
// Best-effort — fall through to normal merge/validate.
|
|
@@ -318,7 +341,7 @@ export class SettingsManager {
|
|
|
318
341
|
const raw = readFileSync(path, "utf-8");
|
|
319
342
|
const parsed = JSON.parse(raw);
|
|
320
343
|
if (parsed && typeof parsed === "object" && !Array.isArray(parsed)) {
|
|
321
|
-
current = parsed;
|
|
344
|
+
current = sanitizeSettingsObject(parsed);
|
|
322
345
|
}
|
|
323
346
|
}
|
|
324
347
|
catch {
|
|
@@ -500,7 +523,7 @@ function parseConfigFile(path) {
|
|
|
500
523
|
const ext = extname(path).toLowerCase();
|
|
501
524
|
const parsed = ext === ".yaml" || ext === ".yml" ? parseYaml(content) : JSON.parse(content);
|
|
502
525
|
if (parsed && typeof parsed === "object" && !Array.isArray(parsed)) {
|
|
503
|
-
return parsed;
|
|
526
|
+
return sanitizeSettingsObject(parsed);
|
|
504
527
|
}
|
|
505
528
|
}
|
|
506
529
|
catch {
|
|
@@ -527,9 +550,17 @@ function resolveConfigPath(jsonPath) {
|
|
|
527
550
|
return null;
|
|
528
551
|
}
|
|
529
552
|
function merge(base, override) {
|
|
530
|
-
const result = {
|
|
553
|
+
const result = {};
|
|
554
|
+
for (const [key, value] of Object.entries(base)) {
|
|
555
|
+
if (isForbiddenSettingKeySegment(key))
|
|
556
|
+
continue;
|
|
557
|
+
result[key] = value;
|
|
558
|
+
}
|
|
531
559
|
for (const [key, value] of Object.entries(override)) {
|
|
532
|
-
if (
|
|
560
|
+
if (isForbiddenSettingKeySegment(key)) {
|
|
561
|
+
continue;
|
|
562
|
+
}
|
|
563
|
+
else if (value === null) {
|
|
533
564
|
delete result[key];
|
|
534
565
|
}
|
|
535
566
|
else if (typeof value === "object" &&
|
|
@@ -366,6 +366,13 @@ export declare const SettingsSchema: z.ZodObject<{
|
|
|
366
366
|
storageDir?: string | undefined;
|
|
367
367
|
maxHistory?: number | undefined;
|
|
368
368
|
}>>;
|
|
369
|
+
worktree: z.ZodDefault<z.ZodObject<{
|
|
370
|
+
branchPrefix: z.ZodDefault<z.ZodEffects<z.ZodEffects<z.ZodString, string, string>, string, string>>;
|
|
371
|
+
}, "strip", z.ZodTypeAny, {
|
|
372
|
+
branchPrefix: string;
|
|
373
|
+
}, {
|
|
374
|
+
branchPrefix?: string | undefined;
|
|
375
|
+
}>>;
|
|
369
376
|
mcpServers: z.ZodDefault<z.ZodEffects<z.ZodRecord<z.ZodString, z.ZodObject<{
|
|
370
377
|
name: z.ZodString;
|
|
371
378
|
command: z.ZodOptional<z.ZodString>;
|
|
@@ -1240,6 +1247,13 @@ export declare const SettingsSchema: z.ZodObject<{
|
|
|
1240
1247
|
storageDir?: string | undefined;
|
|
1241
1248
|
maxHistory?: number | undefined;
|
|
1242
1249
|
}>>;
|
|
1250
|
+
worktree: z.ZodDefault<z.ZodObject<{
|
|
1251
|
+
branchPrefix: z.ZodDefault<z.ZodEffects<z.ZodEffects<z.ZodString, string, string>, string, string>>;
|
|
1252
|
+
}, "strip", z.ZodTypeAny, {
|
|
1253
|
+
branchPrefix: string;
|
|
1254
|
+
}, {
|
|
1255
|
+
branchPrefix?: string | undefined;
|
|
1256
|
+
}>>;
|
|
1243
1257
|
mcpServers: z.ZodDefault<z.ZodEffects<z.ZodRecord<z.ZodString, z.ZodObject<{
|
|
1244
1258
|
name: z.ZodString;
|
|
1245
1259
|
command: z.ZodOptional<z.ZodString>;
|
|
@@ -2114,6 +2128,13 @@ export declare const SettingsSchema: z.ZodObject<{
|
|
|
2114
2128
|
storageDir?: string | undefined;
|
|
2115
2129
|
maxHistory?: number | undefined;
|
|
2116
2130
|
}>>;
|
|
2131
|
+
worktree: z.ZodDefault<z.ZodObject<{
|
|
2132
|
+
branchPrefix: z.ZodDefault<z.ZodEffects<z.ZodEffects<z.ZodString, string, string>, string, string>>;
|
|
2133
|
+
}, "strip", z.ZodTypeAny, {
|
|
2134
|
+
branchPrefix: string;
|
|
2135
|
+
}, {
|
|
2136
|
+
branchPrefix?: string | undefined;
|
|
2137
|
+
}>>;
|
|
2117
2138
|
mcpServers: z.ZodDefault<z.ZodEffects<z.ZodRecord<z.ZodString, z.ZodObject<{
|
|
2118
2139
|
name: z.ZodString;
|
|
2119
2140
|
command: z.ZodOptional<z.ZodString>;
|
package/dist/settings/schema.js
CHANGED
|
@@ -2,6 +2,7 @@
|
|
|
2
2
|
* Settings schema with Zod validation.
|
|
3
3
|
*/
|
|
4
4
|
import { z } from "zod";
|
|
5
|
+
import { DEFAULT_WORKTREE_BRANCH_PREFIX, isValidWorktreeBranchPrefix, normalizeWorktreeBranchPrefix, } from "../git/worktree/slug.js";
|
|
5
6
|
/**
|
|
6
7
|
* Tri-state project capability overlay. Lives in PROJECT settings only and
|
|
7
8
|
* layers over the global baseline (disabledSkills / disabledPlugins /
|
|
@@ -30,6 +31,12 @@ export const CapabilityOverridesSchema = z
|
|
|
30
31
|
pluginHooks: z.record(CapabilityOverrideSchema).optional(),
|
|
31
32
|
})
|
|
32
33
|
.optional();
|
|
34
|
+
const WorktreeBranchPrefixSchema = z
|
|
35
|
+
.string()
|
|
36
|
+
.trim()
|
|
37
|
+
.min(1)
|
|
38
|
+
.refine(isValidWorktreeBranchPrefix, "Invalid git branch prefix")
|
|
39
|
+
.transform((value) => normalizeWorktreeBranchPrefix(value));
|
|
33
40
|
export const SettingsSchema = z
|
|
34
41
|
.object({
|
|
35
42
|
agent: z
|
|
@@ -214,6 +221,11 @@ export const SettingsSchema = z
|
|
|
214
221
|
maxHistory: z.number().default(100),
|
|
215
222
|
})
|
|
216
223
|
.default({}),
|
|
224
|
+
worktree: z
|
|
225
|
+
.object({
|
|
226
|
+
branchPrefix: WorktreeBranchPrefixSchema.default(DEFAULT_WORKTREE_BRANCH_PREFIX),
|
|
227
|
+
})
|
|
228
|
+
.default({}),
|
|
217
229
|
// The record key IS the server name at runtime (MCPManager.connectAll
|
|
218
230
|
// uses Object.entries keys; desktop's persist strips the `name` field and
|
|
219
231
|
// keys by it via stripNameFromServer). So `name` here is optional and
|
|
@@ -3,9 +3,9 @@
|
|
|
3
3
|
*/
|
|
4
4
|
import { readToolDef, readTool } from "./read.js";
|
|
5
5
|
import { writeToolDef, writeTool } from "./write.js";
|
|
6
|
-
import { generateImageToolDef, generateImageTool, isGenerateImageAvailable } from "./generate-image.js";
|
|
6
|
+
import { generateImageToolDef, generateImageTool, isGenerateImageAvailable, } from "./generate-image.js";
|
|
7
7
|
import { editModelCatalogToolDef, editModelCatalogTool } from "./edit-model-catalog.js";
|
|
8
|
-
import { generateVideoToolDef, generateVideoTool, isGenerateVideoAvailable } from "./generate-video.js";
|
|
8
|
+
import { generateVideoToolDef, generateVideoTool, isGenerateVideoAvailable, } from "./generate-video.js";
|
|
9
9
|
import { viewImageToolDef, viewImageTool } from "./view-image.js";
|
|
10
10
|
import { editToolDef, editTool } from "./edit.js";
|
|
11
11
|
import { applyPatchToolDef, applyPatchTool } from "./apply-patch/index.js";
|
|
@@ -16,19 +16,19 @@ import { webSearchToolDef, webSearchTool, isWebSearchAvailable } from "./web-sea
|
|
|
16
16
|
import { webFetchToolDef, webFetchTool } from "./web-fetch.js";
|
|
17
17
|
import { askUserToolDef, askUserTool } from "./ask-user.js";
|
|
18
18
|
import { agentToolDef, agentTool, agentStatusToolDef, agentStatusTool, agentCancelToolDef, agentCancelTool, agentSendInputToolDef, agentSendInputTool, } from "./agent.js";
|
|
19
|
-
import { enterPlanModeToolDef, enterPlanModeTool, exitPlanModeToolDef, exitPlanModeTool } from "./plan.js";
|
|
19
|
+
import { enterPlanModeToolDef, enterPlanModeTool, exitPlanModeToolDef, exitPlanModeTool, } from "./plan.js";
|
|
20
20
|
import { toolSearchToolDef, toolSearchTool } from "./tool-search.js";
|
|
21
21
|
import { todoWriteToolDef, todoWriteTool } from "./task.js";
|
|
22
|
-
import { enterWorktreeToolDef, enterWorktreeTool, exitWorktreeToolDef, exitWorktreeTool } from "./worktree.js";
|
|
22
|
+
import { enterWorktreeToolDef, enterWorktreeTool, exitWorktreeToolDef, exitWorktreeTool, switchSessionWorkspaceToolDef, switchSessionWorkspaceTool, } from "./worktree.js";
|
|
23
23
|
import { sleepToolDef, sleepTool } from "./sleep.js";
|
|
24
24
|
import { configToolDef, configTool } from "./config.js";
|
|
25
25
|
import { notebookEditToolDef, notebookEditTool } from "./notebook-edit.js";
|
|
26
26
|
import { lspToolDef, lspTool } from "./lsp.js";
|
|
27
|
-
import { cronCreateToolDef, cronCreateTool, cronDeleteToolDef, cronDeleteTool, cronListToolDef, cronListTool } from "./cron.js";
|
|
28
|
-
import { driveClaudeCodeToolDef, driveClaudeCodeTool, driveAgentToolDef, driveAgentTool, DRIVE_AGENT_TOOL_TIMEOUT_MS } from "./drive-claude-code.js";
|
|
27
|
+
import { cronCreateToolDef, cronCreateTool, cronDeleteToolDef, cronDeleteTool, cronListToolDef, cronListTool, } from "./cron.js";
|
|
28
|
+
import { driveClaudeCodeToolDef, driveClaudeCodeTool, driveAgentToolDef, driveAgentTool, DRIVE_AGENT_TOOL_TIMEOUT_MS, } from "./drive-claude-code.js";
|
|
29
29
|
import { checkQuotaToolDef, checkQuotaTool } from "./check-quota.js";
|
|
30
30
|
import { skillToolDef, skillTool } from "./skill.js";
|
|
31
|
-
import { mcpToolDef, mcpToolExecute, listMcpResourcesToolDef, listMcpResourcesTool, readMcpResourceToolDef, readMcpResourceTool } from "./mcp-tools.js";
|
|
31
|
+
import { mcpToolDef, mcpToolExecute, listMcpResourcesToolDef, listMcpResourcesTool, readMcpResourceToolDef, readMcpResourceTool, } from "./mcp-tools.js";
|
|
32
32
|
import { replToolDef, replTool } from "./repl.js";
|
|
33
33
|
import { briefToolDef, briefTool } from "./brief.js";
|
|
34
34
|
import { powershellToolDef, powershellTool } from "./powershell.js";
|
|
@@ -344,6 +344,16 @@ export const BUILTIN_TOOLS = [
|
|
|
344
344
|
},
|
|
345
345
|
execute: exitWorktreeTool,
|
|
346
346
|
},
|
|
347
|
+
{
|
|
348
|
+
definition: {
|
|
349
|
+
...switchSessionWorkspaceToolDef,
|
|
350
|
+
source: "builtin",
|
|
351
|
+
permissionDefault: "ask",
|
|
352
|
+
isReadOnly: false,
|
|
353
|
+
isConcurrencySafe: false,
|
|
354
|
+
},
|
|
355
|
+
execute: switchSessionWorkspaceTool,
|
|
356
|
+
},
|
|
347
357
|
// ─── Phase 5: Utility Tools ────────────────────────────────────
|
|
348
358
|
{
|
|
349
359
|
definition: {
|
|
@@ -723,7 +733,7 @@ export const BUILTIN_TOOL_GUARDS = new Map([
|
|
|
723
733
|
[useCredentialToolDef.name, (ctx) => isUseCredentialAvailable(ctx.cwd)],
|
|
724
734
|
// InjectCredential hidden until ≥1 cookie credential exists (browser injection
|
|
725
735
|
// is cookie-only). Also degrades at call time if no browser bridge is wired.
|
|
726
|
-
[injectCredentialToolDef.name, (ctx) => isInjectCredentialAvailable(ctx.cwd)],
|
|
736
|
+
[injectCredentialToolDef.name, (ctx) => isInjectCredentialAvailable(ctx.cwd, ctx.settingsScope)],
|
|
727
737
|
[completeGoalToolDef.name, (ctx) => ctx.hasGoal === true],
|
|
728
738
|
[cancelGoalToolDef.name, (ctx) => ctx.hasGoal === true],
|
|
729
739
|
]);
|
|
@@ -17,9 +17,10 @@
|
|
|
17
17
|
import { MemoryManager } from "../../session/memory.js";
|
|
18
18
|
const VALID_SCOPES = ["user", "dream"];
|
|
19
19
|
const VALID_TYPES = ["user", "feedback", "project", "reference"];
|
|
20
|
+
const VALID_ORIGINS = ["auto", "manual", "dream"];
|
|
20
21
|
function parseScope(raw) {
|
|
21
22
|
if (typeof raw !== "string")
|
|
22
|
-
return
|
|
23
|
+
return 'Error: scope is required ("user" or "dream")';
|
|
23
24
|
if (!VALID_SCOPES.includes(raw)) {
|
|
24
25
|
return `Error: scope must be "user" or "dream", got "${raw}"`;
|
|
25
26
|
}
|
|
@@ -30,6 +31,11 @@ function parseScope(raw) {
|
|
|
30
31
|
function parseLocation(raw) {
|
|
31
32
|
return raw === "global" ? "global" : "project";
|
|
32
33
|
}
|
|
34
|
+
function parseOrigin(raw) {
|
|
35
|
+
return typeof raw === "string" && VALID_ORIGINS.includes(raw)
|
|
36
|
+
? raw
|
|
37
|
+
: undefined;
|
|
38
|
+
}
|
|
33
39
|
function mmFor(ctx, scope, location = "project") {
|
|
34
40
|
// location=global → omit projectDir so the manager points at the global store.
|
|
35
41
|
return new MemoryManager({
|
|
@@ -46,7 +52,7 @@ const LOCATION_SCHEMA = {
|
|
|
46
52
|
export const memoryListToolDef = {
|
|
47
53
|
name: "MemoryList",
|
|
48
54
|
description: "List persistent memory entries from one scope. " +
|
|
49
|
-
"Returns each entry's name, type, and short description (not the full content — use MemoryRead for that). " +
|
|
55
|
+
"Returns each entry's id, origin, counts, name, type, and short description (not the full content — use MemoryRead for that). " +
|
|
50
56
|
"Use this before MemorySave/MemoryDelete to find the exact name to target. " +
|
|
51
57
|
'Scopes: "user" (entries the user owns; you need permission to modify) or "dream" (auto-consolidation workspace; you may freely modify).',
|
|
52
58
|
inputSchema: {
|
|
@@ -73,7 +79,9 @@ export async function memoryListTool(args, ctx) {
|
|
|
73
79
|
if (entries.length === 0)
|
|
74
80
|
return `(no memories in scope "${scope}")`;
|
|
75
81
|
return entries
|
|
76
|
-
.map((e) => `- [${e.type}] ${e.name}
|
|
82
|
+
.map((e) => `- [${e.type}] ${e.name} ` +
|
|
83
|
+
`(id:${e.id ?? "(none)"}, origin:${e.origin ?? "manual"}, use:${e.useCount ?? 0}, updates:${e.updateCount ?? 0}) — ` +
|
|
84
|
+
e.description)
|
|
77
85
|
.join("\n");
|
|
78
86
|
}
|
|
79
87
|
catch (err) {
|
|
@@ -127,15 +135,24 @@ export async function memoryReadTool(args, ctx) {
|
|
|
127
135
|
if (cb) {
|
|
128
136
|
// scope here is always "user"|"dream" (validated above); pending is
|
|
129
137
|
// never tool-readable, so the narrower event type is correct.
|
|
130
|
-
void cb({
|
|
138
|
+
void cb({
|
|
139
|
+
type: "memory_recalled",
|
|
140
|
+
name: entry.name,
|
|
141
|
+
scope: scope,
|
|
142
|
+
location,
|
|
143
|
+
});
|
|
131
144
|
}
|
|
132
145
|
}
|
|
133
146
|
catch {
|
|
134
147
|
// ignore — the read result below is what matters
|
|
135
148
|
}
|
|
136
149
|
return (`name: ${entry.name}\n` +
|
|
150
|
+
`id: ${entry.id ?? ""}\n` +
|
|
137
151
|
`description: ${entry.description}\n` +
|
|
138
152
|
`type: ${entry.type}\n` +
|
|
153
|
+
`origin: ${entry.origin ?? "manual"}\n` +
|
|
154
|
+
`useCount: ${entry.useCount ?? 0}\n` +
|
|
155
|
+
`updateCount: ${entry.updateCount ?? 0}\n` +
|
|
139
156
|
`\n${entry.content}`);
|
|
140
157
|
}
|
|
141
158
|
catch (err) {
|
|
@@ -145,7 +162,7 @@ export async function memoryReadTool(args, ctx) {
|
|
|
145
162
|
// ─── MemorySave ────────────────────────────────────────────────────────────
|
|
146
163
|
export const memorySaveToolDef = {
|
|
147
164
|
name: "MemorySave",
|
|
148
|
-
description: "Create or
|
|
165
|
+
description: "Create or update a memory entry. Before saving, scan the injected memory index and/or call MemoryList; for the same durable topic, update the existing entry by id instead of creating date-stamped variants. " +
|
|
149
166
|
'Saving to scope "user" requires user permission (you will see a confirmation prompt). ' +
|
|
150
167
|
'Saving to scope "dream" is automatic — it is your auto-consolidation workspace. ' +
|
|
151
168
|
"Pick `type` carefully: " +
|
|
@@ -154,7 +171,7 @@ export const memorySaveToolDef = {
|
|
|
154
171
|
"project (non-obvious facts about ongoing work), " +
|
|
155
172
|
"reference (pointers to external resources). " +
|
|
156
173
|
"Pick `location`: global (a lesson/preference true in ANY project) vs project (this repo only). " +
|
|
157
|
-
"If a memory in the injected index is now stale or wrong,
|
|
174
|
+
"If a memory in the injected index is now stale or wrong, update it by id or MemoryDelete it — keep the store correct rather than letting contradictions pile up. " +
|
|
158
175
|
"The `description` is a one-line summary shown in the index; the `content` is the full body.",
|
|
159
176
|
inputSchema: {
|
|
160
177
|
type: "object",
|
|
@@ -165,9 +182,13 @@ export const memorySaveToolDef = {
|
|
|
165
182
|
description: "Which scope to save to",
|
|
166
183
|
},
|
|
167
184
|
location: LOCATION_SCHEMA,
|
|
185
|
+
id: {
|
|
186
|
+
type: "string",
|
|
187
|
+
description: "Stable id from MemoryList/MemoryRead when updating an existing memory. Omit only for genuinely new topics.",
|
|
188
|
+
},
|
|
168
189
|
name: {
|
|
169
190
|
type: "string",
|
|
170
|
-
description: "Short
|
|
191
|
+
description: "Short stable topic identifier; avoid dates, versions, or batch suffixes",
|
|
171
192
|
},
|
|
172
193
|
description: {
|
|
173
194
|
type: "string",
|
|
@@ -192,11 +213,14 @@ export async function memorySaveTool(args, ctx) {
|
|
|
192
213
|
return scope;
|
|
193
214
|
}
|
|
194
215
|
const name = args.name;
|
|
216
|
+
const id = args.id;
|
|
195
217
|
const description = args.description;
|
|
196
218
|
const type = args.type;
|
|
197
219
|
const content = args.content;
|
|
198
220
|
if (typeof name !== "string" || !name)
|
|
199
221
|
return "Error: name is required";
|
|
222
|
+
if (id !== undefined && typeof id !== "string")
|
|
223
|
+
return "Error: id must be a string";
|
|
200
224
|
if (typeof description !== "string")
|
|
201
225
|
return "Error: description is required";
|
|
202
226
|
if (typeof content !== "string")
|
|
@@ -207,12 +231,20 @@ export async function memorySaveTool(args, ctx) {
|
|
|
207
231
|
try {
|
|
208
232
|
const location = parseLocation(args.location);
|
|
209
233
|
const mm = mmFor(ctx, scope, location);
|
|
234
|
+
const dreamLoop = ctx?.__dreamLoop === true;
|
|
235
|
+
const forcedOrigin = dreamLoop
|
|
236
|
+
? "dream"
|
|
237
|
+
: scope === "user"
|
|
238
|
+
? "manual"
|
|
239
|
+
: (parseOrigin(args.origin) ?? "manual");
|
|
210
240
|
const fileName = mm.save({
|
|
241
|
+
id,
|
|
211
242
|
name,
|
|
212
243
|
description,
|
|
213
244
|
type: type,
|
|
214
245
|
content,
|
|
215
|
-
|
|
246
|
+
origin: forcedOrigin,
|
|
247
|
+
}, { forceOrigin: forcedOrigin });
|
|
216
248
|
return `Saved memory "${name}" → ${location}/${scope}/${fileName}`;
|
|
217
249
|
}
|
|
218
250
|
catch (err) {
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
/**
|
|
2
|
-
* Built-in view_image tool —
|
|
3
|
-
* 回传进上下文,让 vision 模型「看」它(对照 codex 的 view_image)。
|
|
2
|
+
* Built-in view_image tool — 把一个本地图片文件或历史图片以 base64 image
|
|
3
|
+
* ContentBlock 回传进上下文,让 vision 模型「看」它(对照 codex 的 view_image)。
|
|
4
4
|
*
|
|
5
5
|
* 典型用法:模型先写 SVG/Mermaid 并用 shell 转成 PNG,再调 view_image(png)
|
|
6
6
|
* 检查图画对没有(标签是否重叠、文字是否溢出),不对就改源再重转。
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
/**
|
|
2
|
-
* Built-in view_image tool —
|
|
3
|
-
* 回传进上下文,让 vision 模型「看」它(对照 codex 的 view_image)。
|
|
2
|
+
* Built-in view_image tool — 把一个本地图片文件或历史图片以 base64 image
|
|
3
|
+
* ContentBlock 回传进上下文,让 vision 模型「看」它(对照 codex 的 view_image)。
|
|
4
4
|
*
|
|
5
5
|
* 典型用法:模型先写 SVG/Mermaid 并用 shell 转成 PNG,再调 view_image(png)
|
|
6
6
|
* 检查图画对没有(标签是否重叠、文字是否溢出),不对就改源再重转。
|
|
@@ -14,6 +14,7 @@
|
|
|
14
14
|
import { readFile, stat } from "node:fs/promises";
|
|
15
15
|
import { extname, isAbsolute, resolve } from "node:path";
|
|
16
16
|
import { capabilitiesFor } from "../../llm/capabilities/index.js";
|
|
17
|
+
import { collectBase64Images, findImageByNumber } from "../../context/compaction.js";
|
|
17
18
|
const MAX_BYTES = 5 * 1024 * 1024; // 5MB —— vision 模型按 tile 计 token,大图无益且撑大请求
|
|
18
19
|
const MEDIA_TYPES = {
|
|
19
20
|
".png": "image/png",
|
|
@@ -24,11 +25,11 @@ const MEDIA_TYPES = {
|
|
|
24
25
|
};
|
|
25
26
|
export const viewImageToolDef = {
|
|
26
27
|
name: "view_image",
|
|
27
|
-
description: "Load
|
|
28
|
-
"
|
|
29
|
-
"
|
|
30
|
-
"
|
|
31
|
-
"
|
|
28
|
+
description: "Load an image into the conversation so you can SEE it (vision). Pass path to view a " +
|
|
29
|
+
"workspace image file, or pass imageNumber to retrieve the original image behind an " +
|
|
30
|
+
"earlier [image #N, already provided] history placeholder. Use exactly one of path or " +
|
|
31
|
+
"imageNumber. File paths support PNG/JPEG/GIF/WebP only; convert SVG/PDF to PNG first. " +
|
|
32
|
+
"Requires a vision-capable model; otherwise the image is skipped.",
|
|
32
33
|
inputSchema: {
|
|
33
34
|
type: "object",
|
|
34
35
|
properties: {
|
|
@@ -36,14 +37,31 @@ export const viewImageToolDef = {
|
|
|
36
37
|
type: "string",
|
|
37
38
|
description: "Path to the image file (absolute, or relative to the working directory).",
|
|
38
39
|
},
|
|
40
|
+
imageNumber: {
|
|
41
|
+
type: "number",
|
|
42
|
+
description: "Image history number N from an earlier [image #N, already provided] placeholder.",
|
|
43
|
+
},
|
|
39
44
|
},
|
|
40
|
-
required: ["path"],
|
|
41
45
|
},
|
|
42
46
|
};
|
|
43
47
|
export async function viewImageTool(args, ctx) {
|
|
44
48
|
const rawPath = args.path;
|
|
49
|
+
const rawImageNumber = args.imageNumber;
|
|
50
|
+
const hasPath = typeof rawPath === "string" && rawPath.trim().length > 0;
|
|
51
|
+
const hasImageNumber = rawImageNumber !== undefined && rawImageNumber !== null;
|
|
52
|
+
if (hasPath === hasImageNumber) {
|
|
53
|
+
return "Error: provide exactly one of path or imageNumber";
|
|
54
|
+
}
|
|
55
|
+
if (hasImageNumber) {
|
|
56
|
+
if (typeof rawImageNumber !== "number" ||
|
|
57
|
+
!Number.isSafeInteger(rawImageNumber) ||
|
|
58
|
+
rawImageNumber <= 0) {
|
|
59
|
+
return "Error: imageNumber must be a positive integer";
|
|
60
|
+
}
|
|
61
|
+
return viewHistoricalImage(rawImageNumber, ctx);
|
|
62
|
+
}
|
|
45
63
|
if (typeof rawPath !== "string" || !rawPath.trim()) {
|
|
46
|
-
return "Error: path
|
|
64
|
+
return "Error: path must be a non-empty string";
|
|
47
65
|
}
|
|
48
66
|
const cwd = ctx?.cwd ?? process.cwd();
|
|
49
67
|
const abs = isAbsolute(rawPath) ? rawPath : resolve(cwd, rawPath);
|
|
@@ -51,13 +69,7 @@ export async function viewImageTool(args, ctx) {
|
|
|
51
69
|
// 缺 llmConfig 时按「非视觉」处理(fail closed),对齐 DEFAULT_CAPABILITY
|
|
52
70
|
// 的保守姿态:运行时 ToolContext.llmConfig 必有,缺失只发生在测试 / 异常
|
|
53
71
|
// 装配下,此时绝不把 base64 读进上下文。
|
|
54
|
-
|
|
55
|
-
if (!ctx?.llmConfig)
|
|
56
|
-
return false;
|
|
57
|
-
const kind = (ctx.llmConfig.providerKind ?? ctx.llmConfig.provider);
|
|
58
|
-
return capabilitiesFor(kind, ctx.llmConfig.model).supportsVision;
|
|
59
|
-
})();
|
|
60
|
-
if (!supportsVision) {
|
|
72
|
+
if (!supportsVision(ctx)) {
|
|
61
73
|
return `[图片未加载: ${abs} —— 当前模型不支持视觉输入,已跳过。切换到 vision 模型后再 view_image。]`;
|
|
62
74
|
}
|
|
63
75
|
// 闸门 2:格式 gate
|
|
@@ -88,9 +100,78 @@ export async function viewImageTool(args, ctx) {
|
|
|
88
100
|
const data = buf.toString("base64");
|
|
89
101
|
const kb = Math.round(buf.length / 1024);
|
|
90
102
|
return {
|
|
91
|
-
contentBlocks: [
|
|
92
|
-
{ type: "image", source: { type: "base64", media_type: mediaType, data } },
|
|
93
|
-
],
|
|
103
|
+
contentBlocks: [{ type: "image", source: { type: "base64", media_type: mediaType, data } }],
|
|
94
104
|
result: `[已加载图片: ${abs} (${mediaType}, ${kb} KB)]`,
|
|
95
105
|
};
|
|
96
106
|
}
|
|
107
|
+
function supportsVision(ctx) {
|
|
108
|
+
if (!ctx?.llmConfig)
|
|
109
|
+
return false;
|
|
110
|
+
const kind = (ctx.llmConfig.providerKind ?? ctx.llmConfig.provider);
|
|
111
|
+
return capabilitiesFor(kind, ctx.llmConfig.model).supportsVision;
|
|
112
|
+
}
|
|
113
|
+
async function viewHistoricalImage(imageNumber, ctx) {
|
|
114
|
+
if (!supportsVision(ctx)) {
|
|
115
|
+
return `[图片未取回: image #${imageNumber} —— 当前模型不支持视觉输入,已跳过。切换到 vision 模型后再 view_image。]`;
|
|
116
|
+
}
|
|
117
|
+
const sessionManager = ctx?.engine?.getSessionManager?.();
|
|
118
|
+
const sessionId = ctx?.sessionId;
|
|
119
|
+
if (!sessionManager || !sessionId) {
|
|
120
|
+
return `Error: 无法取回 image #${imageNumber}: 当前工具上下文没有可读取的 session 历史。`;
|
|
121
|
+
}
|
|
122
|
+
let messages;
|
|
123
|
+
try {
|
|
124
|
+
messages = sessionManager.resume(sessionId).transcript.toMessages();
|
|
125
|
+
}
|
|
126
|
+
catch (err) {
|
|
127
|
+
return `Error: 无法读取 session 历史以取回 image #${imageNumber}: ${err.message}`;
|
|
128
|
+
}
|
|
129
|
+
const images = collectBase64Images(messages);
|
|
130
|
+
const found = findImageByNumber(messages, imageNumber);
|
|
131
|
+
if (!found) {
|
|
132
|
+
return `Error: 未找到 image #${imageNumber}; 当前 session 历史中共有 ${images.length} 张可取回图片。`;
|
|
133
|
+
}
|
|
134
|
+
const block = normalizeImageBlockForReturn(found.block);
|
|
135
|
+
const size = decodedImageBlockBytes(block);
|
|
136
|
+
if (size !== undefined && size > MAX_BYTES) {
|
|
137
|
+
const mb = (size / 1024 / 1024).toFixed(1);
|
|
138
|
+
return `[图片未取回: image #${imageNumber} —— 图片过大 (${mb} MB > 5 MB),请先压缩或缩放原图。]`;
|
|
139
|
+
}
|
|
140
|
+
const kb = size === undefined ? "unknown" : String(Math.round(size / 1024));
|
|
141
|
+
const mediaType = mediaTypeOfImageBlock(block) ?? "base64 image";
|
|
142
|
+
return {
|
|
143
|
+
contentBlocks: [block],
|
|
144
|
+
result: `[已取回 image #${imageNumber}: ${mediaType}, ${kb} KB]`,
|
|
145
|
+
};
|
|
146
|
+
}
|
|
147
|
+
function normalizeImageBlockForReturn(block) {
|
|
148
|
+
if (block.type === "image" && block.source?.type === "base64")
|
|
149
|
+
return block;
|
|
150
|
+
const source = openAIDataUrlImageSource(block);
|
|
151
|
+
return source ? { type: "image", source } : block;
|
|
152
|
+
}
|
|
153
|
+
function decodedImageBlockBytes(block) {
|
|
154
|
+
const data = base64DataOfImageBlock(block);
|
|
155
|
+
if (!data)
|
|
156
|
+
return undefined;
|
|
157
|
+
return Buffer.byteLength(data, "base64");
|
|
158
|
+
}
|
|
159
|
+
function base64DataOfImageBlock(block) {
|
|
160
|
+
if (block.type === "image" && block.source?.type === "base64") {
|
|
161
|
+
return block.source.data;
|
|
162
|
+
}
|
|
163
|
+
return openAIDataUrlImageSource(block)?.data;
|
|
164
|
+
}
|
|
165
|
+
function mediaTypeOfImageBlock(block) {
|
|
166
|
+
if (block.type === "image" && block.source?.type === "base64") {
|
|
167
|
+
return block.source.media_type;
|
|
168
|
+
}
|
|
169
|
+
return openAIDataUrlImageSource(block)?.media_type;
|
|
170
|
+
}
|
|
171
|
+
function openAIDataUrlImageSource(block) {
|
|
172
|
+
const maybeOpenAI = block;
|
|
173
|
+
const match = maybeOpenAI.image_url?.url?.match(/^data:(image\/[^;,]+);base64,(.+)$/i);
|
|
174
|
+
if (maybeOpenAI.type !== "image_url" || !match?.[1] || !match[2])
|
|
175
|
+
return undefined;
|
|
176
|
+
return { type: "base64", media_type: match[1], data: match[2] };
|
|
177
|
+
}
|
|
@@ -3,6 +3,8 @@
|
|
|
3
3
|
*/
|
|
4
4
|
import type { ToolDefinition } from "../../types.js";
|
|
5
5
|
import type { ToolContext } from "../context.js";
|
|
6
|
+
export declare const switchSessionWorkspaceToolDef: ToolDefinition;
|
|
7
|
+
export declare function switchSessionWorkspaceTool(args: Record<string, unknown>, ctx?: ToolContext): Promise<string>;
|
|
6
8
|
export declare const enterWorktreeToolDef: ToolDefinition;
|
|
7
9
|
export declare function enterWorktreeTool(args: Record<string, unknown>, ctx?: ToolContext): Promise<string>;
|
|
8
10
|
export declare const exitWorktreeToolDef: ToolDefinition;
|