diff --git a/ads/aqua/model/model.py b/ads/aqua/model/model.py index 2b5d7108f..c848b2103 100644 --- a/ads/aqua/model/model.py +++ b/ads/aqua/model/model.py @@ -151,6 +151,7 @@ def create( compartment_id: Optional[str] = None, freeform_tags: Optional[Dict] = None, defined_tags: Optional[Dict] = None, + display_name: Optional[str] = None, **kwargs, ) -> DataScienceModel: """ @@ -169,6 +170,8 @@ def create( Freeform tags for the model. defined_tags : Optional[Dict] Defined tags for the model. + display_name: ptional[str] + The display name of the custom model. Returns ------- @@ -204,6 +207,9 @@ def create( **(defined_tags or {}), } + if display_name: + service_model.display_name = display_name + custom_model = ( DataScienceModel() .with_compartment_id(target_compartment) diff --git a/ads/aqua/modeldeployment/deployment.py b/ads/aqua/modeldeployment/deployment.py index 3e93f38ee..c11425f17 100644 --- a/ads/aqua/modeldeployment/deployment.py +++ b/ads/aqua/modeldeployment/deployment.py @@ -214,6 +214,7 @@ def create( ) aqua_model = model_app.create( model_id=create_deployment_details.model_id, + display_name=create_deployment_details.model_name, compartment_id=compartment_id, project_id=project_id, freeform_tags=freeform_tags, @@ -446,6 +447,7 @@ def _create( cmd_var_string = aqua_model.custom_metadata_list.get( AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME ).value + default_cmd_var = shlex.split(cmd_var_string) if default_cmd_var: cmd_var = validate_cmd_var(default_cmd_var, cmd_var) @@ -538,7 +540,6 @@ def _create( ) deployment_params = get_combined_params(config_params, user_params) - params = f"{params} {deployment_params}".strip() if params: env_var.update({"PARAMS": params}) diff --git a/ads/aqua/modeldeployment/entities.py b/ads/aqua/modeldeployment/entities.py index ebce26dc8..5b31f1611 100644 --- a/ads/aqua/modeldeployment/entities.py +++ b/ads/aqua/modeldeployment/entities.py @@ -233,6 +233,9 @@ class CreateModelDeploymentDetails(BaseModel): None, description="The description of the deployment." ) model_id: Optional[str] = Field(None, description="The model OCID to deploy.") + model_name: Optional[str] = Field( + None, description="The model name specified by user to deploy." + ) models: Optional[List[AquaMultiModelRef]] = Field( None, description="List of models for multimodel deployment." )