blaze-performance-tester 3.2.13 → 3.2.16

package/dist/cli.js

@@ -1,4 +1,5 @@
 #!/usr/bin/env ts-node
+import { createRequire } from 'module'; const require = createRequire(import.meta.url);
 var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
@@ -5416,6 +5417,7 @@ import { createRequire as createRequire3 } from "module";
 import * as path from "path";
 import { fileURLToPath as fileURLToPath2 } from "url";
 import * as fs from "fs";
+import { GoogleGenAI } from "@google/genai";
 
 // src/LogAnalyzer.ts
 var LogAnalyzer = class {
@@ -5708,21 +5710,41 @@ async function executeTestPipeline(config) {
     console.log("Pipeline run complete. Parameter tracking successfully mitigated breaking changes.");
   }
 }
-async function generateTestScriptFromPrompt(prompt) {
-  console.log(`
-[AI] Translating natural language to Blaze performance script...`);
-  const generatedCode = `// Auto-generated by Blaze Performance Tester
-// Prompt: ${prompt}
+async function generateTestScriptFromPrompt(prompt, config) {
+  console.log(`\u{1F916} Connecting to Gemini to generate script for: "${prompt}"...`);
+  const ai = new GoogleGenAI({});
+  const systemInstruction = `You are an expert code generation assistant for the "blaze-performance-tester" framework.
+Generate a valid TypeScript performance test script based on the user's prompt and provided configuration options.
+The output must contain ONLY valid TypeScript code without any extra markdown conversational wrappers if possible, or wrapped cleanly.
+Include:
+1. Import statement: import { step } from 'blaze-performance-tester';
+2. An options export object containing: threads, duration, rampUp, rampDown.
+3. A default async function scenario() using await step(...) with the correct fetch request based on the user's instructions.`;
+  const fullPrompt = `User Prompt: ${prompt}
+Configuration Options:
+- threads: ${config.threads}
+- duration: ${config.durationSec}
+- rampUp: ${config.rampUpSec}
+- rampDown: ${config.rampDownSec}
 
-import { step } from 'blaze-performance-tester';
-
-export default async function scenario() {
-  await step('Scenario Execution', async () => {
-    console.log("Executing auto-generated step based on prompt.");
-  });
-}
-`;
-  return generatedCode;
+Generate the complete TypeScript test script now.`;
+  try {
+    const response = await ai.models.generateContent({
+      model: "gemini-2.5-flash",
+      contents: fullPrompt,
+      config: {
+        systemInstruction
+      }
+    });
+    let generatedCode = response.text || "";
+    generatedCode = generatedCode.replace(/^```(?:typescript|ts)?\n?/i, "");
+    generatedCode = generatedCode.replace(/\n?```$/i, "");
+    generatedCode = generatedCode.trim();
+    return generatedCode;
+  } catch (error) {
+    console.error("\u274C Error generating script with Gemini API:", error);
+    throw error;
+  }
 }
 async function runCli() {
   const argv = await yargs_default(hideBin(process.argv)).usage("Usage: blaze-performance-tester <script-path> [options]").option("prompt", {
@@ -5738,7 +5760,7 @@ async function runCli() {
   if (argv.prompt) {
     const promptText = argv.prompt;
     try {
-      const scriptContent = await generateTestScriptFromPrompt(promptText);
+      const scriptContent = await generateTestScriptFromPrompt(promptText, config);
       const dir = path.dirname(config.targetScript);
       if (!fs.existsSync(dir)) {
         fs.mkdirSync(dir, { recursive: true });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "blaze-performance-tester",
-  "version": "3.2.13",
+  "version": "3.2.16",
   "description": "A high-performance, multi-threaded load testing engine built with Rust and QuickJS.",
   "type": "module",
   "main": "dist/index.js",
@@ -17,13 +17,14 @@
     "*.node",
     "blaze.win32-x64-msvc.node"
   ],
-  "scripts": {
+"scripts": {
     "prebuild": "node -e \"if (!fs.existsSync('dist')) fs.mkdirSync('dist')\"",
     "build:rust": "napi build --platform --release --js index.js --dts index.d.ts dist",
-    "build:cli": "esbuild cli.ts --bundle --platform=node --format=esm --outfile=dist/cli.js",
+    "build:cli": "esbuild cli.ts --bundle --platform=node --format=esm --outfile=dist/cli.js --external:@google/genai --banner:js=\"import { createRequire } from 'module'; const require = createRequire(import.meta.url);\"",
     "build": "npm run prebuild && npm run build:rust && npm run build:cli"
   },
   "dependencies": {
+    "@google/genai": "^2.12.0",
     "yargs": "^18.0.0"
   },
   "devDependencies": {