From 7f595c2215210e50450a752184df84d8ccc4d5e8 Mon Sep 17 00:00:00 2001 From: hailin Date: Sat, 10 May 2025 00:11:35 +0800 Subject: [PATCH] . --- scripts/rag_build_query.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/rag_build_query.py b/scripts/rag_build_query.py index e57a3b8..6f62319 100644 --- a/scripts/rag_build_query.py +++ b/scripts/rag_build_query.py @@ -19,7 +19,7 @@ class BGEEmbedding(HuggingFaceEmbedding): prefix = "Represent this sentence for searching relevant passages: " return super()._get_query_embedding(prefix + query) - def _get_query_embeddings(self, queries: List[str]) -> List[List[float]]]: + def _get_query_embeddings(self, queries: List[str]) -> List[List[float]]: # 批量生成嵌入向量 prefix = "Represent this sentence for searching relevant passages: " return super()._get_query_embeddings([prefix + q for q in queries])