adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(workflow): temporarily ignore non-text fields in knowledge retrieval node

Browse Source
This commit is contained in:
mengyonghao
2026-01-06 11:38:03 +08:00
parent 049642ae48
commit 411525687e
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/app/core/workflow/nodes/knowledge/node.py
View File

@@ -203,15 +203,16 @@ class KnowledgeRetrievalNode(BaseNode):
rs2 = vector_service.search_by_full_text(query=query, top_k=kb_config.top_k, rs2 = vector_service.search_by_full_text(query=query, top_k=kb_config.top_k,
indices=indices, indices=indices,
score_threshold=kb_config.similarity_threshold) score_threshold=kb_config.similarity_threshold)
# Deduplicate hybrid retrieval results # Deduplicate hy brid retrieval results
unique_rs = self._deduplicate_docs(rs1, rs2) unique_rs = self._deduplicate_docs(rs1, rs2)
vector_service.reranker = self.get_reranker_model() vector_service.reranker = self.get_reranker_model()
rs.extend(vector_service.rerank(query=query, docs=unique_rs, top_k=kb_config.top_k)) rs.extend(vector_service.rerank(query=query, docs=unique_rs, top_k=kb_config.top_k))
case _: case _:
raise RuntimeError("Unknown retrieval type") raise RuntimeError("Unknown retrieval type")
vector_service.reranker = self.get_reranker_model() vector_service.reranker = self.get_reranker_model()
# TODO：其他重排序方式支持
final_rs = vector_service.rerank(query=query, docs=rs, top_k=self.typed_config.reranker_top_k) final_rs = vector_service.rerank(query=query, docs=rs, top_k=self.typed_config.reranker_top_k)
logger.info( logger.info(
f"Node {self.node_id}: knowledge base retrieval completed, results count: {len(final_rs)}" f"Node {self.node_id}: knowledge base retrieval completed, results count: {len(final_rs)}"
) )
return [chunk.model_dump() for chunk in final_rs] return [chunk.page_content for chunk in final_rs]