diff --git a/ingestion/src/metadata/ingestion/models/es_documents.py b/ingestion/src/metadata/ingestion/models/es_documents.py index 8df6260a121..c7b8c42ac85 100644 --- a/ingestion/src/metadata/ingestion/models/es_documents.py +++ b/ingestion/src/metadata/ingestion/models/es_documents.py @@ -187,6 +187,7 @@ class MlModelESDocument(BaseModel): suggest: List[ESSuggest] service_suggest: List[ESSuggest] = None service: EntityReference + serviceType: str doc_as_upsert: bool = True @@ -211,6 +212,7 @@ class ContainerESDocument(BaseModel): suggest: List[ESSuggest] service_suggest: List[ESSuggest] = None service: EntityReference + serviceType: str doc_as_upsert: bool = True parent: Optional[dict] = None dataModel: Optional[dict] = None diff --git a/ingestion/src/metadata/ingestion/sink/elasticsearch.py b/ingestion/src/metadata/ingestion/sink/elasticsearch.py index c378a33e20b..a1dc3d6454a 100644 --- a/ingestion/src/metadata/ingestion/sink/elasticsearch.py +++ b/ingestion/src/metadata/ingestion/sink/elasticsearch.py @@ -741,6 +741,7 @@ def _create_ml_model_es_doc(record: MlModel, _: OpenMetadata) -> MlModelESDocume followers=followers, service=record.service, service_suggest=[ESSuggest(input=record.service.name, weight=5)], + serviceType=str(record.serviceType.name), ) @@ -778,6 +779,7 @@ def _create_container_es_doc(record: Container, _: OpenMetadata) -> ContainerESD size=record.size, fileFormats=[file_format.value for file_format in record.fileFormats or []], service_suggest=[ESSuggest(input=record.service.name, weight=5)], + serviceType=str(record.serviceType.name), )