Add ignore validation flag while registering model & improve logging

ads/aqua/model/model.py

@@ -19,7 +19,11 @@
     InferenceContainerTypeFamily,
     Tags,
 )
-from ads.aqua.common.errors import AquaRuntimeError, AquaValueError
+from ads.aqua.common.errors import (
+    AquaFileNotFoundError,
+    AquaRuntimeError,
+    AquaValueError,
+)
 from ads.aqua.common.utils import (
     LifecycleStatus,
     _build_resource_identifier,
@@ -1206,7 +1210,9 @@ def _validate_safetensor_format(
             except Exception as ex:
                 logger.error(
                     f"Exception occurred while loading config file from {import_model_details.os_path}"
-                    f"Exception message: {ex}"
+                )
+                logger.error(
+                    ex.reason if isinstance(ex, AquaFileNotFoundError) else str(ex)
                 )
                 if not import_model_details.ignore_model_artifact_check:
                     raise AquaRuntimeError(

tests/unitary/with_extras/aqua/test_model_handler.py

@@ -132,7 +132,7 @@ def test_list(self, mock_list):
 
     @parameterized.expand(
         [
-            (None, None, False, None, None, None, None, None),
+            (None, None, False, None, None, None, None, None, True),
             (
                 "odsc-llm-fine-tuning",
                 None,
@@ -142,8 +142,9 @@ def test_list(self, mock_list):
                 ["test.json"],
                 None,
                 None,
+                False,
             ),
-            (None, "test.gguf", True, None, ["*.json"], None, None, None),
+            (None, "test.gguf", True, None, ["*.json"], None, None, None, False),
             (
                 None,
                 None,
@@ -153,6 +154,7 @@ def test_list(self, mock_list):
                 ["test.json"],
                 None,
                 None,
+                False,
             ),
             (
                 None,
@@ -163,6 +165,7 @@ def test_list(self, mock_list):
                 None,
                 {"ftag1": "fvalue1"},
                 {"dtag1": "dvalue1"},
+                False,
             ),
         ],
     )
@@ -178,6 +181,7 @@ def test_register(
         ignore_patterns,
         freeform_tags,
         defined_tags,
+        ignore_model_artifact_check,
         mock_register,
         mock_finish,
     ):
@@ -201,6 +205,7 @@ def test_register(
                 ignore_patterns=ignore_patterns,
                 freeform_tags=freeform_tags,
                 defined_tags=defined_tags,
+                ignore_model_artifact_check=ignore_model_artifact_check,
             )
         )
         result = self.model_handler.post()
@@ -218,6 +223,7 @@ def test_register(
             ignore_patterns=ignore_patterns,
             freeform_tags=freeform_tags,
             defined_tags=defined_tags,
+            ignore_model_artifact_check=ignore_model_artifact_check,
         )
         assert result["id"] == "test_id"
         assert result["inference_container"] == "odsc-tgi-serving"