From e57a9c06108567af34e9e617d5835a3d80bf4a60 Mon Sep 17 00:00:00 2001
From: Robin Huang <robin.j.huang@gmail.com>
Date: Mon, 6 Jan 2025 15:31:45 -0800
Subject: [PATCH] Ignore banned or deleted versions in latest version. (#124)

---
 services/registry/registry_svc.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/services/registry/registry_svc.go b/services/registry/registry_svc.go
index a65f969..b9af588 100644
--- 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().