eliott/claude-mem
1
0
Fork 0
mirror of https://github.com/thedotmack/claude-mem synced 2026-04-25 17:15:04 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: only apply diversity cap for text queries, not filter-only searches

Browse Source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Alex Newman
2026-04-18 14:30:43 -07:00
parent 64fc4f4065
commit d63471f40d
1 changed files with 23 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/services/worker/SearchManager.ts
View File

@@ -260,8 +260,10 @@ export class SearchManager {
*/
private deduplicateResults<T extends { id: number; content_hash?: string; project?: string; memory_session_id?: string }>(
results: T[],
maxPerProjectSession: number = 5
options: { maxPerProjectSession?: number; skipDiversityCap?: boolean } = {}
): T[] {
const { maxPerProjectSession = 5, skipDiversityCap = false } = options;
// Step 1: Deduplicate by content_hash, keeping first occurrence (highest-scored)
const seenHashes = new Set<string>();
const deduped: T[] = [];
@@ -276,7 +278,19 @@ export class SearchManager {
deduped.push(result);
}
// Step 2: Apply per-project+session diversity cap
// Step 2: Apply per-project+session diversity cap only for text queries.
// Filter-only searches (no text query) should return all matching results
// without artificial truncation per project/session.
if (skipDiversityCap) {
if (deduped.length !== results.length) {
logger.debug('SEARCH', 'Deduplication applied (hash only, diversity cap skipped)', {
original: results.length,
afterHashDedup: deduped.length
});
}
return deduped;
}
const projectSessionCounts = new Map<string, number>();
const diversified: T[] = [];
@@ -468,10 +482,13 @@ export class SearchManager {
}
}
// Bug #1915: Deduplicate results by content hash and apply diversity cap
observations = this.deduplicateResults(observations);
sessions = this.deduplicateResults(sessions);
prompts = this.deduplicateResults(prompts);
// Bug #1915: Deduplicate results by content hash and apply diversity cap.
// Only apply diversity cap for text queries - filter-only searches should
// return all matching results without per-project/session truncation.
const skipDiversityCap = !query;
observations = this.deduplicateResults(observations, { skipDiversityCap });
sessions = this.deduplicateResults(sessions, { skipDiversityCap });
prompts = this.deduplicateResults(prompts, { skipDiversityCap });
const totalResults = observations.length + sessions.length + prompts.length;