@@ -53,7 +53,7 @@ func registerModel(db *database.Handler, regID, modelID uuid.UUID) error {
53
53
entity := Registry {
54
54
RegistrantID : regID ,
55
55
Entity : modelID ,
56
- Type : types .ModelDefinition ,
56
+ Type : types .Model ,
57
57
}
58
58
59
59
byt , err := json .Marshal (entity )
@@ -186,7 +186,6 @@ func (rm *RegistryManager) RegisterEntity(h Host, en Entity) error {
186
186
if err != nil {
187
187
return err
188
188
}
189
-
190
189
191
190
entry := Registry {
192
191
ID : uuid .New (),
@@ -251,7 +250,7 @@ func (rm *RegistryManager) GetModels(db *database.Handler, f types.Filter) ([]v1
251
250
var modelWithCategoriess []modelWithCategories
252
251
finder := db .Model (& v1alpha1.ModelDB {}).
253
252
Select ("model_dbs.*, category_dbs.*" ).
254
- Joins ("JOIN category_dbs ON model_dbs.category_id = category_dbs.id" ) //
253
+ Joins ("JOIN category_dbs ON model_dbs.category_id = category_dbs.id" )
255
254
256
255
// total count before pagination
257
256
var count int64
@@ -304,7 +303,13 @@ func (rm *RegistryManager) GetModels(db *database.Handler, f types.Filter) ([]v1
304
303
}
305
304
306
305
for _ , modelDB := range modelWithCategoriess {
307
- m = append (m , modelDB .ModelDB .GetModel (modelDB .GetCategory (db )))
306
+ model := modelDB .ModelDB .GetModel (modelDB .GetCategory (db ))
307
+ // mod := Entity(model)
308
+ host := rm .GetRegistrant (model )
309
+ model .HostID = host .ID
310
+ model .HostName = host .Hostname
311
+ model .DisplayHostName = host .Hostname
312
+ m = append (m , model )
308
313
}
309
314
return m , count , countUniqueModels (modelWithCategoriess )
310
315
}
0 commit comments