@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,332 @@
|
|
|
1
|
+
import { resolveUserAndWorkspaceScope } from "../../identity.js";
|
|
2
|
+
import { resolveMemoryLimits } from "../../memory/limits.js";
|
|
3
|
+
import { memorySearchDb } from "../../memory/search.js";
|
|
4
|
+
import { memorySearchSessions } from "../../memory/sessions.js";
|
|
5
|
+
import { hasSessionsIndexRows, memorySearchSessionsIndexDb } from "../../memory/sessions-index.js";
|
|
6
|
+
import { filterSessionHitsByVisibility } from "../../memory/sessions-visibility.js";
|
|
7
|
+
import { formatSearchLikeVisibleOutput } from "./memory-visible-output.js";
|
|
8
|
+
function normalizeExact(value) {
|
|
9
|
+
return typeof value === "string" ? value.trim().toLowerCase() : "";
|
|
10
|
+
}
|
|
11
|
+
function classifyQueryMode(query) {
|
|
12
|
+
const normalized = query.trim().toLowerCase();
|
|
13
|
+
if (!normalized) {
|
|
14
|
+
return "contextual";
|
|
15
|
+
}
|
|
16
|
+
const broadSignals = [
|
|
17
|
+
"summarize",
|
|
18
|
+
"summary",
|
|
19
|
+
"overview",
|
|
20
|
+
"around",
|
|
21
|
+
"what values",
|
|
22
|
+
"какие",
|
|
23
|
+
"кратко",
|
|
24
|
+
"обзор",
|
|
25
|
+
"сгруппируй",
|
|
26
|
+
"вокруг",
|
|
27
|
+
];
|
|
28
|
+
if (broadSignals.some((signal) => normalized.includes(signal))) {
|
|
29
|
+
return "contextual";
|
|
30
|
+
}
|
|
31
|
+
const exactSignals = [
|
|
32
|
+
"exact",
|
|
33
|
+
"top-1",
|
|
34
|
+
"only",
|
|
35
|
+
"value only",
|
|
36
|
+
"string only",
|
|
37
|
+
"id",
|
|
38
|
+
"key",
|
|
39
|
+
"marker",
|
|
40
|
+
"token",
|
|
41
|
+
"uuid",
|
|
42
|
+
"login",
|
|
43
|
+
"точн",
|
|
44
|
+
"только",
|
|
45
|
+
"значени",
|
|
46
|
+
];
|
|
47
|
+
return exactSignals.some((signal) => normalized.includes(signal)) ? "exact_value" : "contextual";
|
|
48
|
+
}
|
|
49
|
+
function isMarkerLike(value) {
|
|
50
|
+
const text = value.trim();
|
|
51
|
+
if (!text) {
|
|
52
|
+
return false;
|
|
53
|
+
}
|
|
54
|
+
if (/^[A-Z0-9]+(?:_[A-Z0-9]+){2,}$/.test(text)) {
|
|
55
|
+
return true;
|
|
56
|
+
}
|
|
57
|
+
if (/^[0-9a-f]{8}-[0-9a-f]{4}-[1-5][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i.test(text)) {
|
|
58
|
+
return true;
|
|
59
|
+
}
|
|
60
|
+
if (/^@\w{3,}$/.test(text)) {
|
|
61
|
+
return true;
|
|
62
|
+
}
|
|
63
|
+
return /^[0-9]{6,}$/.test(text);
|
|
64
|
+
}
|
|
65
|
+
function markerBoostForHit(hit) {
|
|
66
|
+
const title = typeof hit?.title === "string" ? hit.title : "";
|
|
67
|
+
const snippet = typeof hit?.snippet === "string" ? hit.snippet : "";
|
|
68
|
+
return isMarkerLike(title) || isMarkerLike(snippet) ? 2 : 0;
|
|
69
|
+
}
|
|
70
|
+
export function registerMemorySearchTool({ ctx, ensureSessionsIndexBootstrapped, }) {
|
|
71
|
+
const api = ctx.api;
|
|
72
|
+
api.registerTool({
|
|
73
|
+
name: "memory_search",
|
|
74
|
+
label: "Memory Search",
|
|
75
|
+
description: "Search memory.\n\nBehavior contract:\n- corpus defaults to \"memory\" (durable items in Postgres).\n- For corpus=\"memory\", retrieval is lexical Postgres FTS (tsquery/ts_rank), not vector semantic retrieval.\n- corpus=\"sessions\" uses Postgres-backed sessions index (DB-first) and returns paths like sessions/<agentId>/<session>.jsonl.\n- Results contain synthetic paths. Use memory_get to read them.\n- If the top result is an exact literal match, content includes \"Top exact match: <value>\" and details.meta.exactTop1/exactTop1Value.\n- Do not claim semantic/vector retrieval unless details.meta explicitly reports it in a future implementation.",
|
|
76
|
+
parameters: {
|
|
77
|
+
type: "object",
|
|
78
|
+
additionalProperties: false,
|
|
79
|
+
properties: {
|
|
80
|
+
query: { type: "string", description: "Search query text." },
|
|
81
|
+
corpus: {
|
|
82
|
+
type: "string",
|
|
83
|
+
description: "Memory corpus (memory|sessions|all). Defaults to memory.",
|
|
84
|
+
enum: ["memory", "sessions", "all", "wiki"],
|
|
85
|
+
},
|
|
86
|
+
maxResults: { type: "number", description: "Max results (capped by configured limits)." },
|
|
87
|
+
minScore: { type: "number", description: "Optional minimum score threshold." },
|
|
88
|
+
},
|
|
89
|
+
},
|
|
90
|
+
async execute(_toolCallId, params) {
|
|
91
|
+
if (ctx.disabledReason) {
|
|
92
|
+
return {
|
|
93
|
+
content: [{ type: "text", text: `anchorclaw: disabled until configured (${ctx.disabledReason})` }],
|
|
94
|
+
details: { disabled: true, error: ctx.disabledReason },
|
|
95
|
+
};
|
|
96
|
+
}
|
|
97
|
+
await ctx.ensureReady();
|
|
98
|
+
const scope = await resolveUserAndWorkspaceScope({
|
|
99
|
+
api,
|
|
100
|
+
pool: ctx.getPool(),
|
|
101
|
+
agentId: api?.runtime?.agentId,
|
|
102
|
+
sessionKey: api?.runtime?.sessionKey,
|
|
103
|
+
configuredExternalId: ctx.cfg?.identity?.externalId,
|
|
104
|
+
});
|
|
105
|
+
const limits = resolveMemoryLimits(ctx.cfg);
|
|
106
|
+
const record = (params ?? {});
|
|
107
|
+
const query = typeof record.query === "string" ? String(record.query) : "";
|
|
108
|
+
const corpus = typeof record.corpus === "string" ? String(record.corpus) : "memory";
|
|
109
|
+
const maxResults = typeof record.maxResults === "number" ? record.maxResults : undefined;
|
|
110
|
+
const minScore = typeof record.minScore === "number" ? record.minScore : undefined;
|
|
111
|
+
const effectiveMax = typeof maxResults === "number" ? maxResults : limits.maxResults;
|
|
112
|
+
const trimmedCorpus = corpus.trim();
|
|
113
|
+
const sessionsVisibility = ctx.cfg?.sessions?.visibility ?? "current";
|
|
114
|
+
const sessionsEnabled = sessionsVisibility !== "off";
|
|
115
|
+
if (trimmedCorpus === "wiki") {
|
|
116
|
+
return {
|
|
117
|
+
content: [
|
|
118
|
+
{
|
|
119
|
+
type: "text",
|
|
120
|
+
text: "anchorclaw: corpus=wiki is not implemented yet. Use the memory-wiki tools (wiki_search/wiki_get) for now.",
|
|
121
|
+
},
|
|
122
|
+
],
|
|
123
|
+
details: { disabled: true, error: "corpus=wiki not implemented" },
|
|
124
|
+
};
|
|
125
|
+
}
|
|
126
|
+
let hits = [];
|
|
127
|
+
let retrievalMode = "fts_memory";
|
|
128
|
+
try {
|
|
129
|
+
if (trimmedCorpus === "sessions") {
|
|
130
|
+
if (!sessionsEnabled) {
|
|
131
|
+
return {
|
|
132
|
+
content: [{ type: "text", text: "anchorclaw: sessions corpus is disabled by config (sessions.visibility=off)" }],
|
|
133
|
+
details: { disabled: true, error: "sessions corpus disabled", visibility: sessionsVisibility },
|
|
134
|
+
};
|
|
135
|
+
}
|
|
136
|
+
await ensureSessionsIndexBootstrapped();
|
|
137
|
+
const indexedHits = await memorySearchSessionsIndexDb({
|
|
138
|
+
pool: ctx.getPool(),
|
|
139
|
+
userId: scope.userId,
|
|
140
|
+
workspaceId: scope.workspaceId,
|
|
141
|
+
limits,
|
|
142
|
+
query,
|
|
143
|
+
maxResults: effectiveMax,
|
|
144
|
+
...(sessionsVisibility === "current" ? { currentAgentId: String(api?.runtime?.agentId ?? "main") } : {}),
|
|
145
|
+
});
|
|
146
|
+
if (indexedHits.length > 0) {
|
|
147
|
+
hits = indexedHits;
|
|
148
|
+
retrievalMode = "sessions_index";
|
|
149
|
+
}
|
|
150
|
+
else {
|
|
151
|
+
const hasIndex = await hasSessionsIndexRows({
|
|
152
|
+
pool: ctx.getPool(),
|
|
153
|
+
userId: scope.userId,
|
|
154
|
+
workspaceId: scope.workspaceId,
|
|
155
|
+
...(sessionsVisibility === "current" ? { currentAgentId: String(api?.runtime?.agentId ?? "main") } : {}),
|
|
156
|
+
});
|
|
157
|
+
hits = hasIndex
|
|
158
|
+
? []
|
|
159
|
+
: await memorySearchSessions({
|
|
160
|
+
query,
|
|
161
|
+
maxResults: effectiveMax,
|
|
162
|
+
agentId: api?.runtime?.agentId,
|
|
163
|
+
limits,
|
|
164
|
+
});
|
|
165
|
+
retrievalMode = hasIndex ? "sessions_index" : "sessions_fallback";
|
|
166
|
+
}
|
|
167
|
+
hits = await filterSessionHitsByVisibility({ api, hits });
|
|
168
|
+
}
|
|
169
|
+
else if (trimmedCorpus === "memory") {
|
|
170
|
+
hits = await memorySearchDb({
|
|
171
|
+
pool: ctx.getPool(),
|
|
172
|
+
userId: scope.userId,
|
|
173
|
+
workspaceId: scope.workspaceId,
|
|
174
|
+
limits,
|
|
175
|
+
query,
|
|
176
|
+
...(typeof maxResults === "number" ? { maxResults } : {}),
|
|
177
|
+
});
|
|
178
|
+
retrievalMode = "fts_memory";
|
|
179
|
+
}
|
|
180
|
+
else if (trimmedCorpus === "all") {
|
|
181
|
+
if (sessionsEnabled) {
|
|
182
|
+
await ensureSessionsIndexBootstrapped();
|
|
183
|
+
}
|
|
184
|
+
const merged = [
|
|
185
|
+
...(await memorySearchDb({
|
|
186
|
+
pool: ctx.getPool(),
|
|
187
|
+
userId: scope.userId,
|
|
188
|
+
workspaceId: scope.workspaceId,
|
|
189
|
+
limits,
|
|
190
|
+
query,
|
|
191
|
+
maxResults: effectiveMax,
|
|
192
|
+
})),
|
|
193
|
+
...(sessionsEnabled
|
|
194
|
+
? await memorySearchSessionsIndexDb({
|
|
195
|
+
pool: ctx.getPool(),
|
|
196
|
+
userId: scope.userId,
|
|
197
|
+
workspaceId: scope.workspaceId,
|
|
198
|
+
limits,
|
|
199
|
+
query,
|
|
200
|
+
maxResults: effectiveMax,
|
|
201
|
+
...(sessionsVisibility === "current" ? { currentAgentId: String(api?.runtime?.agentId ?? "main") } : {}),
|
|
202
|
+
})
|
|
203
|
+
: []),
|
|
204
|
+
];
|
|
205
|
+
if (sessionsEnabled) {
|
|
206
|
+
const hasSessionsHits = merged.some((item) => item?.corpus === "sessions");
|
|
207
|
+
if (!hasSessionsHits) {
|
|
208
|
+
const hasIndex = await hasSessionsIndexRows({
|
|
209
|
+
pool: ctx.getPool(),
|
|
210
|
+
userId: scope.userId,
|
|
211
|
+
workspaceId: scope.workspaceId,
|
|
212
|
+
...(sessionsVisibility === "current" ? { currentAgentId: String(api?.runtime?.agentId ?? "main") } : {}),
|
|
213
|
+
});
|
|
214
|
+
if (!hasIndex) {
|
|
215
|
+
merged.push(...(await memorySearchSessions({
|
|
216
|
+
query,
|
|
217
|
+
maxResults: effectiveMax,
|
|
218
|
+
agentId: api?.runtime?.agentId,
|
|
219
|
+
limits,
|
|
220
|
+
})));
|
|
221
|
+
}
|
|
222
|
+
}
|
|
223
|
+
}
|
|
224
|
+
const mergedForOutput = sessionsEnabled
|
|
225
|
+
? await filterSessionHitsByVisibility({ api, hits: merged })
|
|
226
|
+
: merged;
|
|
227
|
+
mergedForOutput.sort((left, right) => {
|
|
228
|
+
const ls = typeof left?.score === "number" ? left.score : 0;
|
|
229
|
+
const rs = typeof right?.score === "number" ? right.score : 0;
|
|
230
|
+
if (rs !== ls) {
|
|
231
|
+
return rs - ls;
|
|
232
|
+
}
|
|
233
|
+
const lc = left?.corpus === "sessions" ? "sessions" : "memory";
|
|
234
|
+
const rc = right?.corpus === "sessions" ? "sessions" : "memory";
|
|
235
|
+
if (lc !== rc) {
|
|
236
|
+
return lc === "memory" ? -1 : 1;
|
|
237
|
+
}
|
|
238
|
+
const lp = typeof left?.path === "string" ? left.path : "";
|
|
239
|
+
const rp = typeof right?.path === "string" ? right.path : "";
|
|
240
|
+
return lp.localeCompare(rp);
|
|
241
|
+
});
|
|
242
|
+
hits = mergedForOutput.slice(0, effectiveMax);
|
|
243
|
+
retrievalMode = "all_merge_lexical";
|
|
244
|
+
}
|
|
245
|
+
ctx.markSdkSuccess();
|
|
246
|
+
}
|
|
247
|
+
catch (error) {
|
|
248
|
+
ctx.markSdkError(`memory_search:${trimmedCorpus || "unknown"}`, error);
|
|
249
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
250
|
+
return {
|
|
251
|
+
content: [{ type: "text", text: `anchorclaw: memory_search degraded (sdk/runtime error: ${message})` }],
|
|
252
|
+
details: {
|
|
253
|
+
disabled: true,
|
|
254
|
+
error: message,
|
|
255
|
+
degraded: true,
|
|
256
|
+
degradedReason: "sdk_error",
|
|
257
|
+
sdk: { ...ctx.sdkHealth },
|
|
258
|
+
},
|
|
259
|
+
};
|
|
260
|
+
}
|
|
261
|
+
if (trimmedCorpus !== "memory" && trimmedCorpus !== "sessions" && trimmedCorpus !== "all") {
|
|
262
|
+
return {
|
|
263
|
+
content: [{ type: "text", text: `anchorclaw: unsupported corpus (${trimmedCorpus || "empty"})` }],
|
|
264
|
+
details: { disabled: true, error: "unsupported corpus", corpus: trimmedCorpus },
|
|
265
|
+
};
|
|
266
|
+
}
|
|
267
|
+
if (typeof minScore === "number" && Number.isFinite(minScore)) {
|
|
268
|
+
hits = hits.filter((hit) => typeof hit?.score === "number" && hit.score >= minScore);
|
|
269
|
+
}
|
|
270
|
+
const normalizedQuery = normalizeExact(query);
|
|
271
|
+
const queryMode = classifyQueryMode(query);
|
|
272
|
+
if (queryMode === "exact_value") {
|
|
273
|
+
hits.sort((left, right) => {
|
|
274
|
+
const ls = (typeof left?.score === "number" ? left.score : 0) + markerBoostForHit(left);
|
|
275
|
+
const rs = (typeof right?.score === "number" ? right.score : 0) + markerBoostForHit(right);
|
|
276
|
+
if (rs !== ls) {
|
|
277
|
+
return rs - ls;
|
|
278
|
+
}
|
|
279
|
+
const lp = typeof left?.path === "string" ? left.path : "";
|
|
280
|
+
const rp = typeof right?.path === "string" ? right.path : "";
|
|
281
|
+
return lp.localeCompare(rp);
|
|
282
|
+
});
|
|
283
|
+
}
|
|
284
|
+
const topHit = hits[0];
|
|
285
|
+
const topTitle = normalizeExact(topHit?.title);
|
|
286
|
+
const topSnippet = normalizeExact(topHit?.snippet);
|
|
287
|
+
const exactTop1 = Boolean(normalizedQuery &&
|
|
288
|
+
topHit &&
|
|
289
|
+
(topTitle === normalizedQuery || topSnippet === normalizedQuery));
|
|
290
|
+
const exactTop1Value = exactTop1
|
|
291
|
+
? (typeof topHit?.title === "string" && topHit.title.trim()) ||
|
|
292
|
+
(typeof topHit?.snippet === "string" && topHit.snippet.trim()) ||
|
|
293
|
+
null
|
|
294
|
+
: null;
|
|
295
|
+
const recommendedAction = exactTop1 ? "return_exact" : hits.length > 0 ? "inspect_top" : "stop_not_found";
|
|
296
|
+
const visible = formatSearchLikeVisibleOutput({
|
|
297
|
+
hits,
|
|
298
|
+
retrievalMode,
|
|
299
|
+
queryMode,
|
|
300
|
+
exactTop1,
|
|
301
|
+
exactTop1Value,
|
|
302
|
+
recommendedAction,
|
|
303
|
+
provider: "anchorclaw",
|
|
304
|
+
model: "postgres-fts",
|
|
305
|
+
});
|
|
306
|
+
return {
|
|
307
|
+
content: [
|
|
308
|
+
{
|
|
309
|
+
type: "text",
|
|
310
|
+
text: visible,
|
|
311
|
+
},
|
|
312
|
+
],
|
|
313
|
+
details: {
|
|
314
|
+
results: hits,
|
|
315
|
+
count: hits.length,
|
|
316
|
+
meta: {
|
|
317
|
+
retrievalMode,
|
|
318
|
+
queryMode,
|
|
319
|
+
semantic: false,
|
|
320
|
+
exactTop1,
|
|
321
|
+
exactTop1Value,
|
|
322
|
+
recommendedAction,
|
|
323
|
+
},
|
|
324
|
+
...(ctx.sdkHealth.degraded
|
|
325
|
+
? { degraded: true, degradedReason: "sdk_error", sdk: { ...ctx.sdkHealth } }
|
|
326
|
+
: {}),
|
|
327
|
+
},
|
|
328
|
+
};
|
|
329
|
+
},
|
|
330
|
+
});
|
|
331
|
+
}
|
|
332
|
+
//# sourceMappingURL=memory-search.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-search.js","sourceRoot":"","sources":["../../../src/plugin/tools/memory-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,4BAA4B,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAChE,OAAO,EAAE,oBAAoB,EAAE,2BAA2B,EAAE,MAAM,gCAAgC,CAAC;AACnG,OAAO,EAAE,6BAA6B,EAAE,MAAM,qCAAqC,CAAC;AAEpF,OAAO,EAAE,6BAA6B,EAAE,MAAM,4BAA4B,CAAC;AAE3E,SAAS,cAAc,CAAC,KAAc;IACpC,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;AACrE,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAa;IACtC,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC9C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,YAAY,CAAC;IACtB,CAAC;IACD,MAAM,YAAY,GAAG;QACnB,WAAW;QACX,SAAS;QACT,UAAU;QACV,QAAQ;QACR,aAAa;QACb,OAAO;QACP,QAAQ;QACR,OAAO;QACP,YAAY;QACZ,QAAQ;KACT,CAAC;IACF,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;QAC/D,OAAO,YAAY,CAAC;IACtB,CAAC;IACD,MAAM,YAAY,GAAG;QACnB,OAAO;QACP,OAAO;QACP,MAAM;QACN,YAAY;QACZ,aAAa;QACb,IAAI;QACJ,KAAK;QACL,QAAQ;QACR,OAAO;QACP,MAAM;QACN,OAAO;QACP,MAAM;QACN,QAAQ;QACR,SAAS;KACV,CAAC;IACF,OAAO,YAAY,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,YAAY,CAAC;AACnG,CAAC;AAED,SAAS,YAAY,CAAC,KAAa;IACjC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAC1B,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,+BAA+B,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/C,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,4EAA4E,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5F,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAClC,CAAC;AAED,SAAS,iBAAiB,CAAC,GAAQ;IACjC,MAAM,KAAK,GAAG,OAAO,GAAG,EAAE,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IAC9D,MAAM,OAAO,GAAG,OAAO,GAAG,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IACpE,OAAO,YAAY,CAAC,KAAK,CAAC,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC9D,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,EACvC,GAAG,EACH,+BAA+B,GACR;IACvB,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC;IACpB,GAAG,CAAC,YAAY,CAAC;QACf,IAAI,EAAE,eAAe;QACrB,KAAK,EAAE,eAAe;QACtB,WAAW,EACT,6oBAA6oB;QAC/oB,UAAU,EAAE;YACV,IAAI,EAAE,QAAQ;YACd,oBAAoB,EAAE,KAAK;YAC3B,UAAU,EAAE;gBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,oBAAoB,EAAE;gBAC5D,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,0DAA0D;oBACvE,IAAI,EAAE,CAAC,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,CAAC;iBAC5C;gBACD,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,4CAA4C,EAAE;gBACzF,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,mCAAmC,EAAE;aAC/E;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;YACH,MAAM,MAAM,GAAG,mBAAmB,CAAC,GAAG,CAAC,GAAI,CAAC,CAAC;YAC7C,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,EAAE,CAAQ,CAAC;YACrC,MAAM,KAAK,GAAG,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3E,MAAM,MAAM,GAAG,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;YACpF,MAAM,UAAU,GAAG,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAE,MAAM,CAAC,UAAqB,CAAC,CAAC,CAAC,SAAS,CAAC;YACrG,MAAM,QAAQ,GAAG,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAE,MAAM,CAAC,QAAmB,CAAC,CAAC,CAAC,SAAS,CAAC;YAC/F,MAAM,YAAY,GAAG,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC;YACrF,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;YACpC,MAAM,kBAAkB,GAAG,GAAG,CAAC,GAAG,EAAE,QAAQ,EAAE,UAAU,IAAI,SAAS,CAAC;YACtE,MAAM,eAAe,GAAG,kBAAkB,KAAK,KAAK,CAAC;YACrD,IAAI,aAAa,KAAK,MAAM,EAAE,CAAC;gBAC7B,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EACF,2GAA2G;yBAC9G;qBACF;oBACD,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,6BAA6B,EAAE;iBAClE,CAAC;YACJ,CAAC;YACD,IAAI,IAAI,GAAU,EAAE,CAAC;YACrB,IAAI,aAAa,GAAgF,YAAY,CAAC;YAC9G,IAAI,CAAC;gBACH,IAAI,aAAa,KAAK,UAAU,EAAE,CAAC;oBACjC,IAAI,CAAC,eAAe,EAAE,CAAC;wBACrB,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,6EAA6E,EAAE,CAAC;4BAChH,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,0BAA0B,EAAE,UAAU,EAAE,kBAAkB,EAAE;yBAC/F,CAAC;oBACJ,CAAC;oBACD,MAAM,+BAA+B,EAAE,CAAC;oBACxC,MAAM,WAAW,GAAG,MAAM,2BAA2B,CAAC;wBACpD,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE;wBACnB,MAAM,EAAE,KAAK,CAAC,MAAM;wBACpB,WAAW,EAAE,KAAK,CAAC,WAAW;wBAC9B,MAAM;wBACN,KAAK;wBACL,UAAU,EAAE,YAAY;wBACxB,GAAG,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,CAAE,GAAW,EAAE,OAAO,EAAE,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAClH,CAAC,CAAC;oBACH,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC3B,IAAI,GAAG,WAAW,CAAC;wBACnB,aAAa,GAAG,gBAAgB,CAAC;oBACnC,CAAC;yBAAM,CAAC;wBACN,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC;4BAC1C,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE;4BACnB,MAAM,EAAE,KAAK,CAAC,MAAM;4BACpB,WAAW,EAAE,KAAK,CAAC,WAAW;4BAC9B,GAAG,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,CAAE,GAAW,EAAE,OAAO,EAAE,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;yBAClH,CAAC,CAAC;wBACH,IAAI,GAAG,QAAQ;4BACb,CAAC,CAAC,EAAE;4BACJ,CAAC,CAAC,MAAM,oBAAoB,CAAC;gCACzB,KAAK;gCACL,UAAU,EAAE,YAAY;gCACxB,OAAO,EAAG,GAAW,EAAE,OAAO,EAAE,OAAO;gCACvC,MAAM;6BACP,CAAC,CAAC;wBACP,aAAa,GAAG,QAAQ,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,mBAAmB,CAAC;oBACpE,CAAC;oBACD,IAAI,GAAG,MAAM,6BAA6B,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC5D,CAAC;qBAAM,IAAI,aAAa,KAAK,QAAQ,EAAE,CAAC;oBACtC,IAAI,GAAG,MAAM,cAAc,CAAC;wBAC1B,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE;wBACnB,MAAM,EAAE,KAAK,CAAC,MAAM;wBACpB,WAAW,EAAE,KAAK,CAAC,WAAW;wBAC9B,MAAM;wBACN,KAAK;wBACL,GAAG,CAAC,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC1D,CAAC,CAAC;oBACH,aAAa,GAAG,YAAY,CAAC;gBAC/B,CAAC;qBAAM,IAAI,aAAa,KAAK,KAAK,EAAE,CAAC;oBACnC,IAAI,eAAe,EAAE,CAAC;wBACpB,MAAM,+BAA+B,EAAE,CAAC;oBAC1C,CAAC;oBACD,MAAM,MAAM,GAAG;wBACb,GAAG,CAAC,MAAM,cAAc,CAAC;4BACvB,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE;4BACnB,MAAM,EAAE,KAAK,CAAC,MAAM;4BACpB,WAAW,EAAE,KAAK,CAAC,WAAW;4BAC9B,MAAM;4BACN,KAAK;4BACL,UAAU,EAAE,YAAY;yBACzB,CAAC,CAAC;wBACH,GAAG,CAAC,eAAe;4BACjB,CAAC,CAAC,MAAM,2BAA2B,CAAC;gCAChC,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE;gCACnB,MAAM,EAAE,KAAK,CAAC,MAAM;gCACpB,WAAW,EAAE,KAAK,CAAC,WAAW;gCAC9B,MAAM;gCACN,KAAK;gCACL,UAAU,EAAE,YAAY;gCACxB,GAAG,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,CAAE,GAAW,EAAE,OAAO,EAAE,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;6BAClH,CAAC;4BACJ,CAAC,CAAC,EAAE,CAAC;qBACR,CAAC;oBACF,IAAI,eAAe,EAAE,CAAC;wBACpB,MAAM,eAAe,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,UAAU,CAAC,CAAC;wBAChF,IAAI,CAAC,eAAe,EAAE,CAAC;4BACrB,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC;gCAC1C,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE;gCACnB,MAAM,EAAE,KAAK,CAAC,MAAM;gCACpB,WAAW,EAAE,KAAK,CAAC,WAAW;gCAC9B,GAAG,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,CAAE,GAAW,EAAE,OAAO,EAAE,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;6BAClH,CAAC,CAAC;4BACH,IAAI,CAAC,QAAQ,EAAE,CAAC;gCACd,MAAM,CAAC,IAAI,CACT,GAAG,CAAC,MAAM,oBAAoB,CAAC;oCAC7B,KAAK;oCACL,UAAU,EAAE,YAAY;oCACxB,OAAO,EAAG,GAAW,EAAE,OAAO,EAAE,OAAO;oCACvC,MAAM;iCACP,CAAC,CAAC,CACJ,CAAC;4BACJ,CAAC;wBACH,CAAC;oBACH,CAAC;oBACD,MAAM,eAAe,GAAG,eAAe;wBACrC,CAAC,CAAC,MAAM,6BAA6B,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;wBAC5D,CAAC,CAAC,MAAM,CAAC;oBACX,eAAe,CAAC,IAAI,CAAC,CAAC,IAAS,EAAE,KAAU,EAAE,EAAE;wBAC7C,MAAM,EAAE,GAAG,OAAO,IAAI,EAAE,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;wBAC5D,MAAM,EAAE,GAAG,OAAO,KAAK,EAAE,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;wBAC9D,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;4BACd,OAAO,EAAE,GAAG,EAAE,CAAC;wBACjB,CAAC;wBACD,MAAM,EAAE,GAAG,IAAI,EAAE,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC;wBAC/D,MAAM,EAAE,GAAG,KAAK,EAAE,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC;wBAChE,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;4BACd,OAAO,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;wBAClC,CAAC;wBACD,MAAM,EAAE,GAAG,OAAO,IAAI,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC3D,MAAM,EAAE,GAAG,OAAO,KAAK,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC7D,OAAO,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;oBAC9B,CAAC,CAAC,CAAC;oBACH,IAAI,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;oBAC9C,aAAa,GAAG,mBAAmB,CAAC;gBACtC,CAAC;gBACD,GAAG,CAAC,cAAc,EAAE,CAAC;YACvB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,YAAY,CAAC,iBAAiB,aAAa,IAAI,SAAS,EAAE,EAAE,KAAK,CAAC,CAAC;gBACvE,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACvE,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,0DAA0D,OAAO,GAAG,EAAE,CAAC;oBACvG,OAAO,EAAE;wBACP,QAAQ,EAAE,IAAI;wBACd,KAAK,EAAE,OAAO;wBACd,QAAQ,EAAE,IAAI;wBACd,cAAc,EAAE,WAAW;wBAC3B,GAAG,EAAE,EAAE,GAAG,GAAG,CAAC,SAAS,EAAE;qBAC1B;iBACF,CAAC;YACJ,CAAC;YAED,IAAI,aAAa,KAAK,QAAQ,IAAI,aAAa,KAAK,UAAU,IAAI,aAAa,KAAK,KAAK,EAAE,CAAC;gBAC1F,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mCAAmC,aAAa,IAAI,OAAO,GAAG,EAAE,CAAC;oBACjG,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,oBAAoB,EAAE,MAAM,EAAE,aAAa,EAAE;iBAChF,CAAC;YACJ,CAAC;YAED,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC9D,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,OAAO,GAAG,EAAE,KAAK,KAAK,QAAQ,IAAI,GAAG,CAAC,KAAK,IAAI,QAAQ,CAAC,CAAC;YAC5F,CAAC;YACD,MAAM,eAAe,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,SAAS,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAC3C,IAAI,SAAS,KAAK,aAAa,EAAE,CAAC;gBAChC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAS,EAAE,KAAU,EAAE,EAAE;oBAClC,MAAM,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;oBACxF,MAAM,EAAE,GAAG,CAAC,OAAO,KAAK,EAAE,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;oBAC3F,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;wBACd,OAAO,EAAE,GAAG,EAAE,CAAC;oBACjB,CAAC;oBACD,MAAM,EAAE,GAAG,OAAO,IAAI,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC3D,MAAM,EAAE,GAAG,OAAO,KAAK,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC7D,OAAO,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;gBAC9B,CAAC,CAAC,CAAC;YACL,CAAC;YACD,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAQ,CAAC;YAC9B,MAAM,QAAQ,GAAG,cAAc,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC/C,MAAM,UAAU,GAAG,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACnD,MAAM,SAAS,GAAG,OAAO,CACvB,eAAe;gBACb,MAAM;gBACN,CAAC,QAAQ,KAAK,eAAe,IAAI,UAAU,KAAK,eAAe,CAAC,CACnE,CAAC;YACF,MAAM,cAAc,GAAG,SAAS;gBAC9B,CAAC,CAAC,CAAC,OAAO,MAAM,EAAE,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;oBAC1D,CAAC,OAAO,MAAM,EAAE,OAAO,KAAK,QAAQ,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;oBAC9D,IAAI;gBACN,CAAC,CAAC,IAAI,CAAC;YACT,MAAM,iBAAiB,GAAG,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,gBAAgB,CAAC;YAC1G,MAAM,OAAO,GAAG,6BAA6B,CAAC;gBAC5C,IAAI;gBACJ,aAAa;gBACb,SAAS;gBACT,SAAS;gBACT,cAAc;gBACd,iBAAiB;gBACjB,QAAQ,EAAE,YAAY;gBACtB,KAAK,EAAE,cAAc;aACtB,CAAC,CAAC;YACH,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,OAAO;qBACd;iBACF;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE,IAAI;oBACb,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE;wBACJ,aAAa;wBACb,SAAS;wBACT,QAAQ,EAAE,KAAK;wBACf,SAAS;wBACT,cAAc;wBACd,iBAAiB;qBAClB;oBACD,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ;wBACxB,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,GAAG,EAAE,EAAE,GAAG,GAAG,CAAC,SAAS,EAAE,EAAE;wBAC5E,CAAC,CAAC,EAAE,CAAC;iBACR;aACF,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-search.test.d.ts","sourceRoot":"","sources":["../../../src/plugin/tools/memory-search.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,205 @@
|
|
|
1
|
+
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
2
|
+
import { registerMemorySearchTool } from "./memory-search.js";
|
|
3
|
+
const { resolveScopeMock, resolveLimitsMock, memorySearchDbMock, memorySearchSessionsMock, hasSessionsIndexRowsMock, memorySearchSessionsIndexDbMock, filterSessionHitsByVisibilityMock, } = vi.hoisted(() => ({
|
|
4
|
+
resolveScopeMock: vi.fn(async () => ({ userId: "u1", workspaceId: "w1" })),
|
|
5
|
+
resolveLimitsMock: vi.fn(() => ({ maxResults: 10 })),
|
|
6
|
+
memorySearchDbMock: vi.fn(async () => []),
|
|
7
|
+
memorySearchSessionsMock: vi.fn(async () => []),
|
|
8
|
+
hasSessionsIndexRowsMock: vi.fn(async () => false),
|
|
9
|
+
memorySearchSessionsIndexDbMock: vi.fn(async () => []),
|
|
10
|
+
filterSessionHitsByVisibilityMock: vi.fn(async ({ hits }) => hits),
|
|
11
|
+
}));
|
|
12
|
+
vi.mock("../../identity.js", () => ({
|
|
13
|
+
resolveUserAndWorkspaceScope: resolveScopeMock,
|
|
14
|
+
}));
|
|
15
|
+
vi.mock("../../memory/limits.js", () => ({
|
|
16
|
+
resolveMemoryLimits: resolveLimitsMock,
|
|
17
|
+
}));
|
|
18
|
+
vi.mock("../../memory/search.js", () => ({
|
|
19
|
+
memorySearchDb: memorySearchDbMock,
|
|
20
|
+
}));
|
|
21
|
+
vi.mock("../../memory/sessions.js", () => ({
|
|
22
|
+
memorySearchSessions: memorySearchSessionsMock,
|
|
23
|
+
}));
|
|
24
|
+
vi.mock("../../memory/sessions-index.js", () => ({
|
|
25
|
+
hasSessionsIndexRows: hasSessionsIndexRowsMock,
|
|
26
|
+
memorySearchSessionsIndexDb: memorySearchSessionsIndexDbMock,
|
|
27
|
+
}));
|
|
28
|
+
vi.mock("../../memory/sessions-visibility.js", () => ({
|
|
29
|
+
filterSessionHitsByVisibility: filterSessionHitsByVisibilityMock,
|
|
30
|
+
}));
|
|
31
|
+
function buildCtx() {
|
|
32
|
+
const registerTool = vi.fn();
|
|
33
|
+
return {
|
|
34
|
+
ctx: {
|
|
35
|
+
api: {
|
|
36
|
+
registerTool,
|
|
37
|
+
runtime: { agentId: "main", sessionKey: "agent:main:main" },
|
|
38
|
+
},
|
|
39
|
+
disabledReason: null,
|
|
40
|
+
cfg: { sessions: { visibility: "current" } },
|
|
41
|
+
ensureReady: vi.fn(async () => undefined),
|
|
42
|
+
getPool: vi.fn(() => ({ query: vi.fn() })),
|
|
43
|
+
sdkHealth: { degraded: false, reason: null, affectedOperation: null },
|
|
44
|
+
markSdkSuccess: vi.fn(),
|
|
45
|
+
markSdkError: vi.fn(),
|
|
46
|
+
},
|
|
47
|
+
registerTool,
|
|
48
|
+
};
|
|
49
|
+
}
|
|
50
|
+
describe("memory_search tool exactTop1 metadata", () => {
|
|
51
|
+
beforeEach(() => {
|
|
52
|
+
vi.clearAllMocks();
|
|
53
|
+
});
|
|
54
|
+
it("emits exactTop1 metadata and top exact line for literal top-1 match", async () => {
|
|
55
|
+
memorySearchDbMock.mockResolvedValueOnce([
|
|
56
|
+
{
|
|
57
|
+
corpus: "memory",
|
|
58
|
+
path: "db-memory/items/1.md",
|
|
59
|
+
id: "1",
|
|
60
|
+
title: "ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515",
|
|
61
|
+
kind: "note",
|
|
62
|
+
score: 6.7,
|
|
63
|
+
snippet: "ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515",
|
|
64
|
+
},
|
|
65
|
+
]);
|
|
66
|
+
const { ctx, registerTool } = buildCtx();
|
|
67
|
+
registerMemorySearchTool({
|
|
68
|
+
ctx,
|
|
69
|
+
refreshPromptCache: vi.fn(),
|
|
70
|
+
ensureSessionsIndexBootstrapped: vi.fn(async () => undefined),
|
|
71
|
+
});
|
|
72
|
+
const def = registerTool.mock.calls[0]?.[0];
|
|
73
|
+
const result = await def.execute("toolcall-1", {
|
|
74
|
+
query: "ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515",
|
|
75
|
+
corpus: "memory",
|
|
76
|
+
});
|
|
77
|
+
const visible = JSON.parse(result.content[0].text);
|
|
78
|
+
expect(visible.results).toHaveLength(1);
|
|
79
|
+
expect(visible.queryMode).toBe("exact_value");
|
|
80
|
+
expect(visible.topCandidates[0]).toBe("ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515");
|
|
81
|
+
expect(visible.results[0].snippet).toContain("ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515");
|
|
82
|
+
expect(visible.meta.recommendedAction).toBe("return_exact");
|
|
83
|
+
expect(visible.meta.exactTop1Value).toBe("ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515");
|
|
84
|
+
expect(result.details.meta).toMatchObject({
|
|
85
|
+
exactTop1: true,
|
|
86
|
+
exactTop1Value: "ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515",
|
|
87
|
+
recommendedAction: "return_exact",
|
|
88
|
+
});
|
|
89
|
+
});
|
|
90
|
+
it("does not emit exactTop1 metadata for non-literal top-1", async () => {
|
|
91
|
+
memorySearchDbMock.mockResolvedValueOnce([
|
|
92
|
+
{
|
|
93
|
+
corpus: "memory",
|
|
94
|
+
path: "db-memory/items/2.md",
|
|
95
|
+
id: "2",
|
|
96
|
+
title: "RETRIEVAL_MARKER_20260515_D gamma grape",
|
|
97
|
+
kind: "note",
|
|
98
|
+
score: 1.4,
|
|
99
|
+
snippet: "RETRIEVAL_MARKER_20260515_D gamma grape",
|
|
100
|
+
},
|
|
101
|
+
]);
|
|
102
|
+
const { ctx, registerTool } = buildCtx();
|
|
103
|
+
registerMemorySearchTool({
|
|
104
|
+
ctx,
|
|
105
|
+
refreshPromptCache: vi.fn(),
|
|
106
|
+
ensureSessionsIndexBootstrapped: vi.fn(async () => undefined),
|
|
107
|
+
});
|
|
108
|
+
const def = registerTool.mock.calls[0]?.[0];
|
|
109
|
+
const result = await def.execute("toolcall-2", { query: "smoke", corpus: "memory" });
|
|
110
|
+
const visible = JSON.parse(result.content[0].text);
|
|
111
|
+
expect(visible.results).toHaveLength(1);
|
|
112
|
+
expect(visible.queryMode).toBe("contextual");
|
|
113
|
+
expect(visible.meta.recommendedAction).toBe("inspect_top");
|
|
114
|
+
expect(result.details.meta).toMatchObject({
|
|
115
|
+
exactTop1: false,
|
|
116
|
+
exactTop1Value: null,
|
|
117
|
+
recommendedAction: "inspect_top",
|
|
118
|
+
});
|
|
119
|
+
});
|
|
120
|
+
it("emits stop_not_found recommendation when there are no hits", async () => {
|
|
121
|
+
memorySearchDbMock.mockResolvedValueOnce([]);
|
|
122
|
+
const { ctx, registerTool } = buildCtx();
|
|
123
|
+
registerMemorySearchTool({
|
|
124
|
+
ctx,
|
|
125
|
+
refreshPromptCache: vi.fn(),
|
|
126
|
+
ensureSessionsIndexBootstrapped: vi.fn(async () => undefined),
|
|
127
|
+
});
|
|
128
|
+
const def = registerTool.mock.calls[0]?.[0];
|
|
129
|
+
const result = await def.execute("toolcall-3", { query: "nothing-here", corpus: "memory" });
|
|
130
|
+
const visible = JSON.parse(result.content[0].text);
|
|
131
|
+
expect(visible.results).toHaveLength(0);
|
|
132
|
+
expect(visible.queryMode).toBe("contextual");
|
|
133
|
+
expect(visible.meta.recommendedAction).toBe("stop_not_found");
|
|
134
|
+
expect(result.details.meta).toMatchObject({
|
|
135
|
+
exactTop1: false,
|
|
136
|
+
exactTop1Value: null,
|
|
137
|
+
recommendedAction: "stop_not_found",
|
|
138
|
+
});
|
|
139
|
+
});
|
|
140
|
+
it("boosts marker-like candidates in exact_value mode", async () => {
|
|
141
|
+
memorySearchDbMock.mockResolvedValueOnce([
|
|
142
|
+
{
|
|
143
|
+
corpus: "memory",
|
|
144
|
+
path: "db-memory/items/desc.md",
|
|
145
|
+
id: "desc",
|
|
146
|
+
title: "anchorclaw post-restart smoke",
|
|
147
|
+
kind: "note",
|
|
148
|
+
score: 2.0,
|
|
149
|
+
snippet: "anchorclaw post-restart smoke",
|
|
150
|
+
},
|
|
151
|
+
{
|
|
152
|
+
corpus: "memory",
|
|
153
|
+
path: "db-memory/items/marker.md",
|
|
154
|
+
id: "marker",
|
|
155
|
+
title: "ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515",
|
|
156
|
+
kind: "note",
|
|
157
|
+
score: 1.0,
|
|
158
|
+
snippet: "ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515",
|
|
159
|
+
},
|
|
160
|
+
]);
|
|
161
|
+
const { ctx, registerTool } = buildCtx();
|
|
162
|
+
registerMemorySearchTool({
|
|
163
|
+
ctx,
|
|
164
|
+
refreshPromptCache: vi.fn(),
|
|
165
|
+
ensureSessionsIndexBootstrapped: vi.fn(async () => undefined),
|
|
166
|
+
});
|
|
167
|
+
const def = registerTool.mock.calls[0]?.[0];
|
|
168
|
+
const result = await def.execute("toolcall-4", {
|
|
169
|
+
query: "What exact marker did I save?",
|
|
170
|
+
corpus: "memory",
|
|
171
|
+
});
|
|
172
|
+
const visible = JSON.parse(result.content[0].text);
|
|
173
|
+
expect(visible.queryMode).toBe("exact_value");
|
|
174
|
+
expect(visible.topCandidates[0]).toBe("ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515");
|
|
175
|
+
expect(visible.results[0].snippet).toContain("ANCHORCLAW_ACTIVE_MEMORY_MARKER_20260515");
|
|
176
|
+
});
|
|
177
|
+
it("keeps contextual mode for broad marker/id/key summary prompts", async () => {
|
|
178
|
+
memorySearchDbMock.mockResolvedValueOnce([
|
|
179
|
+
{
|
|
180
|
+
corpus: "memory",
|
|
181
|
+
path: "db-memory/items/desc.md",
|
|
182
|
+
id: "desc",
|
|
183
|
+
title: "anchorclaw post-restart smoke",
|
|
184
|
+
kind: "note",
|
|
185
|
+
score: 2.0,
|
|
186
|
+
snippet: "anchorclaw post-restart smoke",
|
|
187
|
+
},
|
|
188
|
+
]);
|
|
189
|
+
const { ctx, registerTool } = buildCtx();
|
|
190
|
+
registerMemorySearchTool({
|
|
191
|
+
ctx,
|
|
192
|
+
refreshPromptCache: vi.fn(),
|
|
193
|
+
ensureSessionsIndexBootstrapped: vi.fn(async () => undefined),
|
|
194
|
+
});
|
|
195
|
+
const def = registerTool.mock.calls[0]?.[0];
|
|
196
|
+
const result = await def.execute("toolcall-5", {
|
|
197
|
+
query: "What marker/id/key values do we have around tests? Summarize briefly.",
|
|
198
|
+
corpus: "memory",
|
|
199
|
+
});
|
|
200
|
+
const visible = JSON.parse(result.content[0].text);
|
|
201
|
+
expect(visible.queryMode).toBe("contextual");
|
|
202
|
+
expect(visible.meta.queryMode).toBe("contextual");
|
|
203
|
+
});
|
|
204
|
+
});
|
|
205
|
+
//# sourceMappingURL=memory-search.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-search.test.js","sourceRoot":"","sources":["../../../src/plugin/tools/memory-search.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAE9D,MAAM,EACJ,gBAAgB,EAChB,iBAAiB,EACjB,kBAAkB,EAClB,wBAAwB,EACxB,wBAAwB,EACxB,+BAA+B,EAC/B,iCAAiC,GAClC,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IACpB,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,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;IACpD,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC;IACzC,wBAAwB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC;IAC/C,wBAAwB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,KAAK,CAAC;IAClD,+BAA+B,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC;IACtD,iCAAiC,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,IAAI,EAAuB,EAAE,EAAE,CAAC,IAAI,CAAC;CACxF,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,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,mBAAmB,EAAE,iBAAiB;CACvC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc,EAAE,kBAAkB;CACnC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,0BAA0B,EAAE,GAAG,EAAE,CAAC,CAAC;IACzC,oBAAoB,EAAE,wBAAwB;CAC/C,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/C,oBAAoB,EAAE,wBAAwB;IAC9C,2BAA2B,EAAE,+BAA+B;CAC7D,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,qCAAqC,EAAE,GAAG,EAAE,CAAC,CAAC;IACpD,6BAA6B,EAAE,iCAAiC;CACjE,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,OAAO,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,iBAAiB,EAAE;aAC5D;YACD,cAAc,EAAE,IAAI;YACpB,GAAG,EAAE,EAAE,QAAQ,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE;YAC5C,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,SAAS,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE;YACrE,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;YACvB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;SACf;QACR,YAAY;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,uCAAuC,EAAE,GAAG,EAAE;IACrD,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QAClF,kBAA0B,CAAC,qBAAqB,CAAC;YAChD;gBACE,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,sBAAsB;gBAC5B,EAAE,EAAE,GAAG;gBACP,KAAK,EAAE,0CAA0C;gBACjD,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,GAAG;gBACV,OAAO,EAAE,0CAA0C;aACpD;SACO,CAAC,CAAC;QACZ,MAAM,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,QAAQ,EAAE,CAAC;QACzC,wBAAwB,CAAC;YACvB,GAAG;YACH,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC3B,+BAA+B,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC;SAC9D,CAAC,CAAC;QACH,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,KAAK,EAAE,0CAA0C;YACjD,MAAM,EAAE,QAAQ;SACjB,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEnD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC9C,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;QAClF,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,0CAA0C,CAAC,CAAC;QACzF,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC5D,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;QACrF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC;YACxC,SAAS,EAAE,IAAI;YACf,cAAc,EAAE,0CAA0C;YAC1D,iBAAiB,EAAE,cAAc;SAClC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACrE,kBAA0B,CAAC,qBAAqB,CAAC;YAChD;gBACE,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,sBAAsB;gBAC5B,EAAE,EAAE,GAAG;gBACP,KAAK,EAAE,yCAAyC;gBAChD,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,GAAG;gBACV,OAAO,EAAE,yCAAyC;aACnD;SACO,CAAC,CAAC;QACZ,MAAM,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,QAAQ,EAAE,CAAC;QACzC,wBAAwB,CAAC;YACvB,GAAG;YACH,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC3B,+BAA+B,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC;SAC9D,CAAC,CAAC;QACH,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,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;QACrF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEnD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC;YACxC,SAAS,EAAE,KAAK;YAChB,cAAc,EAAE,IAAI;YACpB,iBAAiB,EAAE,aAAa;SACjC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QACzE,kBAA0B,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QACtD,MAAM,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,QAAQ,EAAE,CAAC;QACzC,wBAAwB,CAAC;YACvB,GAAG;YACH,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC3B,+BAA+B,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC;SAC9D,CAAC,CAAC;QACH,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,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC5F,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEnD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC;YACxC,SAAS,EAAE,KAAK;YAChB,cAAc,EAAE,IAAI;YACpB,iBAAiB,EAAE,gBAAgB;SACpC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QAChE,kBAA0B,CAAC,qBAAqB,CAAC;YAChD;gBACE,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,yBAAyB;gBAC/B,EAAE,EAAE,MAAM;gBACV,KAAK,EAAE,+BAA+B;gBACtC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,GAAG;gBACV,OAAO,EAAE,+BAA+B;aACzC;YACD;gBACE,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,2BAA2B;gBACjC,EAAE,EAAE,QAAQ;gBACZ,KAAK,EAAE,0CAA0C;gBACjD,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,GAAG;gBACV,OAAO,EAAE,0CAA0C;aACpD;SACO,CAAC,CAAC;QACZ,MAAM,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,QAAQ,EAAE,CAAC;QACzC,wBAAwB,CAAC;YACvB,GAAG;YACH,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC3B,+BAA+B,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC;SAC9D,CAAC,CAAC;QACH,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,KAAK,EAAE,+BAA+B;YACtC,MAAM,EAAE,QAAQ;SACjB,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEnD,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC9C,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;QAClF,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,0CAA0C,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC5E,kBAA0B,CAAC,qBAAqB,CAAC;YAChD;gBACE,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,yBAAyB;gBAC/B,EAAE,EAAE,MAAM;gBACV,KAAK,EAAE,+BAA+B;gBACtC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,GAAG;gBACV,OAAO,EAAE,+BAA+B;aACzC;SACO,CAAC,CAAC;QACZ,MAAM,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,QAAQ,EAAE,CAAC;QACzC,wBAAwB,CAAC;YACvB,GAAG;YACH,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC3B,+BAA+B,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC;SAC9D,CAAC,CAAC;QACH,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,KAAK,EAAE,uEAAuE;YAC9E,MAAM,EAAE,QAAQ;SACjB,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEnD,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-status.d.ts","sourceRoot":"","sources":["../../../src/plugin/tools/memory-status.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAK1D,wBAAgB,wBAAwB,CAAC,EAAE,GAAG,EAAE,EAAE,sBAAsB,QA8IvE"}
|