diff --git a/apps/wpm/services.py b/apps/wpm/services.py
index dd2befd1..7acb71bd 100644
--- a/apps/wpm/services.py
+++ b/apps/wpm/services.py
@@ -997,6 +997,8 @@ def get_batch_dag(batch_number: str, version=1):
                 edges.append({
                     'source': source,
                     'target': target,
+                    "handover": log.handover.id if log.handover else None,
+                    "mlog": log.mlog.id if log.mlog else None,
                     'label': r_dict.get(log.relation_type, ""),  # 使用relation_type作为边的标签
                 })