feat: add the MediaType of layer to inspected result (#226)

Signed-off-by: chlins <[email protected]>

Author: chlins

pkg/backend/inspect.go

@@ -57,6 +57,8 @@ type InspectedModelArtifact struct {
 
 // InspectedModelArtifactLayer is the data structure for model artifact layer that has been inspected.
 type InspectedModelArtifactLayer struct {
+	// MediaType is the media type of the model artifact layer.
+	MediaType string `json:"MediaType"`
 	// Digest is the digest of the model artifact layer.
 	Digest string `json:"Digest"`
 	// Size is the size of the model artifact layer.
@@ -114,9 +116,10 @@ func (b *backend) Inspect(ctx context.Context, target string, cfg *config.Inspec
 
 	for _, layer := range manifest.Layers {
 		inspectedModelArtifact.Layers = append(inspectedModelArtifact.Layers, InspectedModelArtifactLayer{
-			Digest:   layer.Digest.String(),
-			Size:     layer.Size,
-			Filepath: layer.Annotations[modelspec.AnnotationFilepath],
+			MediaType: layer.MediaType,
+			Digest:    layer.Digest.String(),
+			Size:      layer.Size,
+			Filepath:  layer.Annotations[modelspec.AnnotationFilepath],
 		})
 	}
 

pkg/backend/inspect_test.go

@@ -145,6 +145,7 @@ func TestInspect(t *testing.T) {
 	assert.Equal(t, "int8", inspected.Precision)
 	assert.Equal(t, "gptq", inspected.Quantization)
 	assert.Len(t, inspected.Layers, 8)
+	assert.Equal(t, "application/vnd.cnai.model.doc.v1.tar", inspected.Layers[0].MediaType)
 	assert.Equal(t, "sha256:5a96686deb327903f4310e9181ef2ee0bc7261e5181bd23ccdce6c575b6120a2", inspected.Layers[0].Digest)
 	assert.Equal(t, "LICENSE", inspected.Layers[0].Filepath)
 	assert.Equal(t, int64(13312), inspected.Layers[0].Size)