ソースを参照

fix: iteration node in parallel mode token count error (#11539)

Co-authored-by: Novice Lee <novicelee@NoviPro.local>
tags/0.14.0
Novice 10ヶ月前
コミット
1765fe2a29
コミッターのメールアドレスに関連付けられたアカウントが存在しません
1個のファイルの変更1行の追加1行の削除
  1. 1
    1
      api/core/workflow/nodes/iteration/iteration_node.py

+ 1
- 1
api/core/workflow/nodes/iteration/iteration_node.py ファイルの表示

future.add_done_callback(thread_pool.task_done_callback) future.add_done_callback(thread_pool.task_done_callback)
futures.append(future) futures.append(future)
succeeded_count = 0 succeeded_count = 0
empty_count = 0
while True: while True:
try: try:
event = q.get(timeout=1) event = q.get(timeout=1)
parallel_mode_run_id=parallel_mode_run_id, parallel_mode_run_id=parallel_mode_run_id,
): ):
q.put(event) q.put(event)
graph_engine.graph_runtime_state.total_tokens += graph_engine_copy.graph_runtime_state.total_tokens

読み込み中…
キャンセル
保存