add functions to list models defined and services using models

Signed-off-by: Guillaume Lours <[email protected]>

Author: glours

loader/loader_test.go

@@ -32,6 +32,7 @@ import (
 	is "gotest.tools/v3/assert/cmp"
 
 	"github.com/compose-spec/compose-go/v2/types"
+	"github.com/compose-spec/compose-go/v2/utils"
 )
 
 func buildConfigDetails(yaml string, env map[string]string) types.ConfigDetails {
@@ -3899,6 +3900,8 @@ models:
 			ModelVariable:    "MODEL",
 		},
 	})
+	assert.DeepEqual(t, p.ModelNames(), []string{"foo"})
+	assert.Check(t, utils.ArrayContains(p.ServicesWithModels(), []string{"test_array", "test_mapping"}), p.ServicesWithModels())
 }
 
 func TestAttestations(t *testing.T) {

types/project.go

@@ -118,6 +118,16 @@ func (p *Project) ConfigNames() []string {
 	return names
 }
 
+// ModelNames return names for all models in this Compose config
+func (p *Project) ModelNames() []string {
+	var names []string
+	for k := range p.Models {
+		names = append(names, k)
+	}
+	sort.Strings(names)
+	return names
+}
+
 func (p *Project) ServicesWithBuild() []string {
 	servicesBuild := p.Services.Filter(func(s ServiceConfig) bool {
 		return s.Build != nil && s.Build.Context != ""
@@ -139,6 +149,11 @@ func (p *Project) ServicesWithDependsOn() []string {
 	return slices.Collect(maps.Keys(servicesDependsOn))
 }
 
+func (p *Project) ServicesWithModels() []string {
+	servicesModels := p.Services.Filter(func(s ServiceConfig) bool { return len(s.Models) > 0 })
+	return slices.Collect(maps.Keys(servicesModels))
+}
+
 func (p *Project) ServicesWithCapabilities() ([]string, []string, []string) {
 	capabilities := []string{}
 	gpu := []string{}