Don't include node version's status reason in ListAllNodes (#123)

* Don't include status reason in list nodes. * Fixed tests.
Comfy-Org · Jan 6, 2025 · b200f30 · b200f30
1 parent 5753550
commit b200f30
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/integration-tests/registry_integration_test.go b/integration-tests/registry_integration_test.go
@@ -743,6 +743,7 @@ func TestRegistryNodeVersion(t *testing.T) {
 			expectedNode.LatestVersion.DownloadUrl = node.LatestVersion.DownloadUrl // generated
 			expectedNode.LatestVersion.Deprecated = node.LatestVersion.Deprecated   // generated
 			expectedNode.LatestVersion.CreatedAt = node.LatestVersion.CreatedAt     // generated
+			expectedNode.LatestVersion.StatusReason = nil                           // Filtered out
 			expectedNode.Publisher.CreatedAt = node.Publisher.CreatedAt
 			assert.Equal(t, expectedNode, node)
 		}

diff --git a/server/implementation/registry.go b/server/implementation/registry.go
@@ -289,6 +289,7 @@ func (s *DripStrictServerImplementation) ListAllNodes(
 		// attach information of latest version if available
 		if len(dbNode.Edges.Versions) > 0 {
 			apiNode.LatestVersion = mapper.DbNodeVersionToApiNodeVersion(dbNode.Edges.Versions[0])
+			apiNode.LatestVersion.StatusReason = nil
 		}
 
 		// Map publisher information