update fields to retrieve HostID and name

models/meshmodel/core/types/types.go

@@ -6,6 +6,7 @@ const (
 	ComponentDefinition    CapabilityType = "component"
 	PolicyDefinition       CapabilityType = "policy"
 	RelationshipDefinition CapabilityType = "relationship"
+	Model				   CapabilityType = "model"
 )
 
 // Each entity will have it's own Filter implementation via which it exposes the nobs and dials to fetch entities

models/meshmodel/core/v1alpha1/models.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/google/uuid"
 	"github.com/layer5io/meshkit/database"
+	"github.com/layer5io/meshkit/models/meshmodel/core/types"
 	"gorm.io/gorm"
 )
 
@@ -38,6 +39,9 @@ type Model struct {
 	Version     string                 `json:"version"`
 	DisplayName string                 `json:"displayName" gorm:"modelDisplayName"`
 	Category    Category               `json:"category"`
+	HostID	    uuid.UUID			   `json:"hostID"`
+	HostName    string				   `json:"hostName"`
+	DisplayHostName string			   `json:"displayHostName"`
 	Metadata    map[string]interface{} `json:"metadata" yaml:"modelMetadata"`
 }
 type ModelDB struct {
@@ -99,3 +103,10 @@ func (c *Model) GetModelDB() (cmd ModelDB) {
 	cmd.Metadata, _ = json.Marshal(c.Metadata)
 	return
 }
+
+func (r Model) Type() types.CapabilityType {
+	return types.Model
+}
+func (r Model) GetID() uuid.UUID {
+	return r.ID
+}

models/meshmodel/registry.go

@@ -180,6 +180,13 @@ func (rm *RegistryManager) GetEntities(f types.Filter) ([]Entity, *int64, *int)
 			en = append(en, pol)
 		}
 		return en, nil, nil
+	case *v1alpha1.ModelFilter:
+		en := make([]Entity, 0)
+		models, count, _ := rm.GetModels(rm.db, filter)
+		for _, model := range models {
+			en = append(en, model)
+		}
+		return en, &count, nil
 	default:
 		return nil, nil, nil
 	}