hailin
/
faiss_rag_enterprise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-05-10 00:11:35 +08:00
parent 63f1db9d17
commit 7f595c2215
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/rag_build_query.py
View File

@ -19,7 +19,7 @@ class BGEEmbedding(HuggingFaceEmbedding):
prefix = "Represent this sentence for searching relevant passages: " prefix = "Represent this sentence for searching relevant passages: "
return super()._get_query_embedding(prefix + query) 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: " prefix = "Represent this sentence for searching relevant passages: "
return super()._get_query_embeddings([prefix + q for q in queries]) return super()._get_query_embeddings([prefix + q for q in queries])