Skip to content

Commit fcc8911

Browse files
ulixius9claude
andcommitted
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>
1 parent 89c6d21 commit fcc8911

1 file changed

Lines changed: 11 additions & 0 deletions

File tree

ingestion/src/metadata/utils/constants.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,15 +21,19 @@
2121
from metadata.generated.schema.entity.data.dashboardDataModel import DashboardDataModel
2222
from metadata.generated.schema.entity.data.database import Database
2323
from metadata.generated.schema.entity.data.databaseSchema import DatabaseSchema
24+
from metadata.generated.schema.entity.data.directory import Directory
25+
from metadata.generated.schema.entity.data.file import File
2426
from metadata.generated.schema.entity.data.glossary import Glossary
2527
from metadata.generated.schema.entity.data.glossaryTerm import GlossaryTerm
2628
from metadata.generated.schema.entity.data.metric import Metric
2729
from metadata.generated.schema.entity.data.mlmodel import MlModel
2830
from metadata.generated.schema.entity.data.pipeline import Pipeline
2931
from metadata.generated.schema.entity.data.searchIndex import SearchIndex
32+
from metadata.generated.schema.entity.data.spreadsheet import Spreadsheet
3033
from metadata.generated.schema.entity.data.storedProcedure import StoredProcedure
3134
from metadata.generated.schema.entity.data.table import Table
3235
from metadata.generated.schema.entity.data.topic import Topic
36+
from metadata.generated.schema.entity.data.worksheet import Worksheet
3337
from metadata.generated.schema.entity.domains.dataProduct import DataProduct
3438
from metadata.generated.schema.entity.domains.domain import Domain
3539
from metadata.generated.schema.entity.services.apiService import ApiService
@@ -68,6 +72,7 @@
6872
)
6973
from metadata.generated.schema.entity.services.dashboardService import DashboardService
7074
from metadata.generated.schema.entity.services.databaseService import DatabaseService
75+
from metadata.generated.schema.entity.services.driveService import DriveService
7176
from metadata.generated.schema.entity.services.messagingService import MessagingService
7277
from metadata.generated.schema.entity.services.metadataService import MetadataService
7378
from metadata.generated.schema.entity.services.mlmodelService import MlModelService
@@ -132,6 +137,7 @@
132137
"metadataService": MetadataService,
133138
"searchService": SearchService,
134139
"securityService": SecurityService,
140+
"driveService": DriveService,
135141
# Data Asset Entities
136142
"apiCollection": APICollection,
137143
"apiEndpoint": APIEndpoint,
@@ -147,6 +153,11 @@
147153
"searchIndex": SearchIndex,
148154
"mlmodel": MlModel,
149155
"container": Container,
156+
# Drive Entities
157+
"directory": Directory,
158+
"file": File,
159+
"spreadsheet": Spreadsheet,
160+
"worksheet": Worksheet,
150161
# User Entities
151162
"user": User,
152163
"team": Team,

0 commit comments

Comments
 (0)