Return node_id in node versions. (#115)

* Return node_id in node versions. * Update tests.
Comfy-Org · Jan 4, 2025 · 43947e4 · 43947e4
1 parent c46dd20
commit 43947e4
Show file tree

Hide file tree

Showing 4 changed files with 87 additions and 78 deletions.
diff --git a/drip/api.gen.go b/drip/api.gen.go
diff --git a/integration-tests/registry_integration_test.go b/integration-tests/registry_integration_test.go
@@ -620,6 +620,7 @@ func TestRegistryNodeVersion(t *testing.T) {
 			DownloadUrl:  &downloadUrl,
 			Status:       &nodeVersionStatus,
 			StatusReason: proto.String(""),
+			NodeId:       node.Id,
 		}, resVersions200[0], "should be equal")
 
 		// retrieve node versions with the status reason
@@ -668,6 +669,7 @@ func TestRegistryNodeVersion(t *testing.T) {
 			DownloadUrl:  &downloadUrl,
 			Status:       &status,
 			StatusReason: proto.String(""),
+			NodeId:       node.Id,
 		}
 		assert.Equal(t, updatedNodeVersion, res200[0], "should be equal")
 		createdNodeVersion = res200[0]

diff --git a/mapper/node_version.go b/mapper/node_version.go
@@ -79,6 +79,7 @@ func DbNodeVersionToApiNodeVersion(dbNodeVersion *ent.NodeVersion) *drip.NodeVer
 		StatusReason: &dbNodeVersion.StatusReason,
 		DownloadUrl:  &downloadUrl,
 		ComfyNodes:   comfyNodes,
+		NodeId:       &dbNodeVersion.NodeID,
 	}
 	return apiVersion
 }

diff --git a/openapi.yml b/openapi.yml
@@ -2227,6 +2227,9 @@ components:
           type: object 
           additionalProperties:
             $ref: "#/components/schemas/ComfyNode"
+        node_id:
+          type: string
+          description: The unique identifier of the node.
     ComfyNode:
       type: object
       properties: