diff --git a/ingestion/src/metadata/utils/constants.py b/ingestion/src/metadata/utils/constants.py index 9de52b2bce88..2a3053140d44 100644 --- a/ingestion/src/metadata/utils/constants.py +++ b/ingestion/src/metadata/utils/constants.py @@ -21,15 +21,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 @@ -68,6 +72,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 @@ -132,6 +137,7 @@ "metadataService": MetadataService, "searchService": SearchService, "securityService": SecurityService, + "driveService": DriveService, # Data Asset Entities "apiCollection": APICollection, "apiEndpoint": APIEndpoint, @@ -147,6 +153,11 @@ "searchIndex": SearchIndex, "mlmodel": MlModel, "container": Container, + # Drive Entities + "directory": Directory, + "file": File, + "spreadsheet": Spreadsheet, + "worksheet": Worksheet, # User Entities "user": User, "team": Team, diff --git a/ingestion/src/metadata/utils/elasticsearch.py b/ingestion/src/metadata/utils/elasticsearch.py index 36bd1f5ed97b..7e50d1c82fdd 100644 --- a/ingestion/src/metadata/utils/elasticsearch.py +++ b/ingestion/src/metadata/utils/elasticsearch.py @@ -29,6 +29,8 @@ 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 @@ -36,9 +38,11 @@ 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",