fix vessel_id param passing in protocols

This commit is contained in:
Junhan Chang
2025-07-25 17:38:17 +08:00
parent 1b086a85f5
commit 0584bd5e04
10 changed files with 94 additions and 10 deletions

View File

@@ -346,7 +346,16 @@ def generate_add_protocol(
"""
# 🔧 核心修改从字典中提取容器ID
vessel_id = vessel["id"]
# 统一处理vessel参数
if isinstance(vessel, dict):
if "id" not in vessel:
vessel_id = list(vessel.values())[0].get("id", "")
else:
vessel_id = vessel.get("id", "")
vessel_data = vessel.get("data", {})
else:
vessel_id = str(vessel)
vessel_data = G.nodes[vessel_id].get("data", {}) if vessel_id in G.nodes() else {}
# 🔧 修改:更新容器的液体体积(假设有 liquid_volume 字段)
if "data" in vessel and "liquid_volume" in vessel["data"]: