Merge pull request #19 from andyjessop/aj/better-console

Aj/better console

Author: andyjessop

package-lock.json

@@ -37,6 +37,7 @@
 		"@mantine/modals": "^7.16.3",
 		"@mantine/notifications": "^7.16.3",
 		"ai": "^4.1.39",
+		"chalk": "^5.4.1",
 		"dayjs": "^1.11.13",
 		"ejs": "^3.1.10",
 		"eventsource-parser": "^3.0.0",

package.json

@@ -7,6 +7,7 @@ import {
 	log,
 	select,
 } from "@clack/prompts";
+import chalk from "chalk";
 import * as path from "path";
 import {
 	copyDirectory,
@@ -123,17 +124,34 @@ async function main(): Promise<void> {
 	try {
 		formatDirectory(newLocation);
 
-		log.success(`Project successfully generated at ${newLocation}`);
+		log.success(
+			`${chalk.green(`Project successfully generated at ${newLocation}`)}`,
+		);
 	} catch (error) {
 		log.error(JSON.stringify(error, null, 2));
 		log.error("An error occurred while generating the project.");
 		process.exit(1);
 	}
 
-	// Display outro message.
-	log.warning(
-		"In order to use your worker in staging/production, you need to provide an API_KEY secret, and pass that value in requests with the x-api-key header.",
+	const missingSecrets = [];
+
+	if (provider === "openai") {
+		missingSecrets.push(
+			`${chalk.yellow("OPENAI_API_KEY")} (Required for OpenAI integration.)`,
+		);
+	}
+
+	missingSecrets.push(
+		`${chalk.yellow("API_KEY")} (Required to access your worker in staging/production. Use the "${chalk.yellow("x-api-key")}" header in your requests.)`,
 	);
+
+	if (missingSecrets.length > 0) {
+		log.warning(
+			"Please add the following secrets to your account (or .dev.vars for local development):\n- " +
+				missingSecrets.join("\n- "),
+		);
+	}
+
 	outro(`You're all set! Run 'npx nx dev ${projectName}' to start the worker.`);
 }
 

scripts/generate-worker/index.ts

@@ -0,0 +1,3 @@
+<% if (provider === 'openai') { %>
+OPENAI_API_KEY=
+<% } %>

scripts/generate-worker/structured-output/.dev.vars.ejs

@@ -0,0 +1,3 @@
+<% if (provider === 'openai') { %>
+OPENAI_API_KEY=
+<% } %>

scripts/generate-worker/text-generation/.dev.vars.ejs

@@ -0,0 +1,3 @@
+<% if (provider === 'openai') { %>
+OPENAI_API_KEY=
+<% } %>

scripts/generate-worker/tool-calling/.dev.vars.ejs

@@ -0,0 +1,3 @@
+<% if (provider === 'openai') { %>
+OPENAI_API_KEY=
+<% } %>

scripts/generate-worker/workflow-evaluator-optimiser/.dev.vars.ejs

@@ -0,0 +1,3 @@
+<% if (provider === 'openai') { %>
+OPENAI_API_KEY=
+<% } %>

scripts/generate-worker/workflow-orchestrator-workers/.dev.vars.ejs

@@ -0,0 +1,3 @@
+<% if (provider === 'openai') { %>
+OPENAI_API_KEY=
+<% } %>

scripts/generate-worker/workflow-parallelisation/.dev.vars.ejs

@@ -0,0 +1,3 @@
+<% if (provider === 'openai') { %>
+OPENAI_API_KEY=
+<% } %>