diff --git a/api/app/core/workflow/nodes/knowledge/node.py b/api/app/core/workflow/nodes/knowledge/node.py index 0601883d..2a8c5249 100644 --- a/api/app/core/workflow/nodes/knowledge/node.py +++ b/api/app/core/workflow/nodes/knowledge/node.py @@ -233,7 +233,7 @@ class KnowledgeRetrievalNode(BaseNode): } ) ) - case (RetrieveType.HYBRID, RetrieveType.Graph): + case retrieve_type if retrieve_type in (RetrieveType.HYBRID, RetrieveType.Graph): rs1_task = asyncio.to_thread( vector_service.search_by_vector, **{ "query": query,