oracle-ads 2.13.6__py3-none-any.whl → 2.13.7__py3-none-any.whl

ads/aqua/constants.py

@@ -6,9 +6,10 @@
 UNKNOWN_VALUE = ""
 READY_TO_IMPORT_STATUS = "TRUE"
 UNKNOWN_DICT = {}
-README = "README.md"
-LICENSE_TXT = "config/LICENSE.txt"
 DEPLOYMENT_CONFIG = "deployment_config.json"
+FINE_TUNING_CONFIG = "ft_config.json"
+README = "README.md"
+LICENSE = "LICENSE.txt"
 AQUA_MODEL_TOKENIZER_CONFIG = "tokenizer_config.json"
 COMPARTMENT_MAPPING_KEY = "service-model-compartment"
 CONTAINER_INDEX = "container_index.json"
@@ -16,6 +17,7 @@ EVALUATION_REPORT_JSON = "report.json"
 EVALUATION_REPORT_MD = "report.md"
 EVALUATION_REPORT = "report.html"
 UNKNOWN_JSON_STR = "{}"
+UNKNOWN_JSON_LIST = "[]"
 FINE_TUNING_RUNTIME_CONTAINER = "iad.ocir.io/ociodscdev/aqua_ft_cuda121:0.3.17.20"
 DEFAULT_FT_BLOCK_STORAGE_SIZE = 750
 DEFAULT_FT_REPLICA = 1
