fix(monitor): add volc and aliyun jobid (InternLM#338)

* add volc and aliyun jobid * rm workspaceid
yingtongxiong · Sep 25, 2023 · 847cc81 · 847cc81
1 parent 0649655
commit 847cc81
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/internlm/monitor/utils.py b/internlm/monitor/utils.py
@@ -14,8 +14,10 @@ def get_job_id():
     job_id = "none"
     if os.getenv("SLURM_JOB_ID") is not None:
         job_id = os.getenv("SLURM_JOB_ID")
-    elif os.getenv("K8S_WORKSPACE_ID") is not None:
-        job_id = os.getenv("K8S_WORKSPACE_ID")
+    elif os.getenv("KUBERNETES_POD_NAME") is not None:
+        job_id = os.getenv("KUBERNETES_POD_NAME").split("-")[0]
+    elif os.getenv("MLP_TASK_INSTANCE_ID") is not None:
+        job_id = os.getenv("MLP_TASK_ID")
 
     return job_id