feat: 先暂时不改用混合检索
This commit is contained in:
@@ -52,10 +52,12 @@ class MilvusService:
|
||||
dim=env.llama_index_dimension,
|
||||
|
||||
embedding_field="embedding",
|
||||
# text_field="text", # 添加文本字段支持混合检索
|
||||
id_field="id",
|
||||
similarity_metric="COSINE",
|
||||
consistency_level="Strong",
|
||||
overwrite=False,
|
||||
# enable_hybrid=True, # 启用混合检索
|
||||
)
|
||||
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)])
|
||||
|
||||
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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user