@gleanwork/mcp-server-tester 1.0.0 → 1.0.1

package/dist/index.d.cts

@@ -2684,6 +2684,56 @@ interface CLIConfig {
      */
     timeout?: number;
 }
+/**
+ * A cookie to inject into the browser context before running the script.
+ * Matches the shape expected by Playwright's `BrowserContext.addCookies()`.
+ */
+interface BrowserCookie {
+    name: string;
+    value: string;
+    url?: string;
+    domain?: string;
+    path?: string;
+    expires?: number;
+    httpOnly?: boolean;
+    secure?: boolean;
+    sameSite?: 'Strict' | 'Lax' | 'None';
+    partitionKey?: string;
+}
+/**
+ * Configuration for a browser-based host.
+ *
+ * Uses Playwright to launch a Chromium instance, inject auth state,
+ * and execute a user-provided script that drives a web-based MCP host
+ * (e.g., claude.ai).
+ */
+interface BrowserConfig {
+    /**
+     * Path to the browser script (resolved relative to cwd).
+     * The script must default-export an async function
+     * `(page: Page, scenario: string) => Promise<MCPHostSimulationResult>`.
+     */
+    script: string;
+    /**
+     * Timeout in milliseconds for the browser script.
+     * @default 120000 (2 minutes)
+     */
+    timeout?: number;
+    /**
+     * Whether to launch in headless mode.
+     * @default true
+     */
+    headless?: boolean;
+    /**
+     * Path to a Playwright storage state JSON file (cookies + localStorage).
+     * Resolved relative to cwd.
+     */
+    storageState?: string;
+    /**
+     * Extra cookies to inject into the browser context.
+     */
+    cookies?: BrowserCookie[];
+}
 /**
  * Configuration for MCP host simulation
  */
@@ -2729,6 +2779,10 @@ interface MCPHostConfig {
      * CLI host configuration (required for 'cli' host type).
      */
     cli?: CLIConfig;
+    /**
+     * Browser host configuration (required for 'browser' host type).
+     */
+    browser?: BrowserConfig;
 }
 /**
  * A tool call made by the LLM
@@ -2770,6 +2824,11 @@ interface MCPHostSimulationResult {
      * (excludes LLM response time)
      */
     mcpDurationMs?: number;
