ソースを参照
fix: iteration node in parallel mode token count error (#11539)
Co-authored-by: Novice Lee <novicelee@NoviPro.local>
tags/0.14.0
Novice
10ヶ月前
コミッターのメールアドレスに関連付けられたアカウントが存在しません
|
|
|
|
|
|
|
|
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 |