From 6f9773157c6798b2cbba96a0fe80c2d1f1125adf Mon Sep 17 00:00:00 2001
From: wznln <18435084+Xuwznln@users.noreply.github.com>
Date: Wed, 7 May 2025 03:26:22 +0800
Subject: [PATCH] fix: jobadd

---
 unilabos/app/mq.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/unilabos/app/mq.py b/unilabos/app/mq.py
index d0975e91..153f94aa 100644
--- a/unilabos/app/mq.py
+++ b/unilabos/app/mq.py
@@ -55,6 +55,12 @@ class MQTTClient:
             logger.debug("Payload:", json.dumps(payload_json, indent=2, ensure_ascii=False))
             if msg.topic == f"labs/{MQConfig.lab_id}/job/start/":
                 logger.debug("job_add", type(payload_json), payload_json)
+                if "data" not in payload_json:
+                    payload_json["data"] = {}
+                if "action" in payload_json:
+                    payload_json["data"]["action"] = payload_json.pop("action")
+                if "action_kwargs" in payload_json:
+                    payload_json["data"]["action_kwargs"] = payload_json.pop("action_kwargs")
                 job_req = JobAddReq.model_validate(payload_json)
                 data = job_add(job_req)
                 return