Skip to content

Commit d551e89

Browse files
committed
fix: submit workflow using bohrium access key
Signed-off-by: zjgemi <[email protected]>
1 parent 4567683 commit d551e89

File tree

1 file changed

+15
-5
lines changed

1 file changed

+15
-5
lines changed

src/dflow/plugins/bohrium.py

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,10 @@
3838
def _raise_error(res, op):
3939
if res["code"] not in succ_code:
4040
if "error" in res:
41-
raise RuntimeError("%s failed: %s" % (op, res["error"]["msg"]))
41+
if isinstance(res["error"], str):
42+
raise RuntimeError("%s failed: %s" % (op, res["error"]))
43+
else:
44+
raise RuntimeError("%s failed: %s" % (op, res["error"]["msg"]))
4245
elif "message" in res:
4346
raise RuntimeError("%s failed: %s" % (op, res["message"]))
4447
else:
@@ -311,10 +314,17 @@ def get_token(self, retry=1):
311314
"projectId": self.project_id,
312315
}
313316
if self.access_key is not None:
314-
url = self.openapi_url + "/openapi/v1/storage/token"
315-
headers = {"x-app-key": self.app_key}
316-
params["accessKey"] = self.access_key
317-
elif self.ticket is not None:
317+
rsp = requests.get(
318+
self.openapi_url + "/openapi/v1/ticket/get",
319+
headers={"x-app-key": self.app_key},
320+
params={"accessKey": self.access_key}
321+
)
322+
res = json.loads(rsp.text)
323+
_raise_error(res, "get ticket")
324+
self.ticket = res["data"]["ticket"]
325+
config["ticket"] = res["data"]["ticket"]
326+
327+
if self.ticket is not None:
318328
headers["Brm-Ticket"] = config["ticket"]
319329
update_headers()
320330
else:

0 commit comments

Comments
 (0)