@@ -80,6 +80,9 @@ class AquaContainerConfigItem(Serializable):
80
80
default_factory = AquaContainerConfigSpec ,
81
81
description = "Detailed container specification." ,
82
82
)
83
+ usages : Optional [List [str ]] = Field (
84
+ default_factory = list , description = "Supported usages."
85
+ )
83
86
84
87
class Config :
85
88
extra = "allow"
@@ -131,7 +134,7 @@ def from_container_index_json(
131
134
-------
132
135
AquaContainerConfig: The constructed container configuration.
133
136
"""
134
- #TODO: Return this logic back if necessary in the next iteraion.
137
+ # TODO: Return this logic back if necessary in the next iteraion.
135
138
# if not config:
136
139
# config = get_container_config()
137
140
@@ -144,6 +147,7 @@ def from_container_index_json(
144
147
for container in containers :
145
148
platforms = container .get ("platforms" , [])
146
149
model_formats = container .get ("modelFormats" , [])
150
+ usages = container .get ("usages" , [])
147
151
container_spec = (
148
152
config .get (ContainerSpec .CONTAINER_SPEC , {}).get (
149
153
container_type , {}
@@ -160,6 +164,7 @@ def from_container_index_json(
160
164
family = container_type ,
161
165
platforms = platforms ,
162
166
model_formats = model_formats ,
167
+ usages = usages ,
163
168
spec = (
164
169
AquaContainerConfigSpec (
165
170
cli_param = container_spec .get (
0 commit comments