Merge pull request #65 from sanketshinde3001/main

Added a function to remove an ML engine
mindsdb · Nov 6, 2024 · fdf60e7 · fdf60e7
2 parents 09c81ea + 277adab
commit fdf60e7
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/ml_engines/ml_engine.ts b/src/ml_engines/ml_engine.ts
@@ -35,6 +35,16 @@ export default class MLEngine {
     this.connection_data = connection_data;
   }
 
+    /**
+   * Removes a specified mlEngine by its name.
+   * @param {string} mlEngineName - The name of the mlEngine to remove.
+   * @returns {Promise<void>} - Resolves when the mlEngine is successfully removed.
+   * @throws {MindsDbError} - Something went wrong deleting the mlEngine.
+   */
+  async remove(mlEngineName: string): Promise<void> {
+    await this.mlEnginesApiClient.deleteMLEngine(mlEngineName);
+  }
+
   /** Deletes this mlEngine.
    *  @throws {MindsDbError} - Something went wrong deleting the mlEngine.
    */