Skip to content

Commit 9b4cd68

Browse files
committed
🐛 查询永远是空返回
1 parent a27cdd1 commit 9b4cd68

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/nonebot_plugin_nyaturingtest/hippo_mem.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -114,10 +114,12 @@ def retrieve(self, queries: list[str], k: int = 5) -> list[str]:
114114
包含检索结果的Document列表
115115
"""
116116
# 切割(BAAI/bge-m3上限为8192tokens)
117-
queries = []
117+
logger.debug(f"查询文本: {queries}")
118+
splited_queries = []
118119
for query in queries:
119-
queries += _split_text_by_tokens(query, self.tokenizer, max_tokens=8192, overlap=100)
120-
results = self.hippo.retrieve(queries=queries, num_to_retrieve=k)
120+
splited_queries += _split_text_by_tokens(query, self.tokenizer, max_tokens=8192, overlap=100)
121+
logger.debug(f"分割后的查询: {splited_queries}")
122+
results = self.hippo.retrieve(queries=splited_queries, num_to_retrieve=k)
121123
# make ruff happy
122124
assert isinstance(results, list)
123125
docs = [doc for result in results for doc in result.docs]

0 commit comments

Comments
 (0)