@stigg/typescript-mcp 0.1.0-alpha.9 → 0.1.0-beta.10

package/src/logger.ts

@@ -0,0 +1,28 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { pino, type Level, type Logger } from 'pino';
+import pretty from 'pino-pretty';
+
+let _logger: Logger | undefined;
+
+export function configureLogger({ level, pretty: usePretty }: { level: Level; pretty: boolean }): void {
+  _logger = pino(
+    {
+      level,
+      timestamp: pino.stdTimeFunctions.isoTime,
+      formatters: {
+        level(label) {
+          return { level: label };
+        },
+      },
+    },
+    usePretty ? pretty({ colorize: true, levelFirst: true, destination: 2 }) : process.stderr,
+  );
+}
+
+export function getLogger(): Logger {
+  if (!_logger) {
+    throw new Error('Logger has not been configured. Call configureLogger() before using the logger.');
+  }
+  return _logger;
+}

package/src/methods.ts

@@ -34,6 +34,12 @@ export const sdkMethods: SdkMethod[] = [
     httpMethod: 'post',
     httpPath: '/api/v1/customers/{id}/archive',
   },
+  {
+    clientCallName: 'client.v1.customers.checkEntitlement',
+    fullyQualifiedName: 'v1.customers.checkEntitlement',
+    httpMethod: 'get',
+    httpPath: '/api/v1/customers/{id}/entitlements/check',
+  },
   {
     clientCallName: 'client.v1.customers.import',
     fullyQualifiedName: 'v1.customers.import',
@@ -52,6 +58,12 @@ export const sdkMethods: SdkMethod[] = [
     httpMethod: 'post',
     httpPath: '/api/v1/customers',
   },
+  {
+    clientCallName: 'client.v1.customers.retrieveEntitlements',
+    fullyQualifiedName: 'v1.customers.retrieveEntitlements',
+    httpMethod: 'get',
+    httpPath: '/api/v1/customers/{id}/entitlements',
+  },
   {
     clientCallName: 'client.v1.customers.unarchive',
     fullyQualifiedName: 'v1.customers.unarchive',
@@ -88,6 +100,36 @@ export const sdkMethods: SdkMethod[] = [
     httpMethod: 'delete',
     httpPath: '/api/v1/customers/{id}/promotional-entitlements/{featureId}',
   },
+  {
+    clientCallName: 'client.v1.customers.integrations.retrieve',
+    fullyQualifiedName: 'v1.customers.integrations.retrieve',
+    httpMethod: 'get',
+    httpPath: '/api/v1/customers/{id}/integrations/{integrationId}',
+  },
+  {
+    clientCallName: 'client.v1.customers.integrations.update',
+    fullyQualifiedName: 'v1.customers.integrations.update',
+    httpMethod: 'patch',
+    httpPath: '/api/v1/customers/{id}/integrations/{integrationId}',
+  },
+  {
+    clientCallName: 'client.v1.customers.integrations.list',
+    fullyQualifiedName: 'v1.customers.integrations.list',
+    httpMethod: 'get',
+    httpPath: '/api/v1/customers/{id}/integrations',
+  },
+  {
+    clientCallName: 'client.v1.customers.integrations.link',
+    fullyQualifiedName: 'v1.customers.integrations.link',
+    httpMethod: 'post',
+    httpPath: '/api/v1/customers/{id}/integrations',
+  },
+  {
+    clientCallName: 'client.v1.customers.integrations.unlink',
+    fullyQualifiedName: 'v1.customers.integrations.unlink',
+    httpMethod: 'delete',
+    httpPath: '/api/v1/customers/{id}/integrations/{integrationId}',
+  },
   {
     clientCallName: 'client.v1.subscriptions.retrieve',
     fullyQualifiedName: 'v1.subscriptions.retrieve',
@@ -215,107 +257,275 @@ export const sdkMethods: SdkMethod[] = [
     httpPath: '/api/v1/events',
   },
   {
-    clientCallName: 'client.v1.events.features.archiveFeature',
-    fullyQualifiedName: 'v1.events.features.archiveFeature',
+    clientCallName: 'client.v1.credits.getAutoRecharge',
+    fullyQualifiedName: 'v1.credits.getAutoRecharge',
+    httpMethod: 'get',
+    httpPath: '/api/v1/credits/auto-recharge',
+  },
+  {
+    clientCallName: 'client.v1.credits.getUsage',
+    fullyQualifiedName: 'v1.credits.getUsage',
+    httpMethod: 'get',
+    httpPath: '/api/v1/credits/usage',
+  },
+  {
+    clientCallName: 'client.v1.credits.listLedger',
+    fullyQualifiedName: 'v1.credits.listLedger',
+    httpMethod: 'get',
+    httpPath: '/api/v1/credits/ledger',
+  },
+  {
+    clientCallName: 'client.v1.credits.grants.create',
+    fullyQualifiedName: 'v1.credits.grants.create',
+    httpMethod: 'post',
+    httpPath: '/api/v1/credits/grants',
+  },
+  {
+    clientCallName: 'client.v1.credits.grants.list',
+    fullyQualifiedName: 'v1.credits.grants.list',
+    httpMethod: 'get',
+    httpPath: '/api/v1/credits/grants',
+  },
+  {
+    clientCallName: 'client.v1.credits.grants.void',
+    fullyQualifiedName: 'v1.credits.grants.void',
+    httpMethod: 'post',
+    httpPath: '/api/v1/credits/grants/{id}/void',
+  },
+  {
+    clientCallName: 'client.v1.credits.customCurrencies.create',
+    fullyQualifiedName: 'v1.credits.customCurrencies.create',
+    httpMethod: 'post',
+    httpPath: '/api/v1/credits/custom-currencies',
+  },
+  {
+    clientCallName: 'client.v1.credits.customCurrencies.update',
+    fullyQualifiedName: 'v1.credits.customCurrencies.update',
+    httpMethod: 'patch',
+    httpPath: '/api/v1/credits/custom-currencies/{currencyId}',
+  },
+  {
+    clientCallName: 'client.v1.credits.customCurrencies.list',
+    fullyQualifiedName: 'v1.credits.customCurrencies.list',
+    httpMethod: 'get',
+    httpPath: '/api/v1/credits/custom-currencies',
+  },
+  {
+    clientCallName: 'client.v1.credits.customCurrencies.archive',
+    fullyQualifiedName: 'v1.credits.customCurrencies.archive',
+    httpMethod: 'post',
+    httpPath: '/api/v1/credits/custom-currencies/{currencyId}/archive',
+  },
+  {
+    clientCallName: 'client.v1.credits.customCurrencies.listAssociatedEntities',
+    fullyQualifiedName: 'v1.credits.customCurrencies.listAssociatedEntities',
+    httpMethod: 'get',
+    httpPath: '/api/v1/credits/custom-currencies/{currencyId}/associated-entities',
+  },
+  {
+    clientCallName: 'client.v1.credits.customCurrencies.unarchive',
+    fullyQualifiedName: 'v1.credits.customCurrencies.unarchive',
+    httpMethod: 'post',
+    httpPath: '/api/v1/credits/custom-currencies/{currencyId}/unarchive',
+  },
+  {
+    clientCallName: 'client.v1.features.archiveFeature',
+    fullyQualifiedName: 'v1.features.archiveFeature',
     httpMethod: 'post',
     httpPath: '/api/v1/features/{id}/archive',
   },
   {
-    clientCallName: 'client.v1.events.features.createFeature',
-    fullyQualifiedName: 'v1.events.features.createFeature',
+    clientCallName: 'client.v1.features.createFeature',
+    fullyQualifiedName: 'v1.features.createFeature',
     httpMethod: 'post',
     httpPath: '/api/v1/features',
   },
   {
-    clientCallName: 'client.v1.events.features.listFeatures',
-    fullyQualifiedName: 'v1.events.features.listFeatures',
+    clientCallName: 'client.v1.features.listFeatures',
+    fullyQualifiedName: 'v1.features.listFeatures',
     httpMethod: 'get',
     httpPath: '/api/v1/features',
   },
   {
-    clientCallName: 'client.v1.events.features.retrieveFeature',
-    fullyQualifiedName: 'v1.events.features.retrieveFeature',
+    clientCallName: 'client.v1.features.retrieveFeature',
+    fullyQualifiedName: 'v1.features.retrieveFeature',
     httpMethod: 'get',
     httpPath: '/api/v1/features/{id}',
   },
   {
-    clientCallName: 'client.v1.events.features.unarchiveFeature',
-    fullyQualifiedName: 'v1.events.features.unarchiveFeature',
+    clientCallName: 'client.v1.features.unarchiveFeature',
+    fullyQualifiedName: 'v1.features.unarchiveFeature',
     httpMethod: 'post',
     httpPath: '/api/v1/features/{id}/unarchive',
   },
   {
-    clientCallName: 'client.v1.events.features.updateFeature',
-    fullyQualifiedName: 'v1.events.features.updateFeature',
+    clientCallName: 'client.v1.features.updateFeature',
+    fullyQualifiedName: 'v1.features.updateFeature',
     httpMethod: 'patch',
     httpPath: '/api/v1/features/{id}',
   },
   {
-    clientCallName: 'client.v1.events.addons.archiveAddon',
-    fullyQualifiedName: 'v1.events.addons.archiveAddon',
+    clientCallName: 'client.v1.addons.create',
+    fullyQualifiedName: 'v1.addons.create',
+    httpMethod: 'post',
+    httpPath: '/api/v1/addons',
+  },
+  {
+    clientCallName: 'client.v1.addons.retrieve',
+    fullyQualifiedName: 'v1.addons.retrieve',
+    httpMethod: 'get',
+    httpPath: '/api/v1/addons/{id}',
+  },
+  {
+    clientCallName: 'client.v1.addons.update',
+    fullyQualifiedName: 'v1.addons.update',
+    httpMethod: 'patch',
+    httpPath: '/api/v1/addons/{id}',
+  },
+  {
+    clientCallName: 'client.v1.addons.list',
+    fullyQualifiedName: 'v1.addons.list',
+    httpMethod: 'get',
+    httpPath: '/api/v1/addons',
+  },
+  {
+    clientCallName: 'client.v1.addons.archive',
+    fullyQualifiedName: 'v1.addons.archive',
     httpMethod: 'post',
     httpPath: '/api/v1/addons/{id}/archive',
   },
   {
-    clientCallName: 'client.v1.events.addons.createAddon',
-    fullyQualifiedName: 'v1.events.addons.createAddon',
+    clientCallName: 'client.v1.addons.createDraft',
+    fullyQualifiedName: 'v1.addons.createDraft',
     httpMethod: 'post',
-    httpPath: '/api/v1/addons',
+    httpPath: '/api/v1/addons/{id}/draft',
   },
   {
-    clientCallName: 'client.v1.events.addons.listAddons',
-    fullyQualifiedName: 'v1.events.addons.listAddons',
+    clientCallName: 'client.v1.addons.listCharges',
+    fullyQualifiedName: 'v1.addons.listCharges',
     httpMethod: 'get',
-    httpPath: '/api/v1/addons',
+    httpPath: '/api/v1/addons/{id}/charges',
   },
   {
-    clientCallName: 'client.v1.events.addons.publishAddon',
-    fullyQualifiedName: 'v1.events.addons.publishAddon',
+    clientCallName: 'client.v1.addons.publish',
+    fullyQualifiedName: 'v1.addons.publish',
     httpMethod: 'post',
     httpPath: '/api/v1/addons/{id}/publish',
   },
   {
-    clientCallName: 'client.v1.events.addons.retrieveAddon',
-    fullyQualifiedName: 'v1.events.addons.retrieveAddon',
-    httpMethod: 'get',
-    httpPath: '/api/v1/addons/{id}',
+    clientCallName: 'client.v1.addons.removeDraft',
+    fullyQualifiedName: 'v1.addons.removeDraft',
+    httpMethod: 'delete',
+    httpPath: '/api/v1/addons/{id}/draft',
   },
   {
-    clientCallName: 'client.v1.events.addons.updateAddon',
-    fullyQualifiedName: 'v1.events.addons.updateAddon',
+    clientCallName: 'client.v1.addons.entitlements.create',
+    fullyQualifiedName: 'v1.addons.entitlements.create',
+    httpMethod: 'post',
+    httpPath: '/api/v1/addons/{addonId}/entitlements',
+  },
+  {
+    clientCallName: 'client.v1.addons.entitlements.update',
+    fullyQualifiedName: 'v1.addons.entitlements.update',
     httpMethod: 'patch',
-    httpPath: '/api/v1/addons/{id}',
+    httpPath: '/api/v1/addons/{addonId}/entitlements/{id}',
   },
   {
-    clientCallName: 'client.v1.events.addons.draft.createAddonDraft',
-    fullyQualifiedName: 'v1.events.addons.draft.createAddonDraft',
-    httpMethod: 'post',
-    httpPath: '/api/v1/addons/{id}/draft',
+    clientCallName: 'client.v1.addons.entitlements.list',
+    fullyQualifiedName: 'v1.addons.entitlements.list',
+    httpMethod: 'get',
+    httpPath: '/api/v1/addons/{addonId}/entitlements',
   },
   {
-    clientCallName: 'client.v1.events.addons.draft.removeAddonDraft',
-    fullyQualifiedName: 'v1.events.addons.draft.removeAddonDraft',
+    clientCallName: 'client.v1.addons.entitlements.delete',
+    fullyQualifiedName: 'v1.addons.entitlements.delete',
     httpMethod: 'delete',
-    httpPath: '/api/v1/addons/{id}/draft',
+    httpPath: '/api/v1/addons/{addonId}/entitlements/{id}',
   },
   {
-    clientCallName: 'client.v1.events.plans.create',
-    fullyQualifiedName: 'v1.events.plans.create',
+    clientCallName: 'client.v1.plans.create',
+    fullyQualifiedName: 'v1.plans.create',
     httpMethod: 'post',
     httpPath: '/api/v1/plans',
   },
   {
-    clientCallName: 'client.v1.events.plans.retrieve',
-    fullyQualifiedName: 'v1.events.plans.retrieve',
+    clientCallName: 'client.v1.plans.retrieve',
+    fullyQualifiedName: 'v1.plans.retrieve',
     httpMethod: 'get',
     httpPath: '/api/v1/plans/{id}',
   },
   {
-    clientCallName: 'client.v1.events.plans.list',
-    fullyQualifiedName: 'v1.events.plans.list',
+    clientCallName: 'client.v1.plans.update',
+    fullyQualifiedName: 'v1.plans.update',
+    httpMethod: 'patch',
+    httpPath: '/api/v1/plans/{id}',
+  },
+  {
+    clientCallName: 'client.v1.plans.list',
+    fullyQualifiedName: 'v1.plans.list',
     httpMethod: 'get',
     httpPath: '/api/v1/plans',
   },
+  {
+    clientCallName: 'client.v1.plans.archive',
+    fullyQualifiedName: 'v1.plans.archive',
+    httpMethod: 'post',
+    httpPath: '/api/v1/plans/{id}/archive',
+  },
+  {
+    clientCallName: 'client.v1.plans.createDraft',
+    fullyQualifiedName: 'v1.plans.createDraft',
+    httpMethod: 'post',
+    httpPath: '/api/v1/plans/{id}/draft',
+  },
+  {
+    clientCallName: 'client.v1.plans.listCharges',
+    fullyQualifiedName: 'v1.plans.listCharges',
+    httpMethod: 'get',
+    httpPath: '/api/v1/plans/{id}/charges',
+  },
+  {
+    clientCallName: 'client.v1.plans.listOverageCharges',
+    fullyQualifiedName: 'v1.plans.listOverageCharges',
+    httpMethod: 'get',
+    httpPath: '/api/v1/plans/{id}/overage-charges',
+  },
+  {
+    clientCallName: 'client.v1.plans.publish',
+    fullyQualifiedName: 'v1.plans.publish',
+    httpMethod: 'post',
+    httpPath: '/api/v1/plans/{id}/publish',
+  },
+  {
+    clientCallName: 'client.v1.plans.removeDraft',
+    fullyQualifiedName: 'v1.plans.removeDraft',
+    httpMethod: 'delete',
+    httpPath: '/api/v1/plans/{id}/draft',
+  },
+  {
+    clientCallName: 'client.v1.plans.entitlements.create',
+    fullyQualifiedName: 'v1.plans.entitlements.create',
+    httpMethod: 'post',
+    httpPath: '/api/v1/plans/{planId}/entitlements',
+  },
+  {
+    clientCallName: 'client.v1.plans.entitlements.update',
+    fullyQualifiedName: 'v1.plans.entitlements.update',
+    httpMethod: 'patch',
+    httpPath: '/api/v1/plans/{planId}/entitlements/{id}',
+  },
+  {
+    clientCallName: 'client.v1.plans.entitlements.list',
+    fullyQualifiedName: 'v1.plans.entitlements.list',
+    httpMethod: 'get',
+    httpPath: '/api/v1/plans/{planId}/entitlements',
+  },
+  {
+    clientCallName: 'client.v1.plans.entitlements.delete',
+    fullyQualifiedName: 'v1.plans.entitlements.delete',
+    httpMethod: 'delete',
+    httpPath: '/api/v1/plans/{planId}/entitlements/{id}',
+  },
   {
     clientCallName: 'client.v1.usage.history',
     fullyQualifiedName: 'v1.usage.history',
@@ -364,6 +574,66 @@ export const sdkMethods: SdkMethod[] = [
     httpMethod: 'patch',
     httpPath: '/api/v1/products/{id}',
   },
+  {
+    clientCallName: 'client.v1Beta.customers.entitlements.check',
+    fullyQualifiedName: 'v1Beta.customers.entitlements.check',
+    httpMethod: 'get',
+    httpPath: '/api/v1-beta/customers/{id}/entitlements/check',
+  },
+  {
+    clientCallName: 'client.v1Beta.customers.entities.retrieve',
+    fullyQualifiedName: 'v1Beta.customers.entities.retrieve',
+    httpMethod: 'get',
+    httpPath: '/api/v1-beta/customers/{id}/entities/{entityId}',
+  },
+  {
+    clientCallName: 'client.v1Beta.customers.entities.list',
+    fullyQualifiedName: 'v1Beta.customers.entities.list',
+    httpMethod: 'get',
+    httpPath: '/api/v1-beta/customers/{id}/entities',
+  },
+  {
+    clientCallName: 'client.v1Beta.customers.entities.archive',
+    fullyQualifiedName: 'v1Beta.customers.entities.archive',
+    httpMethod: 'post',
+    httpPath: '/api/v1-beta/customers/{id}/entities/archive',
+  },
+  {
+    clientCallName: 'client.v1Beta.customers.entities.unarchive',
+    fullyQualifiedName: 'v1Beta.customers.entities.unarchive',
+    httpMethod: 'post',
+    httpPath: '/api/v1-beta/customers/{id}/entities/unarchive',
+  },
+  {
+    clientCallName: 'client.v1Beta.customers.entities.upsert',
+    fullyQualifiedName: 'v1Beta.customers.entities.upsert',
+    httpMethod: 'put',
+    httpPath: '/api/v1-beta/customers/{id}/entities',
+  },
+  {
+    clientCallName: 'client.v1Beta.customers.assignments.list',
+    fullyQualifiedName: 'v1Beta.customers.assignments.list',
+    httpMethod: 'get',
+    httpPath: '/api/v1-beta/customers/{id}/assignments',
+  },
+  {
+    clientCallName: 'client.v1Beta.customers.assignments.upsert',
+    fullyQualifiedName: 'v1Beta.customers.assignments.upsert',
+    httpMethod: 'put',
+    httpPath: '/api/v1-beta/customers/{id}/assignments',
+  },
+  {
+    clientCallName: 'client.v1Beta.entityTypes.list',
+    fullyQualifiedName: 'v1Beta.entityTypes.list',
+    httpMethod: 'get',
+    httpPath: '/api/v1-beta/entity-types',
+  },
+  {
+    clientCallName: 'client.v1Beta.entityTypes.upsert',
+    fullyQualifiedName: 'v1Beta.entityTypes.upsert',
+    httpMethod: 'put',
+    httpPath: '/api/v1-beta/entity-types',
+  },
 ];
 
 function allowedMethodsForCodeTool(options: McpOptions | undefined): SdkMethod[] | undefined {

package/src/options.ts

@@ -8,19 +8,27 @@ import { readEnv } from './util';
 
 export type CLIOptions = McpOptions & {
   debug: boolean;
+  logFormat: 'json' | 'pretty';
   transport: 'stdio' | 'http';
   port: number | undefined;
   socket: string | undefined;
 };
 
 export type McpOptions = {
+  includeCodeTool?: boolean | undefined;
   includeDocsTools?: boolean | undefined;
   stainlessApiKey?: string | undefined;
+  docsSearchMode?: 'stainless-api' | 'local' | undefined;
+  docsDir?: string | undefined;
   codeAllowHttpGets?: boolean | undefined;
   codeAllowedMethods?: string[] | undefined;
   codeBlockedMethods?: string[] | undefined;
+  codeExecutionMode: McpCodeExecutionMode;
+  customInstructionsPath?: string | undefined;
 };
 
+export type McpCodeExecutionMode = 'stainless-sandbox' | 'local';
+
 export function parseCLIOptions(): CLIOptions {
   const opts = yargs(hideBin(process.argv))
     .option('code-allow-http-gets', {
@@ -40,7 +48,35 @@ export function parseCLIOptions(): CLIOptions {
       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('code-execution-mode', {
+      type: 'string',
+      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('custom-instructions-path', {
+      type: 'string',
+      description: 'Path to custom instructions for the MCP server',
+    })
     .option('debug', { type: 'boolean', description: 'Enable debug logging' })
+    .option('docs-dir', {
+      type: 'string',
+      description:
+        'Path to a directory of local documentation files (markdown/JSON) to include in local docs search.',
+    })
+    .option('docs-search-mode', {
+      type: 'string',
+      choices: ['stainless-api', 'local'],
+      default: 'stainless-api',
+      description:
+        "Where to search documentation; 'stainless-api' uses the Stainless-hosted search API whereas 'local' uses an in-memory search index built from embedded SDK method data and optional local docs files.",
+    })
+    .option('log-format', {
+      type: 'string',
+      choices: ['json', 'pretty'],
+      description: 'Format for log output; defaults to json unless tty is detected',
+    })
     .option('no-tools', {
       type: 'string',
       array: true,
@@ -82,18 +118,29 @@ export function parseCLIOptions(): CLIOptions {
     : argv.tools?.includes(toolType) ? true
     : undefined;
 
+  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';
 
   return {
+    ...(includeCodeTool !== undefined && { includeCodeTool }),
     ...(includeDocsTools !== undefined && { includeDocsTools }),
     debug: !!argv.debug,
     stainlessApiKey: argv.stainlessApiKey,
+    docsSearchMode: argv.docsSearchMode as 'stainless-api' | 'local' | undefined,
+    docsDir: argv.docsDir,
     codeAllowHttpGets: argv.codeAllowHttpGets,
     codeAllowedMethods: argv.codeAllowedMethods,
     codeBlockedMethods: argv.codeBlockedMethods,
+    codeExecutionMode: argv.codeExecutionMode as McpCodeExecutionMode,
+    customInstructionsPath: argv.customInstructionsPath,
     transport,
+    logFormat,
     port: argv.port,
     socket: argv.socket,
   };
@@ -118,12 +165,21 @@ export function parseQueryOptions(defaultOptions: McpOptions, query: unknown): M
   const queryObject = typeof query === 'string' ? qs.parse(query) : query;
   const queryOptions = QueryOptions.parse(queryObject);
 
+  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;
 
   return {
+    ...(codeTool !== undefined && { includeCodeTool: codeTool }),
     ...(docsTools !== undefined && { includeDocsTools: docsTools }),
+    codeExecutionMode: defaultOptions.codeExecutionMode,
+    docsSearchMode: defaultOptions.docsSearchMode,
+    docsDir: defaultOptions.docsDir,
   };
 }