From e62d391e096c6effc85a8b2dffffb3ba5daca92f Mon Sep 17 00:00:00 2001
From: June Rockway <2367+jrockway@users.noreply.github.com>
Date: Fri, 31 Jul 2026 12:06:50 -0400
Subject: [PATCH] Add Nix flake for installing the CLI on macOS and Linux

---
 README.md       | 29 +++++++++++++++++++++++
 flake.lock      | 61 +++++++++++++++++++++++++++++++++++++++++++++++++
 flake.nix       | 50 ++++++++++++++++++++++++++++++++++++++++
 nix/package.nix | 31 +++++++++++++++++++++++++
 4 files changed, 171 insertions(+)
 create mode 100644 flake.lock
 create mode 100644 flake.nix
 create mode 100644 nix/package.nix

diff --git a/README.md b/README.md
index c0acbde..d83847b 100644
--- a/README.md
+++ b/README.md
@@ -27,6 +27,35 @@ brew update
 brew upgrade baseten
 ```
 
+### Nix (macOS and Linux)
+
+This repository is a Nix flake. Run the CLI without installing anything:
+
+```bash
+nix run github:basetenlabs/baseten-cli -- --help
+```
+
+or install it into your profile:
+
+```bash
+nix profile install github:basetenlabs/baseten-cli
+```
+
+To install with [Home Manager](https://github.com/nix-community/home-manager),
+add the flake as an input and put the package in `home.packages`:
+
+```nix
+{
+  inputs.baseten-cli.url = "github:basetenlabs/baseten-cli";
+
+  # In your home-manager configuration (with `inputs` passed through):
+  # home.packages = [ inputs.baseten-cli.packages.${pkgs.system}.default ];
+}
+```
+
+The flake also exposes `overlays.default`, which adds the package as
+`pkgs.baseten`.
+
 ### Prebuilt binaries
 
 Download the [latest release](https://github.com/basetenlabs/baseten-cli/releases/latest) for Windows, macOS, or Linux, extract the archive, and place the `baseten` executable on your `PATH`.
diff --git a/flake.lock b/flake.lock
new file mode 100644
index 0000000..adbeb70
--- /dev/null
+++ b/flake.lock
@@ -0,0 +1,61 @@
+{
+  "nodes": {
+    "flake-utils": {
+      "inputs": {
+        "systems": "systems"
+      },
+      "locked": {
+        "lastModified": 1731533236,
+        "narHash": "sha256-l0KFg5HjrsfsO/JpG+r7fRrqm12kzFHyUHqHCVpMMbI=",
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "rev": "11707dc2f618dd54ca8739b309ec4fc024de578b",
+        "type": "github"
+      },
+      "original": {
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "type": "github"
+      }
+    },
+    "nixpkgs": {
+      "locked": {
+        "lastModified": 1785386831,
+        "narHash": "sha256-sPS3CaXH8RAT3FZRuy4VcV47iuYIWMMfa0GbyJKC3o4=",
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "rev": "21ea275a7c46aef9d4d6ddc962e6d562e9d94183",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nixos",
+        "ref": "nixos-26.05",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
+    "root": {
+      "inputs": {
+        "flake-utils": "flake-utils",
+        "nixpkgs": "nixpkgs"
+      }
+    },
+    "systems": {
+      "locked": {
+        "lastModified": 1681028828,
+        "narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
+        "owner": "nix-systems",
+        "repo": "default",
+        "rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nix-systems",
+        "repo": "default",
+        "type": "github"
+      }
+    }
+  },
+  "root": "root",
+  "version": 7
+}
diff --git a/flake.nix b/flake.nix
new file mode 100644
index 0000000..abf7e36
--- /dev/null
+++ b/flake.nix
@@ -0,0 +1,50 @@
+{
+  description = "CLI for Baseten";
+
+  inputs = {
+    nixpkgs.url = "github:nixos/nixpkgs/nixos-26.05";
+    flake-utils.url = "github:numtide/flake-utils";
+  };
+
+  outputs =
+    {
+      self,
+      nixpkgs,
+      flake-utils,
+    }:
+    flake-utils.lib.eachDefaultSystem (
+      system:
+      let
+        pkgs = import nixpkgs { inherit system; };
+        baseten = pkgs.callPackage ./nix/package.nix {
+          version = if self ? shortRev then "unstable-${self.shortRev}" else "dev";
+        };
+      in
+      {
+        packages = {
+          default = baseten;
+          inherit baseten;
+        };
+
+        checks = {
+          build = baseten;
+          # Smoke test: the built binary runs and reports the expected version.
+          version = pkgs.runCommand "baseten-version-check" { } ''
+            got="$(${pkgs.lib.getExe baseten} version)"
+            want=${pkgs.lib.escapeShellArg baseten.version}
+            if [ "$got" != "$want" ]; then
+              echo "baseten version reported '$got', expected '$want'" >&2
+              exit 1
+            fi
+            ${pkgs.lib.getExe baseten} --help > /dev/null
+            touch $out
+          '';
+        };
+      }
+    )
+    // {
+      overlays.default = final: prev: {
+        baseten = final.callPackage ./nix/package.nix { };
+      };
+    };
+}
diff --git a/nix/package.nix b/nix/package.nix
new file mode 100644
index 0000000..5125a17
--- /dev/null
+++ b/nix/package.nix
@@ -0,0 +1,31 @@
+{
+  lib,
+  buildGoModule,
+  version ? "dev",
+}:
+buildGoModule {
+  pname = "baseten";
+  inherit version;
+
+  src = lib.cleanSource ../.;
+
+  # Update by setting to lib.fakeHash and copying the hash from the build error.
+  vendorHash = "sha256-ghtyvdt80fy6JCw0J20a70fai9fUQyJW5vS8gQibZWE=";
+
+  subPackages = [ "cmd/baseten" ];
+
+  env.CGO_ENABLED = 0;
+
+  ldflags = [
+    "-s"
+    "-w"
+    "-X github.com/basetenlabs/baseten-cli/internal/cmd.Version=${version}"
+  ];
+
+  meta = {
+    description = "CLI for Baseten";
+    homepage = "https://github.com/basetenlabs/baseten-cli";
+    license = lib.licenses.mit;
+    mainProgram = "baseten";
+  };
+}