openmetadata-ingestion 1.6.7.1__py3-none-any.whl → 1.7.0.0rc1__py3-none-any.whl

metadata/generated/schema/api/data/createAPICollection.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createAPICollection.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createAPIEndpoint.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createAPIEndpoint.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createChart.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createChart.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createContainer.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createContainer.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createCustomProperty.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createCustomProperty.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createDashboard.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createDashboard.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createDashboardDataModel.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createDashboardDataModel.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createDatabase.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createDatabase.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createDatabaseSchema.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createDatabaseSchema.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createGlossary.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createGlossary.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createGlossaryTerm.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createGlossaryTerm.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createMetric.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createMetric.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createMlModel.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createMlModel.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createPipeline.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createPipeline.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 
@@ -64,6 +64,10 @@ class CreatePipelineRequest(BaseModel):
         Optional[List[pipeline.Task]],
         Field(None, description='All the tasks that are part of pipeline.'),
     ]
+    state: Annotated[
+        Optional[pipeline.PipelineState],
+        Field(None, description='State of the pipeline.'),
+    ]
     tags: Annotated[
         Optional[List[tagLabel.TagLabel]],
         Field(None, description='Tags for this Pipeline.'),

metadata/generated/schema/api/data/createQuery.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createQuery.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createQueryCostRecord.py

@@ -0,0 +1,35 @@
+# generated by datamodel-codegen:
+#   filename:  api/data/createQueryCostRecord.json
+#   timestamp: 2025-04-02T05:35:35+00:00
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import ConfigDict, Field
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
+
+from ...type import basic, entityReference
+
+
+class CreateQueryCostRecordRequest(BaseModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    timestamp: Annotated[
+        basic.Timestamp,
+        Field(description='Timestamp on which the failure was created.'),
+    ]
+    jsonSchema: Annotated[
+        Optional[str], Field(None, description='Json schema of the query')
+    ]
+    queryReference: Annotated[
+        entityReference.EntityReference, Field(description='Query entity reference')
+    ]
+    cost: Annotated[float, Field(description='Avg query cost per execution')]
+    count: Annotated[float, Field(description='Number of times the query was executed')]
+    totalDuration: Annotated[
+        Optional[float], Field(0, description='Total duration of the query')
+    ]

metadata/generated/schema/api/data/createSearchIndex.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createSearchIndex.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createStoredProcedure.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createStoredProcedure.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createTable.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createTable.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createTableProfile.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createTableProfile.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/createTopic.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/createTopic.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/loadGlossary.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/loadGlossary.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/data/restoreEntity.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/data/restoreEntity.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/dataInsight/__init__.py

@@ -1,3 +1,3 @@
 # generated by datamodel-codegen:
 #   filename:  schema
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00

metadata/generated/schema/api/dataInsight/createDataInsightChart.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/dataInsight/createDataInsightChart.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/dataInsight/custom/__init__.py

@@ -1,3 +1,3 @@
 # generated by datamodel-codegen:
 #   filename:  schema
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00

metadata/generated/schema/api/dataInsight/custom/createDataInsightCustomChart.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/dataInsight/custom/createDataInsightCustomChart.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/dataInsight/kpi/__init__.py

@@ -1,3 +1,3 @@
 # generated by datamodel-codegen:
 #   filename:  schema
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00

metadata/generated/schema/api/dataInsight/kpi/createKpiRequest.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/dataInsight/kpi/createKpiRequest.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/docStore/__init__.py

@@ -1,3 +1,3 @@
 # generated by datamodel-codegen:
 #   filename:  schema
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00

metadata/generated/schema/api/docStore/createDocument.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/docStore/createDocument.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/domains/__init__.py

@@ -1,3 +1,3 @@
 # generated by datamodel-codegen:
 #   filename:  schema
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00

metadata/generated/schema/api/domains/createDataProduct.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/domains/createDataProduct.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/domains/createDomain.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/domains/createDomain.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 
@@ -54,3 +54,10 @@ class CreateDomainRequest(BaseModel):
         Optional[List[tagLabel.TagLabel]],
         Field(None, description='Tags for this Domain.'),
     ]
+    extension: Annotated[
+        Optional[basic.EntityExtension],
+        Field(
+            None,
+            description='Entity extension data with custom attributes added to the entity.',
+        ),
+    ]

metadata/generated/schema/api/feed/__init__.py

@@ -1,3 +1,3 @@
 # generated by datamodel-codegen:
 #   filename:  schema
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00

metadata/generated/schema/api/feed/closeTask.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/feed/closeTask.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/feed/createPost.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/feed/createPost.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/feed/createSuggestion.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/feed/createSuggestion.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/feed/createThread.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/feed/createThread.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/feed/resolveTask.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/feed/resolveTask.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/feed/threadCount.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/feed/threadCount.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 
@@ -40,4 +40,28 @@ class CountOfThreadsRelatedToAnEntity(BaseModel):
         Optional[int],
         Field(None, description='Total count of all the mentions of a user.', ge=0),
     ]
+    totalAnnouncementCount: Annotated[
+        Optional[int],
+        Field(
+            None,
+            description='Total count of all the announcements associated with the entity.',
+            ge=0,
+        ),
+    ]
+    activeAnnouncementCount: Annotated[
+        Optional[int],
+        Field(
+            None,
+            description='Total count of all the active announcements associated with the entity.',
+            ge=0,
+        ),
+    ]
+    inactiveAnnouncementCount: Annotated[
+        Optional[int],
+        Field(
+            None,
+            description='Total count of all the inactive announcements associated with the entity.',
+            ge=0,
+        ),
+    ]
     entityLink: Optional[basic.EntityLink] = None

metadata/generated/schema/api/governance/__init__.py

@@ -1,3 +1,3 @@
 # generated by datamodel-codegen:
 #   filename:  schema
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00

metadata/generated/schema/api/governance/createWorkflowDefinition.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/governance/createWorkflowDefinition.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/governance/createWorkflowInstanceState.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/governance/createWorkflowInstanceState.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/lineage/__init__.py

@@ -1,3 +1,3 @@
 # generated by datamodel-codegen:
 #   filename:  schema
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00

metadata/generated/schema/api/lineage/addLineage.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  api/lineage/addLineage.json
-#   timestamp: 2025-04-02T12:35:09+00:00
+#   timestamp: 2025-04-02T05:35:35+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/api/lineage/esLineageData.py

@@ -0,0 +1,85 @@
+# generated by datamodel-codegen:
+#   filename:  api/lineage/esLineageData.json
+#   timestamp: 2025-04-02T05:35:35+00:00
+
+from __future__ import annotations
+
+from typing import Any, List, Optional
+
+from pydantic import ConfigDict, Field
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
+
+from ...type import basic, entityLineage
+
+
+class RelationshipRef(BaseModel):
+    id: Annotated[
+        Optional[basic.Uuid],
+        Field(None, description='Unique identifier of this entity instance.'),
+    ]
+    fullyQualifiedName: Annotated[
+        Optional[basic.FullyQualifiedEntityName],
+        Field(None, description='FullyQualifiedName of the entity.'),
+    ]
+    fqnHash: Annotated[
+        Optional[str], Field(None, description='FullyQualifiedName Hash of the entity.')
+    ]
+    type: Annotated[Optional[str], Field(None, description='Type of the entity.')]
+
+
+class EsLineageData(BaseModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    fromEntity: Annotated[
+        Optional[RelationshipRef], Field(None, description='From Entity.')
+    ]
+    toEntity: Annotated[
+        Optional[RelationshipRef], Field(None, description='To Entity.')
+    ]
+    pipeline: Annotated[
+        Optional[Any],
+        Field(
+            None, description='Pipeline in case pipeline is present between entities.'
+        ),
+    ]
+    sqlQuery: Annotated[Optional[str], Field(None, description='Sql Query associated.')]
+    columns: Annotated[
+        Optional[List[entityLineage.ColumnLineage]],
+        Field(None, description='Columns associated.'),
+    ]
+    description: Annotated[Optional[str], Field(None, description='Description.')]
+    source: Annotated[Optional[str], Field(None, description='Source of the Lineage.')]
+    docId: Annotated[Optional[str], Field(None, description='Doc Id for the Lineage.')]
+    docUniqueId: Annotated[
+        Optional[str], Field(None, description='Doc Unique Id for the Lineage.')
+    ]
+    pipelineEntityType: Annotated[
+        Optional[str], Field(None, description='Pipeline Entity or Stored procedure.')
+    ]
+    createdAt: Annotated[
+        Optional[basic.Timestamp],
+        Field(
+            None,
+            description='Last update time corresponding to the new version of the entity in Unix epoch time milliseconds.',
+        ),
+    ]
+    createdBy: Annotated[
+        Optional[str], Field(None, description='User who created the node.')
+    ]
+    updatedAt: Annotated[
+        Optional[basic.Timestamp],
+        Field(
+            None,
+            description='Last update time corresponding to the new version of the entity in Unix epoch time milliseconds.',
+        ),
+    ]
+    updatedBy: Annotated[
+        Optional[str], Field(None, description='User who made the update.')
+    ]
+    assetEdges: Annotated[
+        Optional[int],
+        Field(None, description='Asset count in case of child assets lineage.'),
+    ]

metadata/generated/schema/api/lineage/lineageDirection.py

@@ -0,0 +1,12 @@
+# generated by datamodel-codegen:
+#   filename:  api/lineage/lineageDirection.json
+#   timestamp: 2025-04-02T05:35:35+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+
+
+class LineageDirection(Enum):
+    Upstream = 'Upstream'
+    Downstream = 'Downstream'

metadata/generated/schema/api/lineage/nodeInformation.py

@@ -0,0 +1,19 @@
+# generated by datamodel-codegen:
+#   filename:  api/lineage/nodeInformation.json
+#   timestamp: 2025-04-02T05:35:35+00:00
+
+from __future__ import annotations
+
+from typing import Any, Optional
+
+from pydantic import Field
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
+
+from . import searchLineageResult
+
+
+class NodeInformation(BaseModel):
+    entity: Annotated[Optional[Any], Field(None, description='Entity object.')]
+    paging: Optional[searchLineageResult.LayerPaging] = None

metadata/generated/schema/api/lineage/searchLineageRequest.py

@@ -0,0 +1,45 @@
+# generated by datamodel-codegen:
+#   filename:  api/lineage/searchLineageRequest.json
+#   timestamp: 2025-04-02T05:35:35+00:00
+
+from __future__ import annotations
+
+from typing import List, Optional
+
+from pydantic import ConfigDict, Field
+from typing_extensions import Annotated
+
+from metadata.ingestion.models.custom_pydantic import BaseModel
+
+from . import lineageDirection
+
+
+class SearchLineageRequest(BaseModel):
+    model_config = ConfigDict(
+        extra='forbid',
+    )
+    fqn: Annotated[str, Field(description='Entity Fqn to search lineage')]
+    isConnectedVia: Annotated[Optional[bool], Field(None, description='Connected Via')]
+    direction: lineageDirection.LineageDirection
+    directionValue: Annotated[
+        Optional[List[str]], Field(None, description='Lineage Direction Value.')
+    ]
+    upstreamDepth: Annotated[
+        Optional[int], Field(3, description='The upstream depth of the lineage')
+    ]
+    downstreamDepth: Annotated[
+        Optional[int], Field(3, description='The downstream depth of the lineage')
+    ]
+    layerFrom: Annotated[
+        Optional[int], Field(0, description='Layer to start the search from.')
+    ]
+    layerSize: Annotated[
+        Optional[int], Field(1000, description='Size of the search result.')
+    ]
+    queryFilter: Annotated[Optional[str], Field(None, description='Query Filter')]
+    includeDeleted: Annotated[
+        Optional[bool], Field(None, description='Include deleted entities')
+    ]
+    includeSourceFields: Annotated[
+        Optional[List[str]], Field(None, description='Include source fields')
+    ]