@desplega.ai/qa-use 2.11.0 → 2.13.2

package/lib/api/index.ts

@@ -16,7 +16,7 @@ import {
   generateEnhancedTestSummary,
   generateIssueStatistics,
 } from '../../src/utils/summary.js';
-import { getEnv } from '../env/index.js';
+import { getCustomHeaders, getEnv } from '../env/index.js';
 import { type SSEEvent, streamSSE } from './sse.js';
 
 /**
@@ -51,15 +51,21 @@ function formatApiError(
 }
 
 // Re-export new types for external consumers
-export type { IssueType, Severity, IssueReport, TestCreatorDoneIntent };
-export type { EnhancedTestSummary, BlockSummary };
+export type {
+  BlockSummary,
+  EnhancedTestSummary,
+  IssueReport,
+  IssueType,
+  Severity,
+  TestCreatorDoneIntent,
+};
 
 // Re-export utility functions for external consumers
 export {
-  generateEnhancedTestSummary,
+  categorizeIssues,
   formatEnhancedTestReport,
+  generateEnhancedTestSummary,
   generateIssueStatistics,
-  categorizeIssues,
 };
 
 export interface CreateSessionOptions {
@@ -347,6 +353,7 @@ export class ApiClient {
   private readonly client: AxiosInstance;
   private apiKey: string | null = null;
   private appConfigId: string | null = null;
+  private customHeaders: Record<string, string> = {};
 
   constructor(baseUrl?: string) {
     // Use environment variable if available, otherwise use provided baseUrl, finally fall back to production
@@ -365,6 +372,21 @@ export class ApiClient {
     if (envApiKey) {
       this.setApiKey(envApiKey);
     }
+
+    // Auto-load custom headers from environment/config
+    const customHeaders = getCustomHeaders();
+    if (customHeaders) {
+      this.setCustomHeaders(customHeaders);
+    }
+  }
+
+  setCustomHeaders(headers: Record<string, string>): void {
+    for (const [key, value] of Object.entries(headers)) {
+      const lower = key.toLowerCase();
+      if (lower === 'authorization' || lower === 'content-type') continue;
+      this.customHeaders[key] = value;
+      this.client.defaults.headers.common[key] = value;
+    }
   }
 
   setApiKey(apiKey: string): void {
@@ -827,6 +849,7 @@ export class ApiClient {
       const response = await fetch(`${this.getApiUrl()}/vibe-qa/cli/run`, {
         method: 'POST',
         headers: {
+          ...this.customHeaders,
           Authorization: `Bearer ${this.apiKey}`,
           'Content-Type': 'application/json',
           Accept: 'text/event-stream',

package/lib/env/index.ts

@@ -183,6 +183,38 @@ export function logConfigSources(): void {
   }
 }
 
+/**
+ * Get custom headers from config file and QA_USE_HEADERS env var.
+ * Used by API clients that don't go through CLI's loadConfig().
+ *
+ * Priority: config file headers, then QA_USE_HEADERS env var overrides.
+ */
+export function getCustomHeaders(): Record<string, string> | null {
+  const headers: Record<string, string> = {};
+
+  const config = loadConfig();
+  if (config) {
+    const configHeaders = (config as Record<string, unknown>).headers;
+    if (configHeaders && typeof configHeaders === 'object' && !Array.isArray(configHeaders)) {
+      Object.assign(headers, configHeaders);
+    }
+  }
+
+  const envHeaders = process.env.QA_USE_HEADERS;
+  if (envHeaders) {
+    try {
+      const parsed = JSON.parse(envHeaders);
+      if (typeof parsed === 'object' && parsed !== null && !Array.isArray(parsed)) {
+        Object.assign(headers, parsed);
+      }
+    } catch {
+      // Silently ignore invalid JSON
+    }
+  }
+
+  return Object.keys(headers).length > 0 ? headers : null;
+}
+
 /**
  * Get agent session ID from environment if available.
  * Used for auto-linking browser sessions and test runs to agent sessions.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@desplega.ai/qa-use",
-  "version": "2.11.0",
+  "version": "2.13.2",
   "packageManager": "bun@^1.3.4",
   "description": "QA automation tool for browser testing with MCP server support",
   "type": "module",