Throw if no API token provided

Author: erbridge

lib/ReplicateClient.js

@@ -1,5 +1,6 @@
 import fetch from "node-fetch";
 import pkg from "../package.json" assert { type: "json" };
+import { ReplicateError } from "./errors.js";
 import Prediction from "./Prediction.js";
 import { sleep } from "./utils.js";
 
@@ -10,6 +11,10 @@ export default class ReplicateClient {
   constructor({ baseURL, token }) {
     this.baseURL = baseURL || "https://api.replicate.com";
     this.token = token || process.env.REPLICATE_API_TOKEN;
+
+    if (!this.token) {
+      throw new ReplicateError("Missing API token");
+    }
   }
 
   // TODO: Optionally autocancel prediction on exception.

lib/ReplicateClient.test.js

@@ -1,4 +1,5 @@
 import { jest } from "@jest/globals";
+import { ReplicateError } from "./errors.js";
 import { PredictionStatus } from "./Prediction.js";
 import ReplicateClient from "./ReplicateClient.js";
 
@@ -10,6 +11,46 @@ beforeEach(() => {
   client = new ReplicateClient({});
 });
 
+describe("baseURL", () => {
+  it("is baseURL provided to constructor", () => {
+    const client = new ReplicateClient({
+      baseURL: "http://test.host",
+    });
+
+    expect(client.baseURL).toEqual("http://test.host");
+  });
+
+  it("defaults to https://api.replicate.com", () => {
+    const client = new ReplicateClient({});
+
+    expect(client.baseURL).toEqual("https://api.replicate.com");
+  });
+});
+
+describe("token", () => {
+  it("is token provided to constructor", () => {
+    const client = new ReplicateClient({
+      token: "test-token-from-constructor",
+    });
+
+    expect(client.token).toEqual("test-token-from-constructor");
+  });
+
+  it("defaults to REPLICATE_API_TOKEN env var", () => {
+    const client = new ReplicateClient({});
+
+    expect(client.token).toEqual("test-token-from-env");
+  });
+});
+
+describe("constructor()", () => {
+  it("throws when no token is provided or set in env", () => {
+    delete process.env.REPLICATE_API_TOKEN;
+
+    expect(() => new ReplicateClient({})).toThrowError(ReplicateError);
+  });
+});
+
 describe("predict()", () => {
   it("makes request to create prediction", async () => {
     jest

lib/errors.js

@@ -0,0 +1 @@
+export class ReplicateError extends Error {}