all-the-public-replicate-models 1.0.1 → 1.1.1

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "all-the-public-replicate-models",
   "description": "Metadata for all the public models on Replicate, bundled up into an npm package",
-  "version": "1.0.1",
+  "version": "1.1.1",
   "devDependencies": {
     "chai": "^4.3.10",
     "lodash-es": "^4.17.21",
@@ -10,8 +10,15 @@
   },
   "type": "module",
   "main": "index.mjs",
+  "exports": {
+    ".": "./index.mjs",
+    "./lite": "./lite.mjs"
+  },
   "scripts": {
-    "build": "node script/build.js > models.json",
+    "build": "node script/build.js",
     "test": "mocha test.js"
+  },
+  "engines": {
+    "node": ">=18"
   }
 }

package/script/build.js

@@ -1,6 +1,7 @@
 import Replicate from "replicate";
 const replicate = new Replicate();
 import {unset} from "lodash-es";
+import fs from "fs";
 
 async function main () {
   console.error("Fetching all public models from Replicate...")
@@ -17,8 +18,20 @@ async function main () {
     unset(models[i], 'default_example.webhook_completed')
     unset(models[i], 'latest_version.openapi_schema.paths')
   }
+  
+  const lite = models.map(model => {
+    return {
+      url: model.url,
+      owner: model.owner,
+      name: model.name,
+      description: model.description,
+      run_count: model.run_count,
+      cover_image_url: model.cover_image_url
+    }
+  });
 
-  process.stdout.write(JSON.stringify(models, null, 2))
+  fs.writeFileSync('models.json', JSON.stringify(models, null, 2))
+  fs.writeFileSync('models-lite.json', JSON.stringify(lite, null, 2))
 }
 
 main()

package/test.js

@@ -6,8 +6,8 @@ describe('models module', () => {
     expect(models).to.be.an('array');
   });
 
-  it('should have at least 500 objects', () => {
-    expect(models).to.have.lengthOf.at.least(500);
+  it('should have at least 1000 models', () => {
+    expect(models).to.have.lengthOf.at.least(1000);
   });
 
   describe('first object in the array', () => {