chore: tool node support agent logs

pull/12372/head
Joel 1 year ago
parent 18393d2e27
commit 6a2a7aca9b

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

Loading…
Cancel
Save