Addressing review comments

Author: harsh97

ads/aqua/common/utils.py

@@ -17,7 +17,6 @@
 from typing import List, Union
 
 import fsspec
-import ocifs
 from cachetools import TTLCache, cached
 
 import oci
@@ -55,6 +54,7 @@
 from ads.config import AQUA_SERVICE_MODELS_BUCKET, CONDA_BUCKET_NS, TENANCY_OCID
 from ads.model import DataScienceModel, ModelVersionSet
 from oci.data_science.models import JobRun, Model
+from oci.object_storage.models import ObjectSummary
 
 logger = logging.getLogger("ads.aqua")
 
@@ -243,15 +243,20 @@ def list_os_files_with_extension(oss_path: str, extension: str) -> [str]:
     - A list of file paths matching the specified extension.
     """
 
+    oss_client = ObjectStorageDetails.from_path(oss_path)
     signer = default_signer()
 
     # Ensure the extension is prefixed with a dot if not already
+
     if not extension.startswith("."):
         extension = "." + extension
-    fs = ocifs.OCIFileSystem(**signer)
-    files: [str] = fs.ls(oss_path)
+    files: List[ObjectSummary] = oss_client.list_objects().objects
 
-    return [file for file in files if file.endswith(extension)]
+    return [
+        file.name
+        for file in files
+        if file.name.endswith(extension) and "/" not in file.name
+    ]
 
 
 def is_valid_ocid(ocid: str) -> bool:

ads/aqua/config/deployment_config_defaults.json

@@ -1,9 +1,36 @@
 {
+  "configuration": {
+    "VM.Standard.A1.Flex": {
+      "parameters": {},
+      "shape_info": {
+        "configs": [
+          {
+            "memory_in_gbs": 128,
+            "ocpu": 32
+          },
+          {
+            "memory_in_gbs": 256,
+            "ocpu": 64
+          },
+          {
+            "memory_in_gbs": 512,
+            "ocpu": 128
+          },
+          {
+            "memory_in_gbs": 1024,
+            "ocpu": 256
+          }
+        ],
+        "type": "CPU"
+      }
+    }
+  },
   "shape": [
     "VM.GPU.A10.1",
     "VM.GPU.A10.2",
     "BM.GPU.A10.4",
     "BM.GPU4.8",
-    "BM.GPU.A100-v2.8"
+    "BM.GPU.A100-v2.8",
+    "VM.Standard.A1.Flex"
   ]
 }

ads/aqua/model/model.py

@@ -742,12 +742,7 @@ def get_model_files(os_path: str, model_format: ModelFormat) -> [str]:
 
         if model_format == ModelFormat.GGUF:
             model_files.extend(
-                [
-                    os.path.basename(path)
-                    for path in list_os_files_with_extension(
-                        oss_path=os_path, extension=".gguf"
-                    )
-                ]
+                list_os_files_with_extension(oss_path=os_path, extension=".gguf")
             )
         return model_files