Adding model delete and MD delete/deactivate APIs

Author: kumar-shivam-ranjan

ads/aqua/extension/deployment_handler.py

@@ -54,6 +54,14 @@ def get(self, id=""):
         else:
             raise HTTPError(400, f"The request {self.request.path} is invalid.")
 
+    @handle_exceptions
+    def delete(self,model_deployment_id):
+        return self.finish(AquaDeploymentApp().delete(model_deployment_id))
+
+    @handle_exceptions
+    def put(self,model_deployment_id):
+        return self.finish(AquaDeploymentApp().deactivate(model_deployment_id))
+
     @handle_exceptions
     def post(self, *args, **kwargs):
         """

ads/aqua/extension/model_handler.py

@@ -73,6 +73,8 @@ def delete(self, id=""):
         paths = url_parse.path.strip("/")
         if paths.startswith("aqua/model/cache"):
             return self.finish(AquaModelApp().clear_model_list_cache())
+        elif id:
+            return self.finish(AquaModelApp().delete_registered_model(id))
         else:
             raise HTTPError(400, f"The request {self.request.path} is invalid.")
 

ads/aqua/model/model.py

@@ -323,6 +323,15 @@ def get(self, model_id: str, load_model_card: Optional[bool] = True) -> "AquaMod
 
         return model_details
 
+    @telemetry(entry_point="plugin=model&action=delete", name="aqua")
+    def delete_registered_model(self,model_id):
+        ds_model=DataScienceModel.from_id(model_id)
+        is_registered_model=ds_model.freeform_tags.get(Tags.BASE_MODEL_CUSTOM,None)
+        if is_registered_model:
+            return ds_model.delete()
+        else:
+            raise AquaRuntimeError(f"Failed to delete model:{model_id}. Only registered models can be deleted.")
+
     def _fetch_metric_from_metadata(
         self,
         custom_metadata_list: ModelCustomMetadata,

ads/aqua/modeldeployment/deployment.py

@@ -485,6 +485,15 @@ def list(self, **kwargs) -> List["AquaDeployment"]:
 
         return results
 
+
+    @telemetry(entry_point="plugin=deployment&action=delete", name="aqua")
+    def delete(self,model_deployment_id:str):
+        return self.ds_client.delete_model_deployment(model_deployment_id=model_deployment_id).data
+
+    @telemetry(entry_point="plugin=deployment&action=deactivate",name="aqua")
+    def deactivate(self,model_deployment_id:str):
+        return self.ds_client.deactivate_model_deployment(model_deployment_id=model_deployment_id).data
+
     @telemetry(entry_point="plugin=deployment&action=get", name="aqua")
     def get(self, model_deployment_id: str, **kwargs) -> "AquaDeploymentDetail":
         """Gets the information of Aqua model deployment.