claude-memory-layer 1.0.31 → 1.0.33
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +9 -2
- package/dist/cli/index.js +1110 -72
- package/dist/cli/index.js.map +4 -4
- package/dist/core/index.js +414 -25
- package/dist/core/index.js.map +2 -2
- package/dist/hooks/post-tool-use.js +416 -27
- package/dist/hooks/post-tool-use.js.map +2 -2
- package/dist/hooks/semantic-daemon.js +416 -27
- package/dist/hooks/semantic-daemon.js.map +2 -2
- package/dist/hooks/session-end.js +416 -27
- package/dist/hooks/session-end.js.map +2 -2
- package/dist/hooks/session-start.js +416 -27
- package/dist/hooks/session-start.js.map +2 -2
- package/dist/hooks/stop.js +416 -27
- package/dist/hooks/stop.js.map +2 -2
- package/dist/hooks/user-prompt-submit.js +504 -34
- package/dist/hooks/user-prompt-submit.js.map +2 -2
- package/dist/index.js +416 -27
- package/dist/index.js.map +2 -2
- package/dist/mcp/index.js +407 -32
- package/dist/mcp/index.js.map +2 -2
- package/dist/server/api/index.js +850 -44
- package/dist/server/api/index.js.map +3 -3
- package/dist/server/index.js +1073 -64
- package/dist/server/index.js.map +3 -3
- package/dist/services/memory-service.js +416 -27
- package/dist/services/memory-service.js.map +2 -2
- package/dist/ui/assets/js/bootstrap.js +2 -0
- package/dist/ui/assets/js/overview.js +166 -3
- package/dist/ui/assets/js/state.js +3 -0
- package/dist/ui/index.html +20 -0
- package/dist/ui/style.css +193 -0
- package/package.json +15 -2
- package/scripts/postinstall-embedding-backend.cjs +16 -12
- package/AGENTS.md +0 -71
- package/CLAUDE.md +0 -30
- package/HANDOFF.md +0 -92
- package/Memo.txt +0 -558
- package/benchmarks/replay/anonymized-real-sessions.json +0 -48
- package/config/kpi-thresholds.json +0 -7
- package/context.md +0 -636
- package/docs/ARCHITECTURE_COMPARISON_AND_RECOMMENDATIONS.md +0 -627
- package/docs/HERMES_MEMORY_INGESTION_ANALYSIS.md +0 -440
- package/docs/MCP_MEMORY_SERVICE_COMPARATIVE_REVIEW.md +0 -271
- package/docs/MEMORY_USEFULNESS_AUDIT.md +0 -371
- package/docs/MEMORY_USEFULNESS_AUDIT_RAW.json +0 -80
- package/docs/MEMSEARCH_PROJECT_STRUCTURE_ANALYSIS.md +0 -333
- package/docs/MEMU_ADOPTION.md +0 -40
- package/docs/OPERATIONS.md +0 -18
- package/docs/PRODUCT_VALIDATION_MATRIX.md +0 -82
- package/docs/PROJECT_STRUCTURE_ANALYSIS.md +0 -421
- package/docs/REFACTORING_MILESTONES_AND_ISSUES.md +0 -501
- package/docs/REFACTORING_PLAN_THIN_CORE.md +0 -414
- package/docs/REFERENCE_PROJECT_ANALYSES.md +0 -25
- package/docs/SUPERLOCALMEMORY_PROJECT_STRUCTURE_ANALYSIS.md +0 -452
- package/docs/TARGET_ARCHITECTURE_AND_FOLDER_STRUCTURE.md +0 -446
- package/docs/architecture/comparison-index.md +0 -47
- package/docs/reports/codex-real-data-validation-20260505T040447Z.md +0 -46
- package/plan.md +0 -1642
- package/scripts/build.ts +0 -159
- package/scripts/bump-patch-version.sh +0 -18
- package/scripts/delete-unknown-projects.js +0 -154
- package/scripts/fix-sync-gap.js +0 -32
- package/scripts/generate-session-qrels.ts +0 -126
- package/scripts/heartbeat-memory-orchestrator.sh +0 -28
- package/scripts/replay-retrieval-benchmark.ts +0 -69
- package/scripts/report-sync-gap.js +0 -26
- package/scripts/review-queue-auto-resolve.js +0 -21
- package/scripts/sync-gap-auto-heal.sh +0 -17
- package/spec.md +0 -624
- package/specs/20260207-dashboard-upgrade/context.md +0 -38
- package/specs/20260207-dashboard-upgrade/spec.md +0 -96
- package/specs/citations-system/context.md +0 -243
- package/specs/citations-system/plan.md +0 -495
- package/specs/citations-system/spec.md +0 -371
- package/specs/endless-mode/context.md +0 -305
- package/specs/endless-mode/plan.md +0 -620
- package/specs/endless-mode/spec.md +0 -455
- package/specs/entity-edge-model/context.md +0 -401
- package/specs/entity-edge-model/plan.md +0 -459
- package/specs/entity-edge-model/spec.md +0 -391
- package/specs/evidence-aligner-v2/context.md +0 -401
- package/specs/evidence-aligner-v2/plan.md +0 -303
- package/specs/evidence-aligner-v2/spec.md +0 -312
- package/specs/mcp-desktop-integration/context.md +0 -278
- package/specs/mcp-desktop-integration/plan.md +0 -550
- package/specs/mcp-desktop-integration/spec.md +0 -494
- package/specs/memory-utilization-improvements/context.md +0 -145
- package/specs/memory-utilization-improvements/plan.md +0 -361
- package/specs/memory-utilization-improvements/spec.md +0 -361
- package/specs/post-tool-use-hook/context.md +0 -319
- package/specs/post-tool-use-hook/plan.md +0 -469
- package/specs/post-tool-use-hook/spec.md +0 -364
- package/specs/private-tags/context.md +0 -288
- package/specs/private-tags/plan.md +0 -412
- package/specs/private-tags/spec.md +0 -345
- package/specs/progressive-disclosure/context.md +0 -346
- package/specs/progressive-disclosure/plan.md +0 -663
- package/specs/progressive-disclosure/spec.md +0 -415
- package/specs/selective-tool-observation/context.md +0 -100
- package/specs/selective-tool-observation/plan.md +0 -158
- package/specs/selective-tool-observation/spec.md +0 -127
- package/specs/task-entity-system/context.md +0 -297
- package/specs/task-entity-system/plan.md +0 -301
- package/specs/task-entity-system/spec.md +0 -314
- package/specs/thin-core-refactor/context.md +0 -275
- package/specs/thin-core-refactor/plan.md +0 -536
- package/specs/thin-core-refactor/spec.md +0 -465
- package/specs/vector-outbox-v2/context.md +0 -470
- package/specs/vector-outbox-v2/plan.md +0 -562
- package/specs/vector-outbox-v2/spec.md +0 -466
- package/specs/web-viewer-ui/context.md +0 -384
- package/specs/web-viewer-ui/plan.md +0 -797
- package/specs/web-viewer-ui/spec.md +0 -516
- package/src/adapters/claude/capture/index.ts +0 -3
- package/src/adapters/claude/context/index.ts +0 -3
- package/src/adapters/claude/hooks/index.ts +0 -21
- package/src/adapters/claude/hooks/post-tool-use.ts +0 -239
- package/src/adapters/claude/hooks/prompt-injection-policy.ts +0 -104
- package/src/adapters/claude/hooks/semantic-daemon-client.ts +0 -209
- package/src/adapters/claude/hooks/semantic-daemon.ts +0 -283
- package/src/adapters/claude/hooks/session-end.ts +0 -59
- package/src/adapters/claude/hooks/session-start.ts +0 -73
- package/src/adapters/claude/hooks/stop.ts +0 -128
- package/src/adapters/claude/hooks/user-prompt-submit.ts +0 -361
- package/src/adapters/claude/index.ts +0 -4
- package/src/adapters/claude/transcript/index.ts +0 -4
- package/src/adapters/claude/transcript/transcript-reader.ts +0 -57
- package/src/adapters/claude/transcript/turn-reconstructor.ts +0 -65
- package/src/apps/cli/claude-settings-hooks.ts +0 -138
- package/src/apps/cli/codex-import-runner.ts +0 -125
- package/src/apps/cli/codex-validation-output.ts +0 -95
- package/src/apps/cli/hermes-import-runner.ts +0 -130
- package/src/apps/cli/hermes-validation-output.ts +0 -91
- package/src/apps/cli/index.ts +0 -1735
- package/src/apps/cli/mcp-install.ts +0 -106
- package/src/apps/cli/retrieval-disclosure-output.ts +0 -196
- package/src/apps/dashboard/assets/js/bootstrap.js +0 -244
- package/src/apps/dashboard/assets/js/chat.js +0 -373
- package/src/apps/dashboard/assets/js/disclosure.js +0 -232
- package/src/apps/dashboard/assets/js/modals.js +0 -298
- package/src/apps/dashboard/assets/js/overview.js +0 -655
- package/src/apps/dashboard/assets/js/state.js +0 -72
- package/src/apps/dashboard/assets/js/views.js +0 -468
- package/src/apps/dashboard/index.html +0 -543
- package/src/apps/dashboard/index.ts +0 -3
- package/src/apps/dashboard/style.css +0 -1750
- package/src/apps/index.ts +0 -5
- package/src/apps/server/api/chat.ts +0 -244
- package/src/apps/server/api/citations.ts +0 -105
- package/src/apps/server/api/events.ts +0 -137
- package/src/apps/server/api/health.ts +0 -53
- package/src/apps/server/api/index.ts +0 -26
- package/src/apps/server/api/projects.ts +0 -74
- package/src/apps/server/api/search.ts +0 -184
- package/src/apps/server/api/sessions.ts +0 -115
- package/src/apps/server/api/stats.ts +0 -723
- package/src/apps/server/api/turns.ts +0 -143
- package/src/apps/server/api/utils.ts +0 -65
- package/src/apps/server/index.ts +0 -111
- package/src/cli/index.ts +0 -3
- package/src/cli/retrieval-disclosure-output.ts +0 -2
- package/src/compat/index.ts +0 -5
- package/src/core/canonical-key.ts +0 -186
- package/src/core/citation-generator.ts +0 -63
- package/src/core/consolidated-store.ts +0 -356
- package/src/core/consolidation-worker.ts +0 -493
- package/src/core/context-formatter.ts +0 -276
- package/src/core/continuity-manager.ts +0 -341
- package/src/core/db-wrapper.ts +0 -64
- package/src/core/derive/fact-deriver.ts +0 -170
- package/src/core/derive/index.ts +0 -2
- package/src/core/derive/summary-deriver.ts +0 -76
- package/src/core/edge-repo.ts +0 -333
- package/src/core/embedder.ts +0 -4
- package/src/core/engine/embedding-maintenance-service.ts +0 -187
- package/src/core/engine/endless-memory-services.ts +0 -4
- package/src/core/engine/index.ts +0 -19
- package/src/core/engine/memory-engine-services.ts +0 -170
- package/src/core/engine/memory-ingest-service.ts +0 -317
- package/src/core/engine/memory-query-service.ts +0 -173
- package/src/core/engine/memory-runtime-service.ts +0 -162
- package/src/core/engine/memory-service-composition.ts +0 -231
- package/src/core/engine/retrieval-analytics-service.ts +0 -181
- package/src/core/engine/retrieval-disclosure-service.ts +0 -420
- package/src/core/engine/retrieval-orchestrator.ts +0 -377
- package/src/core/engine/retrieval-services.ts +0 -176
- package/src/core/engine/shared-memory-services.ts +0 -4
- package/src/core/entity-repo.ts +0 -349
- package/src/core/event-store.ts +0 -779
- package/src/core/evidence-aligner.ts +0 -635
- package/src/core/external-market-context.ts +0 -582
- package/src/core/graduation-worker.ts +0 -171
- package/src/core/graduation.ts +0 -377
- package/src/core/index.ts +0 -64
- package/src/core/ingest-interceptor.ts +0 -80
- package/src/core/markdown-mirror.ts +0 -70
- package/src/core/matcher.ts +0 -208
- package/src/core/md-mirror.ts +0 -92
- package/src/core/metadata-extractor.ts +0 -203
- package/src/core/model/memory-fact.ts +0 -30
- package/src/core/model/memory-rule.ts +0 -14
- package/src/core/model/memory-summary.ts +0 -21
- package/src/core/model/raw-event.ts +0 -28
- package/src/core/model/retrieval-result.ts +0 -35
- package/src/core/mongo-sync-config.ts +0 -165
- package/src/core/mongo-sync-worker.ts +0 -381
- package/src/core/privacy/filter.ts +0 -190
- package/src/core/privacy/index.ts +0 -20
- package/src/core/privacy/tag-parser.ts +0 -145
- package/src/core/product-validation-matrix.ts +0 -314
- package/src/core/progressive-retriever.ts +0 -414
- package/src/core/registry/project-path.ts +0 -54
- package/src/core/registry/session-registry.ts +0 -69
- package/src/core/replay-evaluator.ts +0 -625
- package/src/core/retrieval-benchmark.ts +0 -117
- package/src/core/retrieval-quality.ts +0 -109
- package/src/core/retriever.ts +0 -800
- package/src/core/session-qrels.ts +0 -360
- package/src/core/shared-event-store.ts +0 -114
- package/src/core/shared-promoter.ts +0 -249
- package/src/core/shared-store.ts +0 -289
- package/src/core/shared-vector-store.ts +0 -203
- package/src/core/sqlite-event-store.ts +0 -1846
- package/src/core/sqlite-wrapper.ts +0 -116
- package/src/core/sync-worker.ts +0 -228
- package/src/core/tag-taxonomy.ts +0 -51
- package/src/core/task/blocker-resolver.ts +0 -333
- package/src/core/task/index.ts +0 -9
- package/src/core/task/task-matcher.ts +0 -240
- package/src/core/task/task-projector.ts +0 -358
- package/src/core/task/task-resolver.ts +0 -421
- package/src/core/turn-state.ts +0 -207
- package/src/core/types.ts +0 -952
- package/src/core/vector-outbox.ts +0 -299
- package/src/core/vector-store.ts +0 -231
- package/src/core/vector-worker.ts +0 -521
- package/src/core/working-set-store.ts +0 -257
- package/src/extensions/endless-memory/endless-memory-services.ts +0 -350
- package/src/extensions/endless-memory/index.ts +0 -1
- package/src/extensions/index.ts +0 -5
- package/src/extensions/mcp/handlers.ts +0 -960
- package/src/extensions/mcp/index.ts +0 -48
- package/src/extensions/mcp/tools.ts +0 -252
- package/src/extensions/shared-memory/index.ts +0 -1
- package/src/extensions/shared-memory/shared-memory-services.ts +0 -211
- package/src/extensions/vector/embedder.ts +0 -234
- package/src/extensions/vector/index.ts +0 -1
- package/src/hooks/post-tool-use.ts +0 -9
- package/src/hooks/semantic-daemon-client.ts +0 -1
- package/src/hooks/semantic-daemon.ts +0 -11
- package/src/hooks/session-end.ts +0 -9
- package/src/hooks/session-start.ts +0 -9
- package/src/hooks/stop.ts +0 -9
- package/src/hooks/user-prompt-submit.ts +0 -9
- package/src/index.ts +0 -13
- package/src/mcp/handlers.ts +0 -2
- package/src/mcp/index.ts +0 -4
- package/src/mcp/tools.ts +0 -2
- package/src/server/api/chat.ts +0 -2
- package/src/server/api/citations.ts +0 -2
- package/src/server/api/events.ts +0 -2
- package/src/server/api/health.ts +0 -2
- package/src/server/api/index.ts +0 -2
- package/src/server/api/projects.ts +0 -2
- package/src/server/api/search.ts +0 -2
- package/src/server/api/sessions.ts +0 -2
- package/src/server/api/stats.ts +0 -2
- package/src/server/api/turns.ts +0 -2
- package/src/server/api/utils.ts +0 -2
- package/src/server/index.ts +0 -2
- package/src/services/bootstrap-organizer.ts +0 -463
- package/src/services/codex-session-history-importer.ts +0 -966
- package/src/services/hermes-session-history-importer.ts +0 -733
- package/src/services/memory-service-config.ts +0 -36
- package/src/services/memory-service-registry.ts +0 -150
- package/src/services/memory-service.ts +0 -688
- package/src/services/session-history-importer.ts +0 -629
- package/tests/README.md +0 -23
- package/tests/adapters/claude/claude-semantic-daemon-adapter.test.ts +0 -54
- package/tests/adapters/claude/claude-transcript-reconstructor.test.ts +0 -98
- package/tests/adapters/claude-hook-prompt-injection-policy.test.ts +0 -99
- package/tests/apps/app-layer-boundary.test.ts +0 -48
- package/tests/apps/claude-settings-hooks.test.ts +0 -107
- package/tests/apps/cli-disclosure-output.test.ts +0 -212
- package/tests/apps/codex-import-runner.test.ts +0 -99
- package/tests/apps/codex-validation-output.test.ts +0 -100
- package/tests/apps/hermes-import-runner.test.ts +0 -99
- package/tests/apps/mcp-install-command.test.ts +0 -59
- package/tests/apps/package-build-entrypoints.test.ts +0 -30
- package/tests/apps/postinstall-embedding-backend.test.ts +0 -185
- package/tests/apps/search-api-disclosure.test.ts +0 -162
- package/tests/apps/stats-api-lightweight.test.ts +0 -67
- package/tests/apps/ui-disclosure-output.test.ts +0 -140
- package/tests/core/bootstrap-organizer.test.ts +0 -111
- package/tests/core/canonical-key.test.ts +0 -101
- package/tests/core/codex-session-history-importer-validation.test.ts +0 -185
- package/tests/core/consolidation-worker.test.ts +0 -75
- package/tests/core/embedding-maintenance-service.test.ts +0 -282
- package/tests/core/evidence-aligner.test.ts +0 -152
- package/tests/core/external-market-context.test.ts +0 -209
- package/tests/core/fact-deriver.test.ts +0 -79
- package/tests/core/hermes-session-history-importer-validation.test.ts +0 -609
- package/tests/core/ingest-interceptor.test.ts +0 -38
- package/tests/core/markdown-mirror.test.ts +0 -85
- package/tests/core/matcher.test.ts +0 -112
- package/tests/core/md-mirror.test.ts +0 -50
- package/tests/core/memory-engine-services.test.ts +0 -240
- package/tests/core/memory-ingest-service.test.ts +0 -296
- package/tests/core/memory-query-service.test.ts +0 -129
- package/tests/core/memory-runtime-service.test.ts +0 -201
- package/tests/core/memory-service-composition.test.ts +0 -192
- package/tests/core/memory-service-config.test.ts +0 -41
- package/tests/core/memory-service-facade.test.ts +0 -30
- package/tests/core/memory-service-registry.test.ts +0 -206
- package/tests/core/product-validation-matrix.test.ts +0 -61
- package/tests/core/project-registry.test.ts +0 -78
- package/tests/core/replay-evaluator.test.ts +0 -181
- package/tests/core/retrieval-analytics-service.test.ts +0 -210
- package/tests/core/retrieval-benchmark.test.ts +0 -93
- package/tests/core/retrieval-disclosure-service.test.ts +0 -264
- package/tests/core/retrieval-orchestrator.test.ts +0 -403
- package/tests/core/retrieval-quality.test.ts +0 -31
- package/tests/core/retrieval-services.test.ts +0 -185
- package/tests/core/retriever-fallback-chain.test.ts +0 -223
- package/tests/core/retriever-strategy-scope.test.ts +0 -164
- package/tests/core/retriever.memu-adoption.test.ts +0 -122
- package/tests/core/session-history-importer-filter.test.ts +0 -78
- package/tests/core/session-qrels.test.ts +0 -250
- package/tests/core/sqlite-event-store-replication.test.ts +0 -127
- package/tests/core/summary-deriver.test.ts +0 -66
- package/tests/extensions/embedder-warning-suppression.test.ts +0 -84
- package/tests/extensions/endless-memory-extension-boundary.test.ts +0 -17
- package/tests/extensions/endless-memory-services.test.ts +0 -325
- package/tests/extensions/mcp-context-tools.test.ts +0 -905
- package/tests/extensions/mcp-extension-boundary.test.ts +0 -21
- package/tests/extensions/mcp-package-build.test.ts +0 -22
- package/tests/extensions/mcp-project-aware-tools.test.ts +0 -102
- package/tests/extensions/shared-memory-extension-boundary.test.ts +0 -24
- package/tests/extensions/shared-memory-services.test.ts +0 -309
- package/tests/extensions/vector-extension-boundary.test.ts +0 -21
- package/tsconfig.json +0 -24
- package/vitest.config.ts +0 -15
|
@@ -1,264 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect } from 'vitest';
|
|
2
|
-
import {
|
|
3
|
-
parseDisclosureResultId,
|
|
4
|
-
parseDisclosureResultRef,
|
|
5
|
-
RetrievalDisclosureService
|
|
6
|
-
} from '../../src/core/engine/retrieval-disclosure-service.js';
|
|
7
|
-
import type { UnifiedRetrievalResult } from '../../src/core/retriever.js';
|
|
8
|
-
import type { MemoryEvent, SharedTroubleshootingEntry } from '../../src/core/types.js';
|
|
9
|
-
|
|
10
|
-
function event(id: string, content: string, timestamp: string, eventType: MemoryEvent['eventType'] = 'user_prompt'): MemoryEvent {
|
|
11
|
-
return {
|
|
12
|
-
id,
|
|
13
|
-
sessionId: 's1',
|
|
14
|
-
eventType,
|
|
15
|
-
content,
|
|
16
|
-
canonicalKey: `canonical/${id}`,
|
|
17
|
-
dedupeKey: `s1:${id}`,
|
|
18
|
-
timestamp: new Date(timestamp),
|
|
19
|
-
metadata: { filePath: `src/${id}.ts` }
|
|
20
|
-
};
|
|
21
|
-
}
|
|
22
|
-
|
|
23
|
-
const timeline = [
|
|
24
|
-
event('e1', 'Earlier user question about checkout memory', '2026-02-24T00:00:00.000Z'),
|
|
25
|
-
event('e2', 'Target assistant answer with checkout fix details', '2026-02-24T00:01:00.000Z', 'agent_response'),
|
|
26
|
-
event('e3', 'Later tool observation after applying the checkout fix', '2026-02-24T00:02:00.000Z', 'tool_observation')
|
|
27
|
-
];
|
|
28
|
-
|
|
29
|
-
const sharedEntry: SharedTroubleshootingEntry = {
|
|
30
|
-
entryId: 'shared-1',
|
|
31
|
-
sourceProjectHash: 'project-a',
|
|
32
|
-
sourceEntryId: 'e-shared',
|
|
33
|
-
title: 'Shared checkout troubleshooting',
|
|
34
|
-
symptoms: ['checkout fails'],
|
|
35
|
-
rootCause: 'stale cache',
|
|
36
|
-
solution: 'clear cache and retry',
|
|
37
|
-
topics: ['checkout'],
|
|
38
|
-
confidence: 0.88,
|
|
39
|
-
usageCount: 3,
|
|
40
|
-
promotedAt: new Date('2026-02-23T00:00:00.000Z'),
|
|
41
|
-
createdAt: new Date('2026-02-23T00:00:00.000Z')
|
|
42
|
-
};
|
|
43
|
-
|
|
44
|
-
function retrievalResult(sharedMemories: SharedTroubleshootingEntry[] = []): UnifiedRetrievalResult {
|
|
45
|
-
return {
|
|
46
|
-
memories: [{ event: timeline[1], score: 0.93, sessionContext: 'short surrounding context' }],
|
|
47
|
-
matchResult: {
|
|
48
|
-
match: { event: timeline[1], score: 0.93 },
|
|
49
|
-
confidence: 'high'
|
|
50
|
-
},
|
|
51
|
-
totalTokens: 42,
|
|
52
|
-
context: 'Target assistant answer with checkout fix details',
|
|
53
|
-
fallbackTrace: ['stage:primary:fast', 'fallback:deep'],
|
|
54
|
-
selectedDebug: [{ eventId: 'e2', score: 0.93, semanticScore: 0.82, lexicalScore: 0.41, recencyScore: 0.12 }],
|
|
55
|
-
candidateDebug: [{ eventId: 'e2', score: 0.93, semanticScore: 0.82, lexicalScore: 0.41, recencyScore: 0.12 }],
|
|
56
|
-
sharedMemories
|
|
57
|
-
};
|
|
58
|
-
}
|
|
59
|
-
|
|
60
|
-
function service(
|
|
61
|
-
result: UnifiedRetrievalResult = retrievalResult(),
|
|
62
|
-
overrides: {
|
|
63
|
-
initialize?: () => Promise<void>;
|
|
64
|
-
getEvent?: (id: string) => Promise<MemoryEvent | null>;
|
|
65
|
-
getSessionEvents?: (sessionId: string) => Promise<MemoryEvent[]>;
|
|
66
|
-
sharedStore?: { get(entryId: string): Promise<SharedTroubleshootingEntry | null> };
|
|
67
|
-
} = {}
|
|
68
|
-
) {
|
|
69
|
-
return new RetrievalDisclosureService({
|
|
70
|
-
initialize: overrides.initialize ?? (async () => {}),
|
|
71
|
-
retrievalOrchestrator: {
|
|
72
|
-
retrieveMemories: async () => result
|
|
73
|
-
},
|
|
74
|
-
eventStore: {
|
|
75
|
-
getEvent: overrides.getEvent ?? (async (id: string) => timeline.find((item) => item.id === id) ?? null),
|
|
76
|
-
getSessionEvents: overrides.getSessionEvents ?? (async (sessionId: string) => timeline.filter((item) => item.sessionId === sessionId))
|
|
77
|
-
},
|
|
78
|
-
sharedStore: overrides.sharedStore
|
|
79
|
-
});
|
|
80
|
-
}
|
|
81
|
-
|
|
82
|
-
describe('RetrievalDisclosureService', () => {
|
|
83
|
-
it('keeps the legacy event-id parser while exposing a discriminated result-ref parser', () => {
|
|
84
|
-
expect(parseDisclosureResultId('event:e2')).toBe('e2');
|
|
85
|
-
expect(parseDisclosureResultId('e2')).toBe('e2');
|
|
86
|
-
expect(parseDisclosureResultId('shared:shared-1')).toBe('shared:shared-1');
|
|
87
|
-
expect(parseDisclosureResultRef('event:e2')).toEqual({ kind: 'event', eventId: 'e2' });
|
|
88
|
-
expect(parseDisclosureResultRef('shared:shared-1')).toEqual({ kind: 'shared', entryId: 'shared-1' });
|
|
89
|
-
});
|
|
90
|
-
|
|
91
|
-
it('search returns spec-aligned compact envelopes with reasons and source refs', async () => {
|
|
92
|
-
const out = await service().search('checkout fix', { topK: 1, sessionId: 's1' });
|
|
93
|
-
|
|
94
|
-
expect(out.meta).toMatchObject({
|
|
95
|
-
total: 1,
|
|
96
|
-
usedVector: true,
|
|
97
|
-
usedKeyword: true,
|
|
98
|
-
fallbackApplied: true
|
|
99
|
-
});
|
|
100
|
-
expect(out.results[0]).toMatchObject({
|
|
101
|
-
id: 'event:e2',
|
|
102
|
-
resultType: 'source',
|
|
103
|
-
score: 0.93,
|
|
104
|
-
snippet: 'Target assistant answer with checkout fix details',
|
|
105
|
-
sourceRef: 'event:e2',
|
|
106
|
-
sessionId: 's1',
|
|
107
|
-
metadata: {
|
|
108
|
-
eventId: 'e2',
|
|
109
|
-
eventType: 'agent_response',
|
|
110
|
-
canonicalKey: 'canonical/e2'
|
|
111
|
-
}
|
|
112
|
-
});
|
|
113
|
-
expect(out.results[0].reasons).toEqual(
|
|
114
|
-
expect.arrayContaining([
|
|
115
|
-
'semantic_match',
|
|
116
|
-
'keyword_match',
|
|
117
|
-
'recent_relevance',
|
|
118
|
-
'continuity_link'
|
|
119
|
-
])
|
|
120
|
-
);
|
|
121
|
-
});
|
|
122
|
-
|
|
123
|
-
it('reports keyword-only fast retrieval without claiming vector usage', async () => {
|
|
124
|
-
const keywordOnly = {
|
|
125
|
-
...retrievalResult(),
|
|
126
|
-
fallbackTrace: ['stage:primary:fast']
|
|
127
|
-
};
|
|
128
|
-
|
|
129
|
-
const out = await service(keywordOnly).search('checkout fix', { strategy: 'fast' });
|
|
130
|
-
|
|
131
|
-
expect(out.meta).toMatchObject({
|
|
132
|
-
usedVector: false,
|
|
133
|
-
usedKeyword: true,
|
|
134
|
-
fallbackApplied: false
|
|
135
|
-
});
|
|
136
|
-
expect(out.results[0].reasons).toContain('keyword_match');
|
|
137
|
-
expect(out.results[0].reasons).not.toContain('semantic_match');
|
|
138
|
-
expect(out.results[0].reasons).not.toContain('summary_fallback');
|
|
139
|
-
});
|
|
140
|
-
|
|
141
|
-
it('search includes shared memories instead of silently dropping them', async () => {
|
|
142
|
-
const out = await service(retrievalResult([sharedEntry])).search('checkout fix', { includeShared: true });
|
|
143
|
-
|
|
144
|
-
expect(out.meta.total).toBe(2);
|
|
145
|
-
expect(out.results.find((item) => item.id === 'shared:shared-1')).toMatchObject({
|
|
146
|
-
resultType: 'rule',
|
|
147
|
-
title: 'Shared checkout troubleshooting',
|
|
148
|
-
snippet: 'clear cache and retry',
|
|
149
|
-
sourceRef: 'shared:shared-1',
|
|
150
|
-
reasons: ['semantic_match'],
|
|
151
|
-
metadata: {
|
|
152
|
-
sourceProjectHash: 'project-a',
|
|
153
|
-
sourceEntryId: 'e-shared'
|
|
154
|
-
}
|
|
155
|
-
});
|
|
156
|
-
});
|
|
157
|
-
|
|
158
|
-
it('expand returns target envelope, surrounding context, and related source references', async () => {
|
|
159
|
-
const expanded = await service().expand('event:e2', { windowSize: 1 });
|
|
160
|
-
|
|
161
|
-
expect(expanded?.target.id).toBe('event:e2');
|
|
162
|
-
expect(expanded?.target.snippet).toContain('checkout fix details');
|
|
163
|
-
expect(expanded?.surroundingFacts?.map((item) => item.id)).toEqual(['event:e1', 'event:e3']);
|
|
164
|
-
expect(expanded?.relatedSources?.[0]).toMatchObject({
|
|
165
|
-
sourceRef: 'event:e1',
|
|
166
|
-
sourceType: 'raw_event',
|
|
167
|
-
eventIds: ['e1']
|
|
168
|
-
});
|
|
169
|
-
expect(expanded?.expandedContext).toContain('[agent_response] Target assistant answer');
|
|
170
|
-
});
|
|
171
|
-
|
|
172
|
-
it('source returns a spec-aligned source reference plus raw drill-down events', async () => {
|
|
173
|
-
const source = await service().source('event:e2');
|
|
174
|
-
|
|
175
|
-
expect(source).toMatchObject({
|
|
176
|
-
sourceRef: 'event:e2',
|
|
177
|
-
sourceType: 'raw_event',
|
|
178
|
-
eventIds: ['e2']
|
|
179
|
-
});
|
|
180
|
-
expect(source?.rawEvents[0].id).toBe('e2');
|
|
181
|
-
expect(source?.rawEvents[0].metadata).toEqual({ filePath: 'src/e2.ts' });
|
|
182
|
-
});
|
|
183
|
-
|
|
184
|
-
it('expand and source keep event drill-down lightweight without full retrieval initialization', async () => {
|
|
185
|
-
let initializeCalls = 0;
|
|
186
|
-
const disclosure = service(retrievalResult(), {
|
|
187
|
-
initialize: async () => {
|
|
188
|
-
initializeCalls += 1;
|
|
189
|
-
throw new Error('full initialization should not be called for drill-down reads');
|
|
190
|
-
}
|
|
191
|
-
});
|
|
192
|
-
|
|
193
|
-
await expect(disclosure.expand('event:e2')).resolves.toMatchObject({
|
|
194
|
-
target: { id: 'event:e2' }
|
|
195
|
-
});
|
|
196
|
-
await expect(disclosure.source('event:e2')).resolves.toMatchObject({
|
|
197
|
-
sourceRef: 'event:e2'
|
|
198
|
-
});
|
|
199
|
-
expect(initializeCalls).toBe(0);
|
|
200
|
-
});
|
|
201
|
-
|
|
202
|
-
it('expands shared disclosure results through the shared store without fake local events', async () => {
|
|
203
|
-
const lookedUpEventIds: string[] = [];
|
|
204
|
-
const expanded = await service(retrievalResult([sharedEntry]), {
|
|
205
|
-
getEvent: async (id: string) => {
|
|
206
|
-
lookedUpEventIds.push(id);
|
|
207
|
-
return timeline.find((item) => item.id === id) ?? null;
|
|
208
|
-
},
|
|
209
|
-
sharedStore: { get: async (entryId: string) => entryId === sharedEntry.entryId ? sharedEntry : null }
|
|
210
|
-
}).expand('shared:shared-1');
|
|
211
|
-
|
|
212
|
-
expect(expanded?.target).toMatchObject({
|
|
213
|
-
id: 'shared:shared-1',
|
|
214
|
-
resultType: 'rule',
|
|
215
|
-
title: 'Shared checkout troubleshooting',
|
|
216
|
-
metadata: {
|
|
217
|
-
sourceProjectHash: 'project-a',
|
|
218
|
-
sourceEntryId: 'e-shared'
|
|
219
|
-
}
|
|
220
|
-
});
|
|
221
|
-
expect(expanded?.relatedSources).toEqual([
|
|
222
|
-
{
|
|
223
|
-
sourceRef: 'shared:shared-1',
|
|
224
|
-
sourceType: 'shared_troubleshooting',
|
|
225
|
-
eventIds: [],
|
|
226
|
-
metadata: {
|
|
227
|
-
sourceProjectHash: 'project-a',
|
|
228
|
-
sourceEntryId: 'e-shared',
|
|
229
|
-
topics: ['checkout']
|
|
230
|
-
}
|
|
231
|
-
}
|
|
232
|
-
]);
|
|
233
|
-
expect(expanded?.expandedContext).toContain('Root cause: stale cache');
|
|
234
|
-
expect(lookedUpEventIds).toEqual([]);
|
|
235
|
-
});
|
|
236
|
-
|
|
237
|
-
it('resolves shared disclosure sources without pretending they are local raw events', async () => {
|
|
238
|
-
const source = await service(retrievalResult([sharedEntry]), {
|
|
239
|
-
sharedStore: { get: async (entryId: string) => entryId === sharedEntry.entryId ? sharedEntry : null }
|
|
240
|
-
}).source('shared:shared-1');
|
|
241
|
-
|
|
242
|
-
expect(source).toMatchObject({
|
|
243
|
-
sourceRef: 'shared:shared-1',
|
|
244
|
-
sourceType: 'shared_troubleshooting',
|
|
245
|
-
eventIds: [],
|
|
246
|
-
rawEvents: [],
|
|
247
|
-
metadata: {
|
|
248
|
-
sourceProjectHash: 'project-a',
|
|
249
|
-
sourceEntryId: 'e-shared',
|
|
250
|
-
topics: ['checkout'],
|
|
251
|
-
rootCause: 'stale cache',
|
|
252
|
-
solution: 'clear cache and retry'
|
|
253
|
-
}
|
|
254
|
-
});
|
|
255
|
-
expect(source?.primaryEvent).toBeUndefined();
|
|
256
|
-
});
|
|
257
|
-
|
|
258
|
-
it('returns null when expand/source cannot resolve the result id', async () => {
|
|
259
|
-
await expect(service().expand('event:missing')).resolves.toBeNull();
|
|
260
|
-
await expect(service().source('event:missing')).resolves.toBeNull();
|
|
261
|
-
await expect(service().expand('shared:missing')).resolves.toBeNull();
|
|
262
|
-
await expect(service().source('shared:missing')).resolves.toBeNull();
|
|
263
|
-
});
|
|
264
|
-
});
|
|
@@ -1,403 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, afterEach } from 'vitest';
|
|
2
|
-
import { RetrievalOrchestrator, type RetrievalAccessStore } from '../../src/core/engine/retrieval-orchestrator.js';
|
|
3
|
-
import type { Retriever, UnifiedRetrievalResult } from '../../src/core/retriever.js';
|
|
4
|
-
import type { MemoryEvent } from '../../src/core/types.js';
|
|
5
|
-
|
|
6
|
-
function event(id: string): MemoryEvent {
|
|
7
|
-
return {
|
|
8
|
-
id,
|
|
9
|
-
sessionId: 's1',
|
|
10
|
-
eventType: 'user_prompt',
|
|
11
|
-
content: 'remembered project context',
|
|
12
|
-
canonicalKey: `test/${id}`,
|
|
13
|
-
dedupeKey: `s1:${id}`,
|
|
14
|
-
timestamp: new Date('2026-02-24T00:00:00.000Z'),
|
|
15
|
-
metadata: {}
|
|
16
|
-
};
|
|
17
|
-
}
|
|
18
|
-
|
|
19
|
-
function retrievalResult(id = 'e1'): UnifiedRetrievalResult {
|
|
20
|
-
const memoryEvent = event(id);
|
|
21
|
-
return {
|
|
22
|
-
memories: [{ event: memoryEvent, score: 0.91 }],
|
|
23
|
-
matchResult: {
|
|
24
|
-
match: { event: memoryEvent, score: 0.91 },
|
|
25
|
-
confidence: 'high'
|
|
26
|
-
},
|
|
27
|
-
totalTokens: 12,
|
|
28
|
-
context: 'remembered project context',
|
|
29
|
-
fallbackTrace: ['stage:primary:fast'],
|
|
30
|
-
selectedDebug: [{ eventId: id, score: 0.91, semanticScore: 0.8, lexicalScore: 0.2, recencyScore: 0.1 }],
|
|
31
|
-
candidateDebug: [{ eventId: id, score: 0.91, semanticScore: 0.8, lexicalScore: 0.2, recencyScore: 0.1 }]
|
|
32
|
-
};
|
|
33
|
-
}
|
|
34
|
-
|
|
35
|
-
function stats() {
|
|
36
|
-
return {
|
|
37
|
-
avgScore: 0.8,
|
|
38
|
-
totalEvaluated: 30,
|
|
39
|
-
totalRetrievals: 40,
|
|
40
|
-
helpful: 20,
|
|
41
|
-
neutral: 5,
|
|
42
|
-
unhelpful: 5
|
|
43
|
-
};
|
|
44
|
-
}
|
|
45
|
-
|
|
46
|
-
function noopAccessStore(): RetrievalAccessStore {
|
|
47
|
-
return {
|
|
48
|
-
incrementAccessCount: async (_eventIds) => {},
|
|
49
|
-
recordRetrieval: async (_eventId, _sessionId, _score, _query) => {}
|
|
50
|
-
};
|
|
51
|
-
}
|
|
52
|
-
|
|
53
|
-
afterEach(() => {
|
|
54
|
-
delete process.env.MEMORY_RERANK_WEIGHT_SEMANTIC;
|
|
55
|
-
delete process.env.MEMORY_RERANK_WEIGHT_LEXICAL;
|
|
56
|
-
delete process.env.MEMORY_RERANK_WEIGHT_RECENCY;
|
|
57
|
-
});
|
|
58
|
-
|
|
59
|
-
describe('RetrievalOrchestrator', () => {
|
|
60
|
-
it('delegates project retrieval with strict project scope and records a trace', async () => {
|
|
61
|
-
let initialized = 0;
|
|
62
|
-
let retrieveArgs: { query: string; options: Record<string, unknown> } | null = null;
|
|
63
|
-
const traces: Array<Record<string, unknown>> = [];
|
|
64
|
-
|
|
65
|
-
const fakeRetriever = {
|
|
66
|
-
setQueryRewriter() {},
|
|
67
|
-
async retrieve(query: string, options: Record<string, unknown>) {
|
|
68
|
-
retrieveArgs = { query, options };
|
|
69
|
-
return retrievalResult('e1');
|
|
70
|
-
}
|
|
71
|
-
} as unknown as Retriever;
|
|
72
|
-
|
|
73
|
-
const orchestrator = new RetrievalOrchestrator({
|
|
74
|
-
initialize: async () => { initialized += 1; },
|
|
75
|
-
retriever: fakeRetriever,
|
|
76
|
-
traceStore: {
|
|
77
|
-
getHelpfulnessStats: async () => stats(),
|
|
78
|
-
recordRetrievalTrace: async (input) => { traces.push(input); }
|
|
79
|
-
},
|
|
80
|
-
accessStore: noopAccessStore(),
|
|
81
|
-
getProjectHash: () => 'project-1',
|
|
82
|
-
hasSharedStore: () => false
|
|
83
|
-
});
|
|
84
|
-
|
|
85
|
-
const out = await orchestrator.retrieveMemories('project query', { topK: 3, sessionId: 's1' });
|
|
86
|
-
|
|
87
|
-
expect(out.memories[0]?.event.id).toBe('e1');
|
|
88
|
-
expect(initialized).toBe(1);
|
|
89
|
-
expect(retrieveArgs?.query).toBe('project query');
|
|
90
|
-
expect(retrieveArgs?.options.projectHash).toBe('project-1');
|
|
91
|
-
expect(retrieveArgs?.options.projectScopeMode).toBe('strict');
|
|
92
|
-
expect(traces[0]).toMatchObject({
|
|
93
|
-
sessionId: 's1',
|
|
94
|
-
projectHash: 'project-1',
|
|
95
|
-
queryText: 'project query',
|
|
96
|
-
strategy: 'auto',
|
|
97
|
-
candidateEventIds: ['e1'],
|
|
98
|
-
selectedEventIds: ['e1'],
|
|
99
|
-
confidence: 'high',
|
|
100
|
-
fallbackTrace: ['stage:primary:fast']
|
|
101
|
-
});
|
|
102
|
-
});
|
|
103
|
-
|
|
104
|
-
it('can skip automatic trace recording for read-only retrieval callers', async () => {
|
|
105
|
-
const traces: Array<Record<string, unknown>> = [];
|
|
106
|
-
const fakeRetriever = {
|
|
107
|
-
setQueryRewriter() {},
|
|
108
|
-
async retrieve() {
|
|
109
|
-
return retrievalResult('readonly-e1');
|
|
110
|
-
}
|
|
111
|
-
} as unknown as Retriever;
|
|
112
|
-
|
|
113
|
-
const orchestrator = new RetrievalOrchestrator({
|
|
114
|
-
initialize: async () => {},
|
|
115
|
-
retriever: fakeRetriever,
|
|
116
|
-
traceStore: {
|
|
117
|
-
getHelpfulnessStats: async () => stats(),
|
|
118
|
-
recordRetrievalTrace: async (input) => { traces.push(input); }
|
|
119
|
-
},
|
|
120
|
-
accessStore: noopAccessStore(),
|
|
121
|
-
getProjectHash: () => 'project-readonly',
|
|
122
|
-
hasSharedStore: () => false
|
|
123
|
-
});
|
|
124
|
-
|
|
125
|
-
await orchestrator.retrieveMemories('secret-bearing read-only query', {
|
|
126
|
-
topK: 3,
|
|
127
|
-
recordTrace: false
|
|
128
|
-
});
|
|
129
|
-
|
|
130
|
-
expect(traces).toEqual([]);
|
|
131
|
-
});
|
|
132
|
-
|
|
133
|
-
it('does not run full runtime initialization for local fast retrieval', async () => {
|
|
134
|
-
let initialized = 0;
|
|
135
|
-
let statsReads = 0;
|
|
136
|
-
let retrieveArgs: { query: string; options: Record<string, unknown> } | null = null;
|
|
137
|
-
const traces: Array<Record<string, unknown>> = [];
|
|
138
|
-
|
|
139
|
-
const fakeRetriever = {
|
|
140
|
-
setQueryRewriter() {},
|
|
141
|
-
async retrieve(query: string, options: Record<string, unknown>) {
|
|
142
|
-
retrieveArgs = { query, options };
|
|
143
|
-
return retrievalResult('fast-e1');
|
|
144
|
-
}
|
|
145
|
-
} as unknown as Retriever;
|
|
146
|
-
|
|
147
|
-
const orchestrator = new RetrievalOrchestrator({
|
|
148
|
-
initialize: async () => { initialized += 1; },
|
|
149
|
-
retriever: fakeRetriever,
|
|
150
|
-
traceStore: {
|
|
151
|
-
getHelpfulnessStats: async () => { statsReads += 1; return stats(); },
|
|
152
|
-
recordRetrievalTrace: async (input) => { traces.push(input); }
|
|
153
|
-
},
|
|
154
|
-
accessStore: noopAccessStore(),
|
|
155
|
-
getProjectHash: () => 'project-fast',
|
|
156
|
-
hasSharedStore: () => false
|
|
157
|
-
});
|
|
158
|
-
|
|
159
|
-
await orchestrator.retrieveMemories('fast keyword query', {
|
|
160
|
-
strategy: 'fast',
|
|
161
|
-
topK: 3,
|
|
162
|
-
minScore: 0.2
|
|
163
|
-
});
|
|
164
|
-
|
|
165
|
-
expect(initialized).toBe(0);
|
|
166
|
-
expect(statsReads).toBe(0);
|
|
167
|
-
expect(retrieveArgs?.options.strategy).toBe('fast');
|
|
168
|
-
expect(retrieveArgs?.options.projectHash).toBe('project-fast');
|
|
169
|
-
expect(traces[0]).toMatchObject({
|
|
170
|
-
strategy: 'fast',
|
|
171
|
-
candidateEventIds: ['fast-e1'],
|
|
172
|
-
selectedEventIds: ['fast-e1']
|
|
173
|
-
});
|
|
174
|
-
});
|
|
175
|
-
|
|
176
|
-
it('keeps fast retrieval lightweight when shared search is requested but unavailable', async () => {
|
|
177
|
-
let initialized = 0;
|
|
178
|
-
let localRetrieveCalls = 0;
|
|
179
|
-
let unifiedRetrieveCalls = 0;
|
|
180
|
-
|
|
181
|
-
const fakeRetriever = {
|
|
182
|
-
setQueryRewriter() {},
|
|
183
|
-
async retrieve(_query: string, options: Record<string, unknown>) {
|
|
184
|
-
localRetrieveCalls += 1;
|
|
185
|
-
expect(options.includeShared).toBe(true);
|
|
186
|
-
expect(options.strategy).toBe('fast');
|
|
187
|
-
return retrievalResult('local-fast-shared-request');
|
|
188
|
-
},
|
|
189
|
-
async retrieveUnified() {
|
|
190
|
-
unifiedRetrieveCalls += 1;
|
|
191
|
-
return retrievalResult('unexpected-unified');
|
|
192
|
-
}
|
|
193
|
-
} as unknown as Retriever;
|
|
194
|
-
|
|
195
|
-
const orchestrator = new RetrievalOrchestrator({
|
|
196
|
-
initialize: async () => { initialized += 1; },
|
|
197
|
-
retriever: fakeRetriever,
|
|
198
|
-
traceStore: {
|
|
199
|
-
getHelpfulnessStats: async () => stats(),
|
|
200
|
-
recordRetrievalTrace: async () => {}
|
|
201
|
-
},
|
|
202
|
-
accessStore: noopAccessStore(),
|
|
203
|
-
getProjectHash: () => 'project-fast',
|
|
204
|
-
hasSharedStore: () => false
|
|
205
|
-
});
|
|
206
|
-
|
|
207
|
-
const out = await orchestrator.retrieveMemories('fast query without shared store', {
|
|
208
|
-
strategy: 'fast',
|
|
209
|
-
includeShared: true,
|
|
210
|
-
topK: 3
|
|
211
|
-
});
|
|
212
|
-
|
|
213
|
-
expect(out.memories[0]?.event.id).toBe('local-fast-shared-request');
|
|
214
|
-
expect(initialized).toBe(0);
|
|
215
|
-
expect(localRetrieveCalls).toBe(1);
|
|
216
|
-
expect(unifiedRetrieveCalls).toBe(0);
|
|
217
|
-
});
|
|
218
|
-
|
|
219
|
-
it('initializes the full runtime for fast retrieval when an actual shared store is available', async () => {
|
|
220
|
-
let initialized = 0;
|
|
221
|
-
let retrieveUnifiedArgs: { query: string; options: Record<string, unknown> } | null = null;
|
|
222
|
-
|
|
223
|
-
const fakeRetriever = {
|
|
224
|
-
setQueryRewriter() {},
|
|
225
|
-
async retrieveUnified(query: string, options: Record<string, unknown>) {
|
|
226
|
-
retrieveUnifiedArgs = { query, options };
|
|
227
|
-
return retrievalResult('shared-fast');
|
|
228
|
-
}
|
|
229
|
-
} as unknown as Retriever;
|
|
230
|
-
|
|
231
|
-
const orchestrator = new RetrievalOrchestrator({
|
|
232
|
-
initialize: async () => { initialized += 1; },
|
|
233
|
-
retriever: fakeRetriever,
|
|
234
|
-
traceStore: {
|
|
235
|
-
getHelpfulnessStats: async () => stats(),
|
|
236
|
-
recordRetrievalTrace: async () => {}
|
|
237
|
-
},
|
|
238
|
-
accessStore: noopAccessStore(),
|
|
239
|
-
getProjectHash: () => 'project-fast',
|
|
240
|
-
hasSharedStore: () => true
|
|
241
|
-
});
|
|
242
|
-
|
|
243
|
-
await orchestrator.retrieveMemories('fast shared query', {
|
|
244
|
-
strategy: 'fast',
|
|
245
|
-
includeShared: true,
|
|
246
|
-
topK: 3
|
|
247
|
-
});
|
|
248
|
-
|
|
249
|
-
expect(initialized).toBe(1);
|
|
250
|
-
expect(retrieveUnifiedArgs?.query).toBe('fast shared query');
|
|
251
|
-
expect(retrieveUnifiedArgs?.options.includeShared).toBe(true);
|
|
252
|
-
});
|
|
253
|
-
|
|
254
|
-
it('uses unified retrieval and normalized configured rerank weights when shared search is enabled', async () => {
|
|
255
|
-
process.env.MEMORY_RERANK_WEIGHT_SEMANTIC = '3';
|
|
256
|
-
process.env.MEMORY_RERANK_WEIGHT_LEXICAL = '1';
|
|
257
|
-
process.env.MEMORY_RERANK_WEIGHT_RECENCY = '1';
|
|
258
|
-
|
|
259
|
-
let retrieveUnifiedArgs: { query: string; options: Record<string, unknown> } | null = null;
|
|
260
|
-
|
|
261
|
-
const fakeRetriever = {
|
|
262
|
-
setQueryRewriter() {},
|
|
263
|
-
async retrieveUnified(query: string, options: Record<string, unknown>) {
|
|
264
|
-
retrieveUnifiedArgs = { query, options };
|
|
265
|
-
return retrievalResult('shared-e1');
|
|
266
|
-
}
|
|
267
|
-
} as unknown as Retriever;
|
|
268
|
-
|
|
269
|
-
const orchestrator = new RetrievalOrchestrator({
|
|
270
|
-
initialize: async () => {},
|
|
271
|
-
retriever: fakeRetriever,
|
|
272
|
-
traceStore: {
|
|
273
|
-
getHelpfulnessStats: async () => stats(),
|
|
274
|
-
recordRetrievalTrace: async () => {}
|
|
275
|
-
},
|
|
276
|
-
accessStore: noopAccessStore(),
|
|
277
|
-
getProjectHash: () => 'project-2',
|
|
278
|
-
hasSharedStore: () => true
|
|
279
|
-
});
|
|
280
|
-
|
|
281
|
-
await orchestrator.retrieveMemories('shared query', {
|
|
282
|
-
includeShared: true,
|
|
283
|
-
intentRewrite: true,
|
|
284
|
-
adaptiveRerank: true
|
|
285
|
-
});
|
|
286
|
-
|
|
287
|
-
expect(retrieveUnifiedArgs?.query).toBe('shared query');
|
|
288
|
-
expect(retrieveUnifiedArgs?.options.includeShared).toBe(true);
|
|
289
|
-
expect(retrieveUnifiedArgs?.options.intentRewrite).toBe(true);
|
|
290
|
-
expect(retrieveUnifiedArgs?.options.rerankWeights).toEqual({
|
|
291
|
-
semantic: 0.6,
|
|
292
|
-
lexical: 0.2,
|
|
293
|
-
recency: 0.2
|
|
294
|
-
});
|
|
295
|
-
});
|
|
296
|
-
|
|
297
|
-
it('formats high-confidence context through the orchestrator facade', () => {
|
|
298
|
-
const fakeRetriever = {
|
|
299
|
-
setQueryRewriter() {}
|
|
300
|
-
} as unknown as Retriever;
|
|
301
|
-
const orchestrator = new RetrievalOrchestrator({
|
|
302
|
-
initialize: async () => {},
|
|
303
|
-
retriever: fakeRetriever,
|
|
304
|
-
traceStore: {
|
|
305
|
-
getHelpfulnessStats: async () => stats(),
|
|
306
|
-
recordRetrievalTrace: async () => {}
|
|
307
|
-
},
|
|
308
|
-
accessStore: noopAccessStore(),
|
|
309
|
-
getProjectHash: () => null,
|
|
310
|
-
hasSharedStore: () => false
|
|
311
|
-
});
|
|
312
|
-
|
|
313
|
-
expect(orchestrator.formatAsContext(retrievalResult())).toContain('High-confidence memory match');
|
|
314
|
-
});
|
|
315
|
-
|
|
316
|
-
it('records prompt memory access through the orchestrator access port without full initialization', async () => {
|
|
317
|
-
let initialized = 0;
|
|
318
|
-
const accessCalls: string[][] = [];
|
|
319
|
-
const fakeRetriever = {
|
|
320
|
-
setQueryRewriter() {}
|
|
321
|
-
} as unknown as Retriever;
|
|
322
|
-
|
|
323
|
-
const orchestrator = new RetrievalOrchestrator({
|
|
324
|
-
initialize: async () => { initialized += 1; },
|
|
325
|
-
retriever: fakeRetriever,
|
|
326
|
-
traceStore: {
|
|
327
|
-
getHelpfulnessStats: async () => stats(),
|
|
328
|
-
recordRetrievalTrace: async () => {}
|
|
329
|
-
},
|
|
330
|
-
accessStore: {
|
|
331
|
-
incrementAccessCount: async (eventIds) => { accessCalls.push(eventIds); },
|
|
332
|
-
recordRetrieval: async () => {}
|
|
333
|
-
},
|
|
334
|
-
getProjectHash: () => null,
|
|
335
|
-
hasSharedStore: () => false
|
|
336
|
-
});
|
|
337
|
-
|
|
338
|
-
await orchestrator.incrementMemoryAccess(['e1', 'e2']);
|
|
339
|
-
|
|
340
|
-
expect(initialized).toBe(0);
|
|
341
|
-
expect(accessCalls).toEqual([['e1', 'e2']]);
|
|
342
|
-
});
|
|
343
|
-
|
|
344
|
-
it('skips prompt memory access tracking for empty event id lists', async () => {
|
|
345
|
-
let initialized = 0;
|
|
346
|
-
let accessCalls = 0;
|
|
347
|
-
const fakeRetriever = {
|
|
348
|
-
setQueryRewriter() {}
|
|
349
|
-
} as unknown as Retriever;
|
|
350
|
-
|
|
351
|
-
const orchestrator = new RetrievalOrchestrator({
|
|
352
|
-
initialize: async () => { initialized += 1; },
|
|
353
|
-
retriever: fakeRetriever,
|
|
354
|
-
traceStore: {
|
|
355
|
-
getHelpfulnessStats: async () => stats(),
|
|
356
|
-
recordRetrievalTrace: async () => {}
|
|
357
|
-
},
|
|
358
|
-
accessStore: {
|
|
359
|
-
incrementAccessCount: async () => { accessCalls += 1; },
|
|
360
|
-
recordRetrieval: async () => {}
|
|
361
|
-
},
|
|
362
|
-
getProjectHash: () => null,
|
|
363
|
-
hasSharedStore: () => false
|
|
364
|
-
});
|
|
365
|
-
|
|
366
|
-
await orchestrator.incrementMemoryAccess([]);
|
|
367
|
-
|
|
368
|
-
expect(initialized).toBe(0);
|
|
369
|
-
expect(accessCalls).toBe(0);
|
|
370
|
-
});
|
|
371
|
-
|
|
372
|
-
it('records retrieval helpfulness events through the orchestrator access port', async () => {
|
|
373
|
-
let initialized = 0;
|
|
374
|
-
const retrievalCalls: Array<{ eventId: string; sessionId: string; score: number; query: string }> = [];
|
|
375
|
-
const fakeRetriever = {
|
|
376
|
-
setQueryRewriter() {}
|
|
377
|
-
} as unknown as Retriever;
|
|
378
|
-
|
|
379
|
-
const orchestrator = new RetrievalOrchestrator({
|
|
380
|
-
initialize: async () => { initialized += 1; },
|
|
381
|
-
retriever: fakeRetriever,
|
|
382
|
-
traceStore: {
|
|
383
|
-
getHelpfulnessStats: async () => stats(),
|
|
384
|
-
recordRetrievalTrace: async () => {}
|
|
385
|
-
},
|
|
386
|
-
accessStore: {
|
|
387
|
-
incrementAccessCount: async () => {},
|
|
388
|
-
recordRetrieval: async (eventId, sessionId, score, query) => {
|
|
389
|
-
retrievalCalls.push({ eventId, sessionId, score, query });
|
|
390
|
-
}
|
|
391
|
-
},
|
|
392
|
-
getProjectHash: () => 'project-1',
|
|
393
|
-
hasSharedStore: () => false
|
|
394
|
-
});
|
|
395
|
-
|
|
396
|
-
await orchestrator.recordRetrieval('e1', 's1', 0.82, 'thin core');
|
|
397
|
-
|
|
398
|
-
expect(initialized).toBe(1);
|
|
399
|
-
expect(retrievalCalls).toEqual([
|
|
400
|
-
{ eventId: 'e1', sessionId: 's1', score: 0.82, query: 'thin core' }
|
|
401
|
-
]);
|
|
402
|
-
});
|
|
403
|
-
});
|
|
@@ -1,31 +0,0 @@
|
|
|
1
|
-
import { describe, expect, it } from 'vitest';
|
|
2
|
-
import {
|
|
3
|
-
isGenericContinuationQuery,
|
|
4
|
-
isLowSignalContextContent
|
|
5
|
-
} from '../../src/core/retrieval-quality.js';
|
|
6
|
-
|
|
7
|
-
describe('retrieval quality guards', () => {
|
|
8
|
-
it('recognizes short generic continuation prompts', () => {
|
|
9
|
-
expect(isGenericContinuationQuery('continue')).toBe(true);
|
|
10
|
-
expect(isGenericContinuationQuery('what next?')).toBe(true);
|
|
11
|
-
expect(isGenericContinuationQuery('다음 추천작업은 뭐야?')).toBe(true);
|
|
12
|
-
expect(isGenericContinuationQuery('이어서 진행해줘')).toBe(true);
|
|
13
|
-
});
|
|
14
|
-
|
|
15
|
-
it('does not classify topic-specific next/recommendation/debug prompts as generic', () => {
|
|
16
|
-
expect(isGenericContinuationQuery('next task for search UI')).toBe(false);
|
|
17
|
-
expect(isGenericContinuationQuery('recommendation for auth design')).toBe(false);
|
|
18
|
-
expect(isGenericContinuationQuery('이 오류 뭐야?')).toBe(false);
|
|
19
|
-
expect(isGenericContinuationQuery('다음 작업은 retrieval-quality.ts에서 뭐야?')).toBe(false);
|
|
20
|
-
});
|
|
21
|
-
|
|
22
|
-
it('detects low-signal context artifacts', () => {
|
|
23
|
-
expect(isLowSignalContextContent('<environment_context><cwd>/repo/app</cwd></environment_context>')).toBe(true);
|
|
24
|
-
expect(isLowSignalContextContent('prefix <environment_context><cwd>/repo/app</cwd></environment_context> suffix')).toBe(true);
|
|
25
|
-
expect(isLowSignalContextContent('<command-name>/model</command-name>\n<local-command-stdout>opus</local-command-stdout>')).toBe(true);
|
|
26
|
-
expect(isLowSignalContextContent('# AGENTS.md instructions for /repo/app <INSTRUCTIONS> ## Skills A skill is local instructions.')).toBe(true);
|
|
27
|
-
expect(isLowSignalContextContent('Understood, stopping here. Let me know when you would like to continue with the design doc review.')).toBe(true);
|
|
28
|
-
expect(isLowSignalContextContent('Implementation note: suppress the stale phrase "Understood, stopping here. Let me know when you would like to continue." from context packs.')).toBe(false);
|
|
29
|
-
expect(isLowSignalContextContent('Merged PR #15 and synced local main.')).toBe(false);
|
|
30
|
-
});
|
|
31
|
-
});
|