diff --git a/api/core/workflow/nodes/knowledge_index/knowledge_index_node.py b/api/core/workflow/nodes/knowledge_index/knowledge_index_node.py index 1d60f530a1..ac3a194cdd 100644 --- a/api/core/workflow/nodes/knowledge_index/knowledge_index_node.py +++ b/api/core/workflow/nodes/knowledge_index/knowledge_index_node.py @@ -46,7 +46,7 @@ class KnowledgeIndexNode(Node[KnowledgeIndexNodeData]): self.index_processor = IndexProcessor() self.summary_index_service = SummaryIndex() - def _run(self) -> NodeRunResult: # type: ignore + def _run(self) -> NodeRunResult: node_data = self.node_data variable_pool = self.graph_runtime_state.variable_pool 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 1aba2737b0..e161796210 100644 --- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py +++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py @@ -134,7 +134,7 @@ class KnowledgeRetrievalNode(LLMUsageTrackingMixin, Node[KnowledgeRetrievalNodeD status=WorkflowNodeExecutionStatus.SUCCEEDED, inputs=variables, process_data={"usage": jsonable_encoder(usage)}, - outputs=outputs, # type: ignore + outputs=outputs, metadata={ WorkflowNodeExecutionMetadataKey.TOTAL_TOKENS: usage.total_tokens, WorkflowNodeExecutionMetadataKey.TOTAL_PRICE: usage.total_price,