MINOR - Add APIs to Ref Map (#19019)

* MINOR - Add APIs to Ref Map

* MINOR - Add APIs to Ref Map

* MINOR - Add APIs to Ref Map

* format
This commit is contained in:
Pere Miquel Brull 2024-12-12 10:40:03 +01:00 committed by GitHub
parent c24bc870d0
commit 17c5ba8698
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 0 deletions

View File

@ -12,6 +12,8 @@
"""
Define constants useful for the metadata ingestion
"""
from metadata.generated.schema.entity.data.apiCollection import APICollection
from metadata.generated.schema.entity.data.apiEndpoint import APIEndpoint
from metadata.generated.schema.entity.data.chart import Chart
from metadata.generated.schema.entity.data.container import Container
from metadata.generated.schema.entity.data.dashboard import Dashboard
@ -131,6 +133,8 @@ ENTITY_REFERENCE_CLASS_MAP = {
"metadataService": MetadataService,
"searchService": SearchService,
# Data Asset Entities
"apiCollection": APICollection,
"apiEndpoint": APIEndpoint,
"table": Table,
"storedProcedure": StoredProcedure,
"database": Database,

View File

@ -18,6 +18,8 @@ from pydantic import BaseModel
from metadata.generated.schema.analytics.reportData import ReportData
from metadata.generated.schema.entity.classification.tag import Tag
from metadata.generated.schema.entity.data.apiCollection import APICollection
from metadata.generated.schema.entity.data.apiEndpoint import APIEndpoint
from metadata.generated.schema.entity.data.chart import Chart
from metadata.generated.schema.entity.data.container import Container
from metadata.generated.schema.entity.data.dashboard import Dashboard
@ -33,6 +35,7 @@ from metadata.generated.schema.entity.data.searchIndex import SearchIndex
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.services.apiService import ApiService
from metadata.generated.schema.entity.services.databaseService import DatabaseService
from metadata.generated.schema.entity.teams.team import Team
from metadata.generated.schema.entity.teams.user import User
@ -42,6 +45,9 @@ logger = utils_logger()
T = TypeVar("T", bound=BaseModel)
ES_INDEX_MAP = {
ApiService.__name__: "api_service_search_index",
APICollection.__name__: "api_collection_search_index",
APIEndpoint.__name__: "api_endpoint_search_index",
Table.__name__: "table_search_index",
Database.__name__: "database_search_index",
DatabaseSchema.__name__: "database_schema_search_index",