feat: Display architecture of agents and images (#212)

lablup · Mar 20, 2022 · 07f59d4 · 07f59d4
1 parent 0514cc9
commit 07f59d4
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 0 deletions.
diff --git a/changes/212.feature.md b/changes/212.feature.md
@@ -0,0 +1 @@
+Display the architecture field in agent and image lists
diff --git a/src/ai/backend/client/cli/admin/agent.py b/src/ai/backend/client/cli/admin/agent.py
@@ -28,6 +28,7 @@ def info(ctx: CLIContext, agent_id: str) -> None:
         agent_fields['id'],
         agent_fields['status'],
         agent_fields['region'],
+        agent_fields['architecture'],
         agent_fields['first_contact'],
         agent_fields['cpu_cur_pct'],
         agent_fields['available_slots'],
@@ -74,6 +75,7 @@ def list(
     fields = [
         agent_fields['id'],
         agent_fields['status'],
+        agent_fields['architecture'],
         agent_fields['scaling_group'],
         agent_fields['region'],
         agent_fields['first_contact'],

diff --git a/src/ai/backend/client/func/image.py b/src/ai/backend/client/func/image.py
@@ -13,6 +13,7 @@
 _default_list_fields_admin = (
     image_fields['name'],
     image_fields['registry'],
+    image_fields['architecture'],
     image_fields['tag'],
     image_fields['digest'],
     image_fields['size_bytes'],

diff --git a/src/ai/backend/client/output/fields.py b/src/ai/backend/client/output/fields.py
@@ -44,6 +44,7 @@
     FieldSpec('status'),
     FieldSpec('status_changed'),
     FieldSpec('region'),
+    FieldSpec('architecture'),
     FieldSpec('scaling_group'),
     FieldSpec('schedulable'),
     FieldSpec('available_slots', formatter=resource_slot_formatter),
@@ -89,6 +90,7 @@
 image_fields = FieldSet([
     FieldSpec('name'),
     FieldSpec('registry'),
+    FieldSpec('architecture'),
     FieldSpec('tag'),
     FieldSpec('digest'),
     FieldSpec('size_bytes', formatter=sizebytes_output_formatter),