Support model version deletion

Author: erbridge

lib/Model.js

@@ -40,6 +40,12 @@ export default class Model extends ReplicateObject {
     return `GET /v1/models/${this.owner}/${this.name}/versions/${this.version}`;
   }
 
+  async deleteVersion() {
+    await this.client.request(
+      `DELETE /v1/models/${this.owner}/${this.name}/versions/${this.version}`
+    );
+  }
+
   async predict(
     input,
     { onUpdate, onTemporaryError } = {},

lib/Model.test.js

@@ -69,6 +69,18 @@ describe("load()", () => {
   });
 });
 
+describe("deleteVersion()", () => {
+  it("makes request to delete model version", async () => {
+    jest.spyOn(client, "request").mockResolvedValue();
+
+    await model.deleteVersion();
+
+    expect(client.request).toHaveBeenCalledWith(
+      "DELETE /v1/models/test-owner/test-name/versions/testversion"
+    );
+  });
+});
+
 describe("predict()", () => {
   it("makes request to create prediction", async () => {
     jest.spyOn(model, "createPrediction").mockResolvedValue(

lib/ReplicateClient.js

@@ -147,7 +147,7 @@ export default class ReplicateClient {
   }
 
   #parseAction(action) {
-    const result = /^(GET|POST) (\/[^\s]+)$/.exec(action);
+    const result = /^(GET|POST|DELETE) (\/[^\s]+)$/.exec(action);
 
     if (!result) {
       throw new ReplicateError(`Invalid action: ${action}`);