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])