diff --git a/singlestoredb/fusion/handlers/models.py b/singlestoredb/fusion/handlers/models.py index 5bb5cefa..acd391c8 100644 --- a/singlestoredb/fusion/handlers/models.py +++ b/singlestoredb/fusion/handlers/models.py @@ -10,9 +10,9 @@ from .utils import get_file_space -class ShowModelsHandler(ShowFilesHandler): +class ShowModelFilesHandler(ShowFilesHandler): """ - SHOW MODELS + SHOW MODEL FILES [ at_path ] [ ] [ ] [ ] [ recursive ] [ extended ]; @@ -53,16 +53,16 @@ class ShowModelsHandler(ShowFilesHandler): -------- The following command lists the models:: - SHOW MODELS; + SHOW MODEL FILES; The following command lists the models with additional information:: - SHOW MODELS EXTENDED; + SHOW MODEL FILES EXTENDED; See Also -------- - * ``UPLOAD MODEL model_name FROM path`` - * ``DOWNLOAD MODEL model_name`` + * ``UPLOAD MODEL FILE model_name FROM path`` + * ``DOWNLOAD MODEL FILE model_name`` """ # noqa: E501 @@ -73,12 +73,12 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]: return super().run(params) -ShowModelsHandler.register(overwrite=True) +ShowModelFilesHandler.register(overwrite=True) -class UploadModelHandler(SQLHandler): +class UploadModelFileHandler(SQLHandler): """ - UPLOAD MODEL model_name + UPLOAD MODEL FILE model_name FROM local_path [ overwrite ]; # Model Name @@ -110,12 +110,12 @@ class UploadModelHandler(SQLHandler): The following command uploads a file to models space and overwrite any existing files at the specified path:: - UPLOAD MODEL model_name + UPLOAD MODEL FILE model_name FROM 'llama3/' OVERWRITE; See Also -------- - * ``DOWNLOAD MODEL model_name`` + * ``DOWNLOAD MODEL FILE model_name`` """ # noqa: E501 @@ -143,12 +143,12 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]: return None -UploadModelHandler.register(overwrite=True) +UploadModelFileHandler.register(overwrite=True) -class DownloadModelHandler(SQLHandler): +class DownloadModelFileHandler(SQLHandler): """ - DOWNLOAD MODEL model_name + DOWNLOAD MODEL FILE model_name [ local_path ] [ overwrite ]; @@ -182,17 +182,17 @@ class DownloadModelHandler(SQLHandler): The following command displays the contents of the file on the standard output:: - DOWNLOAD MODEL llama3; + DOWNLOAD MODEL FILE 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 llama3 + DOWNLOAD MODEL FILE llama3 TO 'local_llama3' OVERWRITE; See Also -------- - * ``UPLOAD MODEL model_name FROM local_path`` + * ``UPLOAD MODEL FILE model_name FROM local_path`` """ # noqa: E501 @@ -211,12 +211,12 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]: return None -DownloadModelHandler.register(overwrite=True) +DownloadModelFileHandler.register(overwrite=True) -class DropModelsHandler(SQLHandler): +class DropModelFileHandler(SQLHandler): """ - DROP MODEL model_name; + DROP MODEL FILE model_name; # Model Name model_name = '' @@ -233,7 +233,7 @@ class DropModelsHandler(SQLHandler): -------- The following commands deletes a model from a model space:: - DROP MODEL llama3; + DROP MODEL FILE llama3; """ # noqa: E501 @@ -247,4 +247,4 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]: return None -DropModelsHandler.register(overwrite=True) +DropModelFileHandler.register(overwrite=True)