yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-09-26 17:34:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix #11214: Ingestion based elastic search index missing serviceType for MLModel and Container (#11215)

Browse Source
This commit is contained in:
Sriharsha Chintalapani 2023-04-22 22:30:37 -07:00 committed by GitHub
parent 947e309bca
commit 9e259be44e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ingestion/src/metadata/ingestion/models/es_documents.py
View File

@ -187,6 +187,7 @@ class MlModelESDocument(BaseModel):
suggest: List[ESSuggest] suggest: List[ESSuggest]
service_suggest: List[ESSuggest] = None service_suggest: List[ESSuggest] = None
service: EntityReference service: EntityReference
serviceType: str
doc_as_upsert: bool = True doc_as_upsert: bool = True
@ -211,6 +212,7 @@ class ContainerESDocument(BaseModel):
suggest: List[ESSuggest] suggest: List[ESSuggest]
service_suggest: List[ESSuggest] = None service_suggest: List[ESSuggest] = None
service: EntityReference service: EntityReference
serviceType: str
doc_as_upsert: bool = True doc_as_upsert: bool = True
parent: Optional[dict] = None parent: Optional[dict] = None
dataModel: Optional[dict] = None dataModel: Optional[dict] = None

2
ingestion/src/metadata/ingestion/sink/elasticsearch.py
View File

@ -741,6 +741,7 @@ def _create_ml_model_es_doc(record: MlModel, _: OpenMetadata) -> MlModelESDocume
followers=followers, followers=followers,
service=record.service, service=record.service,
service_suggest=[ESSuggest(input=record.service.name, weight=5)], 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, size=record.size,
fileFormats=[file_format.value for file_format in record.fileFormats or []], fileFormats=[file_format.value for file_format in record.fileFormats or []],
service_suggest=[ESSuggest(input=record.service.name, weight=5)], service_suggest=[ESSuggest(input=record.service.name, weight=5)],
serviceType=str(record.serviceType.name),
) )