artes 1.7.0 → 1.7.1

package/README.md

@@ -192,6 +192,7 @@ npx artes [options]
 | 🔑 `--aiKey` | API key for the selected AI provider | `artes --ai --aiKey "your-api-key"` |
 | 🔗 `--aiURL` | Local AI endpoint URL (e.g. Ollama, LM Studio). Overrides `--aiModel` and `--aiKey` when set | `artes --ai --aiURL "http://localhost:11434/api/chat"` |
 | 🌍 `--aiLanguage` | Language for AI-generated reports (default: `"English"`) | `artes --ai --aiLanguage "Azerbaijani"` |
+| 🔢 --maxTokens | Maximum tokens for AI response output (default: 4000) | artes --ai --maxTokens 8000 |
 | 📋 `--maxReports` | Maximum number of AI reports to generate per test run (default: `10`) | `artes --ai --maxReports 5` |
 
 
@@ -597,6 +598,7 @@ You can configure Artes by editing the `artes.config.js` file. Below are the def
 | `ai.key`             | `""`              | API key for the selected AI provider.                                                                |
 | `ai.url`             | `""`              | Local AI endpoint URL (e.g. Ollama, LM Studio). Overrides `model` and `key` when set.                |
 | `ai.language`        | `"English"`       | Language for AI-generated reports (e.g. `"Azerbaijani"`, `"German"`).                                |
+| ai.maxTokens | 4000 | Maximum tokens for AI response output |
 | `ai.maxReports`      | `10`              | Maximum number of AI reports to generate per test run.                                               |
 
 ---

package/cucumber.config.js

@@ -203,6 +203,9 @@ key: process.env.AI_KEY
 language: process.env.AI_LANGUAGE
   ? process.env.AI_LANGUAGE
   : artesConfig?.ai?.language || "English",
+  maxTokens: process.env.MAX_TOKENS
+  ? parseInt(process.env.MAX_TOKENS)
+  : artesConfig?.ai?.maxTokens || 4000,
 maxReports: process.env.MAX_REPORTS
   ? parseInt(process.env.MAX_REPORTS)
   : artesConfig?.ai?.maxReports || 10,

package/executer.js

