Updates in create MD request

Author: kumar-shivam-ranjan

ads/aqua/app.py

@@ -454,7 +454,9 @@ def get_config(
 
         return ModelConfigResult(config=config, model_details=oci_model)
 
-    def get_container_image(self, container_type: str = None) -> str:
+    def get_container_image(
+        self, container_type: str = None, container_tag: str = None
+    ) -> str:
         """
         Gets the latest smc container complete image name from the given container type.
 
@@ -463,6 +465,9 @@ def get_container_image(self, container_type: str = None) -> str:
         container_type: str
             type of container, can be either odsc-vllm-serving, odsc-llm-fine-tuning, odsc-llm-evaluate
 
+        container_tag: str
+            tag of container, ex: 0.8.5.post1.1
+
         Returns
         -------
         str:
@@ -476,13 +481,23 @@ def get_container_image(self, container_type: str = None) -> str:
         )
         if not container:
             raise AquaValueError(f"Invalid container type : {container_type}")
-        container_image = (
-            SERVICE_MANAGED_CONTAINER_URI_SCHEME
-            + container.container_name
-            + ":"
-            + container.tag
-        )
-        return container_image
+
+        if container_tag:
+            container_image = (
+                SERVICE_MANAGED_CONTAINER_URI_SCHEME
+                + container.container_name
+                + ":"
+                + container_tag
+            )
+            return container_image
+        else:
+            container_image = (
+                SERVICE_MANAGED_CONTAINER_URI_SCHEME
+                + container.container_name
+                + ":"
+                + container.tag
+            )
+            return container_image
 
     @cached(cache=TTLCache(maxsize=20, ttl=timedelta(minutes=30), timer=datetime.now))
     def list_service_containers(self) -> List[ContainerSummary]:

ads/aqua/modeldeployment/deployment.py

@@ -155,6 +155,7 @@ def create(
                 health_check_port (Optional[int]): Health check port for the Docker container image.
                 env_var (Optional[Dict[str, str]]): Environment variables for deployment.
                 container_family (Optional[str]): Image family of the model deployment container runtime.
+                container_tag (Optional[str]): Image tag of the model deployment container runtime
                 memory_in_gbs (Optional[float]): Memory (in GB) for the selected shape.
                 ocpus (Optional[float]): OCPU count for the selected shape.
                 model_file (Optional[str]): File used for model deployment.
@@ -425,7 +426,10 @@ def _create(
 
         container_image_uri = (
             create_deployment_details.container_image_uri
-            or self.get_container_image(container_type=container_type_key)
+            or self.get_container_image(
+                container_type=container_type_key,
+                container_tag=create_deployment_details.container_tag,
+            )
         )
         if not container_image_uri:
             try:
@@ -631,7 +635,10 @@ def _create_multi(
 
         container_image_uri = (
             create_deployment_details.container_image_uri
-            or self.get_container_image(container_type=container_type_key)
+            or self.get_container_image(
+                container_type=container_type_key,
+                container_tag=create_deployment_details.container_tag,
+            )
         )
         server_port = create_deployment_details.server_port or (
             container_spec.server_port if container_spec else None

ads/aqua/modeldeployment/entities.py

@@ -270,6 +270,9 @@ class CreateModelDeploymentDetails(BaseModel):
     container_family: Optional[str] = Field(
         None, description="Image family of the model deployment container runtime."
     )
+    container_tag: Optional[str] = Field(
+        None, description="Image tag of the model deployment container runtime."
+    )
     memory_in_gbs: Optional[float] = Field(
         None, description="Memory (in GB) for the selected shape."
     )