@kamrankhan027/probe-ai 1.0.0 → 1.0.2

package/dist/index.js

@@ -4508,11 +4508,11 @@ var GraphState = Annotation.Root({
     default: () => []
   })
 });
-function createProbeAgent(apiKey2) {
+function createProbeAgent(apiKey) {
   const model = new ChatGoogleGenerativeAI({
     model: "gemini-3.6-flash",
     temperature: 0.2,
-    apiKey: apiKey2
+    apiKey
   }).bindTools(tools);
   const callModel = async (state) => {
     const response = await model.invoke(state.messages);
@@ -4550,7 +4550,6 @@ marked.use(
     }
   })
 );
-var apiKey = process.env.GEMINI_API_KEY || process.env.GOOGLE_API_KEY;
 function printBanner() {
   console.log(
     boxen(
@@ -4573,29 +4572,34 @@ ${chalk6.yellow("Try prompts like:")}
 }
 async function main() {
   printBanner();
-  if (!apiKey) {
-    console.log(
-      chalk6.yellow(
-        "\n[WARNING] GEMINI_API_KEY not found. Please set GEMINI_API_KEY in your environment or .env file.\n"
-      )
-    );
-  }
-  const app = createProbeAgent(apiKey || "");
-  const messages = [PROBE_SYSTEM_MESSAGE];
   const rl = readline.createInterface({
     input: process.stdin,
     output: process.stdout
   });
-  const promptUser = () => {
+  const promptUser = (questionText) => {
     return new Promise((resolve) => {
-      rl.question(chalk6.bold.green("\nYou: "), (answer) => {
+      rl.question(questionText, (answer) => {
         resolve(answer.trim());
       });
     });
   };
+  let apiKey = process.env.GEMINI_API_KEY || process.env.GOOGLE_API_KEY;
+  if (!apiKey) {
+    console.log(chalk6.yellow("\u{1F511} No GEMINI_API_KEY found in your environment or .env file."));
+    console.log(chalk6.dim("Get a free key from https://aistudio.google.com/\n"));
+    apiKey = await promptUser(chalk6.bold.cyan("Enter your Gemini API Key: "));
+    if (!apiKey) {
+      console.log(chalk6.red("\nError: An API key is required to run ProbeAI. Exiting...\n"));
+      rl.close();
+      process.exit(1);
+    }
+    console.log(chalk6.green("\u2705 API Key registered for this session.\n"));
+  }
+  const app = createProbeAgent(apiKey);
+  const messages = [PROBE_SYSTEM_MESSAGE];
   while (true) {
     try {
-      const input = await promptUser();
+      const input = await promptUser(chalk6.bold.green("\nYou: "));
       if (!input) continue;
       if (input.toLowerCase() === "exit" || input.toLowerCase() === "quit") {
         console.log(chalk6.cyan("\nGoodbye! \u{1F44B}\n"));

package/package.json

@@ -1,19 +1,17 @@
 {
   "name": "@kamrankhan027/probe-ai",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "Autonomous API Schema & Reliability Inspector CLI. Probes OpenAPI specs, generates test suites, executes real-time HTTP calls, and exports Postman collections.",
-  "main": "./dist/index.js",
-  "types": "./dist/index.d.ts",
-  "bin": {
-    "probe-ai": "./dist/index.js"
-  },
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "bin": "dist/index.js",
   "files": [
     "dist"
   ],
   "scripts": {
     "build": "tsup",
     "dev": "tsup --watch",
-    "start": "node ./dist/index.js"
+    "start": "node dist/index.js"
   },
   "keywords": [
     "ai",