elepha 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE.md +375 -0
- package/README.md +135 -0
- package/bin/elepha.js +2 -0
- package/dist/adapters/base.js +600 -0
- package/dist/adapters/base.js.map +1 -0
- package/dist/adapters/claude-code.js +260 -0
- package/dist/adapters/claude-code.js.map +1 -0
- package/dist/adapters/codex.js +499 -0
- package/dist/adapters/codex.js.map +1 -0
- package/dist/adapters/discriminators.js +39 -0
- package/dist/adapters/discriminators.js.map +1 -0
- package/dist/adapters/index.js +6 -0
- package/dist/adapters/index.js.map +1 -0
- package/dist/adapters/internal-command.js +3 -0
- package/dist/adapters/internal-command.js.map +1 -0
- package/dist/cli/backup-wizard.js +97 -0
- package/dist/cli/backup-wizard.js.map +1 -0
- package/dist/cli/capture-service.js +80 -0
- package/dist/cli/capture-service.js.map +1 -0
- package/dist/cli/commands/backfills.js +336 -0
- package/dist/cli/commands/backfills.js.map +1 -0
- package/dist/cli/commands/backup.js +234 -0
- package/dist/cli/commands/backup.js.map +1 -0
- package/dist/cli/commands/config.js +66 -0
- package/dist/cli/commands/config.js.map +1 -0
- package/dist/cli/commands/consent.js +237 -0
- package/dist/cli/commands/consent.js.map +1 -0
- package/dist/cli/commands/daemon-control.js +76 -0
- package/dist/cli/commands/daemon-control.js.map +1 -0
- package/dist/cli/commands/doctor.js +31 -0
- package/dist/cli/commands/doctor.js.map +1 -0
- package/dist/cli/commands/hook.js +24 -0
- package/dist/cli/commands/hook.js.map +1 -0
- package/dist/cli/commands/import.js +611 -0
- package/dist/cli/commands/import.js.map +1 -0
- package/dist/cli/commands/init.js +19 -0
- package/dist/cli/commands/init.js.map +1 -0
- package/dist/cli/commands/inspect.js +54 -0
- package/dist/cli/commands/inspect.js.map +1 -0
- package/dist/cli/commands/install.js +49 -0
- package/dist/cli/commands/install.js.map +1 -0
- package/dist/cli/commands/internal.js +16 -0
- package/dist/cli/commands/internal.js.map +1 -0
- package/dist/cli/commands/mcp.js +12 -0
- package/dist/cli/commands/mcp.js.map +1 -0
- package/dist/cli/commands/projects.js +26 -0
- package/dist/cli/commands/projects.js.map +1 -0
- package/dist/cli/commands/purge.js +216 -0
- package/dist/cli/commands/purge.js.map +1 -0
- package/dist/cli/commands/reingest.js +80 -0
- package/dist/cli/commands/reingest.js.map +1 -0
- package/dist/cli/commands/rekey.js +58 -0
- package/dist/cli/commands/rekey.js.map +1 -0
- package/dist/cli/commands/restore.js +376 -0
- package/dist/cli/commands/restore.js.map +1 -0
- package/dist/cli/commands/rollup.js +130 -0
- package/dist/cli/commands/rollup.js.map +1 -0
- package/dist/cli/commands/sanitize.js +72 -0
- package/dist/cli/commands/sanitize.js.map +1 -0
- package/dist/cli/commands/segment.js +153 -0
- package/dist/cli/commands/segment.js.map +1 -0
- package/dist/cli/commands/self-update.js +26 -0
- package/dist/cli/commands/self-update.js.map +1 -0
- package/dist/cli/commands/start.js +53 -0
- package/dist/cli/commands/start.js.map +1 -0
- package/dist/cli/commands/stats.js +42 -0
- package/dist/cli/commands/stats.js.map +1 -0
- package/dist/cli/commands/status.js +73 -0
- package/dist/cli/commands/status.js.map +1 -0
- package/dist/cli/commands/uninstall.js +23 -0
- package/dist/cli/commands/uninstall.js.map +1 -0
- package/dist/cli/config-wizard.js +104 -0
- package/dist/cli/config-wizard.js.map +1 -0
- package/dist/cli/destructive-op.js +32 -0
- package/dist/cli/destructive-op.js.map +1 -0
- package/dist/cli/import-wizard.js +89 -0
- package/dist/cli/import-wizard.js.map +1 -0
- package/dist/cli/index.js +74 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/init-wizard.js +80 -0
- package/dist/cli/init-wizard.js.map +1 -0
- package/dist/cli/init.js +159 -0
- package/dist/cli/init.js.map +1 -0
- package/dist/cli/launcher-probe.js +71 -0
- package/dist/cli/launcher-probe.js.map +1 -0
- package/dist/cli/project-path.js +16 -0
- package/dist/cli/project-path.js.map +1 -0
- package/dist/cli/purge-wizard.js +200 -0
- package/dist/cli/purge-wizard.js.map +1 -0
- package/dist/cli/restore-wizard.js +101 -0
- package/dist/cli/restore-wizard.js.map +1 -0
- package/dist/cli/shared.js +128 -0
- package/dist/cli/shared.js.map +1 -0
- package/dist/cli/status.js +22 -0
- package/dist/cli/status.js.map +1 -0
- package/dist/cli/wordmark.js +15 -0
- package/dist/cli/wordmark.js.map +1 -0
- package/dist/config/constants.js +115 -0
- package/dist/config/constants.js.map +1 -0
- package/dist/config/env.js +168 -0
- package/dist/config/env.js.map +1 -0
- package/dist/config/memory-config.js +63 -0
- package/dist/config/memory-config.js.map +1 -0
- package/dist/config/paths.js +343 -0
- package/dist/config/paths.js.map +1 -0
- package/dist/config/settings.js +132 -0
- package/dist/config/settings.js.map +1 -0
- package/dist/config/version.js +12 -0
- package/dist/config/version.js.map +1 -0
- package/dist/daemon/failure-window.js +38 -0
- package/dist/daemon/failure-window.js.map +1 -0
- package/dist/daemon/heartbeat.js +40 -0
- package/dist/daemon/heartbeat.js.map +1 -0
- package/dist/daemon/index.js +873 -0
- package/dist/daemon/index.js.map +1 -0
- package/dist/daemon/log-rotation.js +27 -0
- package/dist/daemon/log-rotation.js.map +1 -0
- package/dist/daemon/readability-guard.js +89 -0
- package/dist/daemon/readability-guard.js.map +1 -0
- package/dist/daemon/rollup-service.js +239 -0
- package/dist/daemon/rollup-service.js.map +1 -0
- package/dist/daemon/update-check.js +87 -0
- package/dist/daemon/update-check.js.map +1 -0
- package/dist/daemon/work-queue.js +28 -0
- package/dist/daemon/work-queue.js.map +1 -0
- package/dist/discovery/session-projects.js +242 -0
- package/dist/discovery/session-projects.js.map +1 -0
- package/dist/hooks/codex-trust.js +80 -0
- package/dist/hooks/codex-trust.js.map +1 -0
- package/dist/hooks/common.js +81 -0
- package/dist/hooks/common.js.map +1 -0
- package/dist/hooks/hook-log.js +38 -0
- package/dist/hooks/hook-log.js.map +1 -0
- package/dist/hooks/installer.js +200 -0
- package/dist/hooks/installer.js.map +1 -0
- package/dist/hooks/session-start.js +321 -0
- package/dist/hooks/session-start.js.map +1 -0
- package/dist/hooks/user-prompt-submit.js +248 -0
- package/dist/hooks/user-prompt-submit.js.map +1 -0
- package/dist/install/binary.js +72 -0
- package/dist/install/binary.js.map +1 -0
- package/dist/install/config-file.js +119 -0
- package/dist/install/config-file.js.map +1 -0
- package/dist/install/daemon-health.js +48 -0
- package/dist/install/daemon-health.js.map +1 -0
- package/dist/install/daemon-service.js +309 -0
- package/dist/install/daemon-service.js.map +1 -0
- package/dist/install/doctor.js +180 -0
- package/dist/install/doctor.js.map +1 -0
- package/dist/install/health-checks.js +90 -0
- package/dist/install/health-checks.js.map +1 -0
- package/dist/install/installer.js +411 -0
- package/dist/install/installer.js.map +1 -0
- package/dist/install/launchd-backend.js +261 -0
- package/dist/install/launchd-backend.js.map +1 -0
- package/dist/install/launcher.js +224 -0
- package/dist/install/launcher.js.map +1 -0
- package/dist/install/markers.js +14 -0
- package/dist/install/markers.js.map +1 -0
- package/dist/install/platform.js +29 -0
- package/dist/install/platform.js.map +1 -0
- package/dist/install/present-tools.js +16 -0
- package/dist/install/present-tools.js.map +1 -0
- package/dist/install/self-update.js +120 -0
- package/dist/install/self-update.js.map +1 -0
- package/dist/install/service-backend.js +36 -0
- package/dist/install/service-backend.js.map +1 -0
- package/dist/install/service-manifest.js +36 -0
- package/dist/install/service-manifest.js.map +1 -0
- package/dist/install/status.js +58 -0
- package/dist/install/status.js.map +1 -0
- package/dist/install/systemd-backend.js +225 -0
- package/dist/install/systemd-backend.js.map +1 -0
- package/dist/markers.js +5 -0
- package/dist/markers.js.map +1 -0
- package/dist/mcp/installer.js +131 -0
- package/dist/mcp/installer.js.map +1 -0
- package/dist/mcp/server.js +81 -0
- package/dist/mcp/server.js.map +1 -0
- package/dist/mcp/tools/get-context.js +3 -0
- package/dist/mcp/tools/get-context.js.map +1 -0
- package/dist/mcp/tools/list-projects.js +3 -0
- package/dist/mcp/tools/list-projects.js.map +1 -0
- package/dist/mcp/tools/save-memory.js +3 -0
- package/dist/mcp/tools/save-memory.js.map +1 -0
- package/dist/mcp/tools.js +288 -0
- package/dist/mcp/tools.js.map +1 -0
- package/dist/rendering/raw-turn-renderer.js +85 -0
- package/dist/rendering/raw-turn-renderer.js.map +1 -0
- package/dist/security/file-permissions.js +15 -0
- package/dist/security/file-permissions.js.map +1 -0
- package/dist/security/provider-transcript.js +74 -0
- package/dist/security/provider-transcript.js.map +1 -0
- package/dist/security/sanitize.js +210 -0
- package/dist/security/sanitize.js.map +1 -0
- package/dist/security/self-ingestion.js +47 -0
- package/dist/security/self-ingestion.js.map +1 -0
- package/dist/security/sentinel.js +19 -0
- package/dist/security/sentinel.js.map +1 -0
- package/dist/security/subprocess-allowlist.js +421 -0
- package/dist/security/subprocess-allowlist.js.map +1 -0
- package/dist/serving/instructions.js +33 -0
- package/dist/serving/instructions.js.map +1 -0
- package/dist/serving/lexical-recall.js +249 -0
- package/dist/serving/lexical-recall.js.map +1 -0
- package/dist/serving/session-reader.js +302 -0
- package/dist/serving/session-reader.js.map +1 -0
- package/dist/storage/backfill-runner.js +34 -0
- package/dist/storage/backfill-runner.js.map +1 -0
- package/dist/storage/backup.js +46 -0
- package/dist/storage/backup.js.map +1 -0
- package/dist/storage/candidate-validator.js +77 -0
- package/dist/storage/candidate-validator.js.map +1 -0
- package/dist/storage/consent-store.js +198 -0
- package/dist/storage/consent-store.js.map +1 -0
- package/dist/storage/custom-title-backfill.js +68 -0
- package/dist/storage/custom-title-backfill.js.map +1 -0
- package/dist/storage/db.js +312 -0
- package/dist/storage/db.js.map +1 -0
- package/dist/storage/external-agent-import-purge.js +182 -0
- package/dist/storage/external-agent-import-purge.js.map +1 -0
- package/dist/storage/first-prompt-search-backfill.js +76 -0
- package/dist/storage/first-prompt-search-backfill.js.map +1 -0
- package/dist/storage/first-prompt-search.js +7 -0
- package/dist/storage/first-prompt-search.js.map +1 -0
- package/dist/storage/injection-store.js +37 -0
- package/dist/storage/injection-store.js.map +1 -0
- package/dist/storage/memory-store.js +428 -0
- package/dist/storage/memory-store.js.map +1 -0
- package/dist/storage/project-resolver.js +294 -0
- package/dist/storage/project-resolver.js.map +1 -0
- package/dist/storage/project-store.js +104 -0
- package/dist/storage/project-store.js.map +1 -0
- package/dist/storage/rendered-chars-backfill.js +85 -0
- package/dist/storage/rendered-chars-backfill.js.map +1 -0
- package/dist/storage/resegmentation.js +582 -0
- package/dist/storage/resegmentation.js.map +1 -0
- package/dist/storage/rollup-store.js +246 -0
- package/dist/storage/rollup-store.js.map +1 -0
- package/dist/storage/root-commit-backfill.js +32 -0
- package/dist/storage/root-commit-backfill.js.map +1 -0
- package/dist/storage/sanitize-backfill.js +184 -0
- package/dist/storage/sanitize-backfill.js.map +1 -0
- package/dist/storage/segmentation.js +87 -0
- package/dist/storage/segmentation.js.map +1 -0
- package/dist/storage/session-fields-backfill.js +196 -0
- package/dist/storage/session-fields-backfill.js.map +1 -0
- package/dist/storage/session-read-model.js +130 -0
- package/dist/storage/session-read-model.js.map +1 -0
- package/dist/storage/session-store.js +200 -0
- package/dist/storage/session-store.js.map +1 -0
- package/dist/storage/session-title-backfill.js +75 -0
- package/dist/storage/session-title-backfill.js.map +1 -0
- package/dist/storage/session-title.js +71 -0
- package/dist/storage/session-title.js.map +1 -0
- package/dist/storage/shown-session-list-store.js +33 -0
- package/dist/storage/shown-session-list-store.js.map +1 -0
- package/dist/storage/stats.js +30 -0
- package/dist/storage/stats.js.map +1 -0
- package/dist/storage/turn-store.js +202 -0
- package/dist/storage/turn-store.js.map +1 -0
- package/dist/storage/ulid.js +17 -0
- package/dist/storage/ulid.js.map +1 -0
- package/dist/summarizer/anthropic-call.js +76 -0
- package/dist/summarizer/anthropic-call.js.map +1 -0
- package/dist/summarizer/call-log.js +120 -0
- package/dist/summarizer/call-log.js.map +1 -0
- package/dist/summarizer/haiku-provider.js +136 -0
- package/dist/summarizer/haiku-provider.js.map +1 -0
- package/dist/summarizer/pricing.js +8 -0
- package/dist/summarizer/pricing.js.map +1 -0
- package/dist/summarizer/prompt-cache.js +6 -0
- package/dist/summarizer/prompt-cache.js.map +1 -0
- package/dist/summarizer/prompt.js +47 -0
- package/dist/summarizer/prompt.js.map +1 -0
- package/dist/summarizer/provider-config.js +25 -0
- package/dist/summarizer/provider-config.js.map +1 -0
- package/dist/summarizer/rollup-prompt.js +225 -0
- package/dist/summarizer/rollup-prompt.js.map +1 -0
- package/dist/summarizer/rollup-provider.js +224 -0
- package/dist/summarizer/rollup-provider.js.map +1 -0
- package/dist/types/index.js +10 -0
- package/dist/types/index.js.map +1 -0
- package/dist/util/error.js +5 -0
- package/dist/util/error.js.map +1 -0
- package/dist/util/fs.js +97 -0
- package/dist/util/fs.js.map +1 -0
- package/dist/util/relative-time.js +12 -0
- package/dist/util/relative-time.js.map +1 -0
- package/package.json +81 -0
|
@@ -0,0 +1,249 @@
|
|
|
1
|
+
// Deterministic read-only recall over stored titles and the cached first user
|
|
2
|
+
// prompt. No transcript-derived value can reach a subprocess from this module.
|
|
3
|
+
import { randomUUID } from 'node:crypto';
|
|
4
|
+
import { REMEMBER_DISTINCTIVE_TOKEN_FRACTION, REMEMBER_MATCH_SCORES, REMEMBER_MAX_HITS, REMEMBER_QUERY_FILLER_WORDS, REMEMBER_SCAN_BUDGET_MS, REMEMBER_SESSION_RECENCY_CAP, SESSION_CHAR_BUDGET, } from '../config/constants.js';
|
|
5
|
+
import { escapeShellSyntax } from '../security/sanitize.js';
|
|
6
|
+
import { relativeTime } from '../util/relative-time.js';
|
|
7
|
+
import { DISPLAY_VERBATIM_INSTRUCTIONS, SELECT_HINT, servedContextInstructions } from './instructions.js';
|
|
8
|
+
import { endedAt, hasRealContent, surfaceLabel, titleOf } from './session-reader.js';
|
|
9
|
+
const FILLER_WORDS = new Set(REMEMBER_QUERY_FILLER_WORDS);
|
|
10
|
+
function folded(value) {
|
|
11
|
+
return value
|
|
12
|
+
.normalize('NFKC')
|
|
13
|
+
.replace(/(\p{Lu}+)(\p{Lu}\p{Ll})/gu, '$1 $2')
|
|
14
|
+
.replace(/([\p{Ll}\p{Nd}])(\p{Lu})/gu, '$1 $2')
|
|
15
|
+
.toLowerCase();
|
|
16
|
+
}
|
|
17
|
+
function componentTokens(value) {
|
|
18
|
+
return folded(value).match(/[\p{L}\p{N}]+/gu) ?? [];
|
|
19
|
+
}
|
|
20
|
+
function lexicalTokens(value) {
|
|
21
|
+
const normalized = folded(value);
|
|
22
|
+
const compounds = normalized.match(/[\p{L}\p{N}]+(?:[._/@:+-][\p{L}\p{N}]+)*/gu) ?? [];
|
|
23
|
+
return [...new Set([...compounds, ...componentTokens(normalized)])];
|
|
24
|
+
}
|
|
25
|
+
function phraseText(value) {
|
|
26
|
+
return componentTokens(value).join(' ');
|
|
27
|
+
}
|
|
28
|
+
/** Returns undefined for empty and filler-only queries, so callers never start a broad scan. */
|
|
29
|
+
export function tokenizeRecallQuery(value) {
|
|
30
|
+
const display = value.replace(/\s+/g, ' ').trim();
|
|
31
|
+
const components = componentTokens(value).filter((token) => !FILLER_WORDS.has(token));
|
|
32
|
+
if (components.length === 0) {
|
|
33
|
+
return undefined;
|
|
34
|
+
}
|
|
35
|
+
const usefulCompounds = lexicalTokens(value).filter((token) => {
|
|
36
|
+
const parts = componentTokens(token);
|
|
37
|
+
return parts.some((part) => !FILLER_WORDS.has(part));
|
|
38
|
+
});
|
|
39
|
+
const tokens = [...new Set([...usefulCompounds, ...components])];
|
|
40
|
+
return { components, display, phrase: components.join(' '), tokens };
|
|
41
|
+
}
|
|
42
|
+
function match(texts, query) {
|
|
43
|
+
let exactPhrase = false;
|
|
44
|
+
const matched = new Set();
|
|
45
|
+
for (const text of texts) {
|
|
46
|
+
const targetPhrase = phraseText(text);
|
|
47
|
+
const targetTokens = new Set(lexicalTokens(text));
|
|
48
|
+
if (query.phrase.includes(' ') && targetPhrase.includes(query.phrase)) {
|
|
49
|
+
exactPhrase = true;
|
|
50
|
+
}
|
|
51
|
+
for (const token of query.tokens) {
|
|
52
|
+
if (targetTokens.has(token)) {
|
|
53
|
+
matched.add(token);
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
return { exactPhrase, tokens: matched };
|
|
58
|
+
}
|
|
59
|
+
function compareText(a, b) {
|
|
60
|
+
return a < b ? -1 : a > b ? 1 : 0;
|
|
61
|
+
}
|
|
62
|
+
function compareCandidates(a, b) {
|
|
63
|
+
return compareText(endedAt(b.session), endedAt(a.session)) || a.session.id - b.session.id;
|
|
64
|
+
}
|
|
65
|
+
function idfScore(hit, rarity) {
|
|
66
|
+
let score = 0;
|
|
67
|
+
for (const token of hit.matchedTokens) {
|
|
68
|
+
score += rarity.idfByToken.get(token) ?? 0;
|
|
69
|
+
}
|
|
70
|
+
return score;
|
|
71
|
+
}
|
|
72
|
+
function compareHits(a, b, rarity) {
|
|
73
|
+
return idfScore(b, rarity) - idfScore(a, rarity) || b.tier - a.tier || compareText(b.endedAt, a.endedAt) || a.sessionId - b.sessionId;
|
|
74
|
+
}
|
|
75
|
+
function startsWithElephaCommand(value) {
|
|
76
|
+
return value?.trimStart().toLowerCase().startsWith('elepha:') ?? false;
|
|
77
|
+
}
|
|
78
|
+
function isRecallCommandSession(session) {
|
|
79
|
+
// D80: the title comes from the first non-command turn, so a real title
|
|
80
|
+
// (or custom title) means substantive work — surface it even if the
|
|
81
|
+
// session opened with an elepha: command (first_prompt_search matches).
|
|
82
|
+
if (hasRealContent(session)) {
|
|
83
|
+
return false;
|
|
84
|
+
}
|
|
85
|
+
// No substantive title: a command-only / content-less session. Exclude it
|
|
86
|
+
// when any signal — title, custom title, or opening prompt — is a command.
|
|
87
|
+
return [session.custom_title, session.title, session.first_prompt_search].some(startsWithElephaCommand);
|
|
88
|
+
}
|
|
89
|
+
function hitIdentity(candidate) {
|
|
90
|
+
return {
|
|
91
|
+
project: candidate.project.displayName,
|
|
92
|
+
sessionTitle: titleOf(candidate.session),
|
|
93
|
+
date: endedAt(candidate.session).slice(0, 10),
|
|
94
|
+
tool: surfaceLabel(candidate.session.tool, candidate.session.surface),
|
|
95
|
+
endedAt: endedAt(candidate.session),
|
|
96
|
+
sessionId: candidate.session.id,
|
|
97
|
+
};
|
|
98
|
+
}
|
|
99
|
+
function prepareMetadata(candidate, query) {
|
|
100
|
+
const session = candidate.session;
|
|
101
|
+
const sessionTitles = [session.title, session.custom_title, session.rollup_title].filter((value) => Boolean(value));
|
|
102
|
+
const titleMatch = match(sessionTitles, query);
|
|
103
|
+
const firstTurnMatch = session.first_prompt_search === null
|
|
104
|
+
? { exactPhrase: false, tokens: new Set() }
|
|
105
|
+
: match([session.first_prompt_search], query);
|
|
106
|
+
return { candidate, firstTurnMatch, titleMatch };
|
|
107
|
+
}
|
|
108
|
+
function prepareRarity(prepared, query) {
|
|
109
|
+
const documentFrequency = new Map(query.tokens.map((token) => [token, 0]));
|
|
110
|
+
for (const metadata of prepared) {
|
|
111
|
+
const documentTokens = new Set([...metadata.titleMatch.tokens, ...metadata.firstTurnMatch.tokens]);
|
|
112
|
+
for (const token of documentTokens) {
|
|
113
|
+
documentFrequency.set(token, (documentFrequency.get(token) ?? 0) + 1);
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
const idfByToken = new Map();
|
|
117
|
+
const distinctiveTokens = new Set();
|
|
118
|
+
for (const token of query.tokens) {
|
|
119
|
+
const frequency = documentFrequency.get(token) ?? 0;
|
|
120
|
+
idfByToken.set(token, Math.log((prepared.length + 1) / (frequency + 1)) + 1);
|
|
121
|
+
if (prepared.length > 0 && frequency / prepared.length < REMEMBER_DISTINCTIVE_TOKEN_FRACTION) {
|
|
122
|
+
distinctiveTokens.add(token);
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
return { distinctiveTokens, idfByToken };
|
|
126
|
+
}
|
|
127
|
+
function hitForSession(metadata, query, matchingMode) {
|
|
128
|
+
const { candidate, firstTurnMatch, titleMatch } = metadata;
|
|
129
|
+
const matchedTokens = new Set([...titleMatch.tokens, ...firstTurnMatch.tokens]);
|
|
130
|
+
const matchesQuery = matchingMode === 'strict'
|
|
131
|
+
? query.components.every((token) => matchedTokens.has(token))
|
|
132
|
+
: query.components.some((token) => matchedTokens.has(token));
|
|
133
|
+
if (!matchesQuery) {
|
|
134
|
+
return undefined;
|
|
135
|
+
}
|
|
136
|
+
if (matchedTokens.size === 0) {
|
|
137
|
+
return undefined;
|
|
138
|
+
}
|
|
139
|
+
if (titleMatch.tokens.size > 0) {
|
|
140
|
+
return {
|
|
141
|
+
...hitIdentity(candidate),
|
|
142
|
+
matchedTokens,
|
|
143
|
+
tier: REMEMBER_MATCH_SCORES.title,
|
|
144
|
+
};
|
|
145
|
+
}
|
|
146
|
+
return {
|
|
147
|
+
...hitIdentity(candidate),
|
|
148
|
+
matchedTokens,
|
|
149
|
+
tier: firstTurnMatch.exactPhrase ? REMEMBER_MATCH_SCORES.exactPhrase : REMEMBER_MATCH_SCORES.body,
|
|
150
|
+
};
|
|
151
|
+
}
|
|
152
|
+
function rankAndFloorHits(hits, rarity) {
|
|
153
|
+
const qualifyingHits = rarity.distinctiveTokens.size === 0
|
|
154
|
+
? hits
|
|
155
|
+
: hits.filter((hit) => [...hit.matchedTokens].some((token) => rarity.distinctiveTokens.has(token)));
|
|
156
|
+
const ranked = qualifyingHits.sort((a, b) => compareHits(a, b, rarity));
|
|
157
|
+
return (ranked[0]?.matchedTokens.size ?? 0) >= 2 ? ranked.filter((hit) => hit.matchedTokens.size >= 2) : ranked;
|
|
158
|
+
}
|
|
159
|
+
function coverageLine(reasons, searchedProjects, totalProjects, searchedSessions, totalSessions) {
|
|
160
|
+
if (reasons.length === 0) {
|
|
161
|
+
return undefined;
|
|
162
|
+
}
|
|
163
|
+
return `Partial coverage (${reasons.join(', ')}): searched ${searchedProjects} of ${totalProjects} projects and ${searchedSessions} of ${totalSessions} sessions.`;
|
|
164
|
+
}
|
|
165
|
+
function renderBody(query, hits, coverage, now) {
|
|
166
|
+
const trailer = coverage === undefined ? ['', SELECT_HINT] : ['', coverage, '', SELECT_HINT];
|
|
167
|
+
if (hits.length === 0) {
|
|
168
|
+
return {
|
|
169
|
+
body: escapeShellSyntax([DISPLAY_VERBATIM_INSTRUCTIONS, `No recall matches found for “${query.display}”.`, ...trailer].join('\n')),
|
|
170
|
+
sessionIds: [],
|
|
171
|
+
};
|
|
172
|
+
}
|
|
173
|
+
const nonce = randomUUID();
|
|
174
|
+
const cappedHits = hits.slice(0, REMEMBER_MAX_HITS);
|
|
175
|
+
const resultCapOmitted = hits.length - cappedHits.length;
|
|
176
|
+
const build = (shown, budgetOmitted) => {
|
|
177
|
+
const lines = [
|
|
178
|
+
servedContextInstructions(nonce),
|
|
179
|
+
DISPLAY_VERBATIM_INSTRUCTIONS,
|
|
180
|
+
'',
|
|
181
|
+
`Recall hits for “${query.display}” (${shown.length} shown of ${hits.length}):`,
|
|
182
|
+
...shown.map((hit, index) => `${index + 1}. [${relativeTime(hit.endedAt, now)} | ${hit.tool}] · ${hit.project} · ${hit.sessionTitle}`),
|
|
183
|
+
resultCapOmitted > 0 ? `+${resultCapOmitted} more matches — refine your query.` : undefined,
|
|
184
|
+
budgetOmitted > 0
|
|
185
|
+
? `+${budgetOmitted} eligible recall hits omitted by the 20k-token output budget (${shown.length} shown of ${cappedHits.length}).`
|
|
186
|
+
: undefined,
|
|
187
|
+
...trailer,
|
|
188
|
+
].filter((line) => line !== undefined);
|
|
189
|
+
return escapeShellSyntax(lines.join('\n'));
|
|
190
|
+
};
|
|
191
|
+
let shown = cappedHits.length;
|
|
192
|
+
let body = build(cappedHits, 0);
|
|
193
|
+
while (body.length > SESSION_CHAR_BUDGET && shown > 0) {
|
|
194
|
+
shown -= 1;
|
|
195
|
+
body = build(cappedHits.slice(0, shown), cappedHits.length - shown);
|
|
196
|
+
}
|
|
197
|
+
return { body, sessionIds: cappedHits.slice(0, shown).map((hit) => hit.sessionId) };
|
|
198
|
+
}
|
|
199
|
+
/** Scans recent sessions without an index and returns a fully budgeted injection body. */
|
|
200
|
+
export async function lexicalRecall(reader, projects, query, scope, now, relativeNow, matchingMode) {
|
|
201
|
+
const scanClock = now ?? Date.now;
|
|
202
|
+
const allCandidates = projects
|
|
203
|
+
.flatMap((project) => reader.sessionsFor(project).map((session) => ({ project, session })))
|
|
204
|
+
.filter((candidate) => !isRecallCommandSession(candidate.session))
|
|
205
|
+
.sort(compareCandidates);
|
|
206
|
+
const candidates = allCandidates.slice(0, REMEMBER_SESSION_RECENCY_CAP[scope]);
|
|
207
|
+
const scanStartedAt = scanClock();
|
|
208
|
+
const renderedAt = relativeNow ?? scanStartedAt;
|
|
209
|
+
const prepared = [];
|
|
210
|
+
// Matching is synchronous stored-metadata tokenization, so there is no reparse or I/O to cancel or run concurrently.
|
|
211
|
+
for (const candidate of candidates) {
|
|
212
|
+
if (scanClock() - scanStartedAt > REMEMBER_SCAN_BUDGET_MS) {
|
|
213
|
+
break;
|
|
214
|
+
}
|
|
215
|
+
prepared.push(prepareMetadata(candidate, query));
|
|
216
|
+
}
|
|
217
|
+
const capReached = candidates.length < allCandidates.length;
|
|
218
|
+
const timeBudgetReached = prepared.length < candidates.length;
|
|
219
|
+
const totalByProject = new Map(projects.map((project) => [project.key, 0]));
|
|
220
|
+
for (const candidate of allCandidates) {
|
|
221
|
+
totalByProject.set(candidate.project.key, (totalByProject.get(candidate.project.key) ?? 0) + 1);
|
|
222
|
+
}
|
|
223
|
+
const searchedByProject = new Map(projects.map((project) => [project.key, 0]));
|
|
224
|
+
let notYetIndexedSessions = 0;
|
|
225
|
+
let searchedSessions = 0;
|
|
226
|
+
for (const metadata of prepared) {
|
|
227
|
+
const candidate = metadata.candidate;
|
|
228
|
+
if (candidate.session.first_prompt_search === null) {
|
|
229
|
+
notYetIndexedSessions += 1;
|
|
230
|
+
continue;
|
|
231
|
+
}
|
|
232
|
+
searchedSessions += 1;
|
|
233
|
+
searchedByProject.set(candidate.project.key, (searchedByProject.get(candidate.project.key) ?? 0) + 1);
|
|
234
|
+
}
|
|
235
|
+
const searchedProjects = projects.filter((project) => (searchedByProject.get(project.key) ?? 0) === (totalByProject.get(project.key) ?? 0)).length;
|
|
236
|
+
const reasons = [
|
|
237
|
+
capReached ? 'recency cap' : undefined,
|
|
238
|
+
timeBudgetReached ? 'time budget' : undefined,
|
|
239
|
+
notYetIndexedSessions > 0 ? `${notYetIndexedSessions} not-yet-indexed` : undefined,
|
|
240
|
+
].filter((reason) => reason !== undefined);
|
|
241
|
+
const coverage = coverageLine(reasons, searchedProjects, projects.length, searchedSessions, allCandidates.length);
|
|
242
|
+
const rarity = prepareRarity(prepared, query);
|
|
243
|
+
const hits = prepared.flatMap((metadata) => {
|
|
244
|
+
const hit = hitForSession(metadata, query, matchingMode);
|
|
245
|
+
return hit === undefined ? [] : [hit];
|
|
246
|
+
});
|
|
247
|
+
return renderBody(query, rankAndFloorHits(hits, rarity), coverage, renderedAt);
|
|
248
|
+
}
|
|
249
|
+
//# sourceMappingURL=lexical-recall.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"lexical-recall.js","sourceRoot":"","sources":["../../src/serving/lexical-recall.ts"],"names":[],"mappings":"AAAA,8EAA8E;AAC9E,+EAA+E;AAE/E,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EACH,mCAAmC,EACnC,qBAAqB,EACrB,iBAAiB,EACjB,2BAA2B,EAC3B,uBAAuB,EACvB,4BAA4B,EAC5B,mBAAmB,GACtB,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5D,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAE,6BAA6B,EAAE,WAAW,EAAE,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAC1G,OAAO,EAAE,OAAO,EAAE,cAAc,EAA0C,YAAY,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAgD7H,MAAM,YAAY,GAAG,IAAI,GAAG,CAAS,2BAA2B,CAAC,CAAC;AAElE,SAAS,MAAM,CAAC,KAAa;IACzB,OAAO,KAAK;SACP,SAAS,CAAC,MAAM,CAAC;SACjB,OAAO,CAAC,2BAA2B,EAAE,OAAO,CAAC;SAC7C,OAAO,CAAC,4BAA4B,EAAE,OAAO,CAAC;SAC9C,WAAW,EAAE,CAAC;AACvB,CAAC;AAED,SAAS,eAAe,CAAC,KAAa;IAClC,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC;AACxD,CAAC;AAED,SAAS,aAAa,CAAC,KAAa;IAChC,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IACjC,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,4CAA4C,CAAC,IAAI,EAAE,CAAC;IACvF,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,SAAS,EAAE,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;AACxE,CAAC;AAED,SAAS,UAAU,CAAC,KAAa;IAC7B,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC5C,CAAC;AAED,gGAAgG;AAChG,MAAM,UAAU,mBAAmB,CAAC,KAAa;IAC7C,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;IAClD,MAAM,UAAU,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IACtF,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,MAAM,eAAe,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QAC1D,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QACrC,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,eAAe,EAAE,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;IACjE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;AACzE,CAAC;AAED,SAAS,KAAK,CAAC,KAAuB,EAAE,KAAkB;IACtD,IAAI,WAAW,GAAG,KAAK,CAAC;IACxB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,MAAM,YAAY,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;QACtC,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;QAClD,IAAI,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;YACpE,WAAW,GAAG,IAAI,CAAC;QACvB,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YAC/B,IAAI,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;QACL,CAAC;IACL,CAAC;IACD,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;AAC5C,CAAC;AAED,SAAS,WAAW,CAAC,CAAS,EAAE,CAAS;IACrC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,iBAAiB,CAAC,CAAmB,EAAE,CAAmB;IAC/D,OAAO,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;AAC9F,CAAC;AAED,SAAS,QAAQ,CAAC,GAAc,EAAE,MAAoB;IAClD,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,KAAK,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;QACpC,KAAK,IAAI,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAS,WAAW,CAAC,CAAY,EAAE,CAAY,EAAE,MAAoB;IACjE,OAAO,QAAQ,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,IAAI,WAAW,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC;AAC1I,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAgC;IAC7D,OAAO,KAAK,EAAE,SAAS,EAAE,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC;AAC3E,CAAC;AAED,SAAS,sBAAsB,CAAC,OAAsB;IAClD,wEAAwE;IACxE,oEAAoE;IACpE,wEAAwE;IACxE,IAAI,cAAc,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1B,OAAO,KAAK,CAAC;IACjB,CAAC;IACD,0EAA0E;IAC1E,2EAA2E;IAC3E,OAAO,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;AAC5G,CAAC;AAED,SAAS,WAAW,CAAC,SAA2B;IAC5C,OAAO;QACH,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,WAAW;QACtC,YAAY,EAAE,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;QACxC,IAAI,EAAE,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;QAC7C,IAAI,EAAE,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;QACrE,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC;QACnC,SAAS,EAAE,SAAS,CAAC,OAAO,CAAC,EAAE;KAClC,CAAC;AACN,CAAC;AAED,SAAS,eAAe,CAAC,SAA2B,EAAE,KAAkB;IACpE,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC;IAClC,MAAM,aAAa,GAAG,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;IACrI,MAAM,UAAU,GAAG,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;IAC/C,MAAM,cAAc,GAChB,OAAO,CAAC,mBAAmB,KAAK,IAAI;QAChC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,GAAG,EAAU,EAAE;QACnD,CAAC,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,KAAK,CAAC,CAAC;IACtD,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,UAAU,EAAE,CAAC;AACrD,CAAC;AAED,SAAS,aAAa,CAAC,QAA4B,EAAE,KAAkB;IACnE,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3E,KAAK,MAAM,QAAQ,IAAI,QAAQ,EAAE,CAAC;QAC9B,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,GAAG,QAAQ,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;QACnG,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;YACjC,iBAAiB,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC1E,CAAC;IACL,CAAC;IACD,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC7C,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;IAC5C,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACpD,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7E,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,SAAS,GAAG,QAAQ,CAAC,MAAM,GAAG,mCAAmC,EAAE,CAAC;YAC3F,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,CAAC;IACL,CAAC;IACD,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC;AAC7C,CAAC;AAED,SAAS,aAAa,CAAC,QAA0B,EAAE,KAAkB,EAAE,YAA+B;IAClG,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,UAAU,EAAE,GAAG,QAAQ,CAAC;IAC3D,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;IAChF,MAAM,YAAY,GACd,YAAY,KAAK,QAAQ;QACrB,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC7D,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IACrE,IAAI,CAAC,YAAY,EAAE,CAAC;QAChB,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,IAAI,aAAa,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,IAAI,UAAU,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QAC7B,OAAO;YACH,GAAG,WAAW,CAAC,SAAS,CAAC;YACzB,aAAa;YACb,IAAI,EAAE,qBAAqB,CAAC,KAAK;SACpC,CAAC;IACN,CAAC;IACD,OAAO;QACH,GAAG,WAAW,CAAC,SAAS,CAAC;QACzB,aAAa;QACb,IAAI,EAAE,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,CAAC,qBAAqB,CAAC,IAAI;KACpG,CAAC;AACN,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAiB,EAAE,MAAoB;IAC7D,MAAM,cAAc,GAChB,MAAM,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC;QAC/B,CAAC,CAAC,IAAI;QACN,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,GAAG,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC5G,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;IACxE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,aAAa,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;AACpH,CAAC;AAED,SAAS,YAAY,CACjB,OAAiB,EACjB,gBAAwB,EACxB,aAAqB,EACrB,gBAAwB,EACxB,aAAqB;IAErB,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,OAAO,qBAAqB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,gBAAgB,OAAO,aAAa,iBAAiB,gBAAgB,OAAO,aAAa,YAAY,CAAC;AACvK,CAAC;AAED,SAAS,UAAU,CAAC,KAAkB,EAAE,IAAiB,EAAE,QAA4B,EAAE,GAAW;IAChG,MAAM,OAAO,GAAG,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,WAAW,CAAC,CAAC;IAC7F,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACpB,OAAO;YACH,IAAI,EAAE,iBAAiB,CACnB,CAAC,6BAA6B,EAAE,gCAAgC,KAAK,CAAC,OAAO,IAAI,EAAE,GAAG,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAC5G;YACD,UAAU,EAAE,EAAE;SACjB,CAAC;IACN,CAAC;IAED,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;IAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;IACpD,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;IACzD,MAAM,KAAK,GAAG,CAAC,KAAkB,EAAE,aAAqB,EAAU,EAAE;QAChE,MAAM,KAAK,GAAG;YACV,yBAAyB,CAAC,KAAK,CAAC;YAChC,6BAA6B;YAC7B,EAAE;YACF,oBAAoB,KAAK,CAAC,OAAO,MAAM,KAAK,CAAC,MAAM,aAAa,IAAI,CAAC,MAAM,IAAI;YAC/E,GAAG,KAAK,CAAC,GAAG,CACR,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,KAAK,GAAG,CAAC,MAAM,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,GAAG,CAAC,OAAO,MAAM,GAAG,CAAC,YAAY,EAAE,CAC3H;YACD,gBAAgB,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,gBAAgB,oCAAoC,CAAC,CAAC,CAAC,SAAS;YAC3F,aAAa,GAAG,CAAC;gBACb,CAAC,CAAC,IAAI,aAAa,iEAAiE,KAAK,CAAC,MAAM,aAAa,UAAU,CAAC,MAAM,IAAI;gBAClI,CAAC,CAAC,SAAS;YACf,GAAG,OAAO;SACb,CAAC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;QACvD,OAAO,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC;IAEF,IAAI,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC;IAC9B,IAAI,IAAI,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;IAChC,OAAO,IAAI,CAAC,MAAM,GAAG,mBAAmB,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;QACpD,KAAK,IAAI,CAAC,CAAC;QACX,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,UAAU,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC;IACxE,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;AACxF,CAAC;AAED,0FAA0F;AAC1F,MAAM,CAAC,KAAK,UAAU,aAAa,CAC/B,MAAqB,EACrB,QAAsB,EACtB,KAAkB,EAClB,KAAkB,EAClB,GAA+B,EAC/B,WAA+B,EAC/B,YAA+B;IAE/B,MAAM,SAAS,GAAG,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC;IAClC,MAAM,aAAa,GAAG,QAAQ;SACzB,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;SAC1F,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,sBAAsB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;SACjE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC7B,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,4BAA4B,CAAC,KAAK,CAAC,CAAC,CAAC;IAC/E,MAAM,aAAa,GAAG,SAAS,EAAE,CAAC;IAClC,MAAM,UAAU,GAAG,WAAW,IAAI,aAAa,CAAC;IAChD,MAAM,QAAQ,GAAuB,EAAE,CAAC;IACxC,qHAAqH;IACrH,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACjC,IAAI,SAAS,EAAE,GAAG,aAAa,GAAG,uBAAuB,EAAE,CAAC;YACxD,MAAM;QACV,CAAC;QACD,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;IACrD,CAAC;IACD,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC;IAC5D,MAAM,iBAAiB,GAAG,QAAQ,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;IAC9D,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5E,KAAK,MAAM,SAAS,IAAI,aAAa,EAAE,CAAC;QACpC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IACpG,CAAC;IACD,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/E,IAAI,qBAAqB,GAAG,CAAC,CAAC;IAC9B,IAAI,gBAAgB,GAAG,CAAC,CAAC;IAEzB,KAAK,MAAM,QAAQ,IAAI,QAAQ,EAAE,CAAC;QAC9B,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;QACrC,IAAI,SAAS,CAAC,OAAO,CAAC,mBAAmB,KAAK,IAAI,EAAE,CAAC;YACjD,qBAAqB,IAAI,CAAC,CAAC;YAC3B,SAAS;QACb,CAAC;QACD,gBAAgB,IAAI,CAAC,CAAC;QACtB,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAC1G,CAAC;IACD,MAAM,gBAAgB,GAAG,QAAQ,CAAC,MAAM,CACpC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CACpG,CAAC,MAAM,CAAC;IACT,MAAM,OAAO,GAAG;QACZ,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS;QACtC,iBAAiB,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS;QAC7C,qBAAqB,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,qBAAqB,kBAAkB,CAAC,CAAC,CAAC,SAAS;KACrF,CAAC,MAAM,CAAC,CAAC,MAAM,EAAoB,EAAE,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;IAC7D,MAAM,QAAQ,GAAG,YAAY,CAAC,OAAO,EAAE,gBAAgB,EAAE,QAAQ,CAAC,MAAM,EAAE,gBAAgB,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;IAClH,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC9C,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;QACvC,MAAM,GAAG,GAAG,aAAa,CAAC,QAAQ,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC;QACzD,OAAO,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IACH,OAAO,UAAU,CAAC,KAAK,EAAE,gBAAgB,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;AACnF,CAAC"}
|
|
@@ -0,0 +1,302 @@
|
|
|
1
|
+
// Tool-neutral read path for historical episodes. It owns transcript reparse
|
|
2
|
+
// and the 20k newest-first budget; callers own their transport/envelopes.
|
|
3
|
+
import { randomUUID } from 'node:crypto';
|
|
4
|
+
import { defaultAdapters } from '../adapters/index.js';
|
|
5
|
+
import { AUTO_BRIEF_AGGREGATE_FILE_LIMIT, AUTO_BRIEF_AGGREGATE_SESSION_LIMIT, MAX_GET_SESSION_LAST_N, RECENT_SESSION_WINDOW_MS, SESSION_CHAR_BUDGET, } from '../config/constants.js';
|
|
6
|
+
import { omissionMarker, RAW_TURN_SEPARATOR, renderableRawTurns, renderRawTurn } from '../rendering/raw-turn-renderer.js';
|
|
7
|
+
import { openProviderTranscript } from '../security/provider-transcript.js';
|
|
8
|
+
import { isSubstantive, readProjectSessionAggregates, readProjectSessions, readSessionById, } from '../storage/session-read-model.js';
|
|
9
|
+
import { UNTITLED_EPISODE } from '../storage/session-title.js';
|
|
10
|
+
import { TOOL_METADATA } from '../types/index.js';
|
|
11
|
+
import { dataBlockClose, dataBlockOpen } from './instructions.js';
|
|
12
|
+
function leafStrings(value) {
|
|
13
|
+
if (typeof value === 'string') {
|
|
14
|
+
return [value];
|
|
15
|
+
}
|
|
16
|
+
if (Array.isArray(value)) {
|
|
17
|
+
return value.flatMap(leafStrings);
|
|
18
|
+
}
|
|
19
|
+
if (value && typeof value === 'object') {
|
|
20
|
+
return Object.values(value).flatMap(leafStrings);
|
|
21
|
+
}
|
|
22
|
+
return [];
|
|
23
|
+
}
|
|
24
|
+
function decodedStrings(value) {
|
|
25
|
+
try {
|
|
26
|
+
return leafStrings(JSON.parse(value));
|
|
27
|
+
}
|
|
28
|
+
catch {
|
|
29
|
+
return [];
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
export function surfaceLabel(tool, surface) {
|
|
33
|
+
const displayName = TOOL_METADATA[tool].displayName;
|
|
34
|
+
return surface === 'desktop' ? `${displayName} Desktop` : `${displayName} CLI`;
|
|
35
|
+
}
|
|
36
|
+
export function endedAt(session) {
|
|
37
|
+
return session.last_turn_at ?? session.last_ingested_at ?? session.started_at;
|
|
38
|
+
}
|
|
39
|
+
export function titleOf(session) {
|
|
40
|
+
return session.title?.trim() || UNTITLED_EPISODE;
|
|
41
|
+
}
|
|
42
|
+
export function hasRealContent(session) {
|
|
43
|
+
if (session.custom_title?.trim()) {
|
|
44
|
+
return true;
|
|
45
|
+
}
|
|
46
|
+
const title = session.title?.trim();
|
|
47
|
+
return title !== undefined && title !== '' && title !== UNTITLED_EPISODE;
|
|
48
|
+
}
|
|
49
|
+
export function newestActivity(sessions, { excludeNativeId }) {
|
|
50
|
+
let newest;
|
|
51
|
+
let newestEndedAt = Number.NEGATIVE_INFINITY;
|
|
52
|
+
for (const session of sessions) {
|
|
53
|
+
if (session.native_id === excludeNativeId) {
|
|
54
|
+
continue;
|
|
55
|
+
}
|
|
56
|
+
const sessionEndedAt = Date.parse(endedAt(session));
|
|
57
|
+
if (sessionEndedAt > newestEndedAt) {
|
|
58
|
+
newest = session;
|
|
59
|
+
newestEndedAt = sessionEndedAt;
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
return newest;
|
|
63
|
+
}
|
|
64
|
+
export class SessionReader {
|
|
65
|
+
db;
|
|
66
|
+
openTranscript;
|
|
67
|
+
adapters;
|
|
68
|
+
sessionsMemo = new Map();
|
|
69
|
+
consentedSessionsMemo = new Map();
|
|
70
|
+
constructor(db, adapters = defaultAdapters(), openTranscript = openProviderTranscript) {
|
|
71
|
+
this.db = db;
|
|
72
|
+
this.openTranscript = openTranscript;
|
|
73
|
+
this.adapters = adapters;
|
|
74
|
+
}
|
|
75
|
+
/**
|
|
76
|
+
* Memoized per reader instance, the same pattern as ProjectResolver.list:
|
|
77
|
+
* one operation's repeated reads of a project share a single load. A
|
|
78
|
+
* long-lived caller constructs a fresh reader per request so later
|
|
79
|
+
* requests observe daemon writes.
|
|
80
|
+
*/
|
|
81
|
+
sessionsFor(project) {
|
|
82
|
+
const key = project.projectIds.join(',');
|
|
83
|
+
const cached = this.sessionsMemo.get(key);
|
|
84
|
+
if (cached !== undefined) {
|
|
85
|
+
return cached;
|
|
86
|
+
}
|
|
87
|
+
const rows = readProjectSessions(this.db, project.projectIds);
|
|
88
|
+
this.sessionsMemo.set(key, rows);
|
|
89
|
+
return rows;
|
|
90
|
+
}
|
|
91
|
+
sessionAggregatesFor(projects) {
|
|
92
|
+
const projectIds = [...new Set(projects.flatMap((project) => project.projectIds))];
|
|
93
|
+
return readProjectSessionAggregates(this.db, projectIds);
|
|
94
|
+
}
|
|
95
|
+
/**
|
|
96
|
+
* Reads every session belonging to the already consent-filtered project
|
|
97
|
+
* sets in one newest-first query. The caller owns the consent boundary;
|
|
98
|
+
* this reader only combines its internal project ids.
|
|
99
|
+
*/
|
|
100
|
+
recentConsentedSessions(projects) {
|
|
101
|
+
const projectIds = [...new Set(projects.flatMap((project) => project.projectIds))].sort((a, b) => a - b);
|
|
102
|
+
if (projectIds.length === 0) {
|
|
103
|
+
return [];
|
|
104
|
+
}
|
|
105
|
+
const key = projectIds.join(',');
|
|
106
|
+
const cached = this.consentedSessionsMemo.get(key);
|
|
107
|
+
if (cached !== undefined) {
|
|
108
|
+
return cached;
|
|
109
|
+
}
|
|
110
|
+
const rows = readProjectSessions(this.db, projectIds).filter(hasRealContent);
|
|
111
|
+
this.consentedSessionsMemo.set(key, rows);
|
|
112
|
+
return rows;
|
|
113
|
+
}
|
|
114
|
+
consentedTotal(projects) {
|
|
115
|
+
return this.recentConsentedSessions(projects).length;
|
|
116
|
+
}
|
|
117
|
+
sessionById(id) {
|
|
118
|
+
return readSessionById(this.db, id);
|
|
119
|
+
}
|
|
120
|
+
newestSubstantive(project) {
|
|
121
|
+
return this.sessionsFor(project).find(isSubstantive);
|
|
122
|
+
}
|
|
123
|
+
counts(project, now = Date.now()) {
|
|
124
|
+
const rows = this.sessionsFor(project);
|
|
125
|
+
const sevenDaysAgo = now - RECENT_SESSION_WINDOW_MS;
|
|
126
|
+
return { total: rows.length, recent: rows.filter((row) => Date.parse(endedAt(row)) >= sevenDaysAgo).length };
|
|
127
|
+
}
|
|
128
|
+
storedRecallFieldsFor(sessions) {
|
|
129
|
+
const ids = [...new Set([...sessions].map((session) => session.id))];
|
|
130
|
+
const bySession = new Map(ids.map((id) => [id, new Map()]));
|
|
131
|
+
if (ids.length === 0) {
|
|
132
|
+
return bySession;
|
|
133
|
+
}
|
|
134
|
+
const rows = this.db
|
|
135
|
+
.prepare(`SELECT session_id, turn_index, decisions, files_touched, pending_items
|
|
136
|
+
FROM memories WHERE session_id IN (${ids.map(() => '?').join(',')})
|
|
137
|
+
ORDER BY session_id, turn_index`)
|
|
138
|
+
.all(...ids);
|
|
139
|
+
for (const row of rows) {
|
|
140
|
+
bySession.get(row.session_id)?.set(row.turn_index, {
|
|
141
|
+
decisions: decodedStrings(row.decisions),
|
|
142
|
+
filesTouched: decodedStrings(row.files_touched),
|
|
143
|
+
pendingItems: decodedStrings(row.pending_items),
|
|
144
|
+
});
|
|
145
|
+
}
|
|
146
|
+
return bySession;
|
|
147
|
+
}
|
|
148
|
+
storedTurnRecallFields(session) {
|
|
149
|
+
return this.storedRecallFieldsFor([session]).get(session.id) ?? new Map();
|
|
150
|
+
}
|
|
151
|
+
async turns(session, signal, storedIndexes, bounds) {
|
|
152
|
+
const opened = await this.openTranscript(session.tool, session.source_path);
|
|
153
|
+
if ('reason' in opened) {
|
|
154
|
+
return { reason: opened.reason };
|
|
155
|
+
}
|
|
156
|
+
const { handle } = opened;
|
|
157
|
+
try {
|
|
158
|
+
const indexes = storedIndexes ?? new Set(this.storedTurnRecallFields(session).keys());
|
|
159
|
+
if (indexes.size === 0) {
|
|
160
|
+
return { reason: 'no_stored_turn_indexes' };
|
|
161
|
+
}
|
|
162
|
+
const turns = [];
|
|
163
|
+
const retained = new Map();
|
|
164
|
+
let matchedTurns = 0;
|
|
165
|
+
let renderedTurns = 0;
|
|
166
|
+
let omittedBefore = 0;
|
|
167
|
+
let retainedRenderedChars = 0;
|
|
168
|
+
let highWaterTurns = 0;
|
|
169
|
+
let highWaterRenderedChars = 0;
|
|
170
|
+
for await (const turn of this.adapters[session.tool].parseTurns(session.source_path, undefined, {
|
|
171
|
+
closeTrailingOnIdle: true,
|
|
172
|
+
handle,
|
|
173
|
+
signal,
|
|
174
|
+
})) {
|
|
175
|
+
if (signal?.aborted) {
|
|
176
|
+
return { reason: 'deadline' };
|
|
177
|
+
}
|
|
178
|
+
if (indexes.has(turn.turnIndex)) {
|
|
179
|
+
matchedTurns += 1;
|
|
180
|
+
if (bounds === undefined) {
|
|
181
|
+
turns.push(turn);
|
|
182
|
+
}
|
|
183
|
+
else {
|
|
184
|
+
const rendered = renderRawTurn(turn, renderedTurns + 1);
|
|
185
|
+
if (rendered !== null) {
|
|
186
|
+
renderedTurns += 1;
|
|
187
|
+
const framedLength = dataBlockOpen(bounds.nonce).length + 1 + rendered.length + 1 + dataBlockClose(bounds.nonce).length;
|
|
188
|
+
retainedRenderedChars += framedLength + (retained.size === 0 ? 1 : RAW_TURN_SEPARATOR.length);
|
|
189
|
+
retained.set(turn.turnIndex, { turn, renderedLength: framedLength });
|
|
190
|
+
while ((bounds.lastN !== undefined && retained.size > bounds.lastN) ||
|
|
191
|
+
retainedRenderedChars > bounds.charBudget) {
|
|
192
|
+
const oldest = retained.entries().next().value;
|
|
193
|
+
if (oldest === undefined) {
|
|
194
|
+
break;
|
|
195
|
+
}
|
|
196
|
+
const [oldestIndex, oldestTurn] = oldest;
|
|
197
|
+
const oldestContribution = oldestTurn.renderedLength + (retained.size === 1 ? 1 : RAW_TURN_SEPARATOR.length);
|
|
198
|
+
retained.delete(oldestIndex);
|
|
199
|
+
retainedRenderedChars -= oldestContribution;
|
|
200
|
+
omittedBefore += 1;
|
|
201
|
+
}
|
|
202
|
+
highWaterTurns = Math.max(highWaterTurns, retained.size);
|
|
203
|
+
highWaterRenderedChars = Math.max(highWaterRenderedChars, retainedRenderedChars);
|
|
204
|
+
}
|
|
205
|
+
}
|
|
206
|
+
if (matchedTurns === indexes.size) {
|
|
207
|
+
break;
|
|
208
|
+
}
|
|
209
|
+
}
|
|
210
|
+
}
|
|
211
|
+
if (signal?.aborted) {
|
|
212
|
+
return { reason: 'deadline' };
|
|
213
|
+
}
|
|
214
|
+
if (matchedTurns === 0) {
|
|
215
|
+
return { reason: 'transcript_reparse_empty' };
|
|
216
|
+
}
|
|
217
|
+
if (bounds === undefined) {
|
|
218
|
+
return { turns };
|
|
219
|
+
}
|
|
220
|
+
return {
|
|
221
|
+
turns: [...retained.values()].map((entry) => entry.turn),
|
|
222
|
+
omittedBefore,
|
|
223
|
+
retentionHighWater: { turns: highWaterTurns, renderedChars: highWaterRenderedChars },
|
|
224
|
+
};
|
|
225
|
+
}
|
|
226
|
+
catch {
|
|
227
|
+
return { reason: 'transcript_unreadable' };
|
|
228
|
+
}
|
|
229
|
+
finally {
|
|
230
|
+
await handle.close();
|
|
231
|
+
}
|
|
232
|
+
}
|
|
233
|
+
async render(session, lastN, signal, charBudget = SESSION_CHAR_BUDGET) {
|
|
234
|
+
const nonce = randomUUID();
|
|
235
|
+
const boundedLastN = lastN === undefined ? undefined : Math.min(Math.max(1, Math.trunc(lastN)), MAX_GET_SESSION_LAST_N);
|
|
236
|
+
const parsed = await this.turns(session, signal, undefined, { lastN: boundedLastN, charBudget, nonce });
|
|
237
|
+
return parsed.turns === undefined
|
|
238
|
+
? { reason: parsed.reason }
|
|
239
|
+
: { episode: boundedRender(parsed.turns, boundedLastN, charBudget, nonce, parsed.omittedBefore) };
|
|
240
|
+
}
|
|
241
|
+
aggregate(project) {
|
|
242
|
+
const rows = this.sessionsFor(project).filter(isSubstantive).slice(0, AUTO_BRIEF_AGGREGATE_SESSION_LIMIT);
|
|
243
|
+
const ids = rows.map((row) => row.id);
|
|
244
|
+
if (ids.length === 0) {
|
|
245
|
+
return { files: [], surfaces: [], lastActivity: null };
|
|
246
|
+
}
|
|
247
|
+
const memories = this.db
|
|
248
|
+
.prepare(`SELECT files_touched FROM memories WHERE session_id IN (${ids.map(() => '?').join(',')})`)
|
|
249
|
+
.all(...ids);
|
|
250
|
+
const count = new Map();
|
|
251
|
+
for (const memory of memories) {
|
|
252
|
+
try {
|
|
253
|
+
const files = JSON.parse(memory.files_touched);
|
|
254
|
+
if (Array.isArray(files)) {
|
|
255
|
+
for (const file of files) {
|
|
256
|
+
if (typeof file === 'string') {
|
|
257
|
+
count.set(file, (count.get(file) ?? 0) + 1);
|
|
258
|
+
}
|
|
259
|
+
}
|
|
260
|
+
}
|
|
261
|
+
}
|
|
262
|
+
catch {
|
|
263
|
+
// Stored malformed JSON is not a valid zero; omit it from an aggregate only.
|
|
264
|
+
}
|
|
265
|
+
}
|
|
266
|
+
const files = [...count.entries()]
|
|
267
|
+
.sort((a, b) => b[1] - a[1] || a[0].localeCompare(b[0]))
|
|
268
|
+
.slice(0, AUTO_BRIEF_AGGREGATE_FILE_LIMIT)
|
|
269
|
+
.map(([file]) => file.split('/').filter(Boolean).at(-1) ?? file);
|
|
270
|
+
return { files, surfaces: [...new Set(rows.map((row) => surfaceLabel(row.tool, row.surface)))], lastActivity: endedAt(rows[0]) };
|
|
271
|
+
}
|
|
272
|
+
}
|
|
273
|
+
export function boundedRender(turns, lastN, charBudget = SESSION_CHAR_BUDGET, nonce = randomUUID(), omittedBefore = 0) {
|
|
274
|
+
const pieces = renderableRawTurns(turns, omittedBefore).map((piece) => `${dataBlockOpen(nonce)}\n${piece}\n${dataBlockClose(nonce)}`);
|
|
275
|
+
const eligible = lastN === undefined ? pieces : pieces.slice(-lastN);
|
|
276
|
+
const chosen = [];
|
|
277
|
+
let renderedChars = 0;
|
|
278
|
+
for (let index = eligible.length - 1; index >= 0; index--) {
|
|
279
|
+
const piece = eligible[index];
|
|
280
|
+
if (piece === undefined) {
|
|
281
|
+
continue;
|
|
282
|
+
}
|
|
283
|
+
const addition = piece.length + (chosen.length === 0 ? 1 : RAW_TURN_SEPARATOR.length);
|
|
284
|
+
if (renderedChars + addition > charBudget) {
|
|
285
|
+
break;
|
|
286
|
+
}
|
|
287
|
+
chosen.unshift(piece);
|
|
288
|
+
renderedChars += addition;
|
|
289
|
+
}
|
|
290
|
+
const total = omittedBefore + pieces.length;
|
|
291
|
+
const omitted = total - chosen.length;
|
|
292
|
+
const suffix = omitted === 0 ? '' : `\n${omissionMarker(omitted, chosen.length, total)}\n`;
|
|
293
|
+
return {
|
|
294
|
+
text: `${chosen.join(RAW_TURN_SEPARATOR)}${chosen.length > 0 ? '\n' : ''}${suffix}`,
|
|
295
|
+
returned: chosen.length,
|
|
296
|
+
omitted,
|
|
297
|
+
total,
|
|
298
|
+
renderedChars,
|
|
299
|
+
nonce,
|
|
300
|
+
};
|
|
301
|
+
}
|
|
302
|
+
//# sourceMappingURL=session-reader.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-reader.js","sourceRoot":"","sources":["../../src/serving/session-reader.ts"],"names":[],"mappings":"AAAA,6EAA6E;AAC7E,0EAA0E;AAE1E,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EACH,+BAA+B,EAC/B,kCAAkC,EAClC,sBAAsB,EACtB,wBAAwB,EACxB,mBAAmB,GACtB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAC1H,OAAO,EAAE,sBAAsB,EAAiC,MAAM,oCAAoC,CAAC;AAE3G,OAAO,EACH,aAAa,EAEb,4BAA4B,EAC5B,mBAAmB,EACnB,eAAe,GAElB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAwC,aAAa,EAAiB,MAAM,mBAAmB,CAAC;AACvG,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAgClE,SAAS,WAAW,CAAC,KAAc;IAC/B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5B,OAAO,CAAC,KAAK,CAAC,CAAC;IACnB,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACtC,CAAC;IACD,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACrD,CAAC;IACD,OAAO,EAAE,CAAC;AACd,CAAC;AAED,SAAS,cAAc,CAAC,KAAa;IACjC,IAAI,CAAC;QACD,OAAO,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IAC1C,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,EAAE,CAAC;IACd,CAAC;AACL,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,IAAc,EAAE,OAAiC;IAC1E,MAAM,WAAW,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC;IACpD,OAAO,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,WAAW,UAAU,CAAC,CAAC,CAAC,GAAG,WAAW,MAAM,CAAC;AACnF,CAAC;AAED,MAAM,UAAU,OAAO,CAAC,OAAgF;IACpG,OAAO,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,UAAU,CAAC;AAClF,CAAC;AAED,MAAM,UAAU,OAAO,CAAC,OAAqC;IACzD,OAAO,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,gBAAgB,CAAC;AACrD,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,OAAsD;IACjF,IAAI,OAAO,CAAC,YAAY,EAAE,IAAI,EAAE,EAAE,CAAC;QAC/B,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC;IACpC,OAAO,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,EAAE,IAAI,KAAK,KAAK,gBAAgB,CAAC;AAC7E,CAAC;AAED,MAAM,UAAU,cAAc,CAC1B,QAAiC,EACjC,EAAE,eAAe,EAA+B;IAEhD,IAAI,MAAiC,CAAC;IACtC,IAAI,aAAa,GAAG,MAAM,CAAC,iBAAiB,CAAC;IAC7C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC7B,IAAI,OAAO,CAAC,SAAS,KAAK,eAAe,EAAE,CAAC;YACxC,SAAS;QACb,CAAC;QACD,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;QACpD,IAAI,cAAc,GAAG,aAAa,EAAE,CAAC;YACjC,MAAM,GAAG,OAAO,CAAC;YACjB,aAAa,GAAG,cAAc,CAAC;QACnC,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,MAAM,OAAO,aAAa;IAMD,EAAE;IAEF,cAAc;IAPlB,QAAQ,CAAmC;IAC3C,YAAY,GAAG,IAAI,GAAG,EAA2B,CAAC;IAClD,qBAAqB,GAAG,IAAI,GAAG,EAA2B,CAAC;IAE5E,YACqB,EAAqB,EACtC,QAAQ,GAAqC,eAAe,EAAE,EAC7C,cAAc,GAA6B,sBAAsB;kBAFjE,EAAE;8BAEF,cAAc;QAE/B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC7B,CAAC;IAED;;;;;OAKG;IACH,WAAW,CAAC,OAAmB;QAC3B,MAAM,GAAG,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO,MAAM,CAAC;QAClB,CAAC;QACD,MAAM,IAAI,GAAG,mBAAmB,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;QAC9D,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACjC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,oBAAoB,CAAC,QAA+B;QAChD,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACnF,OAAO,4BAA4B,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAC7D,CAAC;IAED;;;;OAIG;IACH,uBAAuB,CAAC,QAA+B;QACnD,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACzG,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,EAAE,CAAC;QACd,CAAC;QACD,MAAM,GAAG,GAAG,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACnD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO,MAAM,CAAC;QAClB,CAAC;QACD,MAAM,IAAI,GAAG,mBAAmB,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAC7E,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAC1C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,cAAc,CAAC,QAA+B;QAC1C,OAAO,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;IACzD,CAAC;IAED,WAAW,CAAC,EAAU;QAClB,OAAO,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,iBAAiB,CAAC,OAAmB;QACjC,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACzD,CAAC;IAED,MAAM,CAAC,OAAmB,EAAE,GAAG,GAAW,IAAI,CAAC,GAAG,EAAE;QAChD,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACvC,MAAM,YAAY,GAAG,GAAG,GAAG,wBAAwB,CAAC;QACpD,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,YAAY,CAAC,CAAC,MAAM,EAAE,CAAC;IACjH,CAAC;IAED,qBAAqB,CAAC,QAA6C;QAC/D,MAAM,GAAG,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACrE,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,GAAG,EAAkC,CAAC,CAAC,CAAC,CAAC;QAC5F,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnB,OAAO,SAAS,CAAC;QACrB,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACf,OAAO,CACJ;sDACsC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;iDACjC,CACpC;aACA,GAAG,CAAC,GAAG,GAAG,CAMb,CAAC;QACH,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACrB,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE;gBAC/C,SAAS,EAAE,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC;gBACxC,YAAY,EAAE,cAAc,CAAC,GAAG,CAAC,aAAa,CAAC;gBAC/C,YAAY,EAAE,cAAc,CAAC,GAAG,CAAC,aAAa,CAAC;aAClD,CAAC,CAAC;QACP,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,sBAAsB,CAAC,OAAkC;QACrD,OAAO,IAAI,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,KAAK,CACP,OAAsB,EACtB,MAAoB,EACpB,aAAmC,EACnC,MAA6B;QAO7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QAC5E,IAAI,QAAQ,IAAI,MAAM,EAAE,CAAC;YACrB,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;QACrC,CAAC;QACD,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;QAC1B,IAAI,CAAC;YACD,MAAM,OAAO,GAAG,aAAa,IAAI,IAAI,GAAG,CAAC,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YACtF,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACrB,OAAO,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC;YAChD,CAAC;YACD,MAAM,KAAK,GAAiB,EAAE,CAAC;YAC/B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAwB,CAAC;YACjD,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,IAAI,aAAa,GAAG,CAAC,CAAC;YACtB,IAAI,aAAa,GAAG,CAAC,CAAC;YACtB,IAAI,qBAAqB,GAAG,CAAC,CAAC;YAC9B,IAAI,cAAc,GAAG,CAAC,CAAC;YACvB,IAAI,sBAAsB,GAAG,CAAC,CAAC;YAC/B,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,EAAE;gBAC5F,mBAAmB,EAAE,IAAI;gBACzB,MAAM;gBACN,MAAM;aACT,CAAC,EAAE,CAAC;gBACD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;oBAClB,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;gBAClC,CAAC;gBACD,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC9B,YAAY,IAAI,CAAC,CAAC;oBAClB,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;wBACvB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACrB,CAAC;yBAAM,CAAC;wBACJ,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,EAAE,aAAa,GAAG,CAAC,CAAC,CAAC;wBACxD,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;4BACpB,aAAa,IAAI,CAAC,CAAC;4BACnB,MAAM,YAAY,GACd,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;4BACvG,qBAAqB,IAAI,YAAY,GAAG,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;4BAC9F,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC,CAAC;4BACrE,OACI,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,IAAI,QAAQ,CAAC,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC;gCAC5D,qBAAqB,GAAG,MAAM,CAAC,UAAU,EAC3C,CAAC;gCACC,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;gCAC/C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;oCACvB,MAAM;gCACV,CAAC;gCACD,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,GAAG,MAAM,CAAC;gCACzC,MAAM,kBAAkB,GACpB,UAAU,CAAC,cAAc,GAAG,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;gCACtF,QAAQ,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;gCAC7B,qBAAqB,IAAI,kBAAkB,CAAC;gCAC5C,aAAa,IAAI,CAAC,CAAC;4BACvB,CAAC;4BACD,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;4BACzD,sBAAsB,GAAG,IAAI,CAAC,GAAG,CAAC,sBAAsB,EAAE,qBAAqB,CAAC,CAAC;wBACrF,CAAC;oBACL,CAAC;oBACD,IAAI,YAAY,KAAK,OAAO,CAAC,IAAI,EAAE,CAAC;wBAChC,MAAM;oBACV,CAAC;gBACL,CAAC;YACL,CAAC;YACD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBAClB,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;YAClC,CAAC;YACD,IAAI,YAAY,KAAK,CAAC,EAAE,CAAC;gBACrB,OAAO,EAAE,MAAM,EAAE,0BAA0B,EAAE,CAAC;YAClD,CAAC;YACD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACvB,OAAO,EAAE,KAAK,EAAE,CAAC;YACrB,CAAC;YACD,OAAO;gBACH,KAAK,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;gBACxD,aAAa;gBACb,kBAAkB,EAAE,EAAE,KAAK,EAAE,cAAc,EAAE,aAAa,EAAE,sBAAsB,EAAE;aACvF,CAAC;QACN,CAAC;QAAC,MAAM,CAAC;YACL,OAAO,EAAE,MAAM,EAAE,uBAAuB,EAAE,CAAC;QAC/C,CAAC;gBAAS,CAAC;YACP,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACzB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CACR,OAAsB,EACtB,KAAc,EACd,MAAoB,EACpB,UAAU,GAAW,mBAAmB;QAExC,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,YAAY,GAAG,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,sBAAsB,CAAC,CAAC;QACxH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;QACxG,OAAO,MAAM,CAAC,KAAK,KAAK,SAAS;YAC7B,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE;YAC3B,CAAC,CAAC,EAAE,OAAO,EAAE,aAAa,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;IAC1G,CAAC;IAED,SAAS,CAAC,OAAmB;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,kCAAkC,CAAC,CAAC;QAC1G,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtC,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnB,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;QAC3D,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE;aACnB,OAAO,CAAC,2DAA2D,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;aACnG,GAAG,CAAC,GAAG,GAAG,CAAqC,CAAC;QACrD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAC;QACxC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC5B,IAAI,CAAC;gBACD,MAAM,KAAK,GAAY,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;gBACxD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;oBACvB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;wBACvB,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;4BAC3B,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;wBAChD,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACL,6EAA6E;YACjF,CAAC;QACL,CAAC;QACD,MAAM,KAAK,GAAG,CAAC,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC;aAC7B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;aACvD,KAAK,CAAC,CAAC,EAAE,+BAA+B,CAAC;aACzC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;QACrE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACrI,CAAC;CACJ;AAED,MAAM,UAAU,aAAa,CACzB,KAA2B,EAC3B,KAAc,EACd,UAAU,GAAW,mBAAmB,EACxC,KAAK,GAAW,UAAU,EAAE,EAC5B,aAAa,GAAW,CAAC;IAEzB,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,aAAa,CAAC,KAAK,CAAC,KAAK,KAAK,KAAK,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACtI,MAAM,QAAQ,GAAG,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC;IACrE,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,KAAK,IAAI,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;QACxD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC9B,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACtB,SAAS;QACb,CAAC;QACD,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QACtF,IAAI,aAAa,GAAG,QAAQ,GAAG,UAAU,EAAE,CAAC;YACxC,MAAM;QACV,CAAC;QACD,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACtB,aAAa,IAAI,QAAQ,CAAC;IAC9B,CAAC;IACD,MAAM,KAAK,GAAG,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC;IAC5C,MAAM,OAAO,GAAG,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC;IACtC,MAAM,MAAM,GAAG,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC;IAC3F,OAAO;QACH,IAAI,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,MAAM,EAAE;QACnF,QAAQ,EAAE,MAAM,CAAC,MAAM;QACvB,OAAO;QACP,KAAK;QACL,aAAa;QACb,KAAK;KACR,CAAC;AACN,CAAC"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
/** Plans and applies transcript-derived storage backfills. Verification reuses its plan result. */
|
|
2
|
+
export async function planBackfill(db, adapters, deriver) {
|
|
3
|
+
const { sessions, state } = deriver.load(db);
|
|
4
|
+
const changes = [];
|
|
5
|
+
let sessionsMissingTranscript = 0;
|
|
6
|
+
for (const session of sessions) {
|
|
7
|
+
const change = await deriver.derive({ db, adapters, session, state });
|
|
8
|
+
if (change?.transcriptMissing) {
|
|
9
|
+
sessionsMissingTranscript++;
|
|
10
|
+
}
|
|
11
|
+
if (change !== undefined) {
|
|
12
|
+
changes.push(change);
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
return { changes, sessionsScanned: sessions.length, sessionsMissingTranscript };
|
|
16
|
+
}
|
|
17
|
+
export async function applyBackfill(db, adapters, deriver) {
|
|
18
|
+
const plan = await planBackfill(db, adapters, deriver);
|
|
19
|
+
const apply = db.transaction((changes) => {
|
|
20
|
+
let sessionsSkippedConcurrent = 0;
|
|
21
|
+
for (const change of changes) {
|
|
22
|
+
if (!deriver.shouldWrite(change)) {
|
|
23
|
+
continue;
|
|
24
|
+
}
|
|
25
|
+
sessionsSkippedConcurrent += deriver.write(db, change)?.sessionsSkippedConcurrent ?? 0;
|
|
26
|
+
}
|
|
27
|
+
return sessionsSkippedConcurrent;
|
|
28
|
+
});
|
|
29
|
+
const sessionsSkippedConcurrent = apply(plan.changes);
|
|
30
|
+
deriver.recordConcurrentSkips?.(plan, sessionsSkippedConcurrent);
|
|
31
|
+
await deriver.afterApply?.({ db, adapters, plan });
|
|
32
|
+
return plan;
|
|
33
|
+
}
|
|
34
|
+
//# sourceMappingURL=backfill-runner.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"backfill-runner.js","sourceRoot":"","sources":["../../src/storage/backfill-runner.ts"],"names":[],"mappings":"AA+BA,mGAAmG;AACnG,MAAM,CAAC,KAAK,UAAU,YAAY,CAC9B,EAAY,EACZ,QAA0C,EAC1C,OAAgD;IAEhD,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC7C,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,IAAI,yBAAyB,GAAG,CAAC,CAAC;IAElC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QACtE,IAAI,MAAM,EAAE,iBAAiB,EAAE,CAAC;YAC5B,yBAAyB,EAAE,CAAC;QAChC,CAAC;QACD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzB,CAAC;IACL,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,QAAQ,CAAC,MAAM,EAAE,yBAAyB,EAAE,CAAC;AACpF,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CAC/B,EAAY,EACZ,QAA0C,EAC1C,OAAgD;IAEhD,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IACvD,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC,OAAiB,EAAE,EAAE;QAC/C,IAAI,yBAAyB,GAAG,CAAC,CAAC;QAClC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC3B,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/B,SAAS;YACb,CAAC;YACD,yBAAyB,IAAI,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,yBAAyB,IAAI,CAAC,CAAC;QAC3F,CAAC;QACD,OAAO,yBAAyB,CAAC;IACrC,CAAC,CAAC,CAAC;IACH,MAAM,yBAAyB,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtD,OAAO,CAAC,qBAAqB,EAAE,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC;IACjE,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,OAAO,IAAI,CAAC;AAChB,CAAC"}
|