added CUSTOM keyword in existing model commands

Author: mohit-singlestore

singlestoredb/fusion/handlers/models.py

@@ -10,9 +10,9 @@
 from .utils import get_file_space
 
 
-class ShowModelFilesHandler(ShowFilesHandler):
+class ShowCustomModelsHandler(ShowFilesHandler):
     """
-    SHOW MODEL FILES
+    SHOW CUSTOM MODELS
         [ at_path ] [ <like> ]
         [ <order-by> ]
         [ <limit> ] [ recursive ] [ extended ];
@@ -53,16 +53,16 @@ class ShowModelFilesHandler(ShowFilesHandler):
     --------
     The following command lists the models::
 
-        SHOW MODEL FILES;
+        SHOW CUSTOM MODELS;
 
     The following command lists the models with additional information::
 
-        SHOW MODEL FILES EXTENDED;
+        SHOW CUSTOM MODELS EXTENDED;
 
     See Also
     --------
-    * ``UPLOAD MODEL FILE model_name FROM path``
-    * ``DOWNLOAD MODEL FILE model_name``
+    * ``UPLOAD CUSTOM MODEL model_name FROM path``
+    * ``DOWNLOAD CUSTOM MODEL model_name``
 
 
     """  # noqa: E501
@@ -73,12 +73,12 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         return super().run(params)
 
 
-ShowModelFilesHandler.register(overwrite=True)
+ShowCustomModelsHandler.register(overwrite=True)
 
 
-class UploadModelFileHandler(SQLHandler):
+class UploadCustomModelHandler(SQLHandler):
     """
-    UPLOAD MODEL FILE model_name
+    UPLOAD CUSTOM MODEL model_name
         FROM local_path [ overwrite ];
 
     # Model Name
@@ -110,12 +110,12 @@ class UploadModelFileHandler(SQLHandler):
     The following command uploads a file to models space and overwrite any
     existing files at the specified path::
 
-        UPLOAD MODEL FILE model_name
+        UPLOAD CUSTOM MODEL model_name
             FROM 'llama3/' OVERWRITE;
 
     See Also
     --------
-    * ``DOWNLOAD MODEL FILE model_name``
+    * ``DOWNLOAD CUSTOM MODEL model_name``
 
     """  # noqa: E501
 
@@ -143,12 +143,12 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         return None
 
 
-UploadModelFileHandler.register(overwrite=True)
+UploadCustomModelHandler.register(overwrite=True)
 
 
-class DownloadModelFileHandler(SQLHandler):
+class DownloadCustomModelHandler(SQLHandler):
     """
-    DOWNLOAD MODEL FILE model_name
+    DOWNLOAD CUSTOM MODEL model_name
         [ local_path ]
         [ overwrite ];
 
@@ -182,17 +182,17 @@ class DownloadModelFileHandler(SQLHandler):
     The following command displays the contents of the file on the
     standard output::
 
-        DOWNLOAD MODEL FILE llama3;
+        DOWNLOAD CUSTOM MODEL llama3;
 
     The following command downloads a model to a specific location and
     overwrites any existing models folder with the name ``local_llama3`` on the local storage::
 
-        DOWNLOAD MODEL FILE llama3
+        DOWNLOAD CUSTOM MODEL llama3
             TO 'local_llama3' OVERWRITE;
 
     See Also
     --------
-    * ``UPLOAD MODEL FILE model_name FROM local_path``
+    * ``UPLOAD CUSTOM MODEL model_name FROM local_path``
 
     """  # noqa: E501
 
@@ -211,12 +211,12 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         return None
 
 
-DownloadModelFileHandler.register(overwrite=True)
+DownloadCustomModelHandler.register(overwrite=True)
 
 
-class DropModelFileHandler(SQLHandler):
+class DropCustomModelHandler(SQLHandler):
     """
-    DROP MODEL FILE model_name;
+    DROP CUSTOM MODEL model_name;
 
     # Model Name
     model_name = '<model-name>'
@@ -233,7 +233,7 @@ class DropModelFileHandler(SQLHandler):
     --------
     The following commands deletes a model from a model space::
 
-        DROP MODEL FILE llama3;
+        DROP CUSTOM MODEL llama3;
 
     """  # noqa: E501
 
@@ -247,4 +247,4 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         return None
 
 
-DropModelFileHandler.register(overwrite=True)
+DropCustomModelHandler.register(overwrite=True)