+    /**
+     * Token usage from the LLM during simulation.
+     * Populated by SDK-based hosts from the AI SDK response.
+     */
+    usage?: UsageMetrics;
 }
 /**
  * Interface for MCP host simulators.
@@ -3071,15 +3130,15 @@ declare const EvalCaseSchema: z.ZodObject<{
             desktop: "desktop";
         }>>;
         provider: z.ZodOptional<z.ZodEnum<{
-            openai: "openai";
             anthropic: "anthropic";
-            azure: "azure";
+            "vertex-anthropic": "vertex-anthropic";
+            openai: "openai";
             google: "google";
+            azure: "azure";
             mistral: "mistral";
             deepseek: "deepseek";
             openrouter: "openrouter";
             xai: "xai";
-            "vertex-anthropic": "vertex-anthropic";
         }>>;
         apiKeyEnvVar: z.ZodOptional<z.ZodString>;
         model: z.ZodOptional<z.ZodString>;
@@ -3136,11 +3195,11 @@ declare const EvalCaseSchema: z.ZodObject<{
             threshold: z.ZodOptional<z.ZodNumber>;
             reps: z.ZodOptional<z.ZodNumber>;
             provider: z.ZodOptional<z.ZodEnum<{
-                openai: "openai";
                 anthropic: "anthropic";
-                google: "google";
                 "vertex-anthropic": "vertex-anthropic";
                 "anthropic-agent-sdk": "anthropic-agent-sdk";
+                openai: "openai";
+                google: "google";
             }>>;
             model: z.ZodOptional<z.ZodString>;
             apiKeyEnvVar: z.ZodOptional<z.ZodString>;
@@ -3163,11 +3222,11 @@ declare const EvalCaseSchema: z.ZodObject<{
             threshold: z.ZodOptional<z.ZodNumber>;
             reps: z.ZodOptional<z.ZodNumber>;
             provider: z.ZodOptional<z.ZodEnum<{
-                openai: "openai";
                 anthropic: "anthropic";
-                google: "google";
                 "vertex-anthropic": "vertex-anthropic";
                 "anthropic-agent-sdk": "anthropic-agent-sdk";
+                openai: "openai";
+                google: "google";
             }>>;
             model: z.ZodOptional<z.ZodString>;
             apiKeyEnvVar: z.ZodOptional<z.ZodString>;
@@ -3223,15 +3282,15 @@ declare const EvalDatasetSchema: z.ZodObject<{
                 desktop: "desktop";
             }>>;
             provider: z.ZodOptional<z.ZodEnum<{
-                openai: "openai";
                 anthropic: "anthropic";
-                azure: "azure";
+                "vertex-anthropic": "vertex-anthropic";
+                openai: "openai";
                 google: "google";
+                azure: "azure";
                 mistral: "mistral";
                 deepseek: "deepseek";
                 openrouter: "openrouter";
                 xai: "xai";
-                "vertex-anthropic": "vertex-anthropic";
             }>>;
             apiKeyEnvVar: z.ZodOptional<z.ZodString>;
             model: z.ZodOptional<z.ZodString>;
@@ -3288,11 +3347,11 @@ declare const EvalDatasetSchema: z.ZodObject<{
                 threshold: z.ZodOptional<z.ZodNumber>;
                 reps: z.ZodOptional<z.ZodNumber>;
                 provider: z.ZodOptional<z.ZodEnum<{
-                    openai: "openai";
                     anthropic: "anthropic";
-                    google: "google";
                     "vertex-anthropic": "vertex-anthropic";
                     "anthropic-agent-sdk": "anthropic-agent-sdk";
+                    openai: "openai";
+                    google: "google";
                 }>>;
                 model: z.ZodOptional<z.ZodString>;
                 apiKeyEnvVar: z.ZodOptional<z.ZodString>;
@@ -3315,11 +3374,11 @@ declare const EvalDatasetSchema: z.ZodObject<{
                 threshold: z.ZodOptional<z.ZodNumber>;
                 reps: z.ZodOptional<z.ZodNumber>;
                 provider: z.ZodOptional<z.ZodEnum<{
-                    openai: "openai";
                     anthropic: "anthropic";
-                    google: "google";
                     "vertex-anthropic": "vertex-anthropic";
                     "anthropic-agent-sdk": "anthropic-agent-sdk";
+                    openai: "openai";
+                    google: "google";
                 }>>;
                 model: z.ZodOptional<z.ZodString>;
                 apiKeyEnvVar: z.ZodOptional<z.ZodString>;
@@ -3562,6 +3621,8 @@ interface IterationResult {
             name: string;
         }>;
     };
+    /** Token usage from mcp_host LLM simulation in this iteration */
+    hostUsage?: UsageMetrics;
 }
 /**
  * Request data captured from the eval case input.
@@ -3710,6 +3771,11 @@ interface EvalCaseResult {
             name: string;
         }>;
     };
+    /**
+     * Aggregate token usage from mcp_host LLM simulation for this case.
+     * Summed across all iterations. Only populated for mcp_host mode cases.
+     */
+    hostUsage?: UsageMetrics;
 }
 /**
  * Aggregated MCP eval run data
@@ -3759,6 +3825,10 @@ interface MCPEvalRunData {
          * Expectation type breakdown
          */
         expectationBreakdown: ExpectationBreakdown;
+        /**
+         * Aggregate token usage from all mcp_host LLM simulations in this run.
+         */
+        totalHostUsage?: UsageMetrics;
     };
     /**
      * All eval results from this run
@@ -3873,6 +3943,10 @@ interface EvalRunnerResult {
      * Experiment tracking metadata captured at run time.
      */
     metadata?: EvalRunMetadata;
+    /**
+     * Aggregate token usage from all mcp_host LLM simulations across all cases.
+     */
+    totalHostUsage?: UsageMetrics;
 }
 /**
  * Options for running eval dataset

package/dist/index.d.ts

@@ -2684,6 +2684,56 @@ interface CLIConfig {
      */
     timeout?: number;
 }
