chore: tool node support agent logs

This commit is contained in:
Joel 2024-12-27 16:52:06 +08:00
parent 18393d2e27
commit 6a2a7aca9b
1 changed files with 3 additions and 1 deletions

View File

@ -1,6 +1,8 @@
import { BlockEnum } from '@/app/components/workflow/types'
import type { AgentLogItem, AgentLogItemWithChildren, NodeTracing } from '@/types/workflow'
const supportedAgentLogNodes = [BlockEnum.Agent, BlockEnum.Tool]
const listToTree = (logs: AgentLogItem[]) => {
if (!logs || logs.length === 0)
return []
@ -24,7 +26,7 @@ const listToTree = (logs: AgentLogItem[]) => {
}
const format = (list: NodeTracing[]): NodeTracing[] => {
const result: NodeTracing[] = list.map((item) => {
if (item.node_type === BlockEnum.Agent && item.execution_metadata?.agent_log && item.execution_metadata?.agent_log.length > 0)
if (supportedAgentLogNodes.includes(item.node_type) && item.execution_metadata?.agent_log && item.execution_metadata?.agent_log.length > 0)
item.agentLog = listToTree(item.execution_metadata.agent_log)
return item