@stigg/typescript-mcp 0.1.0-alpha.10 → 0.1.0-alpha.12

package/src/code-tool-worker.ts

@@ -138,23 +138,38 @@ const fuse = new Fuse(
     'client.v1.coupons.retrieve',
     'client.v1.coupons.updateCoupon',
     'client.v1.events.report',
-    'client.v1.events.features.archiveFeature',
-    'client.v1.events.features.createFeature',
-    'client.v1.events.features.listFeatures',
-    'client.v1.events.features.retrieveFeature',
-    'client.v1.events.features.unarchiveFeature',
-    'client.v1.events.features.updateFeature',
-    'client.v1.events.addons.archiveAddon',
-    'client.v1.events.addons.createAddon',
-    'client.v1.events.addons.listAddons',
-    'client.v1.events.addons.publishAddon',
-    'client.v1.events.addons.retrieveAddon',
-    'client.v1.events.addons.updateAddon',
-    'client.v1.events.addons.draft.createAddonDraft',
-    'client.v1.events.addons.draft.removeAddonDraft',
-    'client.v1.events.plans.create',
-    'client.v1.events.plans.list',
-    'client.v1.events.plans.retrieve',
+    'client.v1.features.archiveFeature',
+    'client.v1.features.createFeature',
+    'client.v1.features.listFeatures',
+    'client.v1.features.retrieveFeature',
+    'client.v1.features.unarchiveFeature',
+    'client.v1.features.updateFeature',
+    'client.v1.addons.archive',
+    'client.v1.addons.create',
+    'client.v1.addons.createDraft',
+    'client.v1.addons.list',
+    'client.v1.addons.publish',
+    'client.v1.addons.removeDraft',
+    'client.v1.addons.retrieve',
+    'client.v1.addons.setPricing',
+    'client.v1.addons.update',
+    'client.v1.addons.entitlements.create',
+    'client.v1.addons.entitlements.delete',
+    'client.v1.addons.entitlements.list',
+    'client.v1.addons.entitlements.update',
+    'client.v1.plans.archive',
+    'client.v1.plans.create',
+    'client.v1.plans.createDraft',
+    'client.v1.plans.list',
+    'client.v1.plans.publish',
+    'client.v1.plans.removeDraft',
+    'client.v1.plans.retrieve',
+    'client.v1.plans.setPricing',
+    'client.v1.plans.update',
+    'client.v1.plans.entitlements.create',
+    'client.v1.plans.entitlements.delete',
+    'client.v1.plans.entitlements.list',
+    'client.v1.plans.entitlements.update',
     'client.v1.usage.history',
     'client.v1.usage.report',
     'client.v1.products.archiveProduct',

package/src/code-tool.ts

@@ -23,7 +23,7 @@ import { ClientOptions } from '@stigg/typescript';
 
 const prompt = `Runs JavaScript code to interact with the Stigg API.
 
-You are a skilled programmer writing code to interface with the service.
+You are a skilled TypeScript programmer writing code to interface with the service.
 Define an async function named "run" that takes a single parameter of an initialized SDK client and it will be run.
 For example:
 
@@ -39,7 +39,9 @@ You will be returned anything that your function returns, plus the results of an
 Do not add try-catch blocks for single API calls. The tool will handle errors for you.
 Do not add comments unless necessary for generating better code.
 Code will run in a container, and cannot interact with the network outside of the given SDK client.
-Variables will not persist between calls, so make sure to return or log any data you might need later.`;
+Variables will not persist between calls, so make sure to return or log any data you might need later.
+Remember that you are writing TypeScript code, so you need to be careful with your types.
+Always type dynamic key-value stores explicitly as Record<string, YourValueType> instead of {}.`;
 
 /**
  * A tool that runs code against a copy of the SDK.

package/src/methods.ts

@@ -215,107 +215,197 @@ export const sdkMethods: SdkMethod[] = [
     httpPath: '/api/v1/events',
   },
   {
-    clientCallName: 'client.v1.events.features.archiveFeature',
-    fullyQualifiedName: 'v1.events.features.archiveFeature',
+    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/{id}/archive',
+    httpPath: '/api/v1/addons',
   },
   {
-    clientCallName: 'client.v1.events.addons.createAddon',
-    fullyQualifiedName: 'v1.events.addons.createAddon',
-    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.events.addons.listAddons',
-    fullyQualifiedName: 'v1.events.addons.listAddons',
+    clientCallName: 'client.v1.addons.list',
+    fullyQualifiedName: 'v1.addons.list',
     httpMethod: 'get',
     httpPath: '/api/v1/addons',
   },
   {
-    clientCallName: 'client.v1.events.addons.publishAddon',
-    fullyQualifiedName: 'v1.events.addons.publishAddon',
+    clientCallName: 'client.v1.addons.archive',
+    fullyQualifiedName: 'v1.addons.archive',
+    httpMethod: 'post',
+    httpPath: '/api/v1/addons/{id}/archive',
+  },
+  {
+    clientCallName: 'client.v1.addons.createDraft',
+    fullyQualifiedName: 'v1.addons.createDraft',
+    httpMethod: 'post',
+    httpPath: '/api/v1/addons/{id}/draft',
+  },
+  {
+    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',
-    httpMethod: 'patch',
-    httpPath: '/api/v1/addons/{id}',
+    clientCallName: 'client.v1.addons.setPricing',
+    fullyQualifiedName: 'v1.addons.setPricing',
+    httpMethod: 'put',
+    httpPath: '/api/v1/addons/{id}/charges',
   },
   {
-    clientCallName: 'client.v1.events.addons.draft.createAddonDraft',
-    fullyQualifiedName: 'v1.events.addons.draft.createAddonDraft',
+    clientCallName: 'client.v1.addons.entitlements.create',
+    fullyQualifiedName: 'v1.addons.entitlements.create',
     httpMethod: 'post',
-    httpPath: '/api/v1/addons/{id}/draft',
+    httpPath: '/api/v1/addons/{addonId}/entitlements',
+  },
+  {
+    clientCallName: 'client.v1.addons.entitlements.update',
+    fullyQualifiedName: 'v1.addons.entitlements.update',
+    httpMethod: 'patch',
+    httpPath: '/api/v1/addons/{addonId}/entitlements/{id}',
+  },
+  {
+    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.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.setPricing',
+    fullyQualifiedName: 'v1.plans.setPricing',
+    httpMethod: 'put',
+    httpPath: '/api/v1/plans/{id}/charges',
+  },
+  {
+    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',

package/src/options.ts

@@ -14,6 +14,7 @@ export type CLIOptions = McpOptions & {
 };
 
 export type McpOptions = {
+  includeCodeTool?: boolean | undefined;
   includeDocsTools?: boolean | undefined;
   stainlessApiKey?: string | undefined;
   codeAllowHttpGets?: boolean | undefined;
@@ -92,11 +93,13 @@ 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';
 
   return {
+    ...(includeCodeTool !== undefined && { includeCodeTool }),
     ...(includeDocsTools !== undefined && { includeDocsTools }),
     debug: !!argv.debug,
     stainlessApiKey: argv.stainlessApiKey,
@@ -129,13 +132,19 @@ 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 {
-    codeExecutionMode: defaultOptions.codeExecutionMode,
+    ...(codeTool !== undefined && { includeCodeTool: codeTool }),
     ...(docsTools !== undefined && { includeDocsTools: docsTools }),
+    codeExecutionMode: defaultOptions.codeExecutionMode,
   };
 }

package/src/server.ts

@@ -20,7 +20,7 @@ export const newMcpServer = async (stainlessApiKey: string | undefined) =>
   new McpServer(
     {
       name: 'stigg_typescript_api',
-      version: '0.1.0-alpha.10',
+      version: '0.1.0-alpha.12',
     },
     {
       instructions: await getInstructions(stainlessApiKey),
@@ -156,12 +156,16 @@ export async function initMcpServer(params: {
  * Selects the tools to include in the MCP Server based on the provided options.
  */
 export function selectTools(options?: McpOptions): McpTool[] {
-  const includedTools = [
-    codeTool({
-      blockedMethods: blockedMethodsForCodeTool(options),
-      codeExecutionMode: options?.codeExecutionMode ?? 'stainless-sandbox',
-    }),
-  ];
+  const includedTools = [];
+
+  if (options?.includeCodeTool ?? true) {
+    includedTools.push(
+      codeTool({
+        blockedMethods: blockedMethodsForCodeTool(options),
+        codeExecutionMode: options?.codeExecutionMode ?? 'stainless-sandbox',
+      }),
+    );
+  }
   if (options?.includeDocsTools ?? true) {
     includedTools.push(docsSearchTool);
   }