@metronome/mcp 2.2.0 → 3.4.1

package/src/options.ts

@@ -1,140 +1,86 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
 import qs from 'qs';
 import yargs from 'yargs';
 import { hideBin } from 'yargs/helpers';
 import z from 'zod';
-import { endpoints, Filter } from './tools';
-import { ClientCapabilities, knownClients, ClientType } from './compat';
+import { readEnv } from './util';
 
 export type CLIOptions = McpOptions & {
-  list: boolean;
+  debug: boolean;
+  logFormat: 'json' | 'pretty';
   transport: 'stdio' | 'http';
   port: number | undefined;
   socket: string | undefined;
 };
 
 export type McpOptions = {
-  client?: ClientType | undefined;
-  includeDynamicTools?: boolean | undefined;
-  includeAllTools?: boolean | undefined;
-  includeCodeTools?: boolean | undefined;
+  includeCodeTool?: boolean | undefined;
   includeDocsTools?: boolean | undefined;
-  filters?: Filter[] | undefined;
-  capabilities?: Partial<ClientCapabilities> | undefined;
+  stainlessApiKey?: string | undefined;
+  codeAllowHttpGets?: boolean | undefined;
+  codeAllowedMethods?: string[] | undefined;
+  codeBlockedMethods?: string[] | undefined;
+  codeExecutionMode: McpCodeExecutionMode;
 };
 
