Ignore banned or deleted versions in latest version. (#124)

Comfy-Org · Jan 6, 2025 · e57a9c0 · e57a9c0
1 parent b200f30
commit e57a9c0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/services/registry/registry_svc.go b/services/registry/registry_svc.go
@@ -538,7 +538,11 @@ func (s *RegistryService) GetLatestNodeVersion(ctx context.Context, client *ent.
 	nodeVersion, err := client.NodeVersion.
 		Query().
 		Where(nodeversion.NodeIDEQ(nodeId)).
-		//Where(nodeversion.StatusEQ(schema.NodeVersionStatusActive)).
+		Where(nodeversion.StatusIn(
+			schema.NodeVersionStatusActive,
+			schema.NodeVersionStatusFlagged,
+			schema.NodeVersionStatusPending,
+		)).
 		Order(ent.Desc(nodeversion.FieldCreateTime)).
 		WithStorageFile().
 		WithComfyNodes().