From 6e489b74cbae5851a4a108b3ee1b637c004ecaba Mon Sep 17 00:00:00 2001
From: Lu Peng <118394507+lu-ohai@users.noreply.github.com>
Date: Mon, 10 Nov 2025 15:53:08 -0500
Subject: [PATCH] Update model.py

---
 ads/aqua/model/model.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/ads/aqua/model/model.py b/ads/aqua/model/model.py
index b98da08c0..8687f6630 100644
--- a/ads/aqua/model/model.py
+++ b/ads/aqua/model/model.py
@@ -660,7 +660,10 @@ def edit_registered_model(
             raise AquaRuntimeError("Only registered unverified models can be edited.")
 
     def convert_fine_tune(
-        self, model_id: str, delete_model: Optional[bool] = True
+        self,
+        model_id: str,
+        delete_model: Optional[bool] = True,
+        model_display_name: Optional[str] = None,
     ) -> DataScienceModel:
         """Converts legacy fine tuned model to fine tuned model v2.
         1. 'fine_tune_model_version' tag will be added as 'v2' to new fine tuned model.
@@ -672,6 +675,8 @@ def convert_fine_tune(
             The legacy fine tuned model OCID.
         delete_model: bool
             Flag whether to delete the legacy model or not. Defaults to True.
+        model_display_name: str
+            The name of fine tuned model v2 converted. Legacy model's name will be used if not provided. Defaults to None.
 
         Returns
         -------
@@ -722,7 +727,9 @@ def convert_fine_tune(
             .with_model_file_description(
                 json_dict=legacy_fine_tuned_model.model_file_description
             )
-            .with_display_name(legacy_fine_tuned_model.display_name)
+            .with_display_name(
+                model_display_name or legacy_fine_tuned_model.display_name
+            )
             .with_description(legacy_fine_tuned_model.description)
             .with_freeform_tags(**fine_tune_model_v2_tags)
             .with_defined_tags(**(legacy_fine_tuned_model.defined_tags or {}))