@kamrankhan027/probe-ai 1.1.0 → 1.1.2

package/README.md

@@ -7,7 +7,7 @@
 [![npm version](https://img.shields.io/npm/v/@kamrankhan027/probe-ai.svg?style=for-the-badge&color=CB3837&logo=npm)](https://www.npmjs.com/package/@kamrankhan027/probe-ai)
 [![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg?style=for-the-badge)](https://opensource.org/licenses/MIT)
 [![LangGraph](https://img.shields.io/badge/LangGraph.js-1C3C3C?style=for-the-badge&logo=langchain&logoColor=white)](https://langchain-ai.github.io/langgraphjs/)
-[![Gemini](https://img.shields.io/badge/Gemini_3.6_Flash-8E75B2?style=for-the-badge&logo=google&logoColor=white)](https://ai.google.dev/)
+[![Multi-Provider](https://img.shields.io/badge/AI_Engine-Universal-8E75B2?style=for-the-badge)](https://ai.google.dev/)
 [![TypeScript](https://img.shields.io/badge/TypeScript-3178C6?style=for-the-badge&logo=typescript&logoColor=white)](https://www.typescriptlang.org/)
 
 </div>
@@ -18,7 +18,7 @@
 
 ### Option 1: Run Instantly via `npx` (Zero Installation Required)
 ```bash
-npx @kamrankhan027/probe-ai
+npx @kamrankhan027/probe-ai@latest
 ```
 
 ---
@@ -30,13 +30,10 @@ npx @kamrankhan027/probe-ai
 npm install -g @kamrankhan027/probe-ai
 ```
 
-**Using pnpm:**
+**Using pnpm / yarn:**
 ```bash
 pnpm add -g @kamrankhan027/probe-ai
-```
-
-**Using yarn:**
-```bash
+# or
 yarn global add @kamrankhan027/probe-ai
 ```
 
@@ -47,13 +44,20 @@ probe-ai
 
 ---
 
-## 🔑 API Key Setup
+## 🧠 Supported AI Models & Providers
+
+ProbeAI is **universal and provider-agnostic**. On startup, select your preferred engine:
 
-ProbeAI uses Google Gemini for fast, structured reasoning. 
-* **If you have a `.env` file**: Add `GEMINI_API_KEY=your_key_here`
-* **If running via `npx` without a `.env`**: ProbeAI will interactively prompt you to paste your free Gemini API key on startup.
+| Provider | Supported Models | Environment Variable |
+|---|---|---|
+| **🔷 Google Gemini** | `gemini-3.6-flash`, `gemini-3.5-flash`, `gemini-2.5-flash`, `gemini-2.5-pro` | `GEMINI_API_KEY` |
+| **🟢 OpenAI** | `gpt-5.6` (Sol/Terra/Luna), `gpt-4o`, `gpt-4o-mini`, `o3-mini`, `gpt-oss` | `OPENAI_API_KEY` |
+| **🟣 Anthropic Claude** | `claude-sonnet-5`, `claude-opus-5`, `claude-3-5-sonnet-latest`, `claude-3-5-haiku` | `ANTHROPIC_API_KEY` |
+| **⚡ Groq (LPU Engine)** | `llama-3.3-70b-versatile`, `gpt-oss-120b`, `qwen-3.6` (Free & Ultra-Fast) | `GROQ_API_KEY` |
+| **🌐 OpenRouter** | `deepseek/deepseek-r1:free`, `qwen/qwen-2.5-coder-32b-instruct:free`, `mistral-large-3` | `OPENROUTER_API_KEY` |
+| **🦙 Ollama (Local)** | `qwen2.5-coder:7b`, `deepseek-r1`, `llama3.3`, `gemma-4` (100% Offline & Free) | *None needed!* |
 
-*(Get a free key in 10 seconds at [aistudio.google.com](https://aistudio.google.com/)).*
+> 💡 **Custom Model Overrides**: You can pass any custom model string directly via environment variables (e.g. `OPENAI_MODEL=gpt-5.6` or `ANTHROPIC_MODEL=claude-sonnet-5`) or during the interactive startup prompt.
 
 ---
 
@@ -70,25 +74,20 @@ ProbeAI uses Google Gemini for fast, structured reasoning.
 ## 💡 Example Prompts to Try
 
 ```text
-You: Inspect https://bookmark-agent-backend.onrender.com and export a Postman collection
+❯ inspect https://bookmark-agent-backend.onrender.com and export a Postman collection
 ```
 *Auto-maps all endpoints and creates `reports/postman_collection.json`.*
 
 ```text
-You: Test all endpoints on https://bookmark-agent-backend.onrender.com and save an HTML report
+❯ test all endpoints on https://bookmark-agent-backend.onrender.com and save an HTML report
 ```
 *Executes Happy Path, Query Filter, and Error test cases and saves `reports/audit_report.html`.*
 
 ```text
-You: Set auth Bearer my_jwt_token_123 and test protected endpoints
+❯ set auth Bearer my_jwt_token_123 and test protected endpoints
 ```
 *Injects your authorization header into session memory for all subsequent calls.*
 
-```text
-You: Authenticate via POST https://api.example.com/login with {"email":"dev@example.com","password":"secret"}, then test /api/dashboard
-```
-*Extracts the JWT from the login response and tests the protected dashboard route automatically.*
-
 ---
 
 ## 🏗️ Repository Architecture
@@ -97,7 +96,9 @@ You: Authenticate via POST https://api.example.com/login with {"email":"dev@exam
 probe-ai/
 ├── cli/                            # 🚀 Native TypeScript NPM Package (@kamrankhan027/probe-ai)
 │   ├── src/
-│   │   ├── index.ts                # Interactive CLI loop & Marked terminal styling
+│   │   ├── index.ts                # Interactive CLI loop & Modern UI
+│   │   ├── wizard.ts               # Multi-Provider Startup Wizard
+│   │   ├── model-factory.ts        # Universal LangChain chat model factory
 │   │   ├── agent.ts                # @langchain/langgraph ReAct workflow
 │   │   └── tools/                  # OpenAPI, HTTP, Auth, Postman & Report tools
 │   ├── dist/
@@ -111,18 +112,6 @@ probe-ai/
 
 ---
 
-## 🛠️ Local Development
-
-```bash
-git clone https://github.com/kamran-027/probe-ai.git
-cd probe-ai/cli
-npm install
-npm run build
-npm start
-```
-
----
-
 <div align="center">
 Built by <b>Kamran Khan</b> as part of <b>Cadence Labs</b>.
 </div>

package/dist/index.js

@@ -4243,9 +4243,9 @@ var authenticateViaLoginTool = tool2(
     description: "Execute a login or token generation endpoint (e.g. POST /auth/login with credentials), extract the bearer token from JSON response, and save it to the session headers.",
     schema: external_exports.object({
       loginUrl: external_exports.string().describe("The URL of the login/auth endpoint"),
-      method: external_exports.string().optional().describe("HTTP method (default: POST)"),
-      payloadJson: external_exports.string().optional().describe("JSON string payload with email/username/password"),
-      tokenField: external_exports.string().optional().describe("Field name containing the token in JSON response (default: access_token)")
+      method: external_exports.string().default("POST").describe("HTTP method (default: POST)"),
+      payloadJson: external_exports.string().default("{}").describe("JSON string payload with email/username/password"),
+      tokenField: external_exports.string().default("access_token").describe("Field name containing the token in JSON response (default: access_token)")
     })
   }
 );
@@ -4362,9 +4362,9 @@ var executeHttpRequestTool = tool3(
     schema: external_exports.object({
       method: external_exports.string().describe("HTTP method, e.g. GET, POST, DELETE"),
       url: external_exports.string().describe("Target endpoint URL"),
-      headersJson: external_exports.string().optional().describe("Optional JSON string of custom headers"),
-      paramsJson: external_exports.string().optional().describe("Optional JSON string of query parameters"),
-      bodyJson: external_exports.string().optional().describe("Optional JSON string payload body")
+      headersJson: external_exports.string().default("{}").describe("Optional JSON string of custom headers"),
+      paramsJson: external_exports.string().default("{}").describe("Optional JSON string of query parameters"),
+      bodyJson: external_exports.string().default("{}").describe("Optional JSON string payload body")
     })
   }
 );
@@ -4443,7 +4443,7 @@ var exportPostmanCollectionTool = tool4(
     description: "Generate a ready-to-import Postman Collection v2.1.0 JSON file directly from the base URL's OpenAPI spec.",
     schema: external_exports.object({
       baseUrl: external_exports.string().describe("The base URL of the API server"),
-      outputFilename: external_exports.string().optional().describe("Output filename (default: probe_collection.json)")
+      outputFilename: external_exports.string().default("probe_collection.json").describe("Output filename (default: probe_collection.json)")
     })
   }
 );
@@ -4558,8 +4558,8 @@ var saveHtmlAuditReportTool = tool5(
     name: "save_html_audit_report",
     description: "Generate an executive standalone single-page HTML audit report with health scores, latency metrics, and expandable request/response inspection accordions.",
     schema: external_exports.object({
-      filename: external_exports.string().optional().describe("Report filename (default: audit_report.html)"),
-      reportTitle: external_exports.string().optional().describe("Title for the report")
+      filename: external_exports.string().default("audit_report.html").describe("Report filename (default: audit_report.html)"),
+      reportTitle: external_exports.string().default("ProbeAI API Audit Report").describe("Title for the report")
     })
   }
 );
@@ -4709,34 +4709,34 @@ async function resolveProviderConfig() {
     message: "Choose AI Engine Provider:",
     choices: [
       {
-        name: `${chalk7.bold.cyan("\u{1F537} Google Gemini")} ${chalk7.dim("(Gemini 2.5 Flash / 3.6 Flash - Free & Fast)")}`,
+        name: `${chalk7.bold.cyan("\u{1F537} Google Gemini")} ${chalk7.dim("(Gemini 3.6 / 3.5 / 2.5 Flash - Free & Fast)")}`,
         value: "gemini"
       },
       {
-        name: `${chalk7.bold.green("\u{1F7E2} OpenAI")} ${chalk7.dim("(GPT-5.6 / GPT-4o / GPT-4o-mini)")}`,
+        name: `${chalk7.bold.green("\u{1F7E2} OpenAI")} ${chalk7.dim("(GPT-5.6 Sol/Terra, GPT-4o, o3-mini, GPT-OSS)")}`,
         value: "openai"
       },
       {
-        name: `${chalk7.bold.hex("#A855F7")("\u{1F7E3} Anthropic Claude")} ${chalk7.dim("(Sonnet 5 / Claude 3.5 Sonnet)")}`,
+        name: `${chalk7.bold.hex("#A855F7")("\u{1F7E3} Anthropic Claude")} ${chalk7.dim("(Sonnet 5, Opus 5, Claude 3.5 Sonnet)")}`,
         value: "anthropic"
       },
       {
-        name: `${chalk7.bold.yellow("\u26A1 Groq")} ${chalk7.dim("(Llama 3.3 70B - Free tier & Ultra-Fast)")}`,
+        name: `${chalk7.bold.yellow("\u26A1 Groq")} ${chalk7.dim("(Llama 3.3 70B, GPT-OSS 120B, Qwen 3.6)")}`,
         value: "groq"
       },
       {
-        name: `${chalk7.bold.hex("#3B82F6")("\u{1F310} OpenRouter")} ${chalk7.dim("(DeepSeek / Qwen 2.5 - Free tiers)")}`,
+        name: `${chalk7.bold.hex("#3B82F6")("\u{1F310} OpenRouter")} ${chalk7.dim("(DeepSeek V4/R1, Qwen 3.5/2.5 Coder, Mistral)")}`,
         value: "openrouter"
       },
       {
-        name: `${chalk7.bold.white("\u{1F999} Ollama")} ${chalk7.dim("(Local / 100% Offline - No API key needed)")}`,
+        name: `${chalk7.bold.white("\u{1F999} Ollama")} ${chalk7.dim("(Local / Offline: DeepSeek-R1, Qwen Coder, Gemma 4)")}`,
         value: "ollama"
       }
     ]
   });
   if (provider === "ollama") {
     const customModel2 = await input({
-      message: "Enter Ollama model name (default: qwen2.5-coder:7b):",
+      message: "Enter Ollama model (e.g. qwen2.5-coder:7b, deepseek-r1, llama3.3):",
       default: "qwen2.5-coder:7b"
     });
     return { provider: "ollama", modelName: customModel2 };
@@ -4756,8 +4756,15 @@ async function resolveProviderConfig() {
     console.log(chalk7.red("\nError: An API key is required to proceed. Exiting...\n"));
     process.exit(1);
   }
+  const modelHints = {
+    gemini: "gemini-2.5-flash (or gemini-3.6-flash, gemini-2.5-pro)",
+    openai: "gpt-4o (or gpt-5.6, gpt-4o-mini, o3-mini)",
+    anthropic: "claude-3-5-sonnet-latest (or claude-sonnet-5, claude-opus-5)",
+    groq: "llama-3.3-70b-versatile (or qwen-2.5-32b)",
+    openrouter: "qwen/qwen-2.5-coder-32b-instruct:free (or deepseek/deepseek-r1:free)"
+  };
   const customModel = await input({
-    message: "Model ID (press Enter to use provider default):",
+    message: `Model ID [default: ${modelHints[provider]}]:`,
     default: ""
   });
   return {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kamrankhan027/probe-ai",
-  "version": "1.1.0",
+  "version": "1.1.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",