From 3a69075b9ff300d5d26b510c0ef97bf348309691 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Mon, 26 May 2025 11:07:08 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A4=84=E7=90=86dag=E6=97=B6=E8=BF=94?=
 =?UTF-8?q?=E5=9B=9Eedge=E6=9B=B4=E5=A4=9A=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/wpm/services.py | 2 ++
 1 file changed, 2 insertions(+)

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作为边的标签
                 })