@kamrankhan027/probe-ai 1.1.1 → 1.1.3

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")
     })
   }
 );
@@ -4568,6 +4568,7 @@ var saveHtmlAuditReportTool = tool5(
 import { ChatGoogleGenerativeAI } from "@langchain/google-genai";
 import { ChatOpenAI } from "@langchain/openai";
 import { ChatAnthropic } from "@langchain/anthropic";
+import { ChatGroq } from "@langchain/groq";
 import { ChatOllama } from "@langchain/ollama";
 function createChatModel(config) {
   const { provider, apiKey = "", modelName, baseUrl } = config;
@@ -4594,13 +4595,10 @@ function createChatModel(config) {
         maxRetries: 3
       });
     case "groq":
-      return new ChatOpenAI({
+      return new ChatGroq({
         model: modelName || process.env.GROQ_MODEL || "llama-3.3-70b-versatile",
         temperature: 0.2,
         apiKey,
-        configuration: {
-          baseURL: "https://api.groq.com/openai/v1"
-        },
         maxRetries: 3
       });
     case "openrouter":

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kamrankhan027/probe-ai",
-  "version": "1.1.1",
+  "version": "1.1.3",
   "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",
@@ -32,6 +32,7 @@
     "@langchain/anthropic": "^0.3.34",
     "@langchain/core": "^0.3.40",
     "@langchain/google-genai": "^0.1.11",
+    "@langchain/groq": "^0.1.3",
     "@langchain/langgraph": "^0.2.54",
     "@langchain/ollama": "^0.1.6",
     "@langchain/openai": "^0.3.17",