From e9330891582425e05600b81d0ead126906fcff26 Mon Sep 17 00:00:00 2001
From: Ayush Shah <ayush@getcollate.io>
Date: Tue, 16 Sep 2025 08:27:43 +0530
Subject: [PATCH] feat: extend ENTITY_TYPE with additional entity types and
 fields (#23403)

---
 .../org/openmetadata/schema/EntityLink.g4     | 20 +++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/openmetadata-spec/src/main/antlr4/org/openmetadata/schema/EntityLink.g4 b/openmetadata-spec/src/main/antlr4/org/openmetadata/schema/EntityLink.g4
index 4e2ee444beb..72e06578ad2 100644
--- a/openmetadata-spec/src/main/antlr4/org/openmetadata/schema/EntityLink.g4
+++ b/openmetadata-spec/src/main/antlr4/org/openmetadata/schema/EntityLink.g4
@@ -45,6 +45,8 @@ ENTITY_TYPE
     | 'mlmodelService'
     | 'storageService'
     | 'searchService'
+    | 'securityService'
+    | 'driveService'
     | 'webhook'
     | 'mlmodel'
     | 'team'
@@ -54,7 +56,10 @@ ENTITY_TYPE
     | 'policy'
     | 'testSuite'
     | 'testCase'
+    | 'testDefinition'
+    | 'testConnectionDefinition'
     | 'dataInsightChart'
+    | 'dataInsightCustomChart'
     | 'kpi'
     | 'alert'
     | 'container'
@@ -64,6 +69,7 @@ ENTITY_TYPE
     | 'chart'
     | 'domain'
     | 'dataProduct'
+    | 'dataContract'
     | 'sampleData'
     | 'storedProcedure'
     | 'searchIndex'
@@ -81,6 +87,20 @@ ENTITY_TYPE
     | 'apiCollection'
     | 'apiEndpoint'
     | 'apiService'
+    | 'api'
+    | 'metric'
+    | 'type'
+    | 'workflowDefinition'
+    | 'workflow'
+    | 'service'
+    | 'task'
+    | 'report'
+    | 'query'
+    | 'directory'
+    | 'file'
+    | 'spreadsheet'
+    | 'worksheet'
+    | 'webAnalyticEvent'
     ;
 
 ENTITY_FIELD