diff --git a/src/memos/reranker/http_bge.py b/src/memos/reranker/http_bge.py index 32034cf6d..c11644dc6 100644 --- a/src/memos/reranker/http_bge.py +++ b/src/memos/reranker/http_bge.py @@ -154,7 +154,8 @@ def rerank( list[tuple[TextualMemoryItem, float]] Re-ranked items with scores, sorted descending by score. """ - + if len(query) > 8000: + query = query[:500] + "\n" + query[-500:] if not graph_results: return [] diff --git a/src/memos/reranker/http_bge_strategy.py b/src/memos/reranker/http_bge_strategy.py index b0567698c..951893ef5 100644 --- a/src/memos/reranker/http_bge_strategy.py +++ b/src/memos/reranker/http_bge_strategy.py @@ -149,6 +149,8 @@ def rerank( list[tuple[TextualMemoryItem, float]] Re-ranked items with scores, sorted descending by score. """ + if len(query) > 8000: + query = query[:500] + "\n" + query[-500:] if not graph_results: return []