@@ -88,25 +90,25 @@ LLAMA_CPP_INFERENCE_RESTRICTED_PARAMS = {
 TEI_CONTAINER_DEFAULT_HOST = "8080"
 
 OCI_OPERATION_FAILURES = {
-        "list_model_deployments": "Unable to list model deployments. See tips for troubleshooting: ",
-        "list_models": "Unable to list models. See tips for troubleshooting: ",
-        "get_namespace": "Unable to access specified Object Storage Bucket. See tips for troubleshooting: ",
-        "list_log_groups":"Unable to access logs. See tips for troubleshooting: " ,
-        "list_buckets": "Unable to list Object Storage Bucket. See tips for troubleshooting: ",
-        "put_object": "Unable to access or find Object Storage Bucket. See tips for troubleshooting: ",
-        "list_model_version_sets": "Unable to create or fetch model version set. See tips for troubleshooting:",
-        "update_model": "Unable to update model. See tips for troubleshooting: ",
-        "list_data_science_private_endpoints": "Unable to access private endpoint. See tips for troubleshooting:  ",
-        "create_model" : "Unable to register model. See tips for troubleshooting: ",
-        "create_deployment": "Unable to create deployment. See tips for troubleshooting: ",
-        "create_model_version_sets" : "Unable to create model version set. See tips for troubleshooting: ",
-        "create_job": "Unable to create job. See tips for troubleshooting: ",
-        "create_job_run": "Unable to create job run. See tips for troubleshooting: ",
+    "list_model_deployments": "Unable to list model deployments. See tips for troubleshooting: ",
+    "list_models": "Unable to list models. See tips for troubleshooting: ",
+    "get_namespace": "Unable to access specified Object Storage Bucket. See tips for troubleshooting: ",
+    "list_log_groups": "Unable to access logs. See tips for troubleshooting: ",
+    "list_buckets": "Unable to list Object Storage Bucket. See tips for troubleshooting: ",
+    "put_object": "Unable to access or find Object Storage Bucket. See tips for troubleshooting: ",
+    "list_model_version_sets": "Unable to create or fetch model version set. See tips for troubleshooting:",
+    "update_model": "Unable to update model. See tips for troubleshooting: ",
+    "list_data_science_private_endpoints": "Unable to access private endpoint. See tips for troubleshooting:  ",
+    "create_model": "Unable to register model. See tips for troubleshooting: ",
+    "create_deployment": "Unable to create deployment. See tips for troubleshooting: ",
+    "create_model_version_sets": "Unable to create model version set. See tips for troubleshooting: ",
+    "create_job": "Unable to create job. See tips for troubleshooting: ",
+    "create_job_run": "Unable to create job run. See tips for troubleshooting: ",
 }
 
 STATUS_CODE_MESSAGES = {
-        "400": "Could not process your request due to invalid input.",
-        "403": "We're having trouble processing your request with the information provided.",
-        "404": "Authorization Failed: The resource you're looking for isn't accessible.",
-        "408": "Server is taking too long to respond, please try again.",
+    "400": "Could not process your request due to invalid input.",
+    "403": "We're having trouble processing your request with the information provided.",
+    "404": "Authorization Failed: The resource you're looking for isn't accessible.",
+    "408": "Server is taking too long to respond, please try again.",
 }

ads/aqua/evaluation/evaluation.py

@@ -41,13 +41,14 @@ from ads.aqua.common.errors import (
 from ads.aqua.common.utils import (
     extract_id_and_name_from_tag,
     fire_and_forget,
-    get_container_config,
-    get_container_image,
     is_valid_ocid,
     upload_local_to_os,
 )
-from ads.aqua.config.config import get_evaluation_service_config
-from ads.aqua.config.container_config import AquaContainerConfig
+from ads.aqua.config.evaluation.evaluation_service_config import (
+    DEFAULT_EVALUATION_CONTAINER,
+    EvaluationServiceConfig,
+    MetricConfig,
+)
 from ads.aqua.constants import (
     CONSOLE_LINK_RESOURCE_TYPE_MAPPING,
     EVALUATION_REPORT,
@@ -80,7 +81,13 @@ from ads.aqua.evaluation.errors import EVALUATION_JOB_EXIT_CODE_MESSAGE
 from ads.aqua.model.constants import ModelCustomMetadataFields
 from ads.common.auth import default_signer
 from ads.common.object_storage_details import ObjectStorageDetails
-from ads.common.utils import UNKNOWN, get_console_link, get_files, get_log_links
+from ads.common.utils import (
+    UNKNOWN,
+    get_console_link,
+    get_files,
+    get_log_links,
+    read_file,
+)
 from ads.config import (
     AQUA_JOB_SUBNET_ID,
     COMPARTMENT_OCID,
@@ -260,10 +267,10 @@ class AquaEvaluationApp(AquaApp):
                     runtime = ModelDeploymentContainerRuntime.from_dict(
                         evaluation_source.runtime.to_dict()
                     )
-                    inference_config = AquaContainerConfig.from_container_index_json(
-                        config=get_container_config(), enable_spec=True
-                    ).inference
-                    for container in inference_config.values():
+                    inference_config = (
+                        self.get_container_config().to_dict().get("inference")
+                    )
+                    for container in inference_config:
                         if container.name == runtime.image[: runtime.image.rfind(":")]:
                             eval_inference_configuration = (
                                 container.spec.evaluation_configuration
@@ -801,14 +808,13 @@ class AquaEvaluationApp(AquaApp):
 
         return source.display_name
 
-    @staticmethod
-    def _get_evaluation_container(source_id: str) -> str:
+    def _get_evaluation_container(self, source_id: str) -> str:
         # todo: use the source, identify if it is a model or a deployment. If latter, then fetch the base model id
         #   from the deployment object, and call ds_client.get_model() to get model details. Use custom metadata to
         #   get the container_type_key. Pass this key as container_type to get_container_image method.
 
         # fetch image name from config
-        container_image = get_container_image(
+        container_image = self.get_container_image(
             container_type="odsc-llm-evaluate",
         )
         logger.info(f"Aqua Image used for evaluating {source_id} :{container_image}")
@@ -1102,11 +1108,18 @@ class AquaEvaluationApp(AquaApp):
             "loggroup_url": loggroup_url,
         }
 
-    def get_supported_metrics(self) -> dict:
+    def get_supported_metrics(self) -> List[MetricConfig]:
         """Gets a list of supported metrics for evaluation."""
-        return [
-            item.to_dict() for item in get_evaluation_service_config().ui_config.metrics
-        ]
+        containers = self.list_service_containers()
+        container_item = next(
+            c
+            for c in containers
+            if c.is_latest and c.family_name == DEFAULT_EVALUATION_CONTAINER
+        )
+        evaluation_service_config = EvaluationServiceConfig.from_oci_container_config(
+            container_item
+        )
+        return evaluation_service_config.ui_config.metrics
 
     @telemetry(entry_point="plugin=evaluation&action=load_metrics", name="aqua")
     def load_metrics(self, eval_id: str) -> AquaEvalMetrics:
@@ -1130,17 +1143,31 @@ class AquaEvaluationApp(AquaApp):
 
         with tempfile.TemporaryDirectory() as temp_dir:
             logger.info(f"Downloading evaluation artifact: {eval_id}.")
-            DataScienceModel.from_id(eval_id).download_artifact(
-                temp_dir,
-                auth=self._auth,
-            )
+
+            dsc_model = DataScienceModel.from_id(eval_id)
+            if dsc_model.if_model_custom_metadata_artifact_exist(
+                EVALUATION_REPORT_MD
+            ) and dsc_model.if_model_custom_metadata_artifact_exist(
+                EVALUATION_REPORT_JSON
+            ):
+                logger.info(
+                    f"Fetching {EVALUATION_REPORT_MD} and {EVALUATION_REPORT_JSON} from custom metadata..."
+                )
+                dsc_model.get_custom_metadata_artifact(EVALUATION_REPORT_MD, temp_dir)
+                dsc_model.get_custom_metadata_artifact(EVALUATION_REPORT_JSON, temp_dir)
+            else:
+                logger.info("Fetching Evaluation Reports from OSS bucket...")
+                dsc_model.download_artifact(
+                    temp_dir,
+                    auth=self._auth,
+                )
 
             files_in_artifact = get_files(temp_dir)
             md_report_content = self._read_from_artifact(
                 temp_dir, files_in_artifact, EVALUATION_REPORT_MD
             )
 
-            # json report not availiable for failed evaluation
+            # json report not available for failed evaluation
             try:
                 json_report = json.loads(
                     self._read_from_artifact(
@@ -1239,14 +1266,46 @@ class AquaEvaluationApp(AquaApp):
 
         with tempfile.TemporaryDirectory() as temp_dir:
             logger.info(f"Downloading evaluation artifact for {eval_id}.")
-            DataScienceModel.from_id(eval_id).download_artifact(
-                temp_dir,
-                auth=self._auth,
+            dsc_model = DataScienceModel.from_id(eval_id)
+            if_custom_metadata_exists = (
+                dsc_model.if_model_custom_metadata_artifact_exist(EVALUATION_REPORT)
             )
-            content = self._read_from_artifact(
-                temp_dir, get_files(temp_dir), EVALUATION_REPORT
-            )
-
+            if if_custom_metadata_exists:
+                logger.info(f"Fetching {EVALUATION_REPORT} from custom metadata.")
+                dsc_model.get_custom_metadata_artifact(EVALUATION_REPORT, temp_dir)
+            else:
+                logger.info(f"Fetching {EVALUATION_REPORT} from Model artifact.")
+                dsc_model.download_artifact(
+                    temp_dir,
+                    auth=self._auth,
+                )
+            files_in_artifact = get_files(temp_dir)
+            if not len(files_in_artifact):
+                try:
+                    evaluation_output_path = dsc_model.custom_metadata_list.get(
+                        EvaluationCustomMetadata.EVALUATION_OUTPUT_PATH
+                    ).value
+                    report_path = (
+                        evaluation_output_path.rstrip("/")
+                        + "/"
+                        + eval_id
+                        + "/"
+                        + EVALUATION_REPORT
+                    )
+                    logger.info(
+                        f"Fetching {EVALUATION_REPORT} from {report_path} for evaluation {eval_id}"
+                    )
+                    content = read_file(
+                        file_path=report_path, auth=default_signer()
+                    ).encode()
+                except ValueError as err:
+                    raise AquaValueError(
+                        f"{EvaluationCustomMetadata.EVALUATION_OUTPUT_PATH} is missing from custom metadata for the model {eval_id}"
+                    ) from err
+            else:
+                content = self._read_from_artifact(
+                    temp_dir, files_in_artifact, EVALUATION_REPORT
+                )
         report = AquaEvalReport(
             evaluation_id=eval_id, content=base64.b64encode(content).decode()
         )
@@ -1376,11 +1435,10 @@ class AquaEvaluationApp(AquaApp):
 
     @staticmethod
     @fire_and_forget
-    def _delete_job_and_model(job, model):
+    def _delete_job_and_model(job: DataScienceJob, model: DataScienceModel):
         try:
             job.dsc_job.delete(force_delete=True)
             logger.info(f"Deleting Job: {job.job_id} for evaluation {model.id}")
-
             model.delete()
             logger.info(f"Deleting evaluation: {model.id}")
         except oci.exceptions.ServiceError as ex:
@@ -1389,12 +1447,20 @@ class AquaEvaluationApp(AquaApp):
                 f"Exception message: {ex}"
             )
 
-    def load_evaluation_config(self, container: Optional[str] = None) -> Dict:
+    def load_evaluation_config(
+        self, container: Optional[str] = DEFAULT_EVALUATION_CONTAINER
+    ) -> Dict:
         """Loads evaluation config."""
 
         logger.info("Loading evaluation container config.")
         # retrieve the evaluation config by container family name
-        evaluation_config = get_evaluation_service_config(container)
+        containers = self.list_service_containers()
+        container_item = next(
+            c for c in containers if c.is_latest and c.family_name == container
+        )
+        evaluation_config = EvaluationServiceConfig.from_oci_container_config(
+            container_item
+        )
 
         # convert the new config representation to the old one
         return {

ads/aqua/extension/common_handler.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-# Copyright (c) 2024 Oracle and/or its affiliates.
+# Copyright (c) 2025 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 
@@ -12,14 +12,12 @@ from huggingface_hub.utils import LocalTokenNotFoundError
 from tornado.web import HTTPError
 
 from ads.aqua.common.decorator import handle_exceptions
-from ads.aqua.common.errors import AquaResourceAccessError, AquaRuntimeError
+from ads.aqua.common.errors import AquaRuntimeError
 from ads.aqua.common.utils import (
     get_huggingface_login_timeout,
-    known_realm,
 )
 from ads.aqua.extension.base_handler import AquaAPIhandler
 from ads.aqua.extension.errors import Errors
-from ads.aqua.extension.utils import ui_compatability_check
 
 
 class ADSVersionHandler(AquaAPIhandler):
@@ -50,14 +48,7 @@ class CompatibilityCheckHandler(AquaAPIhandler):
             AquaResourceAccessError: raised when aqua is not accessible in the given session/region.
 
         """
-        if ui_compatability_check():
-            return self.finish({"status": "ok"})
-        elif known_realm():
-            return self.finish({"status": "compatible"})
-        else:
-            raise AquaResourceAccessError(
-                "The AI Quick actions extension is not compatible in the given region."
-            )
+        return self.finish({"status": "ok"})
 
 
 class NetworkStatusHandler(AquaAPIhandler):

ads/aqua/extension/common_ws_msg_handler.py

@@ -1,22 +1,18 @@
 #!/usr/bin/env python
 
-# Copyright (c) 2024 Oracle and/or its affiliates.
+# Copyright (c) 2024, 2025 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 import json
 from importlib import metadata
-from typing import List, Union
+from typing import List, Optional, Union
 
 from ads.aqua.common.decorator import handle_exceptions
-from ads.aqua.common.errors import AquaResourceAccessError
-from ads.aqua.common.utils import known_realm
 from ads.aqua.extension.aqua_ws_msg_handler import AquaWSMsgHandler
 from ads.aqua.extension.models.ws_models import (
     AdsVersionResponse,
-    CompatibilityCheckResponse,
     RequestResponseType,
 )
-from ads.aqua.extension.utils import ui_compatability_check
 
 
 class AquaCommonWsMsgHandler(AquaWSMsgHandler):
@@ -28,7 +24,7 @@ class AquaCommonWsMsgHandler(AquaWSMsgHandler):
         super().__init__(message)
 
     @handle_exceptions
-    def process(self) -> Union[AdsVersionResponse, CompatibilityCheckResponse]:
+    def process(self) -> Optional[AdsVersionResponse]:
         request = json.loads(self.message)
         if request.get("kind") == "AdsVersion":
             version = metadata.version("oracle_ads")
@@ -38,20 +34,3 @@ class AquaCommonWsMsgHandler(AquaWSMsgHandler):
                 data=version,
             )
             return response
-        if request.get("kind") == "CompatibilityCheck":
-            if ui_compatability_check():
-                return CompatibilityCheckResponse(
-                    message_id=request.get("message_id"),
-                    kind=RequestResponseType.CompatibilityCheck,
-                    data={"status": "ok"},
-                )
-            elif known_realm():
-                return CompatibilityCheckResponse(
-                    message_id=request.get("message_id"),
-                    kind=RequestResponseType.CompatibilityCheck,
-                    data={"status": "compatible"},
-                )
-            else:
-                raise AquaResourceAccessError(
-                    "The AI Quick actions extension is not compatible in the given region."
-                )

ads/aqua/extension/model_handler.py

@@ -9,12 +9,14 @@ from tornado.web import HTTPError
 
 from ads.aqua.common.decorator import handle_exceptions
 from ads.aqua.common.enums import CustomInferenceContainerTypeFamily
-from ads.aqua.common.errors import AquaRuntimeError, AquaValueError
+from ads.aqua.common.errors import AquaRuntimeError
 from ads.aqua.common.utils import get_hf_model_info, is_valid_ocid, list_hf_models
 from ads.aqua.extension.base_handler import AquaAPIhandler
 from ads.aqua.extension.errors import Errors
 from ads.aqua.model import AquaModelApp
 from ads.aqua.model.entities import AquaModelSummary, HFModelSummary
+from ads.config import SERVICE
+from ads.model.common.utils import MetadataArtifactPathType
 
 
 class AquaModelHandler(AquaAPIhandler):
@@ -37,13 +39,11 @@ class AquaModelHandler(AquaAPIhandler):
                 raise HTTPError(
                     400, Errors.MISSING_REQUIRED_PARAMETER.format("model_format")
                 )
-            
+
             model_format = model_format.upper()
-            
+
             if os_path:
-                return self.finish(
-                    AquaModelApp.get_model_files(os_path, model_format)
-                )
+                return self.finish(AquaModelApp.get_model_files(os_path, model_format))
             elif model_name:
                 return self.finish(
                     AquaModelApp.get_hf_model_files(model_name, model_format)
@@ -82,11 +82,13 @@ class AquaModelHandler(AquaAPIhandler):
         # project_id is no needed.
         project_id = self.get_argument("project_id", default=None)
         model_type = self.get_argument("model_type", default=None)
+        category = self.get_argument("category", default=SERVICE)
         return self.finish(
             AquaModelApp().list(
                 compartment_id=compartment_id,
                 project_id=project_id,
                 model_type=model_type,
+                category=category,
             )
         )
 
@@ -206,6 +208,16 @@ class AquaModelLicenseHandler(AquaAPIhandler):
         return self.finish(AquaModelApp().load_license(model_id))
 
 
+class AquaModelReadmeHandler(AquaAPIhandler):
+    """
+    Handler for fetching model card for AQUA models
+    """
+
+    def get(self, model_id):
+        model_id = model_id.split("/")[0]
+        return self.finish(AquaModelApp().load_readme(model_id).model_dump())
+
+
 class AquaHuggingFaceHandler(AquaAPIhandler):
     """Handler for Aqua Hugging Face REST APIs."""
 
@@ -329,9 +341,50 @@ class AquaModelTokenizerConfigHandler(AquaAPIhandler):
         raise HTTPError(400, f"The request {self.request.path} is invalid.")
 
 
+class AquaModelDefinedMetadataArtifactHandler(AquaAPIhandler):
+    """
+    Handler for Model Defined metadata artifact content
+
+    Raises
+    ------
+    HTTPError
+        Raises HTTPError if inputs are missing or are invalid.
+    """
+
+    @handle_exceptions
+    def get(self, model_id: str, metadata_key: str):
+        """
+        model_id: ocid of the model
+        metadata_key: the metadata key for which artifact content needs to be downloaded.
+        Can be any of Readme, License , FinetuneConfiguration , DeploymentConfiguration
+        """
+
+        return self.finish(
+            AquaModelApp().get_defined_metadata_artifact_content(model_id, metadata_key)
+        )
+
+    @handle_exceptions
+    def post(self, model_id: str, metadata_key: str):
+        input_body = self.get_json_body()
+        path_type = input_body.get("path_type")
+        artifact_path_or_content = input_body.get("artifact_path_or_content")
+        if path_type not in MetadataArtifactPathType.values():
+            raise HTTPError(400, f"Invalid value of path_type: {path_type}")
+        return self.finish(
+            AquaModelApp().create_defined_metadata_artifact(
+                model_id, metadata_key, path_type, artifact_path_or_content
+            )
+        )
+
+
 __handlers__ = [
     ("model/?([^/]*)", AquaModelHandler),
     ("model/?([^/]*)/license", AquaModelLicenseHandler),
+    ("model/?([^/]*)/readme", AquaModelReadmeHandler),
     ("model/?([^/]*)/tokenizer", AquaModelTokenizerConfigHandler),
     ("model/hf/search/?([^/]*)", AquaHuggingFaceHandler),
+    (
+        "model/?([^/]*)/definedMetadata/?([^/]*)",
+        AquaModelDefinedMetadataArtifactHandler,
+    ),
 ]

ads/aqua/extension/models/ws_models.py

@@ -6,6 +6,7 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
+import ads.config
 from ads.aqua.evaluation.entities import AquaEvaluationDetail, AquaEvaluationSummary
 from ads.aqua.model.entities import AquaModel, AquaModelSummary
 from ads.aqua.modeldeployment.entities import AquaDeployment, AquaDeploymentDetail
@@ -57,6 +58,7 @@ class ListModelsRequest(BaseRequest):
     compartment_id: Optional[str] = None
     project_id: Optional[str] = None
     model_type: Optional[str] = None
+    category: str = ads.config.USER
     kind = RequestResponseType.ListDeployments
 
 

ads/aqua/extension/models_ws_msg_handler.py

@@ -32,6 +32,7 @@ class AquaModelWSMsgHandler(AquaWSMsgHandler):
                 compartment_id=request.get("compartment_id"),
                 project_id=request.get("project_id"),
                 model_type=request.get("model_type"),
+                category=request.get("category"),
             )
             response = ListModelsResponse(
                 message_id=request.get("message_id"),

ads/aqua/extension/utils.py

@@ -1,20 +1,17 @@
 #!/usr/bin/env python
-# Copyright (c) 2024 Oracle and/or its affiliates.
+# Copyright (c) 2024, 2025 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 import re
 import traceback
 import uuid
 from dataclasses import fields
-from datetime import datetime, timedelta
 from http.client import responses
 from typing import Dict, Optional
 
-from cachetools import TTLCache, cached
 from tornado.web import HTTPError
 
-from ads.aqua import ODSC_MODEL_COMPARTMENT_OCID, logger
-from ads.aqua.common.utils import fetch_service_compartment
+from ads.aqua import logger
 from ads.aqua.constants import (
     AQUA_TROUBLESHOOTING_LINK,
     OCI_OPERATION_FAILURES,
@@ -36,19 +33,11 @@ def validate_function_parameters(data_class, input_data: Dict):
             )
 
 
-@cached(cache=TTLCache(maxsize=1, ttl=timedelta(minutes=1), timer=datetime.now))
-def ui_compatability_check():
-    """This method caches the service compartment OCID details that is set by either the environment variable or if
-    fetched from the configuration. The cached result is returned when multiple calls are made in quick succession
-    from the UI to avoid multiple config file loads."""
-    return ODSC_MODEL_COMPARTMENT_OCID or fetch_service_compartment()
-
-
 def get_default_error_messages(
     service_payload: dict,
     status_code: str,
     default_msg: str = "Unknown HTTP Error.",
-)-> str:
+) -> str:
     """Method that maps the error messages based on the operation performed or the status codes encountered."""
 
     if service_payload and "operation_name" in service_payload:
@@ -66,14 +55,13 @@ def get_documentation_link(key: str) -> str:
     return f"{AQUA_TROUBLESHOOTING_LINK}#{github_header}"
 
 
-def get_troubleshooting_tips(service_payload: dict,
-                             status_code: str) -> str:
+def get_troubleshooting_tips(service_payload: dict, status_code: str) -> str:
     """Maps authorization errors to potential solutions on Troubleshooting Page per Aqua Documentation on oci-data-science-ai-samples"""
 
     tip = f"For general tips on troubleshooting: {AQUA_TROUBLESHOOTING_LINK}"
 
     if status_code in (404, 400):
-        failed_operation = service_payload.get('operation_name')
+        failed_operation = service_payload.get("operation_name")
 
         if failed_operation in OCI_OPERATION_FAILURES:
             link = get_documentation_link(failed_operation)
@@ -118,14 +106,13 @@ def construct_error(status_code: int, **kwargs) -> ReplyDetails:
 
     tips = get_troubleshooting_tips(service_payload, status_code)
 
-
     reply = ReplyDetails(
-        status = status_code,
-        troubleshooting_tips = tips,
-        message = message,
-        service_payload = service_payload,
-        reason = reason,
-        request_id = str(uuid.uuid4()),
+        status=status_code,
+        troubleshooting_tips=tips,
+        message=message,
+        service_payload=service_payload,
+        reason=reason,
+        request_id=str(uuid.uuid4()),
     )
 
     exc_info = kwargs.get("exc_info")

ads/aqua/finetuning/entities.py

@@ -3,7 +3,7 @@
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 import json
-from typing import List, Literal, Optional, Union
+from typing import Any, Dict, List, Literal, Optional, Union
 
 from pydantic import Field, model_validator
 
@@ -55,6 +55,28 @@ class AquaFineTuningParams(Serializable):
         return data
 
 
+class AquaFineTuningConfig(Serializable):
+    """Represents model's shape list and detailed configuration for fine-tuning.
+
+    Attributes:
+        shape (List[str], optional): A list of shape names (e.g., BM.GPU.A10.4).
+        configuration (Dict[str, Any], optional): Configuration details of fine-tuning.
+    """
+
+    shape: Optional[Dict[str, Any]] = Field(
+        default_factory=dict, description="List of supported shapes for the model."
+    )
+    finetuning_params: Optional[str] = Field(
+        default_factory=str, description="Fine tuning parameters."
+    )
+    configuration: Optional[Dict[str, Any]] = Field(
+        default_factory=dict, description="Configuration details keyed by shape."
+    )
+
+    class Config:
+        extra = "allow"
+
+
 class AquaFineTuningSummary(Serializable):
     """Represents a summary of Aqua Finetuning job."""