瀏覽代碼

Fix the token count at the iteration node (#11235)

Co-authored-by: -LAN- <laipz8200@outlook.com>
suzuki.sh 4 月之前
父節點
當前提交
c3c6a48059
共有 2 個文件被更改,包括 5 次插入3 次删除
  1. 4 1
      api/core/workflow/nodes/iteration/iteration_node.py
  2. 1 2
      web/app/(commonLayout)/apps/AppCard.tsx

+ 4 - 1
api/core/workflow/nodes/iteration/iteration_node.py

@@ -242,7 +242,10 @@ class IterationNode(BaseNode[IterationNodeData]):
                 run_result=NodeRunResult(
                     status=WorkflowNodeExecutionStatus.SUCCEEDED,
                     outputs={"output": outputs},
-                    metadata={NodeRunMetadataKey.ITERATION_DURATION_MAP: iter_run_map},
+                    metadata={
+                        NodeRunMetadataKey.ITERATION_DURATION_MAP: iter_run_map,
+                        NodeRunMetadataKey.TOTAL_TOKENS: graph_engine.graph_runtime_state.total_tokens,
+                    },
                 )
             )
         except IterationNodeError as e:

+ 1 - 2
web/app/(commonLayout)/apps/AppCard.tsx

@@ -9,8 +9,7 @@ import s from './style.module.css'
 import cn from '@/utils/classnames'
 import type { App } from '@/types/app'
 import Confirm from '@/app/components/base/confirm'
-import Toast from '@/app/components/base/toast'
-import { ToastContext } from '@/app/components/base/toast'
+import Toast, { ToastContext } from '@/app/components/base/toast'
 import { copyApp, deleteApp, exportAppConfig, updateAppInfo } from '@/service/apps'
 import DuplicateAppModal from '@/app/components/app/duplicate-modal'
 import type { DuplicateAppModalProps } from '@/app/components/app/duplicate-modal'