+/**
+ * A cookie to inject into the browser context before running the script.
+ * Matches the shape expected by Playwright's `BrowserContext.addCookies()`.
+ */
+interface BrowserCookie {
+    name: string;
+    value: string;
+    url?: string;
+    domain?: string;
+    path?: string;
+    expires?: number;
+    httpOnly?: boolean;
+    secure?: boolean;
+    sameSite?: 'Strict' | 'Lax' | 'None';
+    partitionKey?: string;
+}
+/**
+ * Configuration for a browser-based host.
+ *
+ * Uses Playwright to launch a Chromium instance, inject auth state,
+ * and execute a user-provided script that drives a web-based MCP host
+ * (e.g., claude.ai).
+ */
+interface BrowserConfig {
+    /**
+     * Path to the browser script (resolved relative to cwd).
+     * The script must default-export an async function
+     * `(page: Page, scenario: string) => Promise<MCPHostSimulationResult>`.
+     */
+    script: string;
+    /**
+     * Timeout in milliseconds for the browser script.
+     * @default 120000 (2 minutes)
+     */
+    timeout?: number;
+    /**
+     * Whether to launch in headless mode.
+     * @default true
+     */
+    headless?: boolean;
+    /**
+     * Path to a Playwright storage state JSON file (cookies + localStorage).
+     * Resolved relative to cwd.
+     */
+    storageState?: string;
+    /**
+     * Extra cookies to inject into the browser context.
+     */
+    cookies?: BrowserCookie[];
+}
 /**
  * Configuration for MCP host simulation
  */
@@ -2729,6 +2779,10 @@ interface MCPHostConfig {
      * CLI host configuration (required for 'cli' host type).
      */
     cli?: CLIConfig;
+    /**
+     * Browser host configuration (required for 'browser' host type).
+     */
+    browser?: BrowserConfig;
 }
 /**
  * A tool call made by the LLM
@@ -2770,6 +2824,11 @@ interface MCPHostSimulationResult {
      * (excludes LLM response time)
      */
     mcpDurationMs?: number;
