Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 22 additions & 22 deletions singlestoredb/fusion/handlers/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,9 @@
from .utils import get_file_space


class ShowModelsHandler(ShowFilesHandler):
class ShowModelFilesHandler(ShowFilesHandler):
"""
SHOW MODELS
SHOW MODEL FILES
[ at_path ] [ <like> ]
[ <order-by> ]
[ <limit> ] [ recursive ] [ extended ];
Expand Down Expand Up @@ -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
Expand All @@ -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
Expand Down Expand Up @@ -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

Expand Down Expand Up @@ -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 ];

Expand Down Expand Up @@ -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

Expand All @@ -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 = '<model-name>'
Expand All @@ -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

Expand All @@ -247,4 +247,4 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
return None


DropModelsHandler.register(overwrite=True)
DropModelFileHandler.register(overwrite=True)