Enable fetching all versions of a model

Author: erbridge

lib/ReplicateClient.js

@@ -21,6 +21,39 @@ export default class ReplicateClient {
     }
   }
 
+  async getModelVersions(nameOrParams) {
+    let owner, name;
+
+    if (typeof nameOrParams === "string") {
+      const nameComponents = /^([\w-]+)\/([\w-]+)$/.exec(nameOrParams);
+
+      if (!nameComponents) {
+        throw new ReplicateError(`Invalid name: ${nameOrParams}`);
+      }
+
+      [, owner, name] = nameComponents;
+    } else if (nameOrParams.owner) {
+      owner = nameOrParams.owner;
+      name = nameOrParams.name;
+    } else {
+      const nameComponents = /^([\w-]+)\/([\w-]+)$/.exec(nameOrParams.name);
+
+      if (!nameComponents) {
+        throw new ReplicateError(`Invalid name: ${nameOrParams.name}`);
+      }
+
+      [, owner, name] = nameComponents;
+    }
+
+    const versions = await this.request(
+      `GET /v1/models/${owner}/${name}/versions`
+    );
+
+    return versions.map(
+      (version) => new Model({ owner, name, ...version }, this)
+    );
+  }
+
   model(idOrParams) {
     let owner, name, version;
 

lib/ReplicateClient.test.js

@@ -65,6 +65,61 @@ describe("constructor()", () => {
   });
 });
 
+describe("getModelVersions()", () => {
+  beforeEach(() => {
+    jest
+      .spyOn(client, "request")
+      .mockResolvedValue([
+        { id: "testversion0" },
+        { id: "testversion1" },
+        { id: "testversion2" },
+        { id: "testversion3" },
+        { id: "testversion4" },
+      ]);
+  });
+
+  it("returns array of Models", async () => {
+    const models = await client.getModelVersions("test-owner/test-name");
+
+    expect(models).toBeInstanceOf(Array);
+    expect(models[0]).toBeInstanceOf(Model);
+  });
+
+  it("returns array of Models with version set", async () => {
+    const models = await client.getModelVersions("test-owner/test-name");
+
+    expect(models).toBeInstanceOf(Array);
+    expect(models[0]).toBeInstanceOf(Model);
+    expect(models[0].version).toBe("testversion0");
+  });
+
+  it("parses name string", async () => {
+    const models = await client.getModelVersions("test-owner/test-name");
+
+    expect(models[0].owner).toBe("test-owner");
+    expect(models[0].name).toBe("test-name");
+  });
+
+  it("handles complete params", async () => {
+    const models = await client.getModelVersions({
+      owner: "test-owner",
+      name: "test-name",
+    });
+
+    expect(models[0].owner).toBe("test-owner");
+    expect(models[0].name).toBe("test-name");
+  });
+
+  it("handles params with combined name", async () => {
+    const models = await client.getModelVersions({
+      name: "test-owner/test-name",
+    });
+
+    expect(models[0].owner).toBe("test-owner");
+    expect(models[0].name).toBe("test-name");
+  });
+});
+
 describe("model()", () => {
   it("returns Model", () => {
     const model = client.model("test-owner/test-name@testversion");