fix: fix mcp and virtual-key tests

Author: Radheshg04

.github/workflows/configs/default/config.json

@@ -37,6 +37,19 @@
       "network_config": {
         "default_request_timeout_in_seconds": 300
       }
+    },
+    "anthropic": {
+      "keys": [
+        {
+          "name": "e2e-anthropic-key",
+          "value": "env.ANTHROPIC_API_KEY",
+          "weight": 1,
+          "use_for_batch_api": true
+        }
+      ],
+      "network_config": {
+        "default_request_timeout_in_seconds": 300
+      }
     }
   }
 }

.github/workflows/e2e-tests.yml

@@ -2,8 +2,8 @@ name: E2E Tests
 
 on:
   push:
-    branches: ["02-09-fix_fix_e2e_tests"]
-  
+    branches: ["02-11-fix_fix_mcp_and_virtual-key_tests"]
+
 concurrency:
   group: e2e-${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
   cancel-in-progress: true
@@ -49,8 +49,8 @@ jobs:
       - name: Run E2E UI tests
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
-          # Optional: for SSE MCP tests (e.g. remote proxy). Set in repo secrets.
-          MCP_SSE_URL: ${{ secrets.MCP_SSE_URL }}
+          ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
+          # Optional: for SSE MCP tests (e.g. remote proxy). Set in repo secrets.s
           MCP_SSE_HEADERS: ${{ secrets.MCP_SSE_HEADERS }}
         run: ./.github/workflows/scripts/test-e2e-ui.sh
 

tests/e2e/features/mcp-registry/mcp-registry.data.ts

@@ -1,4 +1,4 @@
-import { resolve } from 'path'
+import { join, resolve } from 'path'
 import { MCPClientConfig, EnvVarLike } from './pages/mcp-registry.page'
 
 /** Normalize header value from env (string or EnvVarLike) to EnvVarLike */
@@ -7,20 +7,62 @@ function toEnvVarLike(v: string | EnvVarLike): EnvVarLike {
   return { value: String(v), env_var: '', from_env: false }
 }
 
-/** Build headers from MCP_SSE_HEADERS JSON (injected in workflow). No defaults in code. */
+/**
+ * Resolve header value: if string starts with "env.", use process.env[VAR_NAME].
+ */
+function resolveHeaderValue(v: EnvVarLike): EnvVarLike {
+  if (v.value.startsWith('env.')) {
+    const envVar = v.value.slice(4)
+    const resolved = process.env[envVar]
+    if (resolved !== undefined) {
+      return { value: resolved, env_var: envVar, from_env: true }
+    }
+  }
+  return v
+}
+
+/**
+ * Parse MCP_SSE_HEADERS: supports single object, array of objects, or concatenated objects.
+ * e.g. {"Authorization":"Bearer ..."},{"ENV_EXA_API_KEY":"..."} → merged into one record
+ */
+function parseSSEHeadersRaw(raw: string): Record<string, string | EnvVarLike> {
+  const trimmed = raw.trim()
+  if (!trimmed) return {}
+  try {
+    const parsed = JSON.parse(trimmed)
+    if (typeof parsed === 'object' && parsed !== null && !Array.isArray(parsed)) {
+      return parsed
+    }
+  } catch {
+    // Fallback: concatenated objects {"a":1},{"b":2} → wrap in [ ] and merge
+  }
+  try {
+    const asArray = JSON.parse('[' + trimmed + ']')
+    if (Array.isArray(asArray) && asArray.every((o) => typeof o === 'object' && o !== null)) {
+      return Object.assign({}, ...asArray)
+    }
+  } catch {
+    // ignore
+  }
+  return {}
+}
+
+/**
+ * Build headers from MCP_SSE_HEADERS JSON (injected in workflow).
+ * Supports: {"Authorization":"Bearer ..."},{"ENV_EXA_API_KEY":"..."} (concatenated objects)
+ * or single object. Values starting with "env." are resolved from process.env.
+ */
 function getSSEHeadersFromEnv(): Record<string, EnvVarLike> {
   const raw = process.env.MCP_SSE_HEADERS
   if (!raw) return {}
-  try {
-    const parsed = JSON.parse(raw) as Record<string, string | EnvVarLike>
-    const out: Record<string, EnvVarLike> = {}
-    for (const [k, v] of Object.entries(parsed)) {
-      if (v !== undefined && v !== null) out[k] = toEnvVarLike(v)
+  const parsed = parseSSEHeadersRaw(raw)
+  const out: Record<string, EnvVarLike> = {}
+  for (const [k, v] of Object.entries(parsed)) {
+    if (v !== undefined && v !== null) {
+      out[k] = resolveHeaderValue(toEnvVarLike(v))
     }
-    return out
-  } catch {
-    return {}
   }
+  return out
 }
 
 /**
@@ -55,7 +97,7 @@ export function createHTTPClientData(overrides: Partial<MCPClientConfig> = {}):
  * When unset, uses local http://localhost:3001/sse and no headers (no secrets in code).
  */
 export function createSSEClientData(overrides: Partial<MCPClientConfig> = {}): MCPClientConfig {
-  const connectionUrl = process.env.MCP_SSE_URL ?? 'http://localhost:3001/sse'
+  const connectionUrl = "https://ts-mcp-sse-proxy.fly.dev/npx%20-y%20exa-mcp-server/sse"
   const headers = getSSEHeadersFromEnv()
   const hasHeaders = headers && Object.keys(headers).length > 0
   return createMCPClientData({
@@ -74,8 +116,10 @@ export function createSSEClientData(overrides: Partial<MCPClientConfig> = {}): M
  */
 export function createSTDIOClientData(overrides: Partial<MCPClientConfig> = {}): MCPClientConfig {
   // Use the built test-tools-server
-  const serverPath = resolve(__dirname, '../../../../examples/mcps/test-tools-server/dist/index.js')
+  const REPO_ROOT = resolve(__dirname, '..', '..', '..', '..')
 
+  // Then for the stdio server:
+  const serverPath = join(REPO_ROOT, 'examples', 'mcps', 'test-tools-server', 'dist', 'index.js')
   return createMCPClientData({
     name: `stdio_client_${Date.now()}`,
     connectionType: 'stdio',