fix(ingestion): register Drive entities in ENTITY_REFERENCE_CLASS_MAP and ES_INDEX_MAP (#28104)

* fix(ingestion): add Drive entities to ENTITY_REFERENCE_CLASS_MAP

The Automator app resolves resource types from resources.type against
ENTITY_REFERENCE_CLASS_MAP. Drive entities (directory, file, spreadsheet,
worksheet) and the driveService were missing, so selecting them in the
Add Automation UI failed at runtime with:

  AutomatorException: Can't get class from resource type: directory

Register the Drive data entities and driveService so the Automator can
paginate and act on them.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

* Adding missng drive entity to ES MAP

---------

Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Co-authored-by: Khairajani <himanshukhairajani8@gmail.com>
Co-authored-by: Himanshu Khairajani <46777429+Khairajani@users.noreply.github.com>

Author: 3 people

ingestion/src/metadata/utils/constants.py

@@ -20,15 +20,19 @@
 from metadata.generated.schema.entity.data.dashboardDataModel import DashboardDataModel
 from metadata.generated.schema.entity.data.database import Database
 from metadata.generated.schema.entity.data.databaseSchema import DatabaseSchema
+from metadata.generated.schema.entity.data.directory import Directory
+from metadata.generated.schema.entity.data.file import File
 from metadata.generated.schema.entity.data.glossary import Glossary
 from metadata.generated.schema.entity.data.glossaryTerm import GlossaryTerm
 from metadata.generated.schema.entity.data.metric import Metric
 from metadata.generated.schema.entity.data.mlmodel import MlModel
 from metadata.generated.schema.entity.data.pipeline import Pipeline
 from metadata.generated.schema.entity.data.searchIndex import SearchIndex
+from metadata.generated.schema.entity.data.spreadsheet import Spreadsheet
 from metadata.generated.schema.entity.data.storedProcedure import StoredProcedure
 from metadata.generated.schema.entity.data.table import Table
 from metadata.generated.schema.entity.data.topic import Topic
+from metadata.generated.schema.entity.data.worksheet import Worksheet
 from metadata.generated.schema.entity.domains.dataProduct import DataProduct
 from metadata.generated.schema.entity.domains.domain import Domain
 from metadata.generated.schema.entity.services.apiService import ApiService
@@ -67,6 +71,7 @@
 )
 from metadata.generated.schema.entity.services.dashboardService import DashboardService
 from metadata.generated.schema.entity.services.databaseService import DatabaseService
+from metadata.generated.schema.entity.services.driveService import DriveService
 from metadata.generated.schema.entity.services.messagingService import MessagingService
 from metadata.generated.schema.entity.services.metadataService import MetadataService
 from metadata.generated.schema.entity.services.mlmodelService import MlModelService
@@ -131,6 +136,7 @@
     "metadataService": MetadataService,
     "searchService": SearchService,
     "securityService": SecurityService,
+    "driveService": DriveService,
     # Data Asset Entities
     "apiCollection": APICollection,
     "apiEndpoint": APIEndpoint,
@@ -146,6 +152,11 @@
     "searchIndex": SearchIndex,
     "mlmodel": MlModel,
     "container": Container,
+    # Drive Entities
+    "directory": Directory,
+    "file": File,
+    "spreadsheet": Spreadsheet,
+    "worksheet": Worksheet,
     # User Entities
     "user": User,
     "team": Team,

ingestion/src/metadata/utils/elasticsearch.py

@@ -29,16 +29,20 @@
 from metadata.generated.schema.entity.data.dashboardDataModel import DashboardDataModel
 from metadata.generated.schema.entity.data.database import Database
 from metadata.generated.schema.entity.data.databaseSchema import DatabaseSchema
+from metadata.generated.schema.entity.data.directory import Directory
+from metadata.generated.schema.entity.data.file import File
 from metadata.generated.schema.entity.data.glossary import Glossary
 from metadata.generated.schema.entity.data.glossaryTerm import GlossaryTerm
 from metadata.generated.schema.entity.data.metric import Metric
 from metadata.generated.schema.entity.data.mlmodel import MlModel
 from metadata.generated.schema.entity.data.pipeline import Pipeline
 from metadata.generated.schema.entity.data.query import Query
 from metadata.generated.schema.entity.data.searchIndex import SearchIndex
+from metadata.generated.schema.entity.data.spreadsheet import Spreadsheet
 from metadata.generated.schema.entity.data.storedProcedure import StoredProcedure
 from metadata.generated.schema.entity.data.table import Table
 from metadata.generated.schema.entity.data.topic import Topic
+from metadata.generated.schema.entity.data.worksheet import Worksheet
 from metadata.generated.schema.entity.services.apiService import ApiService
 from metadata.generated.schema.entity.services.databaseService import DatabaseService
 from metadata.generated.schema.entity.teams.team import Team
@@ -71,6 +75,10 @@
     Tag.__name__: "tag_search_index",
     Classification.__name__: "classification_search_index",
     Container.__name__: "container_search_index",
+    Directory.__name__: "directory_search_index",
+    File.__name__: "file_search_index",
+    Spreadsheet.__name__: "spreadsheet_search_index",
+    Worksheet.__name__: "worksheet_search_index",
     Query.__name__: "query_search_index",
     ReportData.__name__: "entity_report_data_index",
     Metric.__name__: "metric_search_index",