@lightsparkdev/grid-mcp 0.3.0

package/package.json

@@ -0,0 +1,174 @@
+{
+  "name": "@lightsparkdev/grid-mcp",
+  "version": "0.3.0",
+  "description": "The official MCP Server for the Lightspark Grid API",
+  "author": "Lightspark Grid <support@lightspark.com>",
+  "types": "./index.d.ts",
+  "main": "./index.js",
+  "type": "commonjs",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/lightsparkdev/grid-js-sdk.git",
+    "directory": "packages/mcp-server"
+  },
+  "homepage": "https://github.com/lightsparkdev/grid-js-sdk/tree/main/packages/mcp-server#readme",
+  "license": "Apache-2.0",
+  "private": false,
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "@lightsparkdev/grid": "^0.3.0",
+    "@cloudflare/cabidela": "^0.2.4",
+    "@modelcontextprotocol/sdk": "^1.25.2",
+    "@valtown/deno-http-worker": "^0.0.21",
+    "cookie-parser": "^1.4.6",
+    "cors": "^2.8.5",
+    "express": "^5.1.0",
+    "fuse.js": "^7.1.0",
+    "jq-web": "https://github.com/stainless-api/jq-web/releases/download/v0.8.8/jq-web.tar.gz",
+    "morgan": "^1.10.0",
+    "morgan-body": "^2.6.9",
+    "qs": "^6.14.1",
+    "typescript": "5.8.3",
+    "yargs": "^17.7.2",
+    "zod": "^3.25.20",
+    "zod-to-json-schema": "^3.24.5",
+    "zod-validation-error": "^4.0.1"
+  },
+  "bin": {
+    "mcp-server": "./index.js"
+  },
+  "imports": {
+    "@lightsparkdev/grid-mcp": ".",
+    "@lightsparkdev/grid-mcp/*": "./src/*"
+  },
+  "exports": {
+    ".": {
+      "require": {
+        "types": "./index.d.ts",
+        "default": "./index.js"
+      },
+      "types": "./index.d.mts",
+      "default": "./index.mjs"
+    },
+    "./code-tool-types": {
+      "import": "./code-tool-types.mjs",
+      "require": "./code-tool-types.js"
+    },
+    "./code-tool-types.js": {
+      "default": "./code-tool-types.js"
+    },
+    "./code-tool-types.mjs": {
+      "default": "./code-tool-types.mjs"
+    },
+    "./code-tool": {
+      "import": "./code-tool.mjs",
+      "require": "./code-tool.js"
+    },
+    "./code-tool.js": {
+      "default": "./code-tool.js"
+    },
+    "./code-tool.mjs": {
+      "default": "./code-tool.mjs"
+    },
+    "./docs-search-tool": {
+      "import": "./docs-search-tool.mjs",
+      "require": "./docs-search-tool.js"
+    },
+    "./docs-search-tool.js": {
+      "default": "./docs-search-tool.js"
+    },
+    "./docs-search-tool.mjs": {
+      "default": "./docs-search-tool.mjs"
+    },
+    "./headers": {
+      "import": "./headers.mjs",
+      "require": "./headers.js"
+    },
+    "./headers.js": {
+      "default": "./headers.js"
+    },
+    "./headers.mjs": {
+      "default": "./headers.mjs"
+    },
+    "./http": {
+      "import": "./http.mjs",
+      "require": "./http.js"
+    },
+    "./http.js": {
+      "default": "./http.js"
+    },
+    "./http.mjs": {
+      "default": "./http.mjs"
+    },
+    "./index": {
+      "import": "./index.mjs",
+      "require": "./index.js"
+    },
+    "./index.js": {
+      "default": "./index.js"
+    },
+    "./index.mjs": {
+      "default": "./index.mjs"
+    },
+    "./methods": {
+      "import": "./methods.mjs",
+      "require": "./methods.js"
+    },
+    "./methods.js": {
+      "default": "./methods.js"
+    },
+    "./methods.mjs": {
+      "default": "./methods.mjs"
+    },
+    "./options": {
+      "import": "./options.mjs",
+      "require": "./options.js"
+    },
+    "./options.js": {
+      "default": "./options.js"
+    },
+    "./options.mjs": {
+      "default": "./options.mjs"
+    },
+    "./server": {
+      "import": "./server.mjs",
+      "require": "./server.js"
+    },
+    "./server.js": {
+      "default": "./server.js"
+    },
+    "./server.mjs": {
+      "default": "./server.mjs"
+    },
+    "./stdio": {
+      "import": "./stdio.mjs",
+      "require": "./stdio.js"
+    },
+    "./stdio.js": {
+      "default": "./stdio.js"
+    },
+    "./stdio.mjs": {
+      "default": "./stdio.mjs"
+    },
+    "./types": {
+      "import": "./types.mjs",
+      "require": "./types.js"
+    },
+    "./types.js": {
+      "default": "./types.js"
+    },
+    "./types.mjs": {
+      "default": "./types.mjs"
+    }
+  },
+  "scripts": {
+    "test": "jest",
+    "build": "bash ./build",
+    "format": "prettier --write --cache --cache-strategy metadata . !dist",
+    "tsn": "ts-node -r tsconfig-paths/register",
+    "lint": "eslint --ext ts,js .",
+    "fix": "eslint --fix --ext ts,js ."
+  }
+}

