giers10/Heimgeist
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Improve hit deduplication by enforcing a minimum score threshold

Browse Source
This commit is contained in:
Victor Giers
2026-06-15 22:54:25 +02:00
parent 4b0ed88176
commit a5ad7ab261
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/chat_memory.py
View File

@@ -539,11 +539,13 @@ def _trim(value: Any, limit: int) -> str:
return text_value[:limit].rstrip() + "..."
def _dedupe_hits(candidates: Sequence[Dict[str, Any]], top_k: int) -> List[Dict[str, Any]]:
def _dedupe_hits(candidates: Sequence[Dict[str, Any]], top_k: int, *, minimum_score: float) -> List[Dict[str, Any]]:
seen = set()
ordered = sorted(candidates, key=lambda item: (float(item.get("_score") or 0), int(item.get("assistant_message_row_id") or 0)), reverse=True)
hits: List[Dict[str, Any]] = []
for item in ordered:
if float(item.get("_score") or 0) < minimum_score:
continue
key = item.get("turn_key") or (item.get("session_id"), item.get("assistant_message_id"))
if key in seen:
continue