@alexandrgreen/anchorclaw 0.0.3
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/LICENSE +21 -0
- package/NOTICE +9 -0
- package/README.md +252 -0
- package/dist/api.d.ts +3 -0
- package/dist/api.d.ts.map +1 -0
- package/dist/api.js +3 -0
- package/dist/api.js.map +1 -0
- package/dist/config.d.ts +46 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +309 -0
- package/dist/config.js.map +1 -0
- package/dist/config.test.d.ts +2 -0
- package/dist/config.test.d.ts.map +1 -0
- package/dist/config.test.js +100 -0
- package/dist/config.test.js.map +1 -0
- package/dist/identity-policy.d.ts +3 -0
- package/dist/identity-policy.d.ts.map +1 -0
- package/dist/identity-policy.js +7 -0
- package/dist/identity-policy.js.map +1 -0
- package/dist/identity-policy.test.d.ts +2 -0
- package/dist/identity-policy.test.d.ts.map +1 -0
- package/dist/identity-policy.test.js +23 -0
- package/dist/identity-policy.test.js.map +1 -0
- package/dist/identity.d.ts +22 -0
- package/dist/identity.d.ts.map +1 -0
- package/dist/identity.js +113 -0
- package/dist/identity.js.map +1 -0
- package/dist/identity.test.d.ts +2 -0
- package/dist/identity.test.d.ts.map +1 -0
- package/dist/identity.test.js +22 -0
- package/dist/identity.test.js.map +1 -0
- package/dist/importer.d.ts +12 -0
- package/dist/importer.d.ts.map +1 -0
- package/dist/importer.js +297 -0
- package/dist/importer.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +92 -0
- package/dist/index.js.map +1 -0
- package/dist/index.test.d.ts +2 -0
- package/dist/index.test.d.ts.map +1 -0
- package/dist/index.test.js +770 -0
- package/dist/index.test.js.map +1 -0
- package/dist/memory/forget.d.ts +22 -0
- package/dist/memory/forget.d.ts.map +1 -0
- package/dist/memory/forget.js +84 -0
- package/dist/memory/forget.js.map +1 -0
- package/dist/memory/get.d.ts +36 -0
- package/dist/memory/get.d.ts.map +1 -0
- package/dist/memory/get.js +282 -0
- package/dist/memory/get.js.map +1 -0
- package/dist/memory/get.test.d.ts +2 -0
- package/dist/memory/get.test.d.ts.map +1 -0
- package/dist/memory/get.test.js +180 -0
- package/dist/memory/get.test.js.map +1 -0
- package/dist/memory/limits.d.ts +10 -0
- package/dist/memory/limits.d.ts.map +1 -0
- package/dist/memory/limits.js +18 -0
- package/dist/memory/limits.js.map +1 -0
- package/dist/memory/manager.d.ts +93 -0
- package/dist/memory/manager.d.ts.map +1 -0
- package/dist/memory/manager.js +344 -0
- package/dist/memory/manager.js.map +1 -0
- package/dist/memory/manager.test.d.ts +2 -0
- package/dist/memory/manager.test.d.ts.map +1 -0
- package/dist/memory/manager.test.js +201 -0
- package/dist/memory/manager.test.js.map +1 -0
- package/dist/memory/paths.d.ts +12 -0
- package/dist/memory/paths.d.ts.map +1 -0
- package/dist/memory/paths.js +31 -0
- package/dist/memory/paths.js.map +1 -0
- package/dist/memory/paths.test.d.ts +2 -0
- package/dist/memory/paths.test.d.ts.map +1 -0
- package/dist/memory/paths.test.js +35 -0
- package/dist/memory/paths.test.js.map +1 -0
- package/dist/memory/prompt.d.ts +37 -0
- package/dist/memory/prompt.d.ts.map +1 -0
- package/dist/memory/prompt.js +152 -0
- package/dist/memory/prompt.js.map +1 -0
- package/dist/memory/prompt.test.d.ts +2 -0
- package/dist/memory/prompt.test.d.ts.map +1 -0
- package/dist/memory/prompt.test.js +47 -0
- package/dist/memory/prompt.test.js.map +1 -0
- package/dist/memory/read-file-shared.d.ts +17 -0
- package/dist/memory/read-file-shared.d.ts.map +1 -0
- package/dist/memory/read-file-shared.js +53 -0
- package/dist/memory/read-file-shared.js.map +1 -0
- package/dist/memory/read-file-shared.test.d.ts +2 -0
- package/dist/memory/read-file-shared.test.d.ts.map +1 -0
- package/dist/memory/read-file-shared.test.js +82 -0
- package/dist/memory/read-file-shared.test.js.map +1 -0
- package/dist/memory/recall.d.ts +22 -0
- package/dist/memory/recall.d.ts.map +1 -0
- package/dist/memory/recall.js +58 -0
- package/dist/memory/recall.js.map +1 -0
- package/dist/memory/search.d.ts +30 -0
- package/dist/memory/search.d.ts.map +1 -0
- package/dist/memory/search.js +110 -0
- package/dist/memory/search.js.map +1 -0
- package/dist/memory/search.test.d.ts +2 -0
- package/dist/memory/search.test.d.ts.map +1 -0
- package/dist/memory/search.test.js +103 -0
- package/dist/memory/search.test.js.map +1 -0
- package/dist/memory/sessions-index-sync.d.ts +28 -0
- package/dist/memory/sessions-index-sync.d.ts.map +1 -0
- package/dist/memory/sessions-index-sync.js +253 -0
- package/dist/memory/sessions-index-sync.js.map +1 -0
- package/dist/memory/sessions-index-sync.test.d.ts +2 -0
- package/dist/memory/sessions-index-sync.test.d.ts.map +1 -0
- package/dist/memory/sessions-index-sync.test.js +280 -0
- package/dist/memory/sessions-index-sync.test.js.map +1 -0
- package/dist/memory/sessions-index.d.ts +30 -0
- package/dist/memory/sessions-index.d.ts.map +1 -0
- package/dist/memory/sessions-index.js +238 -0
- package/dist/memory/sessions-index.js.map +1 -0
- package/dist/memory/sessions-index.test.d.ts +2 -0
- package/dist/memory/sessions-index.test.d.ts.map +1 -0
- package/dist/memory/sessions-index.test.js +266 -0
- package/dist/memory/sessions-index.test.js.map +1 -0
- package/dist/memory/sessions-visibility.d.ts +18 -0
- package/dist/memory/sessions-visibility.d.ts.map +1 -0
- package/dist/memory/sessions-visibility.js +110 -0
- package/dist/memory/sessions-visibility.js.map +1 -0
- package/dist/memory/sessions-visibility.test.d.ts +2 -0
- package/dist/memory/sessions-visibility.test.d.ts.map +1 -0
- package/dist/memory/sessions-visibility.test.js +137 -0
- package/dist/memory/sessions-visibility.test.js.map +1 -0
- package/dist/memory/sessions.d.ts +26 -0
- package/dist/memory/sessions.d.ts.map +1 -0
- package/dist/memory/sessions.js +250 -0
- package/dist/memory/sessions.js.map +1 -0
- package/dist/memory/sessions.test.d.ts +2 -0
- package/dist/memory/sessions.test.d.ts.map +1 -0
- package/dist/memory/sessions.test.js +257 -0
- package/dist/memory/sessions.test.js.map +1 -0
- package/dist/memory/store.d.ts +25 -0
- package/dist/memory/store.d.ts.map +1 -0
- package/dist/memory/store.js +176 -0
- package/dist/memory/store.js.map +1 -0
- package/dist/migrations-fs.d.ts +5 -0
- package/dist/migrations-fs.d.ts.map +1 -0
- package/dist/migrations-fs.js +19 -0
- package/dist/migrations-fs.js.map +1 -0
- package/dist/migrations.d.ts +17 -0
- package/dist/migrations.d.ts.map +1 -0
- package/dist/migrations.js +44 -0
- package/dist/migrations.js.map +1 -0
- package/dist/migrations.test.d.ts +2 -0
- package/dist/migrations.test.d.ts.map +1 -0
- package/dist/migrations.test.js +66 -0
- package/dist/migrations.test.js.map +1 -0
- package/dist/plugin/capability.d.ts +7 -0
- package/dist/plugin/capability.d.ts.map +1 -0
- package/dist/plugin/capability.js +92 -0
- package/dist/plugin/capability.js.map +1 -0
- package/dist/plugin/capability.test.d.ts +2 -0
- package/dist/plugin/capability.test.d.ts.map +1 -0
- package/dist/plugin/capability.test.js +75 -0
- package/dist/plugin/capability.test.js.map +1 -0
- package/dist/plugin/lifecycle.d.ts +6 -0
- package/dist/plugin/lifecycle.d.ts.map +1 -0
- package/dist/plugin/lifecycle.js +31 -0
- package/dist/plugin/lifecycle.js.map +1 -0
- package/dist/plugin/prompt-cache.d.ts +10 -0
- package/dist/plugin/prompt-cache.d.ts.map +1 -0
- package/dist/plugin/prompt-cache.js +58 -0
- package/dist/plugin/prompt-cache.js.map +1 -0
- package/dist/plugin/runtime-context.d.ts +27 -0
- package/dist/plugin/runtime-context.d.ts.map +1 -0
- package/dist/plugin/runtime-context.js +91 -0
- package/dist/plugin/runtime-context.js.map +1 -0
- package/dist/plugin/runtime-helpers.d.ts +3 -0
- package/dist/plugin/runtime-helpers.d.ts.map +1 -0
- package/dist/plugin/runtime-helpers.js +12 -0
- package/dist/plugin/runtime-helpers.js.map +1 -0
- package/dist/plugin/session-delta-helpers.d.ts +11 -0
- package/dist/plugin/session-delta-helpers.d.ts.map +1 -0
- package/dist/plugin/session-delta-helpers.js +62 -0
- package/dist/plugin/session-delta-helpers.js.map +1 -0
- package/dist/plugin/session-delta.d.ts +12 -0
- package/dist/plugin/session-delta.d.ts.map +1 -0
- package/dist/plugin/session-delta.js +307 -0
- package/dist/plugin/session-delta.js.map +1 -0
- package/dist/plugin/tools/common.d.ts +7 -0
- package/dist/plugin/tools/common.d.ts.map +1 -0
- package/dist/plugin/tools/common.js +2 -0
- package/dist/plugin/tools/common.js.map +1 -0
- package/dist/plugin/tools/index.d.ts +3 -0
- package/dist/plugin/tools/index.d.ts.map +1 -0
- package/dist/plugin/tools/index.js +16 -0
- package/dist/plugin/tools/index.js.map +1 -0
- package/dist/plugin/tools/memory-forget.d.ts +3 -0
- package/dist/plugin/tools/memory-forget.d.ts.map +1 -0
- package/dist/plugin/tools/memory-forget.js +80 -0
- package/dist/plugin/tools/memory-forget.js.map +1 -0
- package/dist/plugin/tools/memory-forget.test.d.ts +2 -0
- package/dist/plugin/tools/memory-forget.test.d.ts.map +1 -0
- package/dist/plugin/tools/memory-forget.test.js +58 -0
- package/dist/plugin/tools/memory-forget.test.js.map +1 -0
- package/dist/plugin/tools/memory-get.d.ts +3 -0
- package/dist/plugin/tools/memory-get.d.ts.map +1 -0
- package/dist/plugin/tools/memory-get.js +136 -0
- package/dist/plugin/tools/memory-get.js.map +1 -0
- package/dist/plugin/tools/memory-recall.d.ts +3 -0
- package/dist/plugin/tools/memory-recall.d.ts.map +1 -0
- package/dist/plugin/tools/memory-recall.js +176 -0
- package/dist/plugin/tools/memory-recall.js.map +1 -0
- package/dist/plugin/tools/memory-recall.test.d.ts +2 -0
- package/dist/plugin/tools/memory-recall.test.d.ts.map +1 -0
- package/dist/plugin/tools/memory-recall.test.js +169 -0
- package/dist/plugin/tools/memory-recall.test.js.map +1 -0
- package/dist/plugin/tools/memory-search.d.ts +3 -0
- package/dist/plugin/tools/memory-search.d.ts.map +1 -0
- package/dist/plugin/tools/memory-search.js +332 -0
- package/dist/plugin/tools/memory-search.js.map +1 -0
- package/dist/plugin/tools/memory-search.test.d.ts +2 -0
- package/dist/plugin/tools/memory-search.test.d.ts.map +1 -0
- package/dist/plugin/tools/memory-search.test.js +205 -0
- package/dist/plugin/tools/memory-search.test.js.map +1 -0
- package/dist/plugin/tools/memory-status.d.ts +3 -0
- package/dist/plugin/tools/memory-status.d.ts.map +1 -0
- package/dist/plugin/tools/memory-status.js +134 -0
- package/dist/plugin/tools/memory-status.js.map +1 -0
- package/dist/plugin/tools/memory-store.d.ts +3 -0
- package/dist/plugin/tools/memory-store.d.ts.map +1 -0
- package/dist/plugin/tools/memory-store.js +88 -0
- package/dist/plugin/tools/memory-store.js.map +1 -0
- package/dist/plugin/tools/memory-store.test.d.ts +2 -0
- package/dist/plugin/tools/memory-store.test.d.ts.map +1 -0
- package/dist/plugin/tools/memory-store.test.js +64 -0
- package/dist/plugin/tools/memory-store.test.js.map +1 -0
- package/dist/plugin/tools/memory-visible-output.d.ts +20 -0
- package/dist/plugin/tools/memory-visible-output.d.ts.map +1 -0
- package/dist/plugin/tools/memory-visible-output.js +61 -0
- package/dist/plugin/tools/memory-visible-output.js.map +1 -0
- package/dist/plugin/types.d.ts +61 -0
- package/dist/plugin/types.d.ts.map +1 -0
- package/dist/plugin/types.js +2 -0
- package/dist/plugin/types.js.map +1 -0
- package/dist/postgres.d.ts +7 -0
- package/dist/postgres.d.ts.map +1 -0
- package/dist/postgres.js +55 -0
- package/dist/postgres.js.map +1 -0
- package/migrations/0001_init.sql +228 -0
- package/migrations/0002_session_index.sql +71 -0
- package/openclaw.plugin.json +314 -0
- package/package.json +63 -0
|
@@ -0,0 +1,134 @@
|
|
|
1
|
+
import { resolveSessionsDirForAgent } from "../../memory/sessions.js";
|
|
2
|
+
import { constants as fsConstants } from "node:fs";
|
|
3
|
+
import fs from "node:fs/promises";
|
|
4
|
+
import path from "node:path";
|
|
5
|
+
export function registerMemoryStatusTool({ ctx }) {
|
|
6
|
+
const api = ctx.api;
|
|
7
|
+
api.registerTool({
|
|
8
|
+
name: "memory_status",
|
|
9
|
+
label: "Memory Status",
|
|
10
|
+
description: "Return runtime health state for AnchorClaw memory operations.\n\nMVP rules:\n- Diagnostics only: use for operator health/debug checks.\n- Do not use this tool for fact lookup, retrieval ranking, or answer selection.\n- It reports SDK degraded state without exposing secrets.",
|
|
11
|
+
parameters: {
|
|
12
|
+
type: "object",
|
|
13
|
+
additionalProperties: false,
|
|
14
|
+
properties: {
|
|
15
|
+
check: {
|
|
16
|
+
type: "boolean",
|
|
17
|
+
description: "When true, performs active health checks (database connectivity/schema + sessions dir accessibility).",
|
|
18
|
+
},
|
|
19
|
+
},
|
|
20
|
+
},
|
|
21
|
+
async execute(_toolCallId, params) {
|
|
22
|
+
if (ctx.disabledReason) {
|
|
23
|
+
return {
|
|
24
|
+
content: [{ type: "text", text: `anchorclaw: disabled until configured (${ctx.disabledReason})` }],
|
|
25
|
+
details: { disabled: true, error: ctx.disabledReason },
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
const record = (params ?? {});
|
|
29
|
+
const activeCheck = record.check === true;
|
|
30
|
+
const base = {
|
|
31
|
+
ok: true,
|
|
32
|
+
mode: activeCheck ? "active" : "cached",
|
|
33
|
+
sdk: { ...ctx.sdkHealth },
|
|
34
|
+
};
|
|
35
|
+
if (activeCheck) {
|
|
36
|
+
const startedAt = Date.now();
|
|
37
|
+
let dbError;
|
|
38
|
+
try {
|
|
39
|
+
await ctx.ensureReady();
|
|
40
|
+
await ctx.getPool().query("SELECT 1");
|
|
41
|
+
const schemaRows = await ctx.getPool().query("SELECT to_regclass('memory_items') AS memory_items, to_regclass('session_index_files') AS session_index_files, to_regclass('session_index_chunks') AS session_index_chunks, to_regclass('schema_migrations') AS schema_migrations");
|
|
42
|
+
const schema = schemaRows.rows[0];
|
|
43
|
+
const schemaOk = Boolean(schema?.memory_items &&
|
|
44
|
+
schema?.session_index_files &&
|
|
45
|
+
schema?.session_index_chunks &&
|
|
46
|
+
schema?.schema_migrations);
|
|
47
|
+
const migrationRows = await ctx.getPool().query("SELECT id FROM schema_migrations ORDER BY id DESC LIMIT 1");
|
|
48
|
+
base.database = {
|
|
49
|
+
ok: schemaOk,
|
|
50
|
+
latencyMs: Math.max(0, Date.now() - startedAt),
|
|
51
|
+
schemaOk,
|
|
52
|
+
migrationVersion: migrationRows.rows[0]?.id ?? null,
|
|
53
|
+
};
|
|
54
|
+
if (!schemaOk) {
|
|
55
|
+
base.ok = false;
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
catch (error) {
|
|
59
|
+
dbError = error instanceof Error ? error.message : String(error);
|
|
60
|
+
base.ok = false;
|
|
61
|
+
base.database = {
|
|
62
|
+
ok: false,
|
|
63
|
+
error: dbError,
|
|
64
|
+
};
|
|
65
|
+
}
|
|
66
|
+
const sessionsVisibility = ctx.cfg?.sessions?.visibility ?? "current";
|
|
67
|
+
const sessionsEnabled = sessionsVisibility !== "off";
|
|
68
|
+
try {
|
|
69
|
+
const agentId = String(api?.runtime?.agentId ?? "main");
|
|
70
|
+
const agentSessionsDir = await resolveSessionsDirForAgent(agentId);
|
|
71
|
+
const stateDir = path.dirname(path.dirname(path.dirname(agentSessionsDir)));
|
|
72
|
+
let exists = false;
|
|
73
|
+
let readable = false;
|
|
74
|
+
try {
|
|
75
|
+
await fs.stat(agentSessionsDir);
|
|
76
|
+
exists = true;
|
|
77
|
+
}
|
|
78
|
+
catch {
|
|
79
|
+
exists = false;
|
|
80
|
+
}
|
|
81
|
+
if (exists) {
|
|
82
|
+
try {
|
|
83
|
+
await fs.access(agentSessionsDir, fsConstants.R_OK);
|
|
84
|
+
readable = true;
|
|
85
|
+
}
|
|
86
|
+
catch {
|
|
87
|
+
readable = false;
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
base.sessions = {
|
|
91
|
+
enabled: sessionsEnabled,
|
|
92
|
+
visibility: sessionsVisibility,
|
|
93
|
+
stateDir,
|
|
94
|
+
agentSessionsDir,
|
|
95
|
+
exists,
|
|
96
|
+
readable,
|
|
97
|
+
};
|
|
98
|
+
}
|
|
99
|
+
catch (error) {
|
|
100
|
+
base.ok = false;
|
|
101
|
+
base.sessions = {
|
|
102
|
+
enabled: sessionsEnabled,
|
|
103
|
+
visibility: sessionsVisibility,
|
|
104
|
+
error: error instanceof Error ? error.message : String(error),
|
|
105
|
+
};
|
|
106
|
+
}
|
|
107
|
+
const pending = Array.from(ctx.sessionDelta.stateByPath.values()).reduce((acc, item) => {
|
|
108
|
+
acc.pendingBytes += item.pendingBytes;
|
|
109
|
+
acc.pendingMessages += item.pendingMessages;
|
|
110
|
+
return acc;
|
|
111
|
+
}, { pendingBytes: 0, pendingMessages: 0 });
|
|
112
|
+
base.index = {
|
|
113
|
+
trackedFiles: ctx.sessionDelta.stateByPath.size,
|
|
114
|
+
pendingBytes: pending.pendingBytes,
|
|
115
|
+
pendingMessages: pending.pendingMessages,
|
|
116
|
+
};
|
|
117
|
+
}
|
|
118
|
+
return {
|
|
119
|
+
content: [
|
|
120
|
+
{
|
|
121
|
+
type: "text",
|
|
122
|
+
text: ctx.sdkHealth.degraded
|
|
123
|
+
? `AnchorClaw memory is degraded (${ctx.sdkHealth.reason ?? "unknown error"}).`
|
|
124
|
+
: activeCheck && !base.ok
|
|
125
|
+
? "AnchorClaw memory active check failed."
|
|
126
|
+
: "AnchorClaw memory is healthy.",
|
|
127
|
+
},
|
|
128
|
+
],
|
|
129
|
+
details: base,
|
|
130
|
+
};
|
|
131
|
+
},
|
|
132
|
+
});
|
|
133
|
+
}
|
|
134
|
+
//# sourceMappingURL=memory-status.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-status.js","sourceRoot":"","sources":["../../../src/plugin/tools/memory-status.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AAGtE,OAAO,EAAE,SAAS,IAAI,WAAW,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,MAAM,UAAU,wBAAwB,CAAC,EAAE,GAAG,EAA0B;IACtE,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC;IACpB,GAAG,CAAC,YAAY,CAAC;QACf,IAAI,EAAE,eAAe;QACrB,KAAK,EAAE,eAAe;QACtB,WAAW,EACT,oRAAoR;QACtR,UAAU,EAAE;YACV,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE;gBACV,KAAK,EAAE;oBACL,IAAI,EAAE,SAAS;oBACf,WAAW,EACT,uGAAuG;iBAC1G;aACF;SACF;QACD,KAAK,CAAC,OAAO,CAAC,WAAmB,EAAE,MAAe;YAChD,IAAI,GAAG,CAAC,cAAc,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,0CAA0C,GAAG,CAAC,cAAc,GAAG,EAAE,CAAC;oBAClG,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,CAAC,cAAc,EAAE;iBACvD,CAAC;YACJ,CAAC;YACD,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,EAAE,CAAwB,CAAC;YACrD,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,KAAK,IAAI,CAAC;YAC1C,MAAM,IAAI,GAA4B;gBACpC,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;gBACvC,GAAG,EAAE,EAAE,GAAG,GAAG,CAAC,SAAS,EAAE;aAC1B,CAAC;YACF,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,IAAI,OAA2B,CAAC;gBAChC,IAAI,CAAC;oBACH,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC;oBACxB,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;oBACtC,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC,KAAK,CAM1C,mOAAmO,CACpO,CAAC;oBACF,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;oBAClC,MAAM,QAAQ,GAAG,OAAO,CACtB,MAAM,EAAE,YAAY;wBAClB,MAAM,EAAE,mBAAmB;wBAC3B,MAAM,EAAE,oBAAoB;wBAC5B,MAAM,EAAE,iBAAiB,CAC5B,CAAC;oBACF,MAAM,aAAa,GAAG,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC,KAAK,CAC7C,2DAA2D,CAC5D,CAAC;oBACF,IAAI,CAAC,QAAQ,GAAG;wBACd,EAAE,EAAE,QAAQ;wBACZ,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;wBAC9C,QAAQ;wBACR,gBAAgB,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,IAAI;qBACpD,CAAC;oBACF,IAAI,CAAC,QAAQ,EAAE,CAAC;wBACd,IAAI,CAAC,EAAE,GAAG,KAAK,CAAC;oBAClB,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACjE,IAAI,CAAC,EAAE,GAAG,KAAK,CAAC;oBAChB,IAAI,CAAC,QAAQ,GAAG;wBACd,EAAE,EAAE,KAAK;wBACT,KAAK,EAAE,OAAO;qBACf,CAAC;gBACJ,CAAC;gBAED,MAAM,kBAAkB,GAAG,GAAG,CAAC,GAAG,EAAE,QAAQ,EAAE,UAAU,IAAI,SAAS,CAAC;gBACtE,MAAM,eAAe,GAAG,kBAAkB,KAAK,KAAK,CAAC;gBACrD,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,MAAM,CAAE,GAAW,EAAE,OAAO,EAAE,OAAO,IAAI,MAAM,CAAC,CAAC;oBACjE,MAAM,gBAAgB,GAAG,MAAM,0BAA0B,CAAC,OAAO,CAAC,CAAC;oBACnE,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;oBAC5E,IAAI,MAAM,GAAG,KAAK,CAAC;oBACnB,IAAI,QAAQ,GAAG,KAAK,CAAC;oBACrB,IAAI,CAAC;wBACH,MAAM,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;wBAChC,MAAM,GAAG,IAAI,CAAC;oBAChB,CAAC;oBAAC,MAAM,CAAC;wBACP,MAAM,GAAG,KAAK,CAAC;oBACjB,CAAC;oBACD,IAAI,MAAM,EAAE,CAAC;wBACX,IAAI,CAAC;4BACH,MAAM,EAAE,CAAC,MAAM,CAAC,gBAAgB,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;4BACpD,QAAQ,GAAG,IAAI,CAAC;wBAClB,CAAC;wBAAC,MAAM,CAAC;4BACP,QAAQ,GAAG,KAAK,CAAC;wBACnB,CAAC;oBACH,CAAC;oBACD,IAAI,CAAC,QAAQ,GAAG;wBACd,OAAO,EAAE,eAAe;wBACxB,UAAU,EAAE,kBAAkB;wBAC9B,QAAQ;wBACR,gBAAgB;wBAChB,MAAM;wBACN,QAAQ;qBACT,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,EAAE,GAAG,KAAK,CAAC;oBAChB,IAAI,CAAC,QAAQ,GAAG;wBACd,OAAO,EAAE,eAAe;wBACxB,UAAU,EAAE,kBAAkB;wBAC9B,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D,CAAC;gBACJ,CAAC;gBAED,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACtE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;oBACZ,GAAG,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC;oBACtC,GAAG,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;oBAC5C,OAAO,GAAG,CAAC;gBACb,CAAC,EACD,EAAE,YAAY,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,CACxC,CAAC;gBACF,IAAI,CAAC,KAAK,GAAG;oBACX,YAAY,EAAE,GAAG,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI;oBAC/C,YAAY,EAAE,OAAO,CAAC,YAAY;oBAClC,eAAe,EAAE,OAAO,CAAC,eAAe;iBACzC,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,QAAQ;4BAC1B,CAAC,CAAC,kCAAkC,GAAG,CAAC,SAAS,CAAC,MAAM,IAAI,eAAe,IAAI;4BAC/E,CAAC,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,EAAE;gCACvB,CAAC,CAAC,wCAAwC;gCAC1C,CAAC,CAAC,+BAA+B;qBACtC;iBACF;gBACD,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-store.d.ts","sourceRoot":"","sources":["../../../src/plugin/tools/memory-store.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAE1D,wBAAgB,uBAAuB,CAAC,EAAE,GAAG,EAAE,kBAAkB,EAAE,EAAE,sBAAsB,QA6F1F"}
|
|
@@ -0,0 +1,88 @@
|
|
|
1
|
+
import { resolveUserAndWorkspaceScope } from "../../identity.js";
|
|
2
|
+
import { memoryStoreDb } from "../../memory/store.js";
|
|
3
|
+
export function registerMemoryStoreTool({ ctx, refreshPromptCache }) {
|
|
4
|
+
const api = ctx.api;
|
|
5
|
+
api.registerTool({
|
|
6
|
+
name: "memory_store",
|
|
7
|
+
label: "Memory Store",
|
|
8
|
+
description: "Store durable long-term memory into Postgres.\n\nMVP rules:\n- Always provide { content }.\n- If you are storing an updateable fact/preference/setting, provide { canonicalKey } so future calls overwrite the same logical item (instead of creating duplicates).\n- Optionally provide { type: \"fact\"|\"note\"|... } (default: \"note\").",
|
|
9
|
+
parameters: {
|
|
10
|
+
type: "object",
|
|
11
|
+
additionalProperties: false,
|
|
12
|
+
required: ["content"],
|
|
13
|
+
properties: {
|
|
14
|
+
content: { type: "string", description: "Markdown/plain text memory content" },
|
|
15
|
+
canonicalKey: {
|
|
16
|
+
type: "string",
|
|
17
|
+
description: "Optional canonical key for upsert (e.g. \"timezone\", \"preferred_language\", \"project_name\"). When set, AnchorClaw updates the existing active item instead of creating duplicates.",
|
|
18
|
+
},
|
|
19
|
+
canonical_key: {
|
|
20
|
+
type: "string",
|
|
21
|
+
description: "Alias for canonicalKey.",
|
|
22
|
+
},
|
|
23
|
+
type: {
|
|
24
|
+
type: "string",
|
|
25
|
+
description: "Optional memory item type (default: note).",
|
|
26
|
+
enum: ["fact", "note"],
|
|
27
|
+
},
|
|
28
|
+
},
|
|
29
|
+
},
|
|
30
|
+
async execute(_toolCallId, params) {
|
|
31
|
+
if (ctx.disabledReason) {
|
|
32
|
+
return {
|
|
33
|
+
content: [{ type: "text", text: `anchorclaw: disabled until configured (${ctx.disabledReason})` }],
|
|
34
|
+
details: { disabled: true, error: ctx.disabledReason },
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
await ctx.ensureReady();
|
|
38
|
+
const scope = await resolveUserAndWorkspaceScope({
|
|
39
|
+
api,
|
|
40
|
+
pool: ctx.getPool(),
|
|
41
|
+
agentId: api?.runtime?.agentId,
|
|
42
|
+
sessionKey: api?.runtime?.sessionKey,
|
|
43
|
+
configuredExternalId: ctx.cfg?.identity?.externalId,
|
|
44
|
+
});
|
|
45
|
+
const record = (params ?? {});
|
|
46
|
+
const content = typeof record.content === "string" ? String(record.content) : "";
|
|
47
|
+
if (!content.trim()) {
|
|
48
|
+
return {
|
|
49
|
+
content: [{ type: "text", text: "anchorclaw: memory_store requires non-empty content" }],
|
|
50
|
+
details: { disabled: true, error: "content is required" },
|
|
51
|
+
};
|
|
52
|
+
}
|
|
53
|
+
const canonicalKey = typeof record.canonicalKey === "string" && record.canonicalKey.trim()
|
|
54
|
+
? String(record.canonicalKey)
|
|
55
|
+
: typeof record.canonical_key === "string" && record.canonical_key.trim()
|
|
56
|
+
? String(record.canonical_key)
|
|
57
|
+
: undefined;
|
|
58
|
+
const type = typeof record.type === "string" ? String(record.type) : undefined;
|
|
59
|
+
const stored = await memoryStoreDb({
|
|
60
|
+
pool: ctx.getPool(),
|
|
61
|
+
userId: scope.userId,
|
|
62
|
+
workspaceId: scope.workspaceId,
|
|
63
|
+
actor: ctx.resolveActor(),
|
|
64
|
+
logger: api.logger,
|
|
65
|
+
input: { content, ...(canonicalKey ? { canonicalKey } : {}), ...(type ? { type } : {}) },
|
|
66
|
+
});
|
|
67
|
+
if (!stored.ok) {
|
|
68
|
+
return {
|
|
69
|
+
content: [{ type: "text", text: `anchorclaw: memory_store failed (${stored.error})` }],
|
|
70
|
+
details: stored,
|
|
71
|
+
};
|
|
72
|
+
}
|
|
73
|
+
refreshPromptCache();
|
|
74
|
+
const visible = {
|
|
75
|
+
ok: true,
|
|
76
|
+
path: stored.path,
|
|
77
|
+
id: stored.id,
|
|
78
|
+
canonicalKey: canonicalKey ?? null,
|
|
79
|
+
type: type ?? null,
|
|
80
|
+
};
|
|
81
|
+
return {
|
|
82
|
+
content: [{ type: "text", text: JSON.stringify(visible, null, 2) }],
|
|
83
|
+
details: stored,
|
|
84
|
+
};
|
|
85
|
+
},
|
|
86
|
+
});
|
|
87
|
+
}
|
|
88
|
+
//# sourceMappingURL=memory-store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-store.js","sourceRoot":"","sources":["../../../src/plugin/tools/memory-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,4BAA4B,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAGtD,MAAM,UAAU,uBAAuB,CAAC,EAAE,GAAG,EAAE,kBAAkB,EAA0B;IACzF,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC;IACpB,GAAG,CAAC,YAAY,CAAC;QACf,IAAI,EAAE,cAAc;QACpB,KAAK,EAAE,cAAc;QACrB,WAAW,EACT,+UAA+U;QACjV,UAAU,EAAE;YACV,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,QAAQ,EAAE,CAAC,SAAS,CAAC;YACrB,UAAU,EAAE;gBACV,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,oCAAoC,EAAE;gBAC9E,YAAY,EAAE;oBACZ,IAAI,EAAE,QAAQ;oBACd,WAAW,EACT,wLAAwL;iBAC3L;gBACD,aAAa,EAAE;oBACb,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,yBAAyB;iBACvC;gBACD,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,4CAA4C;oBACzD,IAAI,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;iBACvB;aACF;SACF;QACD,KAAK,CAAC,OAAO,CAAC,WAAmB,EAAE,MAAe;YAChD,IAAI,GAAG,CAAC,cAAc,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,0CAA0C,GAAG,CAAC,cAAc,GAAG,EAAE,CAAC;oBAClG,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,CAAC,cAAc,EAAE;iBACvD,CAAC;YACJ,CAAC;YACD,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC;YACxB,MAAM,KAAK,GAAG,MAAM,4BAA4B,CAAC;gBAC/C,GAAG;gBACH,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE;gBACnB,OAAO,EAAG,GAAW,EAAE,OAAO,EAAE,OAAO;gBACvC,UAAU,EAAG,GAAW,EAAE,OAAO,EAAE,UAAU;gBAC7C,oBAAoB,EAAE,GAAG,CAAC,GAAG,EAAE,QAAQ,EAAE,UAAU;aACpD,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,EAAE,CAAQ,CAAC;YACrC,MAAM,OAAO,GAAG,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACjF,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;gBACpB,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qDAAqD,EAAE,CAAC;oBACxF,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,qBAAqB,EAAE;iBAC1D,CAAC;YACJ,CAAC;YACD,MAAM,YAAY,GAChB,OAAO,MAAM,CAAC,YAAY,KAAK,QAAQ,IAAI,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE;gBACnE,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC;gBAC7B,CAAC,CAAC,OAAO,MAAM,CAAC,aAAa,KAAK,QAAQ,IAAI,MAAM,CAAC,aAAa,CAAC,IAAI,EAAE;oBACvE,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC;oBAC9B,CAAC,CAAC,SAAS,CAAC;YAClB,MAAM,IAAI,GAAG,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAE/E,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;gBACjC,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE;gBACnB,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,KAAK,EAAE,GAAG,CAAC,YAAY,EAAE;gBACzB,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,KAAK,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE;aACzF,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;gBACf,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,oCAAoC,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC;oBACtF,OAAO,EAAE,MAAM;iBAChB,CAAC;YACJ,CAAC;YAED,kBAAkB,EAAE,CAAC;YAErB,MAAM,OAAO,GAAG;gBACd,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,YAAY,EAAE,YAAY,IAAI,IAAI;gBAClC,IAAI,EAAE,IAAI,IAAI,IAAI;aACnB,CAAC;YAEF,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;gBACnE,OAAO,EAAE,MAAM;aAChB,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-store.test.d.ts","sourceRoot":"","sources":["../../../src/plugin/tools/memory-store.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
2
|
+
import { registerMemoryStoreTool } from "./memory-store.js";
|
|
3
|
+
const { resolveScopeMock, memoryStoreDbMock } = vi.hoisted(() => ({
|
|
4
|
+
resolveScopeMock: vi.fn(async () => ({ userId: "u1", workspaceId: "w1" })),
|
|
5
|
+
memoryStoreDbMock: vi.fn(),
|
|
6
|
+
}));
|
|
7
|
+
vi.mock("../../identity.js", () => ({
|
|
8
|
+
resolveUserAndWorkspaceScope: resolveScopeMock,
|
|
9
|
+
}));
|
|
10
|
+
vi.mock("../../memory/store.js", () => ({
|
|
11
|
+
memoryStoreDb: memoryStoreDbMock,
|
|
12
|
+
}));
|
|
13
|
+
function buildCtx() {
|
|
14
|
+
const registerTool = vi.fn();
|
|
15
|
+
return {
|
|
16
|
+
ctx: {
|
|
17
|
+
api: {
|
|
18
|
+
registerTool,
|
|
19
|
+
logger: { info: vi.fn(), warn: vi.fn(), error: vi.fn() },
|
|
20
|
+
runtime: { agentId: "main", sessionKey: "agent:main:main" },
|
|
21
|
+
},
|
|
22
|
+
disabledReason: null,
|
|
23
|
+
ensureReady: vi.fn(async () => undefined),
|
|
24
|
+
getPool: vi.fn(() => ({ query: vi.fn() })),
|
|
25
|
+
cfg: {},
|
|
26
|
+
resolveActor: vi.fn(() => "tester"),
|
|
27
|
+
},
|
|
28
|
+
registerTool,
|
|
29
|
+
};
|
|
30
|
+
}
|
|
31
|
+
describe("memory_store visible output", () => {
|
|
32
|
+
beforeEach(() => {
|
|
33
|
+
vi.clearAllMocks();
|
|
34
|
+
});
|
|
35
|
+
it("returns compact JSON envelope in content on success", async () => {
|
|
36
|
+
memoryStoreDbMock.mockResolvedValueOnce({
|
|
37
|
+
ok: true,
|
|
38
|
+
id: "11111111-1111-1111-1111-111111111111",
|
|
39
|
+
path: "db-memory/items/11111111-1111-1111-1111-111111111111.md",
|
|
40
|
+
canonicalKey: "preferred_language",
|
|
41
|
+
type: "fact",
|
|
42
|
+
});
|
|
43
|
+
const { ctx, registerTool } = buildCtx();
|
|
44
|
+
registerMemoryStoreTool({
|
|
45
|
+
ctx,
|
|
46
|
+
refreshPromptCache: vi.fn(),
|
|
47
|
+
});
|
|
48
|
+
const def = registerTool.mock.calls[0]?.[0];
|
|
49
|
+
const result = await def.execute("toolcall-1", {
|
|
50
|
+
content: "I prefer TypeScript.",
|
|
51
|
+
canonicalKey: "preferred_language",
|
|
52
|
+
type: "fact",
|
|
53
|
+
});
|
|
54
|
+
const visible = JSON.parse(result.content[0].text);
|
|
55
|
+
expect(visible).toMatchObject({
|
|
56
|
+
ok: true,
|
|
57
|
+
path: "db-memory/items/11111111-1111-1111-1111-111111111111.md",
|
|
58
|
+
id: "11111111-1111-1111-1111-111111111111",
|
|
59
|
+
canonicalKey: "preferred_language",
|
|
60
|
+
type: "fact",
|
|
61
|
+
});
|
|
62
|
+
});
|
|
63
|
+
});
|
|
64
|
+
//# sourceMappingURL=memory-store.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-store.test.js","sourceRoot":"","sources":["../../../src/plugin/tools/memory-store.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAE5D,MAAM,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IAChE,gBAAgB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1E,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC3B,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,4BAA4B,EAAE,gBAAgB;CAC/C,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,aAAa,EAAE,iBAAiB;CACjC,CAAC,CAAC,CAAC;AAEJ,SAAS,QAAQ;IACf,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC7B,OAAO;QACL,GAAG,EAAE;YACH,GAAG,EAAE;gBACH,YAAY;gBACZ,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE;gBACxD,OAAO,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,iBAAiB,EAAE;aAC5D;YACD,cAAc,EAAE,IAAI;YACpB,WAAW,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC;YACzC,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,EAAE,EAAE;YACP,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC;SAC7B;QACR,YAAY;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QAClE,iBAAyB,CAAC,qBAAqB,CAAC;YAC/C,EAAE,EAAE,IAAI;YACR,EAAE,EAAE,sCAAsC;YAC1C,IAAI,EAAE,yDAAyD;YAC/D,YAAY,EAAE,oBAAoB;YAClC,IAAI,EAAE,MAAM;SACb,CAAC,CAAC;QACH,MAAM,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,QAAQ,EAAE,CAAC;QACzC,uBAAuB,CAAC;YACtB,GAAG;YACH,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;SACrB,CAAC,CAAC;QACV,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAE5C,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,YAAY,EAAE;YAC7C,OAAO,EAAE,sBAAsB;YAC/B,YAAY,EAAE,oBAAoB;YAClC,IAAI,EAAE,MAAM;SACb,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YAC5B,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,yDAAyD;YAC/D,EAAE,EAAE,sCAAsC;YAC1C,YAAY,EAAE,oBAAoB;YAClC,IAAI,EAAE,MAAM;SACb,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
type RetrievalMode = string;
|
|
2
|
+
type Hit = {
|
|
3
|
+
path?: string;
|
|
4
|
+
score?: number;
|
|
5
|
+
snippet?: string;
|
|
6
|
+
corpus?: string;
|
|
7
|
+
};
|
|
8
|
+
export declare function formatSearchLikeVisibleOutput(params: {
|
|
9
|
+
hits: Hit[];
|
|
10
|
+
retrievalMode: RetrievalMode;
|
|
11
|
+
queryMode: "exact_value" | "contextual";
|
|
12
|
+
exactTop1: boolean;
|
|
13
|
+
exactTop1Value: string | null;
|
|
14
|
+
recommendedAction: "return_exact" | "inspect_top" | "stop_not_found";
|
|
15
|
+
provider: string;
|
|
16
|
+
model: string;
|
|
17
|
+
broadContext?: boolean;
|
|
18
|
+
}): string;
|
|
19
|
+
export {};
|
|
20
|
+
//# sourceMappingURL=memory-visible-output.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-visible-output.d.ts","sourceRoot":"","sources":["../../../src/plugin/tools/memory-visible-output.ts"],"names":[],"mappings":"AAAA,KAAK,aAAa,GAAG,MAAM,CAAC;AAE5B,KAAK,GAAG,GAAG;IACT,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,wBAAgB,6BAA6B,CAAC,MAAM,EAAE;IACpD,IAAI,EAAE,GAAG,EAAE,CAAC;IACZ,aAAa,EAAE,aAAa,CAAC;IAC7B,SAAS,EAAE,aAAa,GAAG,YAAY,CAAC;IACxC,SAAS,EAAE,OAAO,CAAC;IACnB,cAAc,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,iBAAiB,EAAE,cAAc,GAAG,aAAa,GAAG,gBAAgB,CAAC;IACrE,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB,UA8DA"}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
export function formatSearchLikeVisibleOutput(params) {
|
|
2
|
+
const { hits } = params;
|
|
3
|
+
const topCandidates = hits
|
|
4
|
+
.slice(0, 5)
|
|
5
|
+
.map((hit) => {
|
|
6
|
+
const title = typeof hit.title === "string" ? hit.title.trim() : "";
|
|
7
|
+
const snippet = typeof hit.snippet === "string" ? hit.snippet.trim() : "";
|
|
8
|
+
return title || snippet;
|
|
9
|
+
})
|
|
10
|
+
.filter((value) => value.length > 0);
|
|
11
|
+
const results = hits.map((hit) => {
|
|
12
|
+
const path = typeof hit.path === "string" ? hit.path : "";
|
|
13
|
+
const snippetRaw = typeof hit.snippet === "string" ? hit.snippet : "";
|
|
14
|
+
const startLine = 1;
|
|
15
|
+
const endLine = 1;
|
|
16
|
+
const citation = path ? `${path}#L${startLine}` : "";
|
|
17
|
+
const sourceLabel = path ? `\n\nSource: ${citation}` : "";
|
|
18
|
+
return {
|
|
19
|
+
path,
|
|
20
|
+
startLine,
|
|
21
|
+
endLine,
|
|
22
|
+
score: typeof hit.score === "number" ? hit.score : 0,
|
|
23
|
+
snippet: snippetRaw ? `${snippetRaw}${sourceLabel}` : "",
|
|
24
|
+
source: hit.corpus === "sessions" ? "sessions" : "memory",
|
|
25
|
+
citation,
|
|
26
|
+
corpus: typeof hit.corpus === "string" ? hit.corpus : "memory",
|
|
27
|
+
};
|
|
28
|
+
});
|
|
29
|
+
const envelope = {
|
|
30
|
+
recommendedAction: params.recommendedAction,
|
|
31
|
+
queryMode: params.queryMode,
|
|
32
|
+
exactTop1Value: params.exactTop1Value,
|
|
33
|
+
topCandidates,
|
|
34
|
+
results,
|
|
35
|
+
provider: params.provider,
|
|
36
|
+
model: params.model,
|
|
37
|
+
citations: "auto",
|
|
38
|
+
debug: {
|
|
39
|
+
backend: "anchorclaw",
|
|
40
|
+
effectiveMode: params.retrievalMode,
|
|
41
|
+
hits: results.length,
|
|
42
|
+
},
|
|
43
|
+
meta: {
|
|
44
|
+
retrievalMode: params.retrievalMode,
|
|
45
|
+
queryMode: params.queryMode,
|
|
46
|
+
semantic: false,
|
|
47
|
+
exactTop1: params.exactTop1,
|
|
48
|
+
exactTop1Value: params.exactTop1Value,
|
|
49
|
+
recommendedAction: params.recommendedAction,
|
|
50
|
+
...(params.broadContext
|
|
51
|
+
? {
|
|
52
|
+
broadContext: true,
|
|
53
|
+
notExactEvidence: true,
|
|
54
|
+
note: "empty-query recall is broad context and should not be used as exact marker/id/key evidence",
|
|
55
|
+
}
|
|
56
|
+
: {}),
|
|
57
|
+
},
|
|
58
|
+
};
|
|
59
|
+
return JSON.stringify(envelope, null, 2);
|
|
60
|
+
}
|
|
61
|
+
//# sourceMappingURL=memory-visible-output.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-visible-output.js","sourceRoot":"","sources":["../../../src/plugin/tools/memory-visible-output.ts"],"names":[],"mappings":"AASA,MAAM,UAAU,6BAA6B,CAAC,MAU7C;IACC,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC;IACxB,MAAM,aAAa,GAAG,IAAI;SACvB,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;SACX,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACX,MAAM,KAAK,GAAG,OAAQ,GAAW,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAE,GAAW,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACtF,MAAM,OAAO,GAAG,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1E,OAAO,KAAK,IAAI,OAAO,CAAC;IAC1B,CAAC,CAAC;SACD,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAEvC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QAC/B,MAAM,IAAI,GAAG,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1D,MAAM,UAAU,GAAG,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;QACtE,MAAM,SAAS,GAAG,CAAC,CAAC;QACpB,MAAM,OAAO,GAAG,CAAC,CAAC;QAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,KAAK,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrD,MAAM,WAAW,GAAG,IAAI,CAAC,CAAC,CAAC,eAAe,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1D,OAAO;YACL,IAAI;YACJ,SAAS;YACT,OAAO;YACP,KAAK,EAAE,OAAO,GAAG,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACpD,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,UAAU,GAAG,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE;YACxD,MAAM,EAAE,GAAG,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,QAAQ;YACzD,QAAQ;YACR,MAAM,EAAE,OAAO,GAAG,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ;SAC/D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG;QACf,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;QAC3C,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,cAAc,EAAE,MAAM,CAAC,cAAc;QACrC,aAAa;QACb,OAAO;QACP,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,SAAS,EAAE,MAAM;QACjB,KAAK,EAAE;YACL,OAAO,EAAE,YAAY;YACrB,aAAa,EAAE,MAAM,CAAC,aAAa;YACnC,IAAI,EAAE,OAAO,CAAC,MAAM;SACrB;QACD,IAAI,EAAE;YACJ,aAAa,EAAE,MAAM,CAAC,aAAa;YACnC,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,cAAc,EAAE,MAAM,CAAC,cAAc;YACrC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;YAC3C,GAAG,CAAC,MAAM,CAAC,YAAY;gBACrB,CAAC,CAAC;oBACE,YAAY,EAAE,IAAI;oBAClB,gBAAgB,EAAE,IAAI;oBACtB,IAAI,EAAE,4FAA4F;iBACnG;gBACH,CAAC,CAAC,EAAE,CAAC;SACR;KACF,CAAC;IAEF,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAC3C,CAAC"}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
export type SessionDeltaState = {
|
|
2
|
+
lastSize: number;
|
|
3
|
+
pendingBytes: number;
|
|
4
|
+
pendingMessages: number;
|
|
5
|
+
};
|
|
6
|
+
export type SdkHealthState = {
|
|
7
|
+
degraded: boolean;
|
|
8
|
+
reason?: string;
|
|
9
|
+
affectedOperation?: string;
|
|
10
|
+
lastErrorAt?: string;
|
|
11
|
+
consecutiveSuccesses: number;
|
|
12
|
+
};
|
|
13
|
+
export type SessionDeltaThresholds = {
|
|
14
|
+
deltaBytes: number;
|
|
15
|
+
deltaMessages: number;
|
|
16
|
+
};
|
|
17
|
+
export type MemoryStatusCheckResult = {
|
|
18
|
+
ok: boolean;
|
|
19
|
+
mode: "cached" | "active";
|
|
20
|
+
sdk: SdkHealthState;
|
|
21
|
+
database?: {
|
|
22
|
+
ok: boolean;
|
|
23
|
+
latencyMs?: number;
|
|
24
|
+
schemaOk?: boolean;
|
|
25
|
+
migrationVersion?: string | null;
|
|
26
|
+
error?: string;
|
|
27
|
+
};
|
|
28
|
+
sessions?: {
|
|
29
|
+
enabled: boolean;
|
|
30
|
+
visibility: "off" | "current" | "visible";
|
|
31
|
+
stateDir?: string;
|
|
32
|
+
agentSessionsDir?: string;
|
|
33
|
+
exists?: boolean;
|
|
34
|
+
readable?: boolean;
|
|
35
|
+
error?: string;
|
|
36
|
+
};
|
|
37
|
+
index?: {
|
|
38
|
+
trackedFiles: number;
|
|
39
|
+
pendingBytes: number;
|
|
40
|
+
pendingMessages: number;
|
|
41
|
+
};
|
|
42
|
+
};
|
|
43
|
+
export type PromptCacheState = {
|
|
44
|
+
lines: string[] | null;
|
|
45
|
+
error: string | null;
|
|
46
|
+
refreshPromise: Promise<void> | null;
|
|
47
|
+
};
|
|
48
|
+
export type SessionsIndexState = {
|
|
49
|
+
bootstrapPromise: Promise<void> | null;
|
|
50
|
+
bootstrapped: boolean;
|
|
51
|
+
};
|
|
52
|
+
export type SessionDeltaRuntimeState = {
|
|
53
|
+
pendingFiles: Set<string>;
|
|
54
|
+
timer: ReturnType<typeof setTimeout> | null;
|
|
55
|
+
syncInFlight: Promise<void> | null;
|
|
56
|
+
unsubscribe: (() => void) | null;
|
|
57
|
+
closed: boolean;
|
|
58
|
+
ignoredPathCounts: Map<string, number>;
|
|
59
|
+
stateByPath: Map<string, SessionDeltaState>;
|
|
60
|
+
};
|
|
61
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/plugin/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,iBAAiB,GAAG;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,EAAE,OAAO,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,oBAAoB,EAAE,MAAM,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAAG;IACpC,EAAE,EAAE,OAAO,CAAC;IACZ,IAAI,EAAE,QAAQ,GAAG,QAAQ,CAAC;IAC1B,GAAG,EAAE,cAAc,CAAC;IACpB,QAAQ,CAAC,EAAE;QACT,EAAE,EAAE,OAAO,CAAC;QACZ,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,QAAQ,CAAC,EAAE,OAAO,CAAC;QACnB,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACjC,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,QAAQ,CAAC,EAAE;QACT,OAAO,EAAE,OAAO,CAAC;QACjB,UAAU,EAAE,KAAK,GAAG,SAAS,GAAG,SAAS,CAAC;QAC1C,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,MAAM,CAAC,EAAE,OAAO,CAAC;QACjB,QAAQ,CAAC,EAAE,OAAO,CAAC;QACnB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,KAAK,CAAC,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,eAAe,EAAE,MAAM,CAAC;KACzB,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IACvB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,cAAc,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CACtC,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,gBAAgB,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACvC,YAAY,EAAE,OAAO,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC,YAAY,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC1B,KAAK,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,IAAI,CAAC;IAC5C,YAAY,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACnC,WAAW,EAAE,CAAC,MAAM,IAAI,CAAC,GAAG,IAAI,CAAC;IACjC,MAAM,EAAE,OAAO,CAAC;IAChB,iBAAiB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,WAAW,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;CAC7C,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/plugin/types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../src/postgres.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuB,MAAM,IAAI,CAAC;AACzC,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAEpD,MAAM,MAAM,YAAY,GAAG,EAAE,CAAC,IAAI,CAAC;AAoCnC,wBAAgB,kBAAkB,CAAC,MAAM,EAAE;IAAE,GAAG,EAAE,gBAAgB,CAAA;CAAE,GAAG,EAAE,CAAC,IAAI,CAwB7E"}
|
package/dist/postgres.js
ADDED
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
import pg, {} from "pg";
|
|
2
|
+
function assertSafeIdentifier(value, label) {
|
|
3
|
+
const trimmed = value.trim();
|
|
4
|
+
if (!trimmed) {
|
|
5
|
+
throw new Error(`${label} must be a non-empty identifier`);
|
|
6
|
+
}
|
|
7
|
+
if (!/^[A-Za-z_][A-Za-z0-9_]*$/.test(trimmed)) {
|
|
8
|
+
throw new Error(`${label} must be a simple identifier (letters/numbers/underscore)`);
|
|
9
|
+
}
|
|
10
|
+
return trimmed;
|
|
11
|
+
}
|
|
12
|
+
function resolvePgSslOptions(params) {
|
|
13
|
+
if (params.sslMode === "disable") {
|
|
14
|
+
return false;
|
|
15
|
+
}
|
|
16
|
+
if (params.sslMode === "require") {
|
|
17
|
+
return { rejectUnauthorized: false };
|
|
18
|
+
}
|
|
19
|
+
if (params.sslMode === "verify-full") {
|
|
20
|
+
return {
|
|
21
|
+
rejectUnauthorized: true,
|
|
22
|
+
...(params.sslCa ? { ca: params.sslCa } : {}),
|
|
23
|
+
};
|
|
24
|
+
}
|
|
25
|
+
if (typeof params.ssl === "boolean") {
|
|
26
|
+
return params.ssl;
|
|
27
|
+
}
|
|
28
|
+
return false;
|
|
29
|
+
}
|
|
30
|
+
export function createPostgresPool(params) {
|
|
31
|
+
const { postgres } = params.cfg;
|
|
32
|
+
const ssl = resolvePgSslOptions({
|
|
33
|
+
ssl: postgres.ssl,
|
|
34
|
+
sslMode: postgres.sslMode,
|
|
35
|
+
sslCa: postgres.sslCa,
|
|
36
|
+
});
|
|
37
|
+
const schema = postgres.schema ? assertSafeIdentifier(postgres.schema, "postgres.schema") : undefined;
|
|
38
|
+
const poolMax = postgres.pool?.max ?? 10;
|
|
39
|
+
const connectionTimeoutMillis = postgres.pool?.connectionTimeoutMs ?? 5_000;
|
|
40
|
+
const idleTimeoutMillis = postgres.pool?.idleTimeoutMs ?? 30_000;
|
|
41
|
+
const port = postgres.port ?? 5432;
|
|
42
|
+
return new pg.Pool({
|
|
43
|
+
host: postgres.host,
|
|
44
|
+
port,
|
|
45
|
+
database: postgres.database,
|
|
46
|
+
user: postgres.user,
|
|
47
|
+
...(postgres.password ? { password: postgres.password } : {}),
|
|
48
|
+
ssl,
|
|
49
|
+
max: poolMax,
|
|
50
|
+
connectionTimeoutMillis,
|
|
51
|
+
idleTimeoutMillis,
|
|
52
|
+
...(schema ? { options: `-c search_path=${schema},public` } : {}),
|
|
53
|
+
});
|
|
54
|
+
}
|
|
55
|
+
//# sourceMappingURL=postgres.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"postgres.js","sourceRoot":"","sources":["../src/postgres.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAmB,MAAM,IAAI,CAAC;AAKzC,SAAS,oBAAoB,CAAC,KAAa,EAAE,KAAa;IACxD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAC7B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,GAAG,KAAK,iCAAiC,CAAC,CAAC;IAC7D,CAAC;IACD,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAI,KAAK,CAAC,GAAG,KAAK,2DAA2D,CAAC,CAAC;IACvF,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,mBAAmB,CAAC,MAI5B;IACC,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAAC;IACvC,CAAC;IACD,IAAI,MAAM,CAAC,OAAO,KAAK,aAAa,EAAE,CAAC;QACrC,OAAO;YACL,kBAAkB,EAAE,IAAI;YACxB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC9C,CAAC;IACJ,CAAC;IACD,IAAI,OAAO,MAAM,CAAC,GAAG,KAAK,SAAS,EAAE,CAAC;QACpC,OAAO,MAAM,CAAC,GAAG,CAAC;IACpB,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAAiC;IAClE,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC,GAAG,CAAC;IAChC,MAAM,GAAG,GAAG,mBAAmB,CAAC;QAC9B,GAAG,EAAE,QAAQ,CAAC,GAAG;QACjB,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,KAAK,EAAE,QAAQ,CAAC,KAAK;KACtB,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACtG,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,EAAE,GAAG,IAAI,EAAE,CAAC;IACzC,MAAM,uBAAuB,GAAG,QAAQ,CAAC,IAAI,EAAE,mBAAmB,IAAI,KAAK,CAAC;IAC5E,MAAM,iBAAiB,GAAG,QAAQ,CAAC,IAAI,EAAE,aAAa,IAAI,MAAM,CAAC;IACjE,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,IAAI,IAAI,CAAC;IACnC,OAAO,IAAI,EAAE,CAAC,IAAI,CAAC;QACjB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,IAAI;QACJ,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7D,GAAG;QACH,GAAG,EAAE,OAAO;QACZ,uBAAuB;QACvB,iBAAiB;QACjB,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,kBAAkB,MAAM,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAClE,CAAC,CAAC;AACL,CAAC"}
|