+    /**
+     * Token usage from the LLM during simulation.
+     * Populated by SDK-based hosts from the AI SDK response.
+     */
+    usage?: UsageMetrics;
 }
 /**
  * Interface for MCP host simulators.
@@ -3071,15 +3130,15 @@ declare const EvalCaseSchema: z.ZodObject<{
             desktop: "desktop";
         }>>;
         provider: z.ZodOptional<z.ZodEnum<{
-            openai: "openai";
             anthropic: "anthropic";
-            azure: "azure";
+            "vertex-anthropic": "vertex-anthropic";
+            openai: "openai";
             google: "google";
+            azure: "azure";
             mistral: "mistral";
             deepseek: "deepseek";
             openrouter: "openrouter";
             xai: "xai";
-            "vertex-anthropic": "vertex-anthropic";
         }>>;
         apiKeyEnvVar: z.ZodOptional<z.ZodString>;
         model: z.ZodOptional<z.ZodString>;
@@ -3136,11 +3195,11 @@ declare const EvalCaseSchema: z.ZodObject<{
             threshold: z.ZodOptional<z.ZodNumber>;
             reps: z.ZodOptional<z.ZodNumber>;
             provider: z.ZodOptional<z.ZodEnum<{
-                openai: "openai";
                 anthropic: "anthropic";
-                google: "google";
                 "vertex-anthropic": "vertex-anthropic";
                 "anthropic-agent-sdk": "anthropic-agent-sdk";
+                openai: "openai";
+                google: "google";
             }>>;
             model: z.ZodOptional<z.ZodString>;
             apiKeyEnvVar: z.ZodOptional<z.ZodString>;
@@ -3163,11 +3222,11 @@ declare const EvalCaseSchema: z.ZodObject<{
             threshold: z.ZodOptional<z.ZodNumber>;
             reps: z.ZodOptional<z.ZodNumber>;
             provider: z.ZodOptional<z.ZodEnum<{
-                openai: "openai";
                 anthropic: "anthropic";
-                google: "google";
                 "vertex-anthropic": "vertex-anthropic";
                 "anthropic-agent-sdk": "anthropic-agent-sdk";
+                openai: "openai";
+                google: "google";
             }>>;
             model: z.ZodOptional<z.ZodString>;
             apiKeyEnvVar: z.ZodOptional<z.ZodString>;
@@ -3223,15 +3282,15 @@ declare const EvalDatasetSchema: z.ZodObject<{
                 desktop: "desktop";
             }>>;
             provider: z.ZodOptional<z.ZodEnum<{
-                openai: "openai";
                 anthropic: "anthropic";
-                azure: "azure";
+                "vertex-anthropic": "vertex-anthropic";
+                openai: "openai";
                 google: "google";
+                azure: "azure";
                 mistral: "mistral";
                 deepseek: "deepseek";
                 openrouter: "openrouter";
                 xai: "xai";
-                "vertex-anthropic": "vertex-anthropic";
             }>>;
             apiKeyEnvVar: z.ZodOptional<z.ZodString>;
             model: z.ZodOptional<z.ZodString>;
@@ -3288,11 +3347,11 @@ declare const EvalDatasetSchema: z.ZodObject<{
                 threshold: z.ZodOptional<z.ZodNumber>;
                 reps: z.ZodOptional<z.ZodNumber>;
                 provider: z.ZodOptional<z.ZodEnum<{
-                    openai: "openai";
                     anthropic: "anthropic";
-                    google: "google";
                     "vertex-anthropic": "vertex-anthropic";
                     "anthropic-agent-sdk": "anthropic-agent-sdk";
+                    openai: "openai";
+                    google: "google";
                 }>>;
                 model: z.ZodOptional<z.ZodString>;
                 apiKeyEnvVar: z.ZodOptional<z.ZodString>;
@@ -3315,11 +3374,11 @@ declare const EvalDatasetSchema: z.ZodObject<{
                 threshold: z.ZodOptional<z.ZodNumber>;
                 reps: z.ZodOptional<z.ZodNumber>;
                 provider: z.ZodOptional<z.ZodEnum<{
-                    openai: "openai";
                     anthropic: "anthropic";
-                    google: "google";
                     "vertex-anthropic": "vertex-anthropic";
                     "anthropic-agent-sdk": "anthropic-agent-sdk";
+                    openai: "openai";
+                    google: "google";
                 }>>;
                 model: z.ZodOptional<z.ZodString>;
                 apiKeyEnvVar: z.ZodOptional<z.ZodString>;
@@ -3562,6 +3621,8 @@ interface IterationResult {
             name: string;
         }>;
     };
+    /** Token usage from mcp_host LLM simulation in this iteration */
+    hostUsage?: UsageMetrics;
 }
 /**
  * Request data captured from the eval case input.
@@ -3710,6 +3771,11 @@ interface EvalCaseResult {
             name: string;
         }>;
     };
+    /**
+     * Aggregate token usage from mcp_host LLM simulation for this case.
+     * Summed across all iterations. Only populated for mcp_host mode cases.
+     */
+    hostUsage?: UsageMetrics;
 }
 /**
  * Aggregated MCP eval run data
@@ -3759,6 +3825,10 @@ interface MCPEvalRunData {
          * Expectation type breakdown
          */
         expectationBreakdown: ExpectationBreakdown;
+        /**
+         * Aggregate token usage from all mcp_host LLM simulations in this run.
+         */
+        totalHostUsage?: UsageMetrics;
     };
     /**
      * All eval results from this run
@@ -3873,6 +3943,10 @@ interface EvalRunnerResult {
      * Experiment tracking metadata captured at run time.
      */
     metadata?: EvalRunMetadata;
