Skip to content

Commit b24926f

Browse files
committed
fix: update memory get
1 parent c6c6fbb commit b24926f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/memos/reranker/concat.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,7 @@
11
from typing import Any
2+
import re
3+
4+
_TAG1 = re.compile(r"^\s*\[[^\]]*\]\s*")
25

36

47
def process_source(
@@ -39,7 +42,7 @@ def concat_original_source(
3942
merge_field = ["sources"]
4043
documents = []
4144
for item in graph_results:
42-
memory = getattr(item, "memory", "")
45+
memory = (_TAG1.sub("", m) if isinstance((m := getattr(item, "memory", None)), str) else m)
4346
sources = []
4447
for field in merge_field:
4548
source = getattr(item.metadata, field, "")

0 commit comments

Comments
 (0)