feat: 先暂时不改用混合检索

This commit is contained in:
martsforever
2025-12-03 22:31:44 +08:00
parent 829d723dfb
commit 94bbb7f64a
+7 -1
View File
@@ -52,10 +52,12 @@ class MilvusService:
dim=env.llama_index_dimension, dim=env.llama_index_dimension,
embedding_field="embedding", embedding_field="embedding",
# text_field="text", # 添加文本字段支持混合检索
id_field="id", id_field="id",
similarity_metric="COSINE", similarity_metric="COSINE",
consistency_level="Strong", consistency_level="Strong",
overwrite=False, overwrite=False,
# enable_hybrid=True, # 启用混合检索
) )
return self._milvus_vector_store return self._milvus_vector_store
@@ -92,7 +94,11 @@ class MilvusService:
filters = MetadataFilters(filters=[MetadataFilter(key="parent_code", value=param.kb_code, operator=FilterOperator.IN if isinstance(param.kb_code, list) else FilterOperator.EQ)]) filters = MetadataFilters(filters=[MetadataFilter(key="parent_code", value=param.kb_code, operator=FilterOperator.IN if isinstance(param.kb_code, list) else FilterOperator.EQ)])
retriever = vector_index.as_retriever(filters=filters, similarity_top_k=param.top_k) retriever = vector_index.as_retriever(
retriever_mode="hybrid", # 启用混合模式
filters=filters,
similarity_top_k=param.top_k
)
result_nodes = await retriever.aretrieve(param.question) result_nodes = await retriever.aretrieve(param.question)