changes

Author: mohit-singlestore

singlestoredb/fusion/handlers/models.py

@@ -349,9 +349,9 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
 StopModelHandler.register(overwrite=True)
 
 
-class ShowInferenceAPIsHandler(SQLHandler):
+class ShowModelsHandler(SQLHandler):
     """
-    SHOW INFERENCE APIS ;
+    SHOW MODELS ;
 
     Description
     -----------
@@ -361,13 +361,13 @@ class ShowInferenceAPIsHandler(SQLHandler):
     --------
     The following command lists all inference APIs::
 
-        SHOW INFERENCE APIS;
+        SHOW MODELS;
 
     See Also
     --------
     * ``START MODEL model_name``
     * ``STOP MODEL model_name``
-    * ``DROP INFERENCE API model_name``
+    * ``DROP MODEL model_name``
 
     """  # noqa: E501
 
@@ -390,12 +390,12 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         return res
 
 
-ShowInferenceAPIsHandler.register(overwrite=True)
+ShowModelsHandler.register(overwrite=True)
 
 
-class DropInferenceAPIHandler(SQLHandler):
+class DropModelHandler(SQLHandler):
     """
-    DROP INFERENCE API model_name ;
+    DROP MODEL model_name ;
 
     # Model Name
     model_name = '<model-name>'
@@ -412,13 +412,13 @@ class DropInferenceAPIHandler(SQLHandler):
     --------
     The following command drops an inference API::
 
-        DROP INFERENCE API my_model;
+        DROP MODEL my_model;
 
     See Also
     --------
     * ``START MODEL model_name``
     * ``STOP MODEL model_name``
-    * ``SHOW INFERENCE APIS``
+    * ``SHOW MODELS``
 
     """  # noqa: E501
 
@@ -439,4 +439,4 @@ def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         return res
 
 
-DropInferenceAPIHandler.register(overwrite=True)
+DropModelHandler.register(overwrite=True)

singlestoredb/management/inference_api.py

@@ -353,5 +353,5 @@ def drop(self, model_name: str) -> ModelOperationResult:
         """
         if self._manager is None:
             raise ManagementError(msg='Manager not initialized')
-        res = self._manager._delete(f'models/{model_name}/drop')
+        res = self._manager._delete(f'models/{model_name}')
         return ModelOperationResult.from_drop_response(res.json())