package/server.d.mts

@@ -0,0 +1,30 @@
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { ClientOptions } from '@lightsparkdev/grid';
+import LightsparkGrid from '@lightsparkdev/grid';
+import { McpOptions } from "./options.mjs";
+import { HandlerFunction, McpTool } from "./types.mjs";
+export { McpOptions } from "./options.mjs";
+export { ClientOptions } from '@lightsparkdev/grid';
+export declare const newMcpServer: () => Promise<McpServer>;
+/**
+ * Initializes the provided MCP Server with the given tools and handlers.
+ * If not provided, the default client, tools and handlers will be used.
+ */
+export declare function initMcpServer(params: {
+    server: Server | McpServer;
+    clientOptions?: ClientOptions;
+    mcpOptions?: McpOptions;
+}): Promise<void>;
+/**
+ * Selects the tools to include in the MCP Server based on the provided options.
+ */
+export declare function selectTools(options?: McpOptions): McpTool[];
+/**
+ * Runs the provided handler with the given client and arguments.
+ */
+export declare function executeHandler(handler: HandlerFunction, client: LightsparkGrid, args: Record<string, unknown> | undefined): Promise<import("./types").ToolCallResult>;
+export declare const readEnv: (env: string) => string | undefined;
+export declare const readEnvOrError: (env: string) => string;
+export declare const requireValue: <T>(value: T | undefined, description: string) => T;
+//# sourceMappingURL=server.d.mts.map

package/server.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.mts","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":"OAEO,EAAE,MAAM,EAAE,MAAM,2CAA2C;OAC3D,EAAE,SAAS,EAAE,MAAM,yCAAyC;OAM5D,EAAE,aAAa,EAAE,MAAM,qBAAqB;OAC5C,cAAc,MAAM,qBAAqB;OAGzC,EAAE,UAAU,EAAE;OAEd,EAAE,eAAe,EAAE,OAAO,EAAE;OAE5B,EAAE,UAAU,EAAE;OACd,EAAE,aAAa,EAAE,MAAM,qBAAqB;AAsCnD,eAAO,MAAM,YAAY,0BAUtB,CAAC;AAEJ;;;GAGG;AACH,wBAAsB,aAAa,CAAC,MAAM,EAAE;IAC1C,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,UAAU,CAAC,EAAE,UAAU,CAAC;CACzB,iBAoEA;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,OAAO,CAAC,EAAE,UAAU,GAAG,OAAO,EAAE,CAU3D;AAED;;GAEG;AACH,wBAAsB,cAAc,CAClC,OAAO,EAAE,eAAe,EACxB,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,6CAG1C;AAED,eAAO,MAAM,OAAO,GAAI,KAAK,MAAM,KAAG,MAAM,GAAG,SAO9C,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,KAAK,MAAM,KAAG,MAM5C,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,CAAC,EAAE,OAAO,CAAC,GAAG,SAAS,EAAE,aAAa,MAAM,KAAG,CAK3E,CAAC"}

