| ) | ) | ||||
| elif self.node_type == NodeType.DATASOURCE.value and "datasource_info" in self.execution_metadata_dict: | elif self.node_type == NodeType.DATASOURCE.value and "datasource_info" in self.execution_metadata_dict: | ||||
| datasource_info = self.execution_metadata_dict["datasource_info"] | datasource_info = self.execution_metadata_dict["datasource_info"] | ||||
| extras["icon"] = datasource_info["icon"] | |||||
| extras["icon"] = datasource_info.get("icon") | |||||
| return extras | return extras | ||||
| def _get_offload_by_type(self, type_: ExecutionOffLoadType) -> Optional["WorkflowNodeExecutionOffload"]: | def _get_offload_by_type(self, type_: ExecutionOffLoadType) -> Optional["WorkflowNodeExecutionOffload"]: |