Initial client

Author: bfirsh

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "editor.formatOnSave": true
+}

README.md

@@ -8,43 +8,55 @@ This is a design document for figuring out the design of the future JavaScript c
 
 Install with `npm install replicate`
 
+Set your API token as `REPLICATE_API_TOKEN` in your environment.
+
+To run a prediction and return its 
+
 ```jsx
-import Replicate from "@replicate/client";
+import replicate from "@replicate/client";
 
-const replicate = new Replicate({
-  // get your token from https://replicate.com/account
-  auth: "<YOUR TOKEN HERE>",
-  userAgent: "my-app/1.2.3",
+let output = await replicate.predict({
+  version: "<MODEL VERSION>",
+  input: {
+    // your model inputs need to be set here
+    prompt: "painting of a cat by andy warhol",
+  },
 });
 
-const prediction = await replicate.createAndAwaitPrediction({
+// "https://replicate.delivery/pbxt/lGWovsQZ7jZuNtPvofMth1rSeCcVn5xes8dWWdWZ64MlTi7gA/out-0.png"
+```
+
+Or, you can run a prediction in the background to do more advanced stuff:
+
+```jsx
+let prediction = await replicate.predictions.create({
   version: "<MODEL VERSION>",
   input: {
-    // your model inputs need to be set here
     prompt: "painting of a cat by andy warhol",
   },
 });
 
-// {
-//   id: "ufawqhfynnddngldkgtslldrkq",
-//   version: "5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa",
-//   urls: {
-//     get: "https://api.replicate.com/v1/predictions/ufawqhfynnddngldkgtslldrkq",
-//     cancel: "https://api.replicate.com/v1/predictions/ufawqhfynnddngldkgtslldrkq/cancel"
-//   },
-//   created_at: "2020-02-02T00:00:00.00000Z",
-//   started_at: "2022-02-02T00:01:00.00000Z",
-//   completed_at: "2022-02-02T00:02:00.00000Z",
-//   status: "starting",
-//   input: {
-//     text: "Alice"
-//   },
-//   output: [
-//     "https://replicate.com/api/models/stability-ai/stable-diffusion/files/9c3b6fe4-2d37-4571-a17a-83951b1cb120/out-0.png"
-//   ],
-//   error: null,
-//   metrics: {}
-// }
+prediction.status
+// "starting"
+
+await prediction.reload()
+prediction.status
+// "processing"
+
+prediction.logs
+// Using seed: 53168
+//  0%|          | 0/50 [00:00<?, ?it/s]
+//  2%|▏         | 1/50 [00:00<00:12,  3.83it/s]
+// ...
+
+await prediction.wait()
+
+prediction.status
+// "succeeded"
+
+prediction.output
+// "https://replicate.delivery/pbxt/lGWovsQZ7jZuNtPvofMth1rSeCcVn5xes8dWWdWZ64MlTi7gA/out-0.png"
+
 ```
 
 ## Features

index.js

@@ -0,0 +1,3 @@
+import Client from "./lib/client.js";
+const replicate = new Client({});
+export default replicate;

lib/client.js

@@ -0,0 +1,71 @@
+import fetch from "node-fetch";
+import pkg from "../package.json" assert { type: "json" };
+
+class Prediction {
+  static fromJSON(json) {
+    const prediction = new Prediction();
+    Object.assign(prediction, json);
+    // TODO: make prediction.version a Version object
+    return prediction;
+  }
+}
+
+class PredictionCollection {
+  constructor({ client }) {
+    this.client = client;
+  }
+
+  async create({ version, input }) {
+    const response = await this.client._request({
+      method: "POST",
+      path: `/v1/predictions`,
+      body: {
+        version,
+        input,
+      },
+    });
+    return Prediction.fromJSON(response);
+  }
+
+  async get({ id }) {
+    const response = await this.client._request({
+      method: "GET",
+      path: `/v1/predictions/${id}`,
+    });
+    return Prediction.fromJSON(response);
+  }
+
+  async list() {
+    const response = await this.client._request({
+      method: "GET",
+      path: `/v1/predictions`,
+    });
+    // TODO pagination
+    return response.results.map((prediction) =>
+      Prediction.fromJSON(prediction)
+    );
+  }
+}
+
+export default class Client {
+  constructor({ token }) {
+    this.token = token || process.env.REPLICATE_API_TOKEN;
+    this.predictions = new PredictionCollection({ client: this });
+  }
+
+  async _request({ method, path, body }) {
+    const response = await fetch(`https://api.replicate.com${path}`, {
+      method: method,
+      headers: this._headers(),
+      body: JSON.stringify(body),
+    });
+    return await response.json();
+  }
+
+  _headers() {
+    return {
+      Authorization: `Token ${this.token}`,
+      "User-Agent": `replicate-js@${pkg.version}`,
+    };
+  }
+}

package-lock.json

@@ -0,0 +1,23 @@
+{
+  "name": "replicate",
+  "version": "0.0.1",
+  "description": "This is a design document for figuring out the design of the future JavaScript client for Replicate’s HTTP API. Let’s design the API here, before we write any code. README-driven development!",
+  "main": "index.js",
+  "type": "module",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/replicate/replicate-js.git"
+  },
+  "author": "",
+  "license": "Apache-2.0",
+  "bugs": {
+    "url": "https://github.com/replicate/replicate-js/issues"
+  },
+  "homepage": "https://github.com/replicate/replicate-js#readme",
+  "dependencies": {
+    "node-fetch": "^3.3.0"
+  }
+}