package/server.d.ts

@@ -0,0 +1,30 @@
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { ClientOptions } from '@lightsparkdev/grid';
+import LightsparkGrid from '@lightsparkdev/grid';
+import { McpOptions } from "./options.js";
+import { HandlerFunction, McpTool } from "./types.js";
+export { McpOptions } from "./options.js";
+export { ClientOptions } from '@lightsparkdev/grid';
+export declare const newMcpServer: () => Promise<McpServer>;
+/**
+ * Initializes the provided MCP Server with the given tools and handlers.
+ * If not provided, the default client, tools and handlers will be used.
+ */
+export declare function initMcpServer(params: {
+    server: Server | McpServer;
+    clientOptions?: ClientOptions;
+    mcpOptions?: McpOptions;
+}): Promise<void>;
+/**
+ * Selects the tools to include in the MCP Server based on the provided options.
+ */
+export declare function selectTools(options?: McpOptions): McpTool[];
+/**
+ * Runs the provided handler with the given client and arguments.
+ */
+export declare function executeHandler(handler: HandlerFunction, client: LightsparkGrid, args: Record<string, unknown> | undefined): Promise<import("./types").ToolCallResult>;
+export declare const readEnv: (env: string) => string | undefined;
+export declare const readEnvOrError: (env: string) => string;
+export declare const requireValue: <T>(value: T | undefined, description: string) => T;
+//# sourceMappingURL=server.d.ts.map

package/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":"OAEO,EAAE,MAAM,EAAE,MAAM,2CAA2C;OAC3D,EAAE,SAAS,EAAE,MAAM,yCAAyC;OAM5D,EAAE,aAAa,EAAE,MAAM,qBAAqB;OAC5C,cAAc,MAAM,qBAAqB;OAGzC,EAAE,UAAU,EAAE;OAEd,EAAE,eAAe,EAAE,OAAO,EAAE;OAE5B,EAAE,UAAU,EAAE;OACd,EAAE,aAAa,EAAE,MAAM,qBAAqB;AAsCnD,eAAO,MAAM,YAAY,0BAUtB,CAAC;AAEJ;;;GAGG;AACH,wBAAsB,aAAa,CAAC,MAAM,EAAE;IAC1C,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,UAAU,CAAC,EAAE,UAAU,CAAC;CACzB,iBAoEA;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,OAAO,CAAC,EAAE,UAAU,GAAG,OAAO,EAAE,CAU3D;AAED;;GAEG;AACH,wBAAsB,cAAc,CAClC,OAAO,EAAE,eAAe,EACxB,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,6CAG1C;AAED,eAAO,MAAM,OAAO,GAAI,KAAK,MAAM,KAAG,MAAM,GAAG,SAO9C,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,KAAK,MAAM,KAAG,MAM5C,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,CAAC,EAAE,OAAO,CAAC,GAAG,SAAS,EAAE,aAAa,MAAM,KAAG,CAK3E,CAAC"}

package/server.js

