Fix: Find Related: State all getting garbage

routers/tool_find_related.py

@@ -319,21 +319,24 @@ def _sort_similarity(data: dict,
                      state: State = State.opened) -> list:
     duplicates = []
     embeddings = data['embeddings']
-    true_indices = None
+    mask_opened = data["opened"]
 
-    if state != State.all:
+    if state == State.all:
+        mask = mask_opened | data["closed"]
+    else:
         mask = data[state.value]
-        embeddings = embeddings[mask]
-        true_indices = mask.nonzero(as_tuple=True)[0]
+
+    embeddings = embeddings[mask]
+    true_indices = mask.nonzero(as_tuple=True)[0]
 
     ret = util.semantic_search(
         query_emb, embeddings, top_k=limit, score_function=util.dot_score)
 
     for score in ret[0]:
         corpus_id = score['corpus_id']
-        number = true_indices[corpus_id].item(
-        ) if true_indices is not None else corpus_id
-        text = f"#{number}: {data['titles'][number]}"
+        number = true_indices[corpus_id].item()
+        closed_char = "" if mask_opened[number] else "~~"
+        text = f"{closed_char}#{number}{closed_char}: {data['titles'][number]}"
         duplicates.append(text)
 
     return duplicates
@@ -390,7 +393,8 @@ if __name__ == "__main__":
             val['embeddings'] = val['embeddings'].to(torch.device('cuda'))
 
         # 'blender/blender/111434' must print #96153, #83604 and #79762
-        related1 = find_relatedness('blender', 111434, limit=20)
+        related1 = find_relatedness(
+            'blender', 111434, limit=20, state=State.all)
         related2 = find_relatedness('blender-addons', 104399, limit=20)
 
         print("These are the 20 most related issues:")

routers/tool_find_related_cache.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:46c42973a8caaa2f0d4a76ebc6ff16c0b8df927c9b16ba645c3f7155cce84f6a
-size 723382452
+oid sha256:178cabd52e35e69184b0e49a0bdae18478e99d1b5cec6f590840e7d7c65576d8
+size 723396066