@@ -62,6 +62,7 @@ const flags = {
   aiModel: args.includes("--aiModel"),
   aiKey: args.includes("--aiKey"),
   aiLanguage: args.includes("--aiLanguage"),
+  maxTokens: args.includes("--maxTokens"),
   maxReports: args.includes("--maxReports"),
   features: args.includes("--features"),
   stepDef: args.includes("--stepDef"),
@@ -99,6 +100,7 @@ const aiURL = getArgValue("--aiURL");
 const aiModel = getArgValue("--aiModel");
 const aiKey = getArgValue("--aiKey");
 const aiLanguage = getArgValue("--aiLanguage");
+const maxTokens = getArgValue("--maxTokens")
 const maxReports = getArgValue("--maxReports");
 const featureFiles = getArgValue("--features");
 const features = flags.features && featureFiles;
@@ -163,6 +165,7 @@ flags.aiURL ? (process.env.AI_URL = aiURL) : "";
 flags.aiModel ? (process.env.AI_MODEL = aiModel) : "";
 flags.aiKey ? (process.env.AI_KEY = aiKey) : "";
 flags.aiLanguage ? (process.env.AI_LANGUAGE = aiLanguage) : "";
+flags.maxTokens ? (process.env.MAX_TOKENS = maxTokens) : "";
 flags.maxReports ? (process.env.MAX_REPORTS = maxReports) : "";
 
 flags.headless &&

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "artes",
-  "version": "1.7.0",
+  "version": "1.7.1",
   "description": "The simplest way to automate UI and API tests using Cucumber-style steps.",
   "main": "index.js",
   "scripts": {

package/src/helper/controller/aiBugReporter.js

@@ -30,13 +30,11 @@ function resolveProvider(aiFlag = "gemini 2.5 flash") {
 }
 
 
-async function callAI({ prompt, aiFlag, apiKey }) {
+async function callAI({ prompt, aiFlag, apiKey, maxTokens }) {
   const { provider, modelId } = resolveProvider(aiFlag);
-
-  console.log(`Using ${provider.name} — model: ${modelId}`);
-
+  
   const url  = provider.buildUrl(modelId, apiKey);
-  const body = provider.buildBody(prompt, modelId);
+  const body = provider.buildBody(prompt, modelId, maxTokens);
 
   const headers = { "Content-Type": "application/json" };
 
@@ -118,7 +116,6 @@ function buildPickleContext(pickle = {}) {
 }
 
 async function callLocalAI({ prompt, url, apiKey }) {
-  console.log(`Using AI — endpoint: ${url}`);
 
   const headers = { "Content-Type": "application/json" };
   if (apiKey) headers["Authorization"] = `Bearer ${apiKey}`;
@@ -215,7 +212,7 @@ if (url) {
     return callLocalAI({ prompt, url, apiKey });
   }
 
-  return callAI({ prompt, aiFlag, apiKey });
+  return callAI({ prompt, aiFlag, apiKey, maxTokens });
 }
 
 async function generatePassedSummary({ pickleCtx, response, language, aiFlag, apiKey, url }) {
@@ -256,7 +253,7 @@ async function generatePassedSummary({ pickleCtx, response, language, aiFlag, ap
     return callLocalAI({ prompt, url, apiKey });
   }
 
-  return callAI({ prompt, aiFlag, apiKey });
+  return callAI({ prompt, aiFlag, apiKey, maxTokens });
 }
 
 
@@ -266,7 +263,7 @@ let _reportCount = 0;
 
 
 
-const DEFAULT_DELAY_MS = 1000;
+const DEFAULT_DELAY_MS = 3000;
 
 async function attachAiBugReport({
   result,
@@ -278,6 +275,7 @@ async function attachAiBugReport({
   url,
   maxReports = 10,
   delayMs = DEFAULT_DELAY_MS,  
+  maxTokens
 }) {
   try {
     if (!aiKey && !url) {
@@ -307,6 +305,7 @@ async function attachAiBugReport({
         aiFlag: aiModel,
         apiKey: aiKey,
         url,
+        maxTokens
       });
       attachmentLabel = "Test Summary";
     } else {
@@ -318,7 +317,8 @@ async function attachAiBugReport({
         language,
         aiFlag: aiModel,
         apiKey: aiKey,
-        url,   
+        url,
+        maxTokens 
       });
       attachmentLabel = "Bug Report";
     }

package/src/helper/controller/aiProvider.js

@@ -1,27 +1,24 @@
-
 const PROVIDERS = [
-
   {
     name    : "Gemini",
     keywords: ["gemini"],
     models  : {
-      "2.5 flash lite" : "gemini-2.5-flash-lite",   
-      "2.5 flash"      : "gemini-2.5-flash",         
-      "2.5 pro"        : "gemini-2.5-pro",           
-      "2.0 flash"      : "gemini-2.0-flash",         
-      default          : "gemini-2.5-flash",         
+      "2.5 flash lite" : "gemini-2.5-flash-lite",
+      "2.5 flash"      : "gemini-2.5-flash",
+      "2.5 pro"        : "gemini-2.5-pro",
+      "2.0 flash"      : "gemini-2.0-flash",
+      default          : "gemini-2.5-flash",
     },
     authStyle : "queryparam",
     buildUrl  : (modelId, apiKey) =>
       `https://generativelanguage.googleapis.com/v1beta/models/${modelId}:generateContent?key=${apiKey}`,
-    buildBody : (prompt) => ({
+    buildBody : (prompt, modelId, maxTokens = 4000) => ({
       contents        : [{ parts: [{ text: prompt }] }],
-      generationConfig: { temperature: 0.6, maxOutputTokens: 4000 },
+      generationConfig: { temperature: 0.6, maxOutputTokens: maxTokens },
     }),
     parseResp : (data) => data?.candidates?.[0]?.content?.parts?.[0]?.text ?? "",
   },
 
-
   {
     name    : "OpenAI",
     keywords: ["openai", "chatgpt", "gpt"],
@@ -39,16 +36,15 @@ const PROVIDERS = [
     authKey   : "Authorization",
     authValue : (apiKey) => `Bearer ${apiKey}`,
     buildUrl  : () => "https://api.openai.com/v1/chat/completions",
-    buildBody : (prompt, modelId) => ({
+    buildBody : (prompt, modelId, maxTokens = 4000) => ({
       model      : modelId,
-      max_tokens : 4000,
+      max_tokens : maxTokens,
       temperature: 0.6,
       messages   : [{ role: "user", content: prompt }],
     }),
     parseResp : (data) => data?.choices?.[0]?.message?.content ?? "",
   },
 
-
   {
     name    : "Claude",
     keywords: ["claude", "anthropic"],
@@ -65,16 +61,15 @@ const PROVIDERS = [
     authValue        : (apiKey) => apiKey,
     buildExtraHeaders: () => ({ "anthropic-version": "2023-06-01" }),
     buildUrl         : () => "https://api.anthropic.com/v1/messages",
-    buildBody        : (prompt, modelId) => ({
+    buildBody        : (prompt, modelId, maxTokens = 4000) => ({
       model     : modelId,
-      max_tokens: 4000,
+      max_tokens: maxTokens,
       messages  : [{ role: "user", content: prompt }],
     }),
     parseResp : (data) =>
       (data?.content ?? []).filter((b) => b.type === "text").map((b) => b.text).join(""),
   },
 
-
   {
     name    : "Mistral",
     keywords: ["mistral"],
@@ -90,16 +85,15 @@ const PROVIDERS = [
     authKey   : "Authorization",
     authValue : (apiKey) => `Bearer ${apiKey}`,
     buildUrl  : () => "https://api.mistral.ai/v1/chat/completions",
-    buildBody : (prompt, modelId) => ({
+    buildBody : (prompt, modelId, maxTokens = 4000) => ({
       model      : modelId,
-      max_tokens : 4000,
+      max_tokens : maxTokens,
       temperature: 0.6,
       messages   : [{ role: "user", content: prompt }],
     }),
     parseResp : (data) => data?.choices?.[0]?.message?.content ?? "",
   },
 
-
   {
     name    : "Groq",
     keywords: ["groq"],
@@ -114,16 +108,15 @@ const PROVIDERS = [
     authKey   : "Authorization",
     authValue : (apiKey) => `Bearer ${apiKey}`,
     buildUrl  : () => "https://api.groq.com/openai/v1/chat/completions",
-    buildBody : (prompt, modelId) => ({
+    buildBody : (prompt, modelId, maxTokens = 4000) => ({
       model      : modelId,
-      max_tokens : 4000,
+      max_tokens : maxTokens,
       temperature: 0.6,
       messages   : [{ role: "user", content: prompt }],
     }),
     parseResp : (data) => data?.choices?.[0]?.message?.content ?? "",
   },
 
-
   {
     name    : "Cohere",
     keywords: ["cohere", "command"],
@@ -136,14 +129,14 @@ const PROVIDERS = [
     authKey   : "Authorization",
     authValue : (apiKey) => `Bearer ${apiKey}`,
     buildUrl  : () => "https://api.cohere.com/v2/chat",
-    buildBody : (prompt, modelId) => ({
-      model   : modelId,
-      messages: [{ role: "user", content: prompt }],
+    buildBody : (prompt, modelId, maxTokens = 4000) => ({
+      model         : modelId,
+      max_tokens    : maxTokens,
+      messages      : [{ role: "user", content: prompt }],
     }),
     parseResp : (data) => data?.message?.content?.[0]?.text ?? "",
   },
 
-
   {
     name    : "DeepSeek",
     keywords: ["deepseek"],
@@ -156,9 +149,9 @@ const PROVIDERS = [
     authKey   : "Authorization",
     authValue : (apiKey) => `Bearer ${apiKey}`,
     buildUrl  : () => "https://api.deepseek.com/chat/completions",
-    buildBody : (prompt, modelId) => ({
+    buildBody : (prompt, modelId, maxTokens = 4000) => ({
       model      : modelId,
-      max_tokens : 4000,
+      max_tokens : maxTokens,
       temperature: 0.6,
       messages   : [{ role: "user", content: prompt }],
     }),

package/src/helper/controller/getEnvInfo.js

@@ -55,6 +55,14 @@ async function getEnvInfo() {
     Parallel_Runner: cucumberConfig.default.parallel,
     Timeout: cucumberConfig.default.timeout ?? "N/A",
 
+    // ── AI Config ─────────────────────────
+    AI_Enabled:      cucumberConfig.default.ai        ?? false,
+    AI_URL:          cucumberConfig.default.url        ?? "N/A",
+    AI_Model:        cucumberConfig.default.model      ?? "gemini 2.5 flash",
+    AI_Language:     cucumberConfig.default.language   ?? "English",
+    AI_Max_Tokens:   cucumberConfig.default.maxTokens  ?? "4000",
+    AI_Max_Reports:  cucumberConfig.default.maxReports ?? 10,
+
     // ── Git ─────────────────────────────────
     Git_Branch: process.env.GIT_BRANCH ?? process.env.BRANCH_NAME ?? "N/A",
     Git_Commit: process.env.GIT_COMMIT ?? process.env.GIT_SHA ?? "N/A",

package/src/helper/executers/helper.js

@@ -154,6 +154,9 @@ function showAIHelp() {
 
             🌍 --aiLanguage    Language for AI-generated reports (default: "English")
                 Usage:   artes --ai --aiLanguage "Azerbaijani"
+               
+            🔢 --maxTokens     Maximum tokens for AI-generated reports (default: 4000)
+                Usage:   artes --ai --maxTokens 8000
 
             📋 --maxReports    Maximum number of AI reports to generate per test run (default: 10)
                 Usage:   artes --ai --maxReports 5

package/src/helper/executers/projectCreator.js

@@ -55,9 +55,10 @@ function createProject(createYes, noDeps) {
     // ai: {
     //   ai: false,                  // boolean - Enable AI-generated bug reports and test summaries
     //   url: "",                    // string  - Local AI endpoint URL (e.g., Ollama, LM Studio). Overrides model/key when set
-    //   model: "gpt-4o",            // string  - AI model to use (e.g., "gpt-4o", "gemini 2.5 flash", "claude sonnet", "mistral large")
+    //   model: "gemini 2.5 flash",  // string  - AI model to use (e.g., "gpt-4o", "gemini 2.5 flash", "claude sonnet", "mistral large")
     //   key: "",                    // string  - API key for the selected AI provider
-    //   language: "English",        // string  - Language for generated reports (e.g., "English", "Azerbaijani")
+    //   language: "English",        // string  - Language for generated reports (e.g., "English", "Azerbaijani"),
+    //   maxTokens: 4000,            // number  - Maximum tokens for AI-generated reports (default: 4000)
     //   maxReports: 10,             // number  - Maximum number of AI reports to generate per test run
     // },
     // timeout : 0,                  // number - Test timeout in seconds

package/src/hooks/hooks.js

@@ -159,27 +159,28 @@ AfterStep(async function ({ pickleStep }) {
 });
 
 After(async function ({ result, pickle }) {
-  if (typeof projectHooks.After === "function") {
-    await projectHooks.After();
-  }
-
-  
   const shouldReport =
-    cucumberConfig.default.successReport || result?.status !== Status.PASSED;
-    
-  await attachResponse(allure.attachment);
+  cucumberConfig.default.successReport || result?.status !== Status.PASSED;
+  
+await attachResponse(allure.attachment);
 
 if (shouldReport && cucumberConfig.ai.ai) {
-    await attachAiBugReport({
-      result,
-      pickle,
-      response: context.response,
-      language: cucumberConfig.ai.language,
-      url: cucumberConfig.ai.url,
-      aiModel: cucumberConfig.ai.model,
-      aiKey: cucumberConfig.ai.key,
-      maxReports: cucumberConfig.ai.maxReports
-    });
+  await attachAiBugReport({
+    result,
+    pickle,
+    response: context.response,
+    language: cucumberConfig.ai.language,
+    url: cucumberConfig.ai.url,
+    aiModel: cucumberConfig.ai.model,
+    aiKey: cucumberConfig.ai.key,
+    maxReports: cucumberConfig.ai.maxReports,
+    maxTokens: cucumberConfig.ai.maxTokens
+  });
+}
+
+
+  if (typeof projectHooks.After === "function") {
+    await projectHooks.After();
   }