@@ -0,0 +1,160 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.requireValue = exports.readEnvOrError = exports.readEnv = exports.newMcpServer = void 0;
+exports.initMcpServer = initMcpServer;
+exports.selectTools = selectTools;
+exports.executeHandler = executeHandler;
+const mcp_js_1 = require("@modelcontextprotocol/sdk/server/mcp.js");
+const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
+const grid_1 = __importDefault(require("@lightsparkdev/grid"));
+const code_tool_1 = require("./code-tool.js");
+const docs_search_tool_1 = __importDefault(require("./docs-search-tool.js"));
+const methods_1 = require("./methods.js");
+async function getInstructions() {
+    // This API key is optional; providing it allows the server to fetch instructions for unreleased versions.
+    const stainlessAPIKey = (0, exports.readEnv)('STAINLESS_API_KEY');
+    const response = await fetch((0, exports.readEnv)('CODE_MODE_INSTRUCTIONS_URL') ?? 'https://api.stainless.com/api/ai/instructions/grid', {
+        method: 'GET',
+        headers: { ...(stainlessAPIKey && { Authorization: stainlessAPIKey }) },
+    });
+    let instructions;
+    if (!response.ok) {
+        console.warn('Warning: failed to retrieve MCP server instructions. Proceeding with default instructions...');
+        instructions = `
+      This is the grid MCP server. You will use Code Mode to help the user perform
+      actions. You can use search_docs tool to learn about how to take action with this server. Then,
+      you will write TypeScript code using the execute tool take action. It is CRITICAL that you be
+      thoughtful and deliberate when executing code. Always try to entirely solve the problem in code
+      block: it can be as long as you need to get the job done!
+    `;
+    }
+    instructions ??= (await response.json()).instructions;
+    instructions = `
+    The current time in Unix timestamps is ${Date.now()}.
+
+    ${instructions}
+  `;
+    return instructions;
+}
+const newMcpServer = async () => new mcp_js_1.McpServer({
+    name: 'lightsparkdev_grid_api',
+    version: '0.3.0',
+}, {
+    instructions: await getInstructions(),
+    capabilities: { tools: {}, logging: {} },
+});
+exports.newMcpServer = newMcpServer;
+/**
+ * Initializes the provided MCP Server with the given tools and handlers.
+ * If not provided, the default client, tools and handlers will be used.
+ */
+async function initMcpServer(params) {
+    const server = params.server instanceof mcp_js_1.McpServer ? params.server.server : params.server;
+    const logAtLevel = (level) => (message, ...rest) => {
+        void server.sendLoggingMessage({
+            level,
+            data: { message, rest },
+        });
+    };
+    const logger = {
+        debug: logAtLevel('debug'),
+        info: logAtLevel('info'),
+        warn: logAtLevel('warning'),
+        error: logAtLevel('error'),
+    };
+    let client = new grid_1.default({
+        logger,
+        ...params.clientOptions,
+        defaultHeaders: {
+            ...params.clientOptions?.defaultHeaders,
+            'X-Stainless-MCP': 'true',
+        },
+    });
+    const providedTools = selectTools(params.mcpOptions);
+    const toolMap = Object.fromEntries(providedTools.map((mcpTool) => [mcpTool.tool.name, mcpTool]));
+    server.setRequestHandler(types_js_1.ListToolsRequestSchema, async () => {
+        return {
+            tools: providedTools.map((mcpTool) => mcpTool.tool),
+        };
+    });
+    server.setRequestHandler(types_js_1.CallToolRequestSchema, async (request) => {
+        const { name, arguments: args } = request.params;
+        const mcpTool = toolMap[name];
+        if (!mcpTool) {
+            throw new Error(`Unknown tool: ${name}`);
+        }
+        return executeHandler(mcpTool.handler, client, args);
+    });
+    server.setRequestHandler(types_js_1.SetLevelRequestSchema, async (request) => {
+        const { level } = request.params;
+        switch (level) {
+            case 'debug':
+                client = client.withOptions({ logLevel: 'debug' });
+                break;
+            case 'info':
+                client = client.withOptions({ logLevel: 'info' });
+                break;
+            case 'notice':
+            case 'warning':
+                client = client.withOptions({ logLevel: 'warn' });
+                break;
+            case 'error':
+                client = client.withOptions({ logLevel: 'error' });
+                break;
+            default:
+                client = client.withOptions({ logLevel: 'off' });
+                break;
+        }
+        return {};
+    });
+}
+/**
+ * Selects the tools to include in the MCP Server based on the provided options.
+ */
+function selectTools(options) {
+    const includedTools = [
+        (0, code_tool_1.codeTool)({
+            blockedMethods: (0, methods_1.blockedMethodsForCodeTool)(options),
+        }),
+    ];
+    if (options?.includeDocsTools ?? true) {
+        includedTools.push(docs_search_tool_1.default);
+    }
+    return includedTools;
+}
+/**
+ * Runs the provided handler with the given client and arguments.
+ */
+async function executeHandler(handler, client, args) {
+    return await handler(client, args || {});
+}
+const readEnv = (env) => {
+    if (typeof globalThis.process !== 'undefined') {
+        return globalThis.process.env?.[env]?.trim();
+    }
+    else if (typeof globalThis.Deno !== 'undefined') {
+        return globalThis.Deno.env?.get?.(env)?.trim();
+    }
+    return;
+};
+exports.readEnv = readEnv;
+const readEnvOrError = (env) => {
+    let envValue = (0, exports.readEnv)(env);
+    if (envValue === undefined) {
+        throw new Error(`Environment variable ${env} is not set`);
+    }
+    return envValue;
+};
+exports.readEnvOrError = readEnvOrError;
+const requireValue = (value, description) => {
+    if (value === undefined) {
+        throw new Error(`Missing required value: ${description}`);
+    }
+    return value;
+};
+exports.requireValue = requireValue;
+//# sourceMappingURL=server.js.map