-const CAPABILITY_CHOICES = [
-  'top-level-unions',
-  'valid-json',
-  'refs',
-  'unions',
-  'formats',
-  'tool-name-length',
-] as const;
-
-type Capability = (typeof CAPABILITY_CHOICES)[number];
-
-function parseCapabilityValue(cap: string): { name: Capability; value?: number } {
-  if (cap.startsWith('tool-name-length=')) {
-    const parts = cap.split('=');
-    if (parts.length === 2) {
-      const length = parseInt(parts[1]!, 10);
-      if (!isNaN(length)) {
-        return { name: 'tool-name-length', value: length };
-      }
-      throw new Error(`Invalid tool-name-length value: ${parts[1]}. Expected a number.`);
-    }
-    throw new Error(`Invalid format for tool-name-length. Expected tool-name-length=N.`);
-  }
-  if (!CAPABILITY_CHOICES.includes(cap as Capability)) {
-    throw new Error(`Unknown capability: ${cap}. Valid capabilities are: ${CAPABILITY_CHOICES.join(', ')}`);
-  }
-  return { name: cap as Capability };
-}
+export type McpCodeExecutionMode = 'stainless-sandbox' | 'local';
 
 export function parseCLIOptions(): CLIOptions {
   const opts = yargs(hideBin(process.argv))
-    .option('tools', {
-      type: 'string',
-      array: true,
-      choices: ['dynamic', 'all', 'code', 'docs'],
-      description: 'Use dynamic tools or all tools',
-    })
-    .option('no-tools', {
-      type: 'string',
-      array: true,
-      choices: ['dynamic', 'all', 'code', 'docs'],
-      description: 'Do not use any dynamic or all tools',
-    })
-    .option('tool', {
-      type: 'string',
-      array: true,
-      description: 'Include tools matching the specified names',
-    })
-    .option('resource', {
-      type: 'string',
-      array: true,
-      description: 'Include tools matching the specified resources',
+    .option('code-allow-http-gets', {
+      type: 'boolean',
+      description:
+        'Allow all code tool methods that map to HTTP GET operations. If all code-allow-* flags are unset, then everything is allowed.',
     })
-    .option('operation', {
+    .option('code-allowed-methods', {
       type: 'string',
       array: true,
-      choices: ['read', 'write'],
-      description: 'Include tools matching the specified operations',
+      description:
+        'Methods to explicitly allow for code tool. Evaluated as regular expressions against method fully qualified names. If all code-allow-* flags are unset, then everything is allowed.',
     })
-    .option('tag', {
+    .option('code-blocked-methods', {
       type: 'string',
       array: true,
-      description: 'Include tools with the specified tags',
+      description:
+        'Methods to explicitly block for code tool. Evaluated as regular expressions against method fully qualified names. If all code-allow-* flags are unset, then everything is allowed.',
     })
-    .option('no-tool', {
+    .option('code-execution-mode', {
       type: 'string',
-      array: true,
-      description: 'Exclude tools matching the specified names',
+      choices: ['stainless-sandbox', 'local'],
+      default: 'stainless-sandbox',
+      description:
+        "Where to run code execution in code tool; 'stainless-sandbox' will execute code in Stainless-hosted sandboxes whereas 'local' will execute code locally on the MCP server machine.",
     })
-    .option('no-resource', {
+    .option('debug', { type: 'boolean', description: 'Enable debug logging' })
+    .option('log-format', {
       type: 'string',
-      array: true,
-      description: 'Exclude tools matching the specified resources',
+      choices: ['json', 'pretty'],
+      description: 'Format for log output; defaults to json unless tty is detected',
     })
-    .option('no-operation', {
-      type: 'string',
-      array: true,
-      description: 'Exclude tools matching the specified operations',
-    })
-    .option('no-tag', {
+    .option('no-tools', {
       type: 'string',
       array: true,
-      description: 'Exclude tools with the specified tags',
-    })
-    .option('list', {
-      type: 'boolean',
-      description: 'List all tools and exit',
+      choices: ['code', 'docs'],
+      description: 'Tools to explicitly disable',
     })
-    .option('client', {
-      type: 'string',
-      choices: Object.keys(knownClients),
-      description: 'Specify the MCP client being used',
+    .option('port', {
+      type: 'number',
+      default: 3000,
+      description: 'Port to serve on if using http transport',
     })
-    .option('capability', {
+    .option('socket', { type: 'string', description: 'Unix socket to serve on if using http transport' })
+    .option('stainless-api-key', {
       type: 'string',
-      array: true,
-      description: 'Specify client capabilities',
-      coerce: (values: string[]) => {
-        return values.flatMap((v) => v.split(','));
-      },
+      default: readEnv('STAINLESS_API_KEY'),
+      description:
+        'API key for Stainless. Used to authenticate requests to Stainless-hosted tools endpoints.',
     })
-    .option('no-capability', {
+    .option('tools', {
       type: 'string',
       array: true,
-      description: 'Unset client capabilities',
-      choices: CAPABILITY_CHOICES,
-      coerce: (values: string[]) => {
-        return values.flatMap((v) => v.split(','));
-      },
-    })
-    .option('describe-capabilities', {
-      type: 'boolean',
-      description: 'Print detailed explanation of client capabilities and exit',
+      choices: ['code', 'docs'],
+      description: 'Tools to explicitly enable',
     })
     .option('transport', {
       type: 'string',
@@ -142,133 +88,37 @@ export function parseCLIOptions(): CLIOptions {
       default: 'stdio',
       description: 'What transport to use; stdio for local servers or http for remote servers',
     })
-    .option('port', {
-      type: 'number',
-      description: 'Port to serve on if using http transport',
-    })
-    .option('socket', {
-      type: 'string',
-      description: 'Unix socket to serve on if using http transport',
-    })
+    .env('MCP_SERVER')
+    .version(true)
     .help();
 
-  for (const [command, desc] of examples()) {
-    opts.example(command, desc);
-  }
-
   const argv = opts.parseSync();
 
-  // Handle describe-capabilities flag
-  if (argv.describeCapabilities) {
-    console.log(getCapabilitiesExplanation());
-    process.exit(0);
-  }
-
-  const filters: Filter[] = [];
-
-  // Helper function to support comma-separated values
-  const splitValues = (values: string[] | undefined): string[] => {
-    if (!values) return [];
-    return values.flatMap((v) => v.split(','));
-  };
-
-  for (const tag of splitValues(argv.tag)) {
-    filters.push({ type: 'tag', op: 'include', value: tag });
-  }
-
-  for (const tag of splitValues(argv.noTag)) {
-    filters.push({ type: 'tag', op: 'exclude', value: tag });
-  }
-
-  for (const resource of splitValues(argv.resource)) {
-    filters.push({ type: 'resource', op: 'include', value: resource });
-  }
-
-  for (const resource of splitValues(argv.noResource)) {
-    filters.push({ type: 'resource', op: 'exclude', value: resource });
-  }
-
-  for (const tool of splitValues(argv.tool)) {
-    filters.push({ type: 'tool', op: 'include', value: tool });
-  }
-
-  for (const tool of splitValues(argv.noTool)) {
-    filters.push({ type: 'tool', op: 'exclude', value: tool });
-  }
-
-  for (const operation of splitValues(argv.operation)) {
-    filters.push({ type: 'operation', op: 'include', value: operation });
-  }
-
-  for (const operation of splitValues(argv.noOperation)) {
-    filters.push({ type: 'operation', op: 'exclude', value: operation });
-  }
-
-  // Parse client capabilities
-  const clientCapabilities: Partial<ClientCapabilities> = {};
-
-  // Apply individual capability overrides
-  if (Array.isArray(argv.capability)) {
-    for (const cap of argv.capability) {
-      const parsedCap = parseCapabilityValue(cap);
-      if (parsedCap.name === 'top-level-unions') {
-        clientCapabilities.topLevelUnions = true;
-      } else if (parsedCap.name === 'valid-json') {
-        clientCapabilities.validJson = true;
-      } else if (parsedCap.name === 'refs') {
-        clientCapabilities.refs = true;
-      } else if (parsedCap.name === 'unions') {
-        clientCapabilities.unions = true;
-      } else if (parsedCap.name === 'formats') {
-        clientCapabilities.formats = true;
-      } else if (parsedCap.name === 'tool-name-length') {
-        clientCapabilities.toolNameLength = parsedCap.value;
-      }
-    }
-  }
-
-  // Handle no-capability options to unset capabilities
-  if (Array.isArray(argv.noCapability)) {
-    for (const cap of argv.noCapability) {
-      if (cap === 'top-level-unions') {
-        clientCapabilities.topLevelUnions = false;
-      } else if (cap === 'valid-json') {
-        clientCapabilities.validJson = false;
-      } else if (cap === 'refs') {
-        clientCapabilities.refs = false;
-      } else if (cap === 'unions') {
-        clientCapabilities.unions = false;
-      } else if (cap === 'formats') {
-        clientCapabilities.formats = false;
-      } else if (cap === 'tool-name-length') {
-        clientCapabilities.toolNameLength = undefined;
-      }
-    }
-  }
-
-  const shouldIncludeToolType = (toolType: 'dynamic' | 'all' | 'code' | 'docs') =>
+  const shouldIncludeToolType = (toolType: 'code' | 'docs') =>
     argv.noTools?.includes(toolType) ? false
     : argv.tools?.includes(toolType) ? true
     : undefined;
 
-  const includeDynamicTools = shouldIncludeToolType('dynamic');
-  const includeAllTools = shouldIncludeToolType('all');
-  const includeCodeTools = shouldIncludeToolType('code');
+  const includeCodeTool = shouldIncludeToolType('code');
   const includeDocsTools = shouldIncludeToolType('docs');
 
   const transport = argv.transport as 'stdio' | 'http';
+  const logFormat =
+    argv.logFormat ? (argv.logFormat as 'json' | 'pretty')
+    : process.stderr.isTTY ? 'pretty'
+    : 'json';
 
-  const client = argv.client as ClientType;
   return {
-    client: client && client !== 'infer' && knownClients[client] ? client : undefined,
-    includeDynamicTools,
-    includeAllTools,
-    includeCodeTools,
-    includeDocsTools,
-    filters,
-    capabilities: clientCapabilities,
-    list: argv.list || false,
+    ...(includeCodeTool !== undefined && { includeCodeTool }),
+    ...(includeDocsTools !== undefined && { includeDocsTools }),
+    debug: !!argv.debug,
+    stainlessApiKey: argv.stainlessApiKey,
+    codeAllowHttpGets: argv.codeAllowHttpGets,
+    codeAllowedMethods: argv.codeAllowedMethods,
+    codeBlockedMethods: argv.codeBlockedMethods,
+    codeExecutionMode: argv.codeExecutionMode as McpCodeExecutionMode,
     transport,
+    logFormat,
     port: argv.port,
     socket: argv.socket,
   };
@@ -284,190 +134,28 @@ const coerceArray = <T extends z.ZodTypeAny>(zodType: T) =>
   );
 
 const QueryOptions = z.object({
-  tools: coerceArray(z.enum(['dynamic', 'all', 'code', 'docs'])).describe('Specify which MCP tools to use'),
-  no_tools: coerceArray(z.enum(['dynamic', 'all', 'code', 'docs'])).describe(
-    'Specify which MCP tools to not use.',
-  ),
+  tools: coerceArray(z.enum(['code', 'docs'])).describe('Specify which MCP tools to use'),
+  no_tools: coerceArray(z.enum(['code', 'docs'])).describe('Specify which MCP tools to not use.'),
   tool: coerceArray(z.string()).describe('Include tools matching the specified names'),
-  resource: coerceArray(z.string()).describe('Include tools matching the specified resources'),
-  operation: coerceArray(z.enum(['read', 'write'])).describe(
-    'Include tools matching the specified operations',
-  ),
-  tag: coerceArray(z.string()).describe('Include tools with the specified tags'),
-  no_tool: coerceArray(z.string()).describe('Exclude tools matching the specified names'),
-  no_resource: coerceArray(z.string()).describe('Exclude tools matching the specified resources'),
-  no_operation: coerceArray(z.enum(['read', 'write'])).describe(
-    'Exclude tools matching the specified operations',
-  ),
-  no_tag: coerceArray(z.string()).describe('Exclude tools with the specified tags'),
-  client: ClientType.optional().describe('Specify the MCP client being used'),
-  capability: coerceArray(z.string()).describe('Specify client capabilities'),
-  no_capability: coerceArray(z.enum(CAPABILITY_CHOICES)).describe('Unset client capabilities'),
 });
 
 export function parseQueryOptions(defaultOptions: McpOptions, query: unknown): McpOptions {
   const queryObject = typeof query === 'string' ? qs.parse(query) : query;
   const queryOptions = QueryOptions.parse(queryObject);
 
-  const filters: Filter[] = [...(defaultOptions.filters ?? [])];
-
-  for (const resource of queryOptions.resource || []) {
-    filters.push({ type: 'resource', op: 'include', value: resource });
-  }
-  for (const operation of queryOptions.operation || []) {
-    filters.push({ type: 'operation', op: 'include', value: operation });
-  }
-  for (const tag of queryOptions.tag || []) {
-    filters.push({ type: 'tag', op: 'include', value: tag });
-  }
-  for (const tool of queryOptions.tool || []) {
-    filters.push({ type: 'tool', op: 'include', value: tool });
-  }
-  for (const resource of queryOptions.no_resource || []) {
-    filters.push({ type: 'resource', op: 'exclude', value: resource });
-  }
-  for (const operation of queryOptions.no_operation || []) {
-    filters.push({ type: 'operation', op: 'exclude', value: operation });
-  }
-  for (const tag of queryOptions.no_tag || []) {
-    filters.push({ type: 'tag', op: 'exclude', value: tag });
-  }
-  for (const tool of queryOptions.no_tool || []) {
-    filters.push({ type: 'tool', op: 'exclude', value: tool });
-  }
-
-  // Parse client capabilities
-  const clientCapabilities: Partial<ClientCapabilities> = { ...defaultOptions.capabilities };
-
-  for (const cap of queryOptions.capability || []) {
-    const parsed = parseCapabilityValue(cap);
-    if (parsed.name === 'top-level-unions') {
-      clientCapabilities.topLevelUnions = true;
-    } else if (parsed.name === 'valid-json') {
-      clientCapabilities.validJson = true;
-    } else if (parsed.name === 'refs') {
-      clientCapabilities.refs = true;
-    } else if (parsed.name === 'unions') {
-      clientCapabilities.unions = true;
-    } else if (parsed.name === 'formats') {
-      clientCapabilities.formats = true;
-    } else if (parsed.name === 'tool-name-length') {
-      clientCapabilities.toolNameLength = parsed.value;
-    }
-  }
-
-  for (const cap of queryOptions.no_capability || []) {
-    if (cap === 'top-level-unions') {
-      clientCapabilities.topLevelUnions = false;
-    } else if (cap === 'valid-json') {
-      clientCapabilities.validJson = false;
-    } else if (cap === 'refs') {
-      clientCapabilities.refs = false;
-    } else if (cap === 'unions') {
-      clientCapabilities.unions = false;
-    } else if (cap === 'formats') {
-      clientCapabilities.formats = false;
-    } else if (cap === 'tool-name-length') {
-      clientCapabilities.toolNameLength = undefined;
-    }
-  }
-
-  let dynamicTools: boolean | undefined =
-    queryOptions.no_tools && queryOptions.no_tools?.includes('dynamic') ? false
-    : queryOptions.tools?.includes('dynamic') ? true
-    : defaultOptions.includeDynamicTools;
-
-  let allTools: boolean | undefined =
-    queryOptions.no_tools && queryOptions.no_tools?.includes('all') ? false
-    : queryOptions.tools?.includes('all') ? true
-    : defaultOptions.includeAllTools;
+  let codeTool: boolean | undefined =
+    queryOptions.no_tools && queryOptions.no_tools?.includes('code') ? false
+    : queryOptions.tools?.includes('code') ? true
+    : defaultOptions.includeCodeTool;
 
   let docsTools: boolean | undefined =
     queryOptions.no_tools && queryOptions.no_tools?.includes('docs') ? false
     : queryOptions.tools?.includes('docs') ? true
     : defaultOptions.includeDocsTools;
 
-  let codeTools: boolean | undefined =
-    queryOptions.no_tools && queryOptions.no_tools?.includes('code') ? false
-    : queryOptions.tools?.includes('code') && defaultOptions.includeCodeTools ? true
-    : defaultOptions.includeCodeTools;
-
   return {
-    client: queryOptions.client ?? defaultOptions.client,
-    includeDynamicTools: dynamicTools,
-    includeAllTools: allTools,
-    includeCodeTools: codeTools,
-    includeDocsTools: docsTools,
-    filters,
-    capabilities: clientCapabilities,
+    ...(codeTool !== undefined && { includeCodeTool: codeTool }),
+    ...(docsTools !== undefined && { includeDocsTools: docsTools }),
+    codeExecutionMode: defaultOptions.codeExecutionMode,
   };
 }
-
-function getCapabilitiesExplanation(): string {
-  return `
-Client Capabilities Explanation:
-
-Different Language Models (LLMs) and the MCP clients that use them have varying limitations in how they handle tool schemas. Capability flags allow you to inform the MCP server about these limitations.
-
-When a capability flag is set to false, the MCP server will automatically adjust the tool schemas to work around that limitation, ensuring broader compatibility.
-
-Available Capabilities:
-
-# top-level-unions
-Some clients/LLMs do not support JSON schemas with a union type (anyOf) at the root level. If a client lacks this capability, the MCP server splits tools with top-level unions into multiple separate tools, one for each variant in the union.
-
-# refs
-Some clients/LLMs do not support $ref pointers for schema reuse. If a client lacks this capability, the MCP server automatically inlines all references ($defs) directly into the schema. Properties that would cause circular references are removed during this process.
-
-# valid-json
-Some clients/LLMs may incorrectly send arguments as a JSON-encoded string instead of a proper JSON object. If a client *has* this capability, the MCP server will attempt to parse string values as JSON if the initial validation against the schema fails.
-
-# unions
-Some clients/LLMs do not support union types (anyOf) in JSON schemas. If a client lacks this capability, the MCP server removes all anyOf fields and uses only the first variant as the schema.
-
-# formats
-Some clients/LLMs do not support the 'format' keyword in JSON Schema specifications. If a client lacks this capability, the MCP server removes all format fields and appends the format information to the field's description in parentheses.
-
-# tool-name-length=N
-Some clients/LLMs impose a maximum length on tool names. If this capability is set, the MCP server will automatically truncate tool names exceeding the specified length (N), ensuring uniqueness by appending numbers if necessary.
-
-Client Presets (--client):
-Presets like '--client=openai-agents' or '--client=cursor' automatically configure these capabilities based on current known limitations of those clients, simplifying setup.
-
-Current presets:
-${JSON.stringify(knownClients, null, 2)}
-  `;
-}
-
-function examples(): [string, string][] {
-  const firstEndpoint = endpoints[0]!;
-  const secondEndpoint =
-    endpoints.find((e) => e.metadata.resource !== firstEndpoint.metadata.resource) || endpoints[1];
-  const tag = endpoints.find((e) => e.metadata.tags.length > 0)?.metadata.tags[0];
-  const otherEndpoint = secondEndpoint || firstEndpoint;
-
-  return [
-    [
-      `--tool="${firstEndpoint.tool.name}" ${secondEndpoint ? `--tool="${secondEndpoint.tool.name}"` : ''}`,
-      'Include tools by name',
-    ],
-    [
-      `--resource="${firstEndpoint.metadata.resource}" --operation="read"`,
-      'Filter by resource and operation',
-    ],
-    [
-      `--resource="${otherEndpoint.metadata.resource}*" --no-tool="${otherEndpoint.tool.name}"`,
-      'Use resource wildcards and exclusions',
-    ],
-    [`--client="cursor"`, 'Adjust schemas to be more compatible with Cursor'],
-    [
-      `--capability="top-level-unions" --capability="tool-name-length=40"`,
-      'Specify individual client capabilities',
-    ],
-    [
-      `--client="cursor" --no-capability="tool-name-length"`,
-      'Use cursor client preset but remove tool name length limit',
-    ],
-    ...(tag ? [[`--tag="${tag}"`, 'Filter based on tags'] as [string, string]] : []),
-  ];
-}