From 7662aad1b42b2024341f0fec53e34b8656dd48dd Mon Sep 17 00:00:00 2001
From: Matthew Anderson <42154938+matoszz@users.noreply.github.com>
Date: Thu, 4 Jun 2026 22:20:02 -0500
Subject: [PATCH] add the integration edge ref

---
 .../integrations/operations/ingest_internalpolicy_persist.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/internal/integrations/operations/ingest_internalpolicy_persist.go b/internal/integrations/operations/ingest_internalpolicy_persist.go
index 0d46c96481..dfd874b107 100644
--- a/internal/integrations/operations/ingest_internalpolicy_persist.go
+++ b/internal/integrations/operations/ingest_internalpolicy_persist.go
@@ -3,6 +3,7 @@ package operations
 
 import (
 	"context"
+	"slices"
 
 	ent "github.com/theopenlane/core/internal/ent/generated"
 	"github.com/theopenlane/core/internal/ent/generated/internalpolicy"
@@ -14,6 +15,10 @@ func persistInternalPolicyInput(ctx context.Context, db *ent.Client, integration
 		return ErrIngestUpsertKeyMissing
 	}
 
+	if integration.ID != "" && !slices.Contains(createInput.IntegrationIDs, integration.ID) {
+		createInput.IntegrationIDs = append(createInput.IntegrationIDs, integration.ID)
+	}
+
 	return persistRoundTripUpsert(
 		ctx,
 		createInput,