package/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;;AAwEtF,sCAwEC;AAKD,kCAUC;AAKD,wCAMC;AAvKD,oEAAoE;AACpE,iEAI4C;AAE5C,+DAAiD;AACjD,8CAAuC;AACvC,6EAAgD;AAEhD,0CAAsD;AAMtD,KAAK,UAAU,eAAe;IAC5B,0GAA0G;IAC1G,MAAM,eAAe,GAAG,IAAA,eAAO,EAAC,mBAAmB,CAAC,CAAC;IACrD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,IAAA,eAAO,EAAC,4BAA4B,CAAC,IAAI,oDAAoD,EAC7F;QACE,MAAM,EAAE,KAAK;QACb,OAAO,EAAE,EAAE,GAAG,CAAC,eAAe,IAAI,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,EAAE;KACxE,CACF,CAAC;IAEF,IAAI,YAAgC,CAAC;IACrC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,OAAO,CAAC,IAAI,CACV,8FAA8F,CAC/F,CAAC;QAEF,YAAY,GAAG;;;;;;KAMd,CAAC;IACJ,CAAC;IAED,YAAY,KAAM,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA8B,CAAC,YAAY,CAAC;IACpF,YAAY,GAAG;6CAC4B,IAAI,CAAC,GAAG,EAAE;;MAEjD,YAAY;GACf,CAAC;IAEF,OAAO,YAAY,CAAC;AACtB,CAAC;AAEM,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE,CACrC,IAAI,kBAAS,CACX;IACE,IAAI,EAAE,wBAAwB;IAC9B,OAAO,EAAE,OAAO;CACjB,EACD;IACE,YAAY,EAAE,MAAM,eAAe,EAAE;IACrC,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;CACzC,CACF,CAAC;AAVS,QAAA,YAAY,gBAUrB;AAEJ;;;GAGG;AACI,KAAK,UAAU,aAAa,CAAC,MAInC;IACC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,YAAY,kBAAS,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;IAEzF,MAAM,UAAU,GACd,CAAC,KAA6C,EAAE,EAAE,CAClD,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QACtC,KAAK,MAAM,CAAC,kBAAkB,CAAC;YAC7B,KAAK;YACL,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SACxB,CAAC,CAAC;IACL,CAAC,CAAC;IACJ,MAAM,MAAM,GAAG;QACb,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC;QAC1B,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC;QACxB,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC;QAC3B,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC;KAC3B,CAAC;IAEF,IAAI,MAAM,GAAG,IAAI,cAAc,CAAC;QAC9B,MAAM;QACN,GAAG,MAAM,CAAC,aAAa;QACvB,cAAc,EAAE;YACd,GAAG,MAAM,CAAC,aAAa,EAAE,cAAc;YACvC,iBAAiB,EAAE,MAAM;SAC1B;KACF,CAAC,CAAC;IAEH,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAEjG,MAAM,CAAC,iBAAiB,CAAC,iCAAsB,EAAE,KAAK,IAAI,EAAE;QAC1D,OAAO;YACL,KAAK,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;SACpD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,iBAAiB,CAAC,gCAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QACjD,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,OAAO,cAAc,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,iBAAiB,CAAC,gCAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QACjC,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,OAAO;gBACV,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBACnD,MAAM;YACR,KAAK,MAAM;gBACT,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;gBAClD,MAAM;YACR,KAAK,QAAQ,CAAC;YACd,KAAK,SAAS;gBACZ,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;gBAClD,MAAM;YACR,KAAK,OAAO;gBACV,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBACnD,MAAM;YACR;gBACE,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;gBACjD,MAAM;QACV,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAgB,WAAW,CAAC,OAAoB;IAC9C,MAAM,aAAa,GAAG;QACpB,IAAA,oBAAQ,EAAC;YACP,cAAc,EAAE,IAAA,mCAAyB,EAAC,OAAO,CAAC;SACnD,CAAC;KACH,CAAC;IACF,IAAI,OAAO,EAAE,gBAAgB,IAAI,IAAI,EAAE,CAAC;QACtC,aAAa,CAAC,IAAI,CAAC,0BAAc,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,cAAc,CAClC,OAAwB,EACxB,MAAsB,EACtB,IAAyC;IAEzC,OAAO,MAAM,OAAO,CAAC,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;AAC3C,CAAC;AAEM,MAAM,OAAO,GAAG,CAAC,GAAW,EAAsB,EAAE;IACzD,IAAI,OAAQ,UAAkB,CAAC,OAAO,KAAK,WAAW,EAAE,CAAC;QACvD,OAAQ,UAAkB,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC;IACxD,CAAC;SAAM,IAAI,OAAQ,UAAkB,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QAC3D,OAAQ,UAAkB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC;IAC1D,CAAC;IACD,OAAO;AACT,CAAC,CAAC;AAPW,QAAA,OAAO,WAOlB;AAEK,MAAM,cAAc,GAAG,CAAC,GAAW,EAAU,EAAE;IACpD,IAAI,QAAQ,GAAG,IAAA,eAAO,EAAC,GAAG,CAAC,CAAC;IAC5B,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,aAAa,CAAC,CAAC;IAC5D,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AANW,QAAA,cAAc,kBAMzB;AAEK,MAAM,YAAY,GAAG,CAAI,KAAoB,EAAE,WAAmB,EAAK,EAAE;IAC9E,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,2BAA2B,WAAW,EAAE,CAAC,CAAC;IAC5D,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AALW,QAAA,YAAY,gBAKvB"}