+    /**
+     * Aggregate token usage from all mcp_host LLM simulations across all cases.
+     */
+    totalHostUsage?: UsageMetrics;
 }
 /**
  * Options for running eval dataset

package/dist/index.js

@@ -4384,7 +4384,7 @@ function escapeHtml(text) {
 
 // package.json
 var package_default = {
-  version: "1.0.0"};
+  version: "1.0.1"};
 
 // src/mcp/clientFactory.ts
 function getRetryAfterDelayMs(err) {
@@ -6931,6 +6931,12 @@ function createVercelOrchestrator() {
         });
         const totalDurationMs = Date.now() - llmStart;
         const llmDurationMs = totalDurationMs - mcpDurationMs;
+        const hostUsage = result.usage ? {
+          inputTokens: result.usage.promptTokens ?? 0,
+          outputTokens: result.usage.completionTokens ?? 0,
+          totalCostUsd: 0,
+          durationMs: llmDurationMs
+        } : void 0;
         const conversationHistory = (result.steps ?? []).map((step) => ({
           role: step.toolCalls?.length > 0 ? "tool" : "assistant",
           content: step.toolCalls?.length > 0 ? JSON.stringify(step.toolResults) : step.text ?? ""
@@ -6942,7 +6948,8 @@ function createVercelOrchestrator() {
           scenario,
           llmDurationMs,
           mcpDurationMs,
-          conversationHistory
+          conversationHistory,
+          usage: hostUsage
         };
       } catch (err) {
         return {
@@ -7278,6 +7285,32 @@ async function execFileNoThrow(file, args) {
   }
 }
 
+// src/utils/usageUtils.ts
+function optionalSum(a, b) {
+  if (a === void 0 && b === void 0) return void 0;
+  return (a ?? 0) + (b ?? 0);
+}
+function sumUsage(a, b) {
+  if (!a && !b) return void 0;
+  if (!a) return b ? { ...b } : void 0;
+  if (!b) return { ...a };
+  return {
+    inputTokens: a.inputTokens + b.inputTokens,
+    outputTokens: a.outputTokens + b.outputTokens,
+    totalCostUsd: a.totalCostUsd + b.totalCostUsd,
+    durationMs: a.durationMs + b.durationMs,
+    durationApiMs: optionalSum(a.durationApiMs, b.durationApiMs),
+    cacheReadInputTokens: optionalSum(
+      a.cacheReadInputTokens,
+      b.cacheReadInputTokens
+    ),
+    cacheCreationInputTokens: optionalSum(
+      a.cacheCreationInputTokens,
+      b.cacheCreationInputTokens
+    )
+  };
+}
+
 // src/evals/evalRunner.ts
 async function executeToolCall(evalCase, mcp) {
   const mode = evalCase.mode || "direct";
@@ -7523,6 +7556,7 @@ async function runSingleIteration(evalCase, context, options) {
       };
     }
   }
+  const hostUsage = isMCPHostSimulationResult(response) && response.usage ? response.usage : void 0;
   return {
     id: evalCase.id,
     datasetName: options.datasetName ?? "single-case",
@@ -7539,7 +7573,8 @@ async function runSingleIteration(evalCase, context, options) {
     tags: evalCase.tags,
     toolPrecision,
     toolRecall,
-    mcpHostTrace
+    mcpHostTrace,
+    hostUsage
   };
 }
 function isInfrastructureError(err) {
@@ -7575,7 +7610,8 @@ async function runEvalCase(evalCase, context, options = {}) {
         durationMs: result.durationMs,
         error: result.error,
         isInfrastructureError: infraError,
-        mcpHostTrace: result.mcpHostTrace
+        mcpHostTrace: result.mcpHostTrace,
+        hostUsage: result.hostUsage
       });
     } catch (err) {
       const errorMessage = err instanceof Error ? err.message : String(err);
@@ -7608,6 +7644,10 @@ async function runEvalCase(evalCase, context, options = {}) {
     durationMs: 0,
     tags: evalCase.tags
   };
+  const totalHostUsage = iterationResults.reduce(
+    (acc, r) => sumUsage(acc, r.hostUsage),
+    void 0
+  );
   return {
     ...baseResult,
     pass: assertionPassRate >= threshold,
@@ -7616,7 +7656,8 @@ async function runEvalCase(evalCase, context, options = {}) {
     infrastructureErrorRate,
     iterationResults,
     infrastructureErrorCount: infraErrors.length,
-    durationMs: iterationResults.reduce((sum, r) => sum + r.durationMs, 0)
+    durationMs: iterationResults.reduce((sum, r) => sum + r.durationMs, 0),
+    hostUsage: totalHostUsage
   };
 }
 function wilsonCI(k, n) {
@@ -7726,13 +7767,18 @@ async function runEvalDataset(options, context) {
     ...mcpHostModel !== void 0 && { mcpHostModel },
     ...judgeModel !== void 0 && { judgeModel }
   };
+  const runHostUsage = caseResults.reduce(
+    (acc, r) => sumUsage(acc, r.hostUsage),
+    void 0
+  );
   const result = {
     total,
     passed,
     failed: total - passed,
     caseResults,
     durationMs: Date.now() - startTime,
-    metadata
+    metadata,
+    totalHostUsage: runHostUsage
   };
   if (baselineResultsFrom) {
     try {