Update test script

SubstrateLabs · Apr 12, 2024 · ff5ca9b · ff5ca9b
1 parent c03fae1
commit ff5ca9b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1,082 deletions.
diff --git a/examples/kitchen-sink.ts b/examples/kitchen-sink.ts
@@ -137,7 +137,7 @@ const examples = [
       prompt: "what are these paintings of and who made them?",
       image_uris: [
         "https://media.substrate.run/docs-fuji-red.jpg",
-        "https://media.substrate.run/docs-fuji-blue.png",
+        "https://media.substrate.run/docs-fuji-blue.jpg",
       ],
     }),
     envs: ALL_ENVS,
@@ -168,7 +168,7 @@ const examples = [
         "hokusai futuristic supercell spiral cloud with glowing core over turbulent ocean",
       store: "hosted",
     }),
-    envs: [STAGING_V0, STAGING_V1],
+    envs: [STAGING_V0, PRODUCTION_V0],
   },
   {
     node: new MultiGenerateImage({
@@ -177,7 +177,7 @@ const examples = [
       store: "hosted",
       num_images: 2,
     }),
-    envs: [STAGING_V0, STAGING_V1],
+    envs: [STAGING_V0, PRODUCTION_V0],
   },
   {
     node: new GenerativeEditImage({
@@ -480,10 +480,15 @@ const examples = [
   },
 ];
 
+const noColor = process.argv.includes("--no-color");
 const ok = (message: string, ...rest: any[]) =>
-  console.log(`\x1b[32m✓ ${message}\x1b[0m`, ...rest);
+  noColor
+    ? console.log(`✓ ${message}`, ...rest)
+    : console.log(`\x1b[32m✓ ${message}\x1b[0m`, ...rest);
 const error = (message: string, ...rest: any[]) =>
-  console.log(`\x1b[31mx ${message}\x1b[0m`, ...rest);
+  noColor
+    ? console.error(`x ${message}`, ...rest)
+    : console.error(`\x1b[31mx ${message}\x1b[0m`, ...rest);
 
 async function main() {
   const SUBSTRATE_API_KEY = process.env["SUBSTRATE_API_KEY"];
@@ -513,5 +518,6 @@ async function main() {
       }
     }
   }
+  console.info("🌀 done.");
 }
 main();