package/server.mjs

@@ -0,0 +1,147 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { CallToolRequestSchema, ListToolsRequestSchema, SetLevelRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
+import LightsparkGrid from '@lightsparkdev/grid';
+import { codeTool } from "./code-tool.mjs";
+import docsSearchTool from "./docs-search-tool.mjs";
+import { blockedMethodsForCodeTool } from "./methods.mjs";
+async function getInstructions() {
+    // This API key is optional; providing it allows the server to fetch instructions for unreleased versions.
+    const stainlessAPIKey = readEnv('STAINLESS_API_KEY');
+    const response = await fetch(readEnv('CODE_MODE_INSTRUCTIONS_URL') ?? 'https://api.stainless.com/api/ai/instructions/grid', {
+        method: 'GET',
+        headers: { ...(stainlessAPIKey && { Authorization: stainlessAPIKey }) },
+    });
+    let instructions;
+    if (!response.ok) {
+        console.warn('Warning: failed to retrieve MCP server instructions. Proceeding with default instructions...');
+        instructions = `
+      This is the grid MCP server. You will use Code Mode to help the user perform
+      actions. You can use search_docs tool to learn about how to take action with this server. Then,
+      you will write TypeScript code using the execute tool take action. It is CRITICAL that you be
+      thoughtful and deliberate when executing code. Always try to entirely solve the problem in code
+      block: it can be as long as you need to get the job done!
+    `;
+    }
+    instructions ??= (await response.json()).instructions;
+    instructions = `
+    The current time in Unix timestamps is ${Date.now()}.
+
+    ${instructions}
+  `;
+    return instructions;
+}
+export const newMcpServer = async () => new McpServer({
+    name: 'lightsparkdev_grid_api',
+    version: '0.3.0',
+}, {
+    instructions: await getInstructions(),
+    capabilities: { tools: {}, logging: {} },
+});
+/**
+ * Initializes the provided MCP Server with the given tools and handlers.
+ * If not provided, the default client, tools and handlers will be used.
+ */
+export async function initMcpServer(params) {
+    const server = params.server instanceof McpServer ? params.server.server : params.server;
+    const logAtLevel = (level) => (message, ...rest) => {
+        void server.sendLoggingMessage({
+            level,
+            data: { message, rest },
+        });
+    };
+    const logger = {
+        debug: logAtLevel('debug'),
+        info: logAtLevel('info'),
+        warn: logAtLevel('warning'),
+        error: logAtLevel('error'),
+    };
+    let client = new LightsparkGrid({
+        logger,
+        ...params.clientOptions,
+        defaultHeaders: {
+            ...params.clientOptions?.defaultHeaders,
+            'X-Stainless-MCP': 'true',
+        },
+    });
+    const providedTools = selectTools(params.mcpOptions);
+    const toolMap = Object.fromEntries(providedTools.map((mcpTool) => [mcpTool.tool.name, mcpTool]));
+    server.setRequestHandler(ListToolsRequestSchema, async () => {
+        return {
+            tools: providedTools.map((mcpTool) => mcpTool.tool),
+        };
+    });
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        const { name, arguments: args } = request.params;
+        const mcpTool = toolMap[name];
+        if (!mcpTool) {
+            throw new Error(`Unknown tool: ${name}`);
+        }
+        return executeHandler(mcpTool.handler, client, args);
+    });
+    server.setRequestHandler(SetLevelRequestSchema, async (request) => {
+        const { level } = request.params;
+        switch (level) {
+            case 'debug':
+                client = client.withOptions({ logLevel: 'debug' });
+                break;
+            case 'info':
+                client = client.withOptions({ logLevel: 'info' });
+                break;
+            case 'notice':
+            case 'warning':
+                client = client.withOptions({ logLevel: 'warn' });
+                break;
+            case 'error':
+                client = client.withOptions({ logLevel: 'error' });
+                break;
+            default:
+                client = client.withOptions({ logLevel: 'off' });
+                break;
+        }
+        return {};
+    });
+}
+/**
+ * Selects the tools to include in the MCP Server based on the provided options.
+ */
+export function selectTools(options) {
+    const includedTools = [
+        codeTool({
+            blockedMethods: blockedMethodsForCodeTool(options),
+        }),
+    ];
+    if (options?.includeDocsTools ?? true) {
+        includedTools.push(docsSearchTool);
+    }
+    return includedTools;
+}
+/**
+ * Runs the provided handler with the given client and arguments.
+ */
+export async function executeHandler(handler, client, args) {
+    return await handler(client, args || {});
+}
+export const readEnv = (env) => {
+    if (typeof globalThis.process !== 'undefined') {
+        return globalThis.process.env?.[env]?.trim();
+    }
+    else if (typeof globalThis.Deno !== 'undefined') {
+        return globalThis.Deno.env?.get?.(env)?.trim();
+    }
+    return;
+};
+export const readEnvOrError = (env) => {
+    let envValue = readEnv(env);
+    if (envValue === undefined) {
+        throw new Error(`Environment variable ${env} is not set`);
+    }
+    return envValue;
+};
+export const requireValue = (value, description) => {
+    if (value === undefined) {
+        throw new Error(`Missing required value: ${description}`);
+    }
+    return value;
+};
+//# sourceMappingURL=server.mjs.map

package/server.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.mjs","sourceRoot":"","sources":["src/server.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAG/E,EAAE,SAAS,EAAE,MAAM,yCAAyC;OAC5D,EACL,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,GACtB,MAAM,oCAAoC;OAEpC,cAAc,MAAM,qBAAqB;OACzC,EAAE,QAAQ,EAAE;OACZ,cAAc;OAEd,EAAE,yBAAyB,EAAE;AAMpC,KAAK,UAAU,eAAe;IAC5B,0GAA0G;IAC1G,MAAM,eAAe,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACrD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,OAAO,CAAC,4BAA4B,CAAC,IAAI,oDAAoD,EAC7F;QACE,MAAM,EAAE,KAAK;QACb,OAAO,EAAE,EAAE,GAAG,CAAC,eAAe,IAAI,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,EAAE;KACxE,CACF,CAAC;IAEF,IAAI,YAAgC,CAAC;IACrC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,OAAO,CAAC,IAAI,CACV,8FAA8F,CAC/F,CAAC;QAEF,YAAY,GAAG;;;;;;KAMd,CAAC;IACJ,CAAC;IAED,YAAY,KAAM,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA8B,CAAC,YAAY,CAAC;IACpF,YAAY,GAAG;6CAC4B,IAAI,CAAC,GAAG,EAAE;;MAEjD,YAAY;GACf,CAAC;IAEF,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE,CACrC,IAAI,SAAS,CACX;IACE,IAAI,EAAE,wBAAwB;IAC9B,OAAO,EAAE,OAAO;CACjB,EACD;IACE,YAAY,EAAE,MAAM,eAAe,EAAE;IACrC,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;CACzC,CACF,CAAC;AAEJ;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,MAInC;IACC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,YAAY,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;IAEzF,MAAM,UAAU,GACd,CAAC,KAA6C,EAAE,EAAE,CAClD,CAAC,OAAe,EAAE,GAAG,IAAe,EAAE,EAAE;QACtC,KAAK,MAAM,CAAC,kBAAkB,CAAC;YAC7B,KAAK;YACL,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SACxB,CAAC,CAAC;IACL,CAAC,CAAC;IACJ,MAAM,MAAM,GAAG;QACb,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC;QAC1B,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC;QACxB,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC;QAC3B,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC;KAC3B,CAAC;IAEF,IAAI,MAAM,GAAG,IAAI,cAAc,CAAC;QAC9B,MAAM;QACN,GAAG,MAAM,CAAC,aAAa;QACvB,cAAc,EAAE;YACd,GAAG,MAAM,CAAC,aAAa,EAAE,cAAc;YACvC,iBAAiB,EAAE,MAAM;SAC1B;KACF,CAAC,CAAC;IAEH,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAEjG,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QAC1D,OAAO;YACL,KAAK,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;SACpD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QACjD,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,OAAO,cAAc,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QACjC,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,OAAO;gBACV,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBACnD,MAAM;YACR,KAAK,MAAM;gBACT,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;gBAClD,MAAM;YACR,KAAK,QAAQ,CAAC;YACd,KAAK,SAAS;gBACZ,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;gBAClD,MAAM;YACR,KAAK,OAAO;gBACV,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBACnD,MAAM;YACR;gBACE,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;gBACjD,MAAM;QACV,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,OAAoB;IAC9C,MAAM,aAAa,GAAG;QACpB,QAAQ,CAAC;YACP,cAAc,EAAE,yBAAyB,CAAC,OAAO,CAAC;SACnD,CAAC;KACH,CAAC;IACF,IAAI,OAAO,EAAE,gBAAgB,IAAI,IAAI,EAAE,CAAC;QACtC,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAwB,EACxB,MAAsB,EACtB,IAAyC;IAEzC,OAAO,MAAM,OAAO,CAAC,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,CAAC,MAAM,OAAO,GAAG,CAAC,GAAW,EAAsB,EAAE;IACzD,IAAI,OAAQ,UAAkB,CAAC,OAAO,KAAK,WAAW,EAAE,CAAC;QACvD,OAAQ,UAAkB,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC;IACxD,CAAC;SAAM,IAAI,OAAQ,UAAkB,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QAC3D,OAAQ,UAAkB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC;IAC1D,CAAC;IACD,OAAO;AACT,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,GAAW,EAAU,EAAE;IACpD,IAAI,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;IAC5B,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,aAAa,CAAC,CAAC;IAC5D,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,YAAY,GAAG,CAAI,KAAoB,EAAE,WAAmB,EAAK,EAAE;IAC9E,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,2BAA2B,WAAW,EAAE,CAAC,CAAC;IAC5D,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC"}

package/src/code-tool-types.ts

@@ -0,0 +1,16 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { ClientOptions } from '@lightsparkdev/grid';
+
+export type WorkerInput = {
+  project_name: string;
+  code: string;
+  client_opts: ClientOptions;
+  intent?: string | undefined;
+};
+export type WorkerOutput = {
+  is_error: boolean;
+  result: unknown | null;
+  log_lines: string[];
+  err_lines: string[];
+};