mirror of
https://github.com/langgenius/dify.git
synced 2026-04-03 18:00:30 -04:00
48 lines
2.0 KiB
Python
48 lines
2.0 KiB
Python
"""
|
|
Parser for tool nodes that captures tool-specific metadata.
|
|
"""
|
|
|
|
from graphon.enums import WorkflowNodeExecutionMetadataKey
|
|
from graphon.graph_events import GraphNodeEventBase
|
|
from graphon.nodes.base.node import Node
|
|
from graphon.nodes.tool.entities import ToolNodeData
|
|
from opentelemetry.trace import Span
|
|
|
|
from extensions.otel.parser.base import DefaultNodeOTelParser, safe_json_dumps
|
|
from extensions.otel.semconv.gen_ai import ToolAttributes
|
|
|
|
|
|
class ToolNodeOTelParser:
|
|
"""Parser for tool nodes that captures tool-specific metadata."""
|
|
|
|
def __init__(self) -> None:
|
|
self._delegate = DefaultNodeOTelParser()
|
|
|
|
def parse(
|
|
self, *, node: Node, span: "Span", error: Exception | None, result_event: GraphNodeEventBase | None = None
|
|
) -> None:
|
|
self._delegate.parse(node=node, span=span, error=error, result_event=result_event)
|
|
|
|
tool_data = getattr(node, "_node_data", None)
|
|
if not isinstance(tool_data, ToolNodeData):
|
|
return
|
|
|
|
span.set_attribute(ToolAttributes.TOOL_NAME, node.title)
|
|
span.set_attribute(ToolAttributes.TOOL_TYPE, tool_data.provider_type.value)
|
|
|
|
# Extract tool info from metadata (consistent with aliyun_trace)
|
|
tool_info = {}
|
|
if result_event and result_event.node_run_result:
|
|
node_run_result = result_event.node_run_result
|
|
if node_run_result.metadata:
|
|
tool_info = node_run_result.metadata.get(WorkflowNodeExecutionMetadataKey.TOOL_INFO, {})
|
|
|
|
if tool_info:
|
|
span.set_attribute(ToolAttributes.TOOL_DESCRIPTION, safe_json_dumps(tool_info))
|
|
|
|
if result_event and result_event.node_run_result and result_event.node_run_result.inputs:
|
|
span.set_attribute(ToolAttributes.TOOL_CALL_ARGUMENTS, safe_json_dumps(result_event.node_run_result.inputs))
|
|
|
|
if result_event and result_event.node_run_result and result_event.node_run_result.outputs:
|
|
span.set_attribute(ToolAttributes.TOOL_CALL_RESULT, safe_json_dumps(result_event.node_run_result.outputs))
|