feat: Support new status of session and kernel

lablup · Nov 13, 2024 · ae8dcb4 · ae8dcb4
1 parent 0479273
commit ae8dcb4
Show file tree

Hide file tree

Showing 9 changed files with 19 additions and 2 deletions.
diff --git a/react/src/components/ComputeSessionNodeItems/SessionStatusTag.tsx b/react/src/components/ComputeSessionNodeItems/SessionStatusTag.tsx
@@ -16,6 +16,7 @@ interface SessionStatusTagProps {
 const statusTagColor = {
   //prepare
   RESTARTING: 'blue',
+  PREPARED: 'blue',
   PREPARING: 'blue',
   PULLING: 'blue',
   //running

diff --git a/react/src/components/SessionList.tsx b/react/src/components/SessionList.tsx
@@ -115,6 +115,7 @@ const SessionList: React.FC<SessionListProps> = ({
   const statusTagColor = {
     //prepare
     RESTARTING: 'blue',
+    PREPARED: 'blue',
     PREPARING: 'blue',
     PULLING: 'blue',
     //running

diff --git a/react/src/components/SessionListColums/SessionInfoCell.tsx b/react/src/components/SessionListColums/SessionInfoCell.tsx
@@ -20,13 +20,14 @@ const isRunningStatus = (status: string = '') => {
     'TERMINATING',
     'PENDING',
     'SCHEDULED',
+    'PREPARED',
     'PREPARING',
     'PULLING',
   ].includes(status);
 };
 
 const isPreparing = (status: string = '') => {
-  return ['RESTARTING', 'PREPARING', 'PULLING'].includes(status);
+  return ['RESTARTING', 'PREPARED', 'PREPARING', 'PULLING'].includes(status);
 };
 
 const SessionInfoCell: React.FC<{

diff --git a/react/src/pages/SessionListPage.tsx b/react/src/pages/SessionListPage.tsx
@@ -15,6 +15,7 @@ const RUNNINGS = [
   'TERMINATING',
   'PENDING',
   'SCHEDULED',
+  'PREPARED',
   'PREPARING',
   'PULLING',
 ];

diff --git a/src/components/backend-ai-edu-applauncher.ts b/src/components/backend-ai-edu-applauncher.ts
@@ -258,6 +258,7 @@ export default class BackendAiEduApplauncher extends BackendAIPage {
         'TERMINATING',
         'PENDING',
         'SCHEDULED',
+        'PREPARED',
         'PREPARING',
         'PULLING',
       ].join(',');
@@ -267,6 +268,7 @@ export default class BackendAiEduApplauncher extends BackendAIPage {
         'RESTARTING',
         'TERMINATING',
         'PENDING',
+        'PREPARED',
         'PREPARING',
         'PULLING',
       ].join(',');

diff --git a/src/components/backend-ai-session-list.ts b/src/components/backend-ai-session-list.ts
@@ -578,7 +578,12 @@ export default class BackendAISessionList extends BackendAIPage {
   }
 
   _isPreparing(status) {
-    const preparingStatuses = ['RESTARTING', 'PREPARING', 'PULLING'];
+    const preparingStatuses = [
+      'RESTARTING',
+      'PREPARED',
+      'PREPARING',
+      'PULLING',
+    ];
     if (preparingStatuses.indexOf(status) === -1) {
       return false;
     }
@@ -760,6 +765,7 @@ export default class BackendAISessionList extends BackendAIPage {
           'TERMINATING',
           'PENDING',
           'SCHEDULED',
+          'PREPARED',
           'PREPARING',
           'PULLING',
           'ERROR',
@@ -775,6 +781,7 @@ export default class BackendAISessionList extends BackendAIPage {
           'TERMINATING',
           'PENDING',
           'SCHEDULED',
+          'PREPARED',
           'PREPARING',
           'PULLING',
         ];

diff --git a/src/components/backend-ai-session-view.ts b/src/components/backend-ai-session-view.ts
@@ -334,6 +334,7 @@ export default class BackendAISessionView extends BackendAIPage {
         'TERMINATING',
         'PENDING',
         'SCHEDULED',
+        'PREPARED',
         'PREPARING',
         'PULLING',
         'TERMINATED',
@@ -346,6 +347,7 @@ export default class BackendAISessionView extends BackendAIPage {
         'RESTARTING',
         'TERMINATING',
         'PENDING',
+        'PREPARED',
         'PREPARING',
         'PULLING',
         'TERMINATED',

diff --git a/src/pipeline/components/pipeline-job-list.ts b/src/pipeline/components/pipeline-job-list.ts
@@ -298,6 +298,7 @@ export default class PipelineJobList extends BackendAIPage {
             [
               'PENDING',
               'SCHEDULED',
+              'PREPARED',
               'PREPARING',
               'BUILDING',
               'PULLING',

diff --git a/src/pipeline/lib/pipeline-utils.ts b/src/pipeline/lib/pipeline-utils.ts
@@ -106,6 +106,7 @@ export default class PipelineUtils extends LitElement {
       case 'BUILDING':
       case 'PULLING':
       case 'SCHEDULED':
+      case 'PREPARED':
       case 'PREPARING':
       case 'PENDING':
         color = 'yellow';