Add support for new deployment endpoints

Author: aron

index.d.ts

@@ -194,6 +194,25 @@ declare module "replicate" {
         deployment_owner: string,
         deployment_name: string
       ): Promise<Deployment>;
+      create(deployment_config: {
+        name: string;
+        model: string;
+        version: string;
+        hardware: string;
+        min_instances: number;
+        max_instances: number;
+      }): Promise<Deployment>;
+      update(
+        deployment_owner: string,
+        deployment_name: string,
+        deployment_config: {
+          version: string;
+          hardware: string;
+          min_instances: number;
+          max_instances: number;
+        }
+      ): Promise<Deployment>;
+      list(): Promise<Page<Deployment>>;
     };
 
     hardware: {

index.js

@@ -66,6 +66,9 @@ class Replicate {
 
     this.deployments = {
       get: deployments.get.bind(this),
+      create: deployments.create.bind(this),
+      update: deployments.update.bind(this),
+      list: deployments.list.bind(this),
       predictions: {
         create: deployments.predictions.create.bind(this),
       },

lib/deployments.js

@@ -57,9 +57,82 @@ async function getDeployment(deployment_owner, deployment_name) {
   return response.json();
 }
 
+/**
+ * @typedef {Object} DeploymentCreateRequest - Request body for `deployments.create`
+ * @property {string} name - the name of the deployment
+ * @property {string} model - the full name of the model that you want to deploy e.g. stability-ai/sdxl
+ * @property {string} version - the 64-character string ID of the model version that you want to deploy
+ * @property {string} hardware - the SKU for the hardware used to run the model, via `replicate.hardware.list()`
+ * @property {number} min_instances - the minimum number of instances for scaling
+ * @property {number} max_instances - the maximum number of instances for scaling
+ */
+
+/**
+ * Create a deployment
+ *
+ * @param {DeploymentCreateRequest} config - Required. The deployment config.
+ * @returns {Promise<object>} Resolves with the deployment data
+ */
+async function createDeployment(deployment_config) {
+  const response = await this.request("/deployments", {
+    method: "POST",
+    data: deployment_config,
+  });
+
+  return response.json();
+}
+
+/**
+ * @typedef {Object} DeploymentUpdateRequest - Request body for `deployments.update`
+ * @property {string} version - the 64-character string ID of the model version that you want to deploy
+ * @property {string} hardware - the SKU for the hardware used to run the model, via `replicate.hardware.list()`
+ * @property {number} min_instances - the minimum number of instances for scaling
+ * @property {number} max_instances - the maximum number of instances for scaling
+ */
+
+/**
+ * Update an existing deployment
+ *
+ * @param {string} deployment_owner - Required. The username of the user or organization who owns the deployment
+ * @param {string} deployment_name - Required. The name of the deployment
+ * @param {DeploymentUpdateRequest} deployment_config - Required. The deployment changes.
+ * @returns {Promise<object>} Resolves with the deployment data
+ */
+async function updateDeployment(
+  deployment_owner,
+  deployment_name,
+  deployment_config
+) {
+  const response = await this.request(
+    `/deployments/${deployment_owner}/${deployment_name}`,
+    {
+      method: "PATCH",
+      data: deployment_config,
+    }
+  );
+
+  return response.json();
+}
+
+/**
+ * List all deployments
+ *
+ * @returns {Promise<object>} - Resolves with a page of deployments
+ */
+async function listDeployments() {
+  const response = await this.request("/deployments", {
+    method: "GET",
+  });
+
+  return response.json();
+}
+
 module.exports = {
   predictions: {
     create: createPrediction,
   },
   get: getDeployment,
+  create: createDeployment,
+  update: updateDeployment,
+  list: listDeployments,
 };

tsconfig.json

@@ -5,7 +5,5 @@
     "strict": true,
     "allowJs": true
   },
-  "exclude": [
-    "**/node_modules"
-  ]
-}
+  "exclude": ["**/node_modules", "integration"]
+}