openmetadata-ingestion 1.5.8.0__py3-none-any.whl → 1.5.9.0__py3-none-any.whl

metadata/generated/schema/type/bulkOperationResult.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/bulkOperationResult.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/changeEvent.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/changeEvent.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/changeEventType.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/changeEventType.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/collectionDescriptor.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/collectionDescriptor.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/csvDocumentation.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/csvDocumentation.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/csvErrorType.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/csvErrorType.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/csvFile.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/csvFile.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/csvImportResult.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/csvImportResult.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/customProperties/__init__.py

@@ -1,3 +1,3 @@
 # generated by datamodel-codegen:
 #   filename:  schema
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00

metadata/generated/schema/type/customProperties/complexTypes.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/customProperties/complexTypes.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/customProperties/enumConfig.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/customProperties/enumConfig.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/customProperties/tableConfig.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/customProperties/tableConfig.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/customProperty.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/customProperty.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/dailyCount.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/dailyCount.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/databaseConnectionConfig.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/databaseConnectionConfig.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/entityHierarchy.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/entityHierarchy.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/entityHistory.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/entityHistory.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/entityLineage.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/entityLineage.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/entityReference.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/entityReference.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/entityReferenceList.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/entityReferenceList.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/entityRelationship.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/entityRelationship.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/entityUsage.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/entityUsage.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/filterPattern.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/filterPattern.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/function.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/function.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/include.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/include.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/jdbcConnection.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/jdbcConnection.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/lifeCycle.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/lifeCycle.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/paging.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/paging.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/profile.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/profile.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/queryParserData.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/queryParserData.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/reaction.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/reaction.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/schedule.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/schedule.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/schema.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/schema.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/tableQuery.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/tableQuery.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/tableUsageCount.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/tableUsageCount.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/tagLabel.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/tagLabel.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/usageDetails.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/usageDetails.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/usageRequest.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/usageRequest.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/generated/schema/type/votes.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  type/votes.json
-#   timestamp: 2024-10-24T15:47:03+00:00
+#   timestamp: 2024-10-29T14:34:36+00:00
 
 from __future__ import annotations
 

metadata/ingestion/ometa/mixins/suggestions_mixin.py

@@ -13,8 +13,13 @@ Mixin class containing Suggestions specific methods
 
 To be used by OpenMetadata class
 """
-from metadata.generated.schema.entity.feed.suggestion import Suggestion
+from typing import Union
+
+from metadata.generated.schema.entity.feed.suggestion import Suggestion, SuggestionType
+from metadata.generated.schema.type import basic
+from metadata.generated.schema.type.basic import FullyQualifiedEntityName
 from metadata.ingestion.ometa.client import REST
+from metadata.ingestion.ometa.utils import model_str
 from metadata.utils.logger import ometa_logger
 
 logger = ometa_logger()
@@ -30,12 +35,50 @@ class OMetaSuggestionsMixin:
     client: REST
 
     def update_suggestion(self, suggestion: Suggestion) -> Suggestion:
-        """
-        Update an existing Suggestion with new fields
-        """
+        """Update an existing Suggestion with new fields"""
         resp = self.client.put(
             f"{self.get_suffix(Suggestion)}/{str(suggestion.root.id.root)}",
             data=suggestion.model_dump_json(),
         )
 
         return Suggestion(**resp)
+
+    def accept_suggestion(self, suggestion_id: Union[str, basic.Uuid]) -> None:
+        """Accept a given suggestion"""
+        self.client.put(
+            f"{self.get_suffix(Suggestion)}/{model_str(suggestion_id)}/accept",
+        )
+
+    def reject_suggestion(self, suggestion_id: Union[str, basic.Uuid]) -> None:
+        """Reject a given suggestion"""
+        self.client.put(
+            f"{self.get_suffix(Suggestion)}/{model_str(suggestion_id)}/reject",
+        )
+
+    def accept_all_suggestions(
+        self,
+        fqn: Union[str, FullyQualifiedEntityName],
+        user_id: Union[str, basic.Uuid],
+        suggestion_type: SuggestionType = SuggestionType.SuggestDescription,
+    ) -> None:
+        """Accept all suggestions"""
+        self.client.put(
+            f"{self.get_suffix(Suggestion)}/accept-all?"
+            f"userId={model_str(user_id)}&"
+            f"entityFQN={model_str(fqn)}&"
+            f"suggestionType={suggestion_type.value}",
+        )
+
+    def reject_all_suggestions(
+        self,
+        fqn: Union[str, FullyQualifiedEntityName],
+        user_id: Union[str, basic.Uuid],
+        suggestion_type: SuggestionType = SuggestionType.SuggestDescription,
+    ) -> None:
+        """Accept all suggestions"""
+        self.client.put(
+            f"{self.get_suffix(Suggestion)}/reject-all?"
+            f"userId={model_str(user_id)}&"
+            f"entityFQN={model_str(fqn)}&"
+            f"suggestionType={suggestion_type.value}",
+        )

metadata/ingestion/ometa/ometa_api.py

@@ -19,6 +19,7 @@ from typing import Dict, Generic, Iterable, List, Optional, Type, TypeVar, Union
 
 from pydantic import BaseModel
 
+from metadata.generated.schema.api.createBot import CreateBot
 from metadata.generated.schema.api.services.ingestionPipelines.createIngestionPipeline import (
     CreateIngestionPipelineRequest,
 )
@@ -172,13 +173,16 @@ class OpenMetadata(
 
         return route
 
-    def get_module_path(self, entity: Type[T]) -> str:
+    def get_module_path(self, entity: Type[T]) -> Optional[str]:
         """
         Based on the entity, return the module path
         it is found inside generated
         """
         if issubclass(entity, CreateIngestionPipelineRequest):
             return "services.ingestionPipelines"
+        if issubclass(entity, CreateBot):
+            # Bots schemas don't live inside any subdirectory
+            return None
         return entity.__module__.split(".")[-2]
 
     def get_create_entity_type(self, entity: Type[T]) -> Type[C]:

metadata/ingestion/ometa/routes.py

@@ -21,6 +21,7 @@ from metadata.generated.schema.api.classification.createClassification import (
     CreateClassificationRequest,
 )
 from metadata.generated.schema.api.classification.createTag import CreateTagRequest
+from metadata.generated.schema.api.createBot import CreateBot
 from metadata.generated.schema.api.data.createAPICollection import (
     CreateAPICollectionRequest,
 )
@@ -211,7 +212,8 @@ ROUTES = {
     User.__name__: "/users",
     CreateUserRequest.__name__: "/users",
     AuthenticationMechanism.__name__: "/users/auth-mechanism",
-    Bot.__name__: "/bots",  # We won't allow bot creation from the client
+    Bot.__name__: "/bots",
+    CreateBot.__name__: "/bots",
     # Roles
     Role.__name__: "/roles",
     CreateRoleRequest.__name__: "/roles",