Adding inference container uri for edit model

Author: kumar-shivam-ranjan

ads/aqua/extension/model_handler.py

@@ -163,10 +163,12 @@ def put(self, id):
             raise HTTPError(400, Errors.NO_INPUT_DATA)
 
         inference_container = input_data.get("inference_container")
+        inference_container_uri=input_data.get("inference_container_uri")
         inference_containers = AquaModelApp.list_valid_inference_containers()
         if (
             inference_container is not None
             and inference_container not in inference_containers
+            and inference_container != InferenceContainerTypeFamily.AQUA_TEI_CONTAINER_FAMILY
         ):
             raise HTTPError(
                 400, Errors.INVALID_VALUE_OF_PARAMETER.format("inference_container")
@@ -176,7 +178,7 @@ def put(self, id):
         task = input_data.get("task")
         app = AquaModelApp()
         self.finish(
-            app.edit_registered_model(id, inference_container, enable_finetuning, task)
+            app.edit_registered_model(id, inference_container, inference_container_uri,enable_finetuning, task)
         )
         app.clear_model_details_cache(model_id=id)
 

ads/aqua/model/model.py

@@ -377,7 +377,7 @@ def delete_model(self, model_id):
             )
 
     @telemetry(entry_point="plugin=model&action=delete", name="aqua")
-    def edit_registered_model(self, id, inference_container, enable_finetuning, task):
+    def edit_registered_model(self, id, inference_container,inference_container_uri, enable_finetuning, task):
         """Edits the default config of unverified registered model.
 
         Parameters
@@ -386,6 +386,8 @@ def edit_registered_model(self, id, inference_container, enable_finetuning, task
             The model OCID.
         inference_container: str.
             The inference container family name
+        inference_container_uri: str
+            The inference container uri for embedding models
         enable_finetuning: str
             Flag to enable or disable finetuning over the model. Defaults to None
         task:
@@ -414,6 +416,13 @@ def edit_registered_model(self, id, inference_container, enable_finetuning, task
                         description="Deployment container mapping for SMC",
                         replace=True,
                     )
+                if inference_container_uri:
+                    custom_metadata_list.add(
+                        key=ModelCustomMetadataFields.DEPLOYMENT_CONTAINER_URI,
+                        value=inference_container_uri,
+                        category=MetadataCustomCategory.OTHER,
+                        description=f"Inference container URI for {ds_model.display_name}"
+                    )
                 if enable_finetuning is not None:
                     if enable_finetuning.lower() == "true":
                         custom_metadata_list.add(