From 69f66b0bd8fc3c21a00d968b71fd9c6246419771 Mon Sep 17 00:00:00 2001 From: -LAN- Date: Fri, 18 Jul 2025 03:41:33 +0800 Subject: [PATCH] refactor(knowledge_retrieval_node): Remove unused cast Signed-off-by: -LAN- --- .../nodes/knowledge_retrieval/knowledge_retrieval_node.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py index 3b96cecf81..4e9a38f552 100644 --- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py +++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py @@ -151,9 +151,8 @@ class KnowledgeRetrievalNode(BaseNode): return "1" def _run(self) -> NodeRunResult: # type: ignore - node_data = cast(KnowledgeRetrievalNodeData, self._node_data) # extract variables - variable = self.graph_runtime_state.variable_pool.get(node_data.query_variable_selector) + variable = self.graph_runtime_state.variable_pool.get(self._node_data.query_variable_selector) if not isinstance(variable, StringSegment): return NodeRunResult( status=WorkflowNodeExecutionStatus.FAILED, @@ -194,7 +193,7 @@ class KnowledgeRetrievalNode(BaseNode): # retrieve knowledge try: - results = self._fetch_dataset_retriever(node_data=node_data, query=query) + results = self._fetch_dataset_retriever(node_data=self._node_data, query=query) outputs = {"result": ArrayObjectSegment(value=results)} return NodeRunResult( status=WorkflowNodeExecutionStatus.SUCCEEDED,