feat: Allows field_ref to be used as a key value in FieldSet (#1756)

Backported-from: main
Backported-to: 23.09

changes/1756.feature.md

@@ -0,0 +1 @@
+In order to be able to use not only alt_name but also field_ref when using the --format option of session list, add values to FieldSet.

src/ai/backend/client/output/types.py

@@ -104,7 +104,9 @@ def _default_formatter(self) -> AbstractOutputFormatter:
 
 class FieldSet(UserDict, Mapping[str, FieldSpec]):
     def __init__(self, fields: Sequence[FieldSpec]) -> None:
-        super().__init__({f.alt_name: f for f in fields})
+        fields_set = {f.alt_name: f for f in fields}
+        fields_set.update({f.field_ref: fields_set[f.alt_name] for f in fields})
+        super().__init__(fields_set)
 
 
 T = TypeVar("T")

tests/client/output/test_types.py

@@ -44,7 +44,7 @@ def test_fieldspec_init():
             ]
         ),
     )
-    assert f.field_ref == "key_foo { bar baz }"
+    assert f.field_ref == "key_foo { bar baz baz }"
     assert f.field_name == "key_foo"
     assert f.humanized_name == "Key Foo"
     assert f.subfields["bar"].field_ref == "bar"
@@ -65,7 +65,7 @@ def test_fieldspec_init():
             ]
         ),
     )
-    assert f.field_ref == "key_foo { bar { kaz } }"
+    assert f.field_ref == "key_foo { bar { kaz } bar { kaz } }"
     assert f.field_name == "key_foo"
     assert f.humanized_name == "Key Foo"
     assert f.subfields["bar"].field_ref == "bar { kaz }"