feat: enterprise OTEL telemetry exporter (squash merge from feat/otel-telemetry-ee)

api/dify_graph/nodes/tool/entities.py

@@ -5,7 +5,7 @@ from pydantic_core.core_schema import ValidationInfo
 
 from core.tools.entities.tool_entities import ToolProviderType
 from dify_graph.entities.base_node_data import BaseNodeData
-from dify_graph.enums import NodeType
+from dify_graph.enums import BuiltinNodeTypes, NodeType
 
 
 class ToolEntity(BaseModel):
@@ -33,7 +33,7 @@ class ToolEntity(BaseModel):
 
 
 class ToolNodeData(BaseNodeData, ToolEntity):
-    type: NodeType = NodeType.TOOL
+    type: NodeType = BuiltinNodeTypes.TOOL
 
     class ToolInput(BaseModel):
         # TODO: check this type

api/dify_graph/nodes/tool/tool_node.py

@@ -9,7 +9,7 @@ from core.tools.tool_engine import ToolEngine
 from core.tools.utils.message_transformer import ToolFileMessageTransformer
 from dify_graph.entities.graph_config import NodeConfigDict
 from dify_graph.enums import (
-    NodeType,
+    BuiltinNodeTypes,
     SystemVariableKey,
     WorkflowNodeExecutionMetadataKey,
     WorkflowNodeExecutionStatus,
@@ -42,7 +42,7 @@ class ToolNode(Node[ToolNodeData]):
     Tool Node
     """
 
-    node_type = NodeType.TOOL
+    node_type = BuiltinNodeTypes.TOOL
 
     def __init__(
         self,
@@ -65,6 +65,10 @@ class ToolNode(Node[ToolNodeData]):
     def version(cls) -> str:
         return "1"
 
+    def populate_start_event(self, event) -> None:
+        event.provider_id = self.node_data.provider_id
+        event.provider_type = self.node_data.provider_type
+
     def _run(self) -> Generator[NodeEventBase, None, None]:
         """
         Run the tool node
@@ -77,7 +81,9 @@ class ToolNode(Node[ToolNodeData]):
         tool_info = {
             "provider_type": self.node_data.provider_type.value,
             "provider_id": self.node_data.provider_id,
+            "tool_name": self.node_data.tool_name,
             "plugin_unique_identifier": self.node_data.plugin_unique_identifier,
+            "credential_id": self.node_data.credential_id,
         }
 
         # get tool runtime