@beeper/desktop-mcp 4.2.2 → 4.6.0

package/code-tool-worker.js

@@ -4,40 +4,242 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+const node_path_1 = __importDefault(require("node:path"));
 const node_util_1 = __importDefault(require("node:util"));
+const fuse_js_1 = __importDefault(require("fuse.js"));
+const typescript_1 = __importDefault(require("typescript"));
 const desktop_api_1 = require("@beeper/desktop-api");
+function getRunFunctionSource(code) {
+    const sourceFile = typescript_1.default.createSourceFile('code.ts', code, typescript_1.default.ScriptTarget.Latest, true);
+    const printer = typescript_1.default.createPrinter();
+    for (const statement of sourceFile.statements) {
+        // Check for top-level function declarations
+        if (typescript_1.default.isFunctionDeclaration(statement)) {
+            if (statement.name?.text === 'run') {
+                return {
+                    type: 'declaration',
+                    client: statement.parameters[0]?.name.getText(),
+                    code: printer.printNode(typescript_1.default.EmitHint.Unspecified, statement.body, sourceFile),
+                };
+            }
+        }
+        // Check for variable declarations: const run = () => {} or const run = function() {}
+        if (typescript_1.default.isVariableStatement(statement)) {
+            for (const declaration of statement.declarationList.declarations) {
+                if (typescript_1.default.isIdentifier(declaration.name) &&
+                    declaration.name.text === 'run' &&
+                    // Check if it's initialized with a function
+                    declaration.initializer &&
+                    (typescript_1.default.isFunctionExpression(declaration.initializer) || typescript_1.default.isArrowFunction(declaration.initializer))) {
+                    return {
+                        type: 'expression',
+                        client: declaration.initializer.parameters[0]?.name.getText(),
+                        code: printer.printNode(typescript_1.default.EmitHint.Unspecified, declaration.initializer, sourceFile),
+                    };
+                }
+            }
+        }
+    }
+    return null;
+}
+function getTSDiagnostics(code) {
+    const functionSource = getRunFunctionSource(code);
+    const codeWithImport = [
+        'import { BeeperDesktop } from "@beeper/desktop-api";',
+        functionSource.type === 'declaration' ?
+            `async function run(${functionSource.client}: BeeperDesktop)`
+            : `const run: (${functionSource.client}: BeeperDesktop) => Promise<unknown> =`,
+        functionSource.code,
+    ].join('\n');
+    const sourcePath = node_path_1.default.resolve('code.ts');
+    const ast = typescript_1.default.createSourceFile(sourcePath, codeWithImport, typescript_1.default.ScriptTarget.Latest, true);
+    const options = typescript_1.default.getDefaultCompilerOptions();
+    options.target = typescript_1.default.ScriptTarget.Latest;
+    options.module = typescript_1.default.ModuleKind.NodeNext;
+    options.moduleResolution = typescript_1.default.ModuleResolutionKind.NodeNext;
+    const host = typescript_1.default.createCompilerHost(options, true);
+    const newHost = {
+        ...host,
+        getSourceFile: (...args) => {
+            if (node_path_1.default.resolve(args[0]) === sourcePath) {
+                return ast;
+            }
+            return host.getSourceFile(...args);
+        },
+        readFile: (...args) => {
+            if (node_path_1.default.resolve(args[0]) === sourcePath) {
+                return codeWithImport;
+            }
+            return host.readFile(...args);
+        },
+        fileExists: (...args) => {
+            if (node_path_1.default.resolve(args[0]) === sourcePath) {
+                return true;
+            }
+            return host.fileExists(...args);
+        },
+    };
+    const program = typescript_1.default.createProgram({
+        options,
+        rootNames: [sourcePath],
+        host: newHost,
+    });
+    const diagnostics = typescript_1.default.getPreEmitDiagnostics(program, ast);
+    return diagnostics.map((d) => {
+        const message = typescript_1.default.flattenDiagnosticMessageText(d.messageText, '\n');
+        if (!d.file || !d.start)
+            return `- ${message}`;
+        const { line: lineNumber } = typescript_1.default.getLineAndCharacterOfPosition(d.file, d.start);
+        const line = codeWithImport.split('\n').at(lineNumber)?.trim();
+        return line ? `- ${message}\n    ${line}` : `- ${message}`;
+    });
+}
+const fuse = new fuse_js_1.default([
+    'client.focus',
+    'client.search',
+    'client.accounts.list',
+    'client.accounts.contacts.list',
+    'client.accounts.contacts.search',
+    'client.chats.archive',
+    'client.chats.create',
+    'client.chats.list',
+    'client.chats.retrieve',
+    'client.chats.search',
+    'client.chats.reminders.create',
+    'client.chats.reminders.delete',
+    'client.chats.messages.reactions.add',
+    'client.chats.messages.reactions.delete',
+    'client.messages.list',
+    'client.messages.search',
+    'client.messages.send',
+    'client.messages.update',
+    'client.assets.download',
+    'client.assets.serve',
+    'client.assets.upload',
+    'client.assets.uploadBase64',
+    'client.info.retrieve',
+], { threshold: 1, shouldSort: true });
+function getMethodSuggestions(fullyQualifiedMethodName) {
+    return fuse
+        .search(fullyQualifiedMethodName)
+        .map(({ item }) => item)
+        .slice(0, 5);
+}
+const proxyToObj = new WeakMap();
+const objToProxy = new WeakMap();
+function makeSdkProxy(obj, { path, isBelievedBad = false }) {
+    let proxy = objToProxy.get(obj);
+    if (!proxy) {
+        proxy = new Proxy(obj, {
+            get(target, prop, receiver) {
+                const propPath = [...path, String(prop)];
+                const value = Reflect.get(target, prop, receiver);
+                if (isBelievedBad || (!(prop in target) && value === undefined)) {
+                    // If we're accessing a path that doesn't exist, it will probably eventually error.
+                    // Let's proxy it and mark it bad so that we can control the error message.
+                    // We proxy an empty class so that an invocation or construction attempt is possible.
+                    return makeSdkProxy(class {
+                    }, { path: propPath, isBelievedBad: true });
+                }
+                if (value !== null && (typeof value === 'object' || typeof value === 'function')) {
+                    return makeSdkProxy(value, { path: propPath, isBelievedBad });
+                }
+                return value;
+            },
+            apply(target, thisArg, args) {
+                if (isBelievedBad || typeof target !== 'function') {
+                    const fullyQualifiedMethodName = path.join('.');
+                    const suggestions = getMethodSuggestions(fullyQualifiedMethodName);
+                    throw new Error(`${fullyQualifiedMethodName} is not a function. Did you mean: ${suggestions.join(', ')}`);
+                }
+                return Reflect.apply(target, proxyToObj.get(thisArg) ?? thisArg, args);
+            },
+            construct(target, args, newTarget) {
+                if (isBelievedBad || typeof target !== 'function') {
+                    const fullyQualifiedMethodName = path.join('.');
+                    const suggestions = getMethodSuggestions(fullyQualifiedMethodName);
+                    throw new Error(`${fullyQualifiedMethodName} is not a constructor. Did you mean: ${suggestions.join(', ')}`);
+                }
+                return Reflect.construct(target, args, newTarget);
+            },
+        });
+        objToProxy.set(obj, proxy);
+        proxyToObj.set(proxy, obj);
+    }
+    return proxy;
+}
+function parseError(code, error) {
+    if (!(error instanceof Error))
+        return;
+    const message = error.name ? `${error.name}: ${error.message}` : error.message;
+    try {
+        // Deno uses V8; the first "<anonymous>:LINE:COLUMN" is the top of stack.
+        const lineNumber = error.stack?.match(/<anonymous>:([0-9]+):[0-9]+/)?.[1];
+        // -1 for the zero-based indexing
+        const line = lineNumber &&
+            code
+                .split('\n')
+                .at(parseInt(lineNumber, 10) - 1)
+                ?.trim();
+        return line ? `${message}\n  at line ${lineNumber}\n    ${line}` : message;
+    }
+    catch {
+        return message;
+    }
+}
 const fetch = async (req) => {
     const { opts, code } = (await req.json());
+    const runFunctionSource = code ? getRunFunctionSource(code) : null;
+    if (!runFunctionSource) {
+        const message = code ?
+            'The code is missing a top-level `run` function.'
+            : 'The code argument is missing. Provide one containing a top-level `run` function.';
+        return Response.json({
+            is_error: true,
+            result: `${message} Write code within this template:\n\n\`\`\`\nasync function run(client) {\n  // Fill this out\n}\n\`\`\``,
+            log_lines: [],
+            err_lines: [],
+        }, { status: 400, statusText: 'Code execution error' });
+    }
+    const diagnostics = getTSDiagnostics(code);
+    if (diagnostics.length > 0) {
+        return Response.json({
+            is_error: true,
+            result: `The code contains TypeScript diagnostics:\n${diagnostics.join('\n')}`,
+            log_lines: [],
+            err_lines: [],
+        }, { status: 400, statusText: 'Code execution error' });
+    }
     const client = new desktop_api_1.BeeperDesktop({
         ...opts,
     });
-    const logLines = [];
-    const errLines = [];
+    const log_lines = [];
+    const err_lines = [];
     const console = {
         log: (...args) => {
-            logLines.push(node_util_1.default.format(...args));
+            log_lines.push(node_util_1.default.format(...args));
         },
         error: (...args) => {
-            errLines.push(node_util_1.default.format(...args));
+            err_lines.push(node_util_1.default.format(...args));
         },
     };
     try {
         let run_ = async (client) => { };
-        eval(`
-      ${code}
-      run_ = run;
-    `);
-        const result = await run_(client);
+        eval(`${code}\nrun_ = run;`);
+        const result = await run_(makeSdkProxy(client, { path: ['client'] }));
         return Response.json({
+            is_error: false,
             result,
-            logLines,
-            errLines,
+            log_lines,
+            err_lines,
         });
     }
     catch (e) {
-        const message = e instanceof Error ? e.message : undefined;
         return Response.json({
-            message,
+            is_error: true,
+            result: parseError(code, e),
+            log_lines,
+            err_lines,
         }, { status: 400, statusText: 'Code execution error' });
     }
 };

package/code-tool-worker.js.map

@@ -1 +1 @@
-{"version":3,"file":"code-tool-worker.js","sourceRoot":"","sources":["src/code-tool-worker.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;AAEtF,0DAA6B;AAE7B,qDAAoD;AAEpD,MAAM,KAAK,GAAG,KAAK,EAAE,GAAY,EAAqB,EAAE;IACtD,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAgB,CAAC;IACzD,MAAM,MAAM,GAAG,IAAI,2BAAa,CAAC;QAC/B,GAAG,IAAI;KACR,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,OAAO,GAAG;QACd,GAAG,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE;YAC1B,QAAQ,CAAC,IAAI,CAAC,mBAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;QACD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE;YAC5B,QAAQ,CAAC,IAAI,CAAC,mBAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;KACF,CAAC;IACF,IAAI,CAAC;QACH,IAAI,IAAI,GAAG,KAAK,EAAE,MAAW,EAAE,EAAE,GAAE,CAAC,CAAC;QACrC,IAAI,CAAC;QACD,IAAI;;KAEP,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,CAAC;QAClC,OAAO,QAAQ,CAAC,IAAI,CAAC;YACnB,MAAM;YACN,QAAQ;YACR,QAAQ;SACe,CAAC,CAAC;IAC7B,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3D,OAAO,QAAQ,CAAC,IAAI,CAClB;YACE,OAAO;SACc,EACvB,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,sBAAsB,EAAE,CACpD,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAEF,kBAAe,EAAE,KAAK,EAAE,CAAC"}
+{"version":3,"file":"code-tool-worker.js","sourceRoot":"","sources":["src/code-tool-worker.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;;;AAEtF,0DAA6B;AAC7B,0DAA6B;AAC7B,sDAA2B;AAC3B,4DAA4B;AAE5B,qDAAmE;AAEnE,SAAS,oBAAoB,CAAC,IAAY;IAKxC,MAAM,UAAU,GAAG,oBAAE,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,EAAE,oBAAE,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IACtF,MAAM,OAAO,GAAG,oBAAE,CAAC,aAAa,EAAE,CAAC;IAEnC,KAAK,MAAM,SAAS,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;QAC9C,4CAA4C;QAC5C,IAAI,oBAAE,CAAC,qBAAqB,CAAC,SAAS,CAAC,EAAE,CAAC;YACxC,IAAI,SAAS,CAAC,IAAI,EAAE,IAAI,KAAK,KAAK,EAAE,CAAC;gBACnC,OAAO;oBACL,IAAI,EAAE,aAAa;oBACnB,MAAM,EAAE,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE;oBAC/C,IAAI,EAAE,OAAO,CAAC,SAAS,CAAC,oBAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,SAAS,CAAC,IAAK,EAAE,UAAU,CAAC;iBAC9E,CAAC;YACJ,CAAC;QACH,CAAC;QAED,qFAAqF;QACrF,IAAI,oBAAE,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,CAAC;YACtC,KAAK,MAAM,WAAW,IAAI,SAAS,CAAC,eAAe,CAAC,YAAY,EAAE,CAAC;gBACjE,IACE,oBAAE,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC;oBACjC,WAAW,CAAC,IAAI,CAAC,IAAI,KAAK,KAAK;oBAC/B,4CAA4C;oBAC5C,WAAW,CAAC,WAAW;oBACvB,CAAC,oBAAE,CAAC,oBAAoB,CAAC,WAAW,CAAC,WAAW,CAAC,IAAI,oBAAE,CAAC,eAAe,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,EACjG,CAAC;oBACD,OAAO;wBACL,IAAI,EAAE,YAAY;wBAClB,MAAM,EAAE,WAAW,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE;wBAC7D,IAAI,EAAE,OAAO,CAAC,SAAS,CAAC,oBAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC,WAAW,EAAE,UAAU,CAAC;qBACtF,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAY;IACpC,MAAM,cAAc,GAAG,oBAAoB,CAAC,IAAI,CAAE,CAAC;IACnD,MAAM,cAAc,GAAG;QACrB,sDAAsD;QACtD,cAAc,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;YACrC,sBAAsB,cAAc,CAAC,MAAM,kBAAkB;YAC/D,CAAC,CAAC,eAAe,cAAc,CAAC,MAAM,wCAAwC;QAC9E,cAAc,CAAC,IAAI;KACpB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACb,MAAM,UAAU,GAAG,mBAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3C,MAAM,GAAG,GAAG,oBAAE,CAAC,gBAAgB,CAAC,UAAU,EAAE,cAAc,EAAE,oBAAE,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC1F,MAAM,OAAO,GAAG,oBAAE,CAAC,yBAAyB,EAAE,CAAC;IAC/C,OAAO,CAAC,MAAM,GAAG,oBAAE,CAAC,YAAY,CAAC,MAAM,CAAC;IACxC,OAAO,CAAC,MAAM,GAAG,oBAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;IACxC,OAAO,CAAC,gBAAgB,GAAG,oBAAE,CAAC,oBAAoB,CAAC,QAAQ,CAAC;IAC5D,MAAM,IAAI,GAAG,oBAAE,CAAC,kBAAkB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAClD,MAAM,OAAO,GAAgB;QAC3B,GAAG,IAAI;QACP,aAAa,EAAE,CAAC,GAAG,IAAI,EAAE,EAAE;YACzB,IAAI,mBAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,EAAE,CAAC;gBACzC,OAAO,GAAG,CAAC;YACb,CAAC;YACD,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,CAAC;QACrC,CAAC;QACD,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,EAAE;YACpB,IAAI,mBAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,EAAE,CAAC;gBACzC,OAAO,cAAc,CAAC;YACxB,CAAC;YACD,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;QAChC,CAAC;QACD,UAAU,EAAE,CAAC,GAAG,IAAI,EAAE,EAAE;YACtB,IAAI,mBAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,EAAE,CAAC;gBACzC,OAAO,IAAI,CAAC;YACd,CAAC;YACD,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,CAAC;QAClC,CAAC;KACF,CAAC;IACF,MAAM,OAAO,GAAG,oBAAE,CAAC,aAAa,CAAC;QAC/B,OAAO;QACP,SAAS,EAAE,CAAC,UAAU,CAAC;QACvB,IAAI,EAAE,OAAO;KACd,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,oBAAE,CAAC,qBAAqB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IAC3D,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC3B,MAAM,OAAO,GAAG,oBAAE,CAAC,4BAA4B,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QACrE,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK;YAAE,OAAO,KAAK,OAAO,EAAE,CAAC;QAC/C,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,oBAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;QAC/E,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC;QAC/D,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,OAAO,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,OAAO,EAAE,CAAC;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,IAAI,GAAG,IAAI,iBAAI,CACnB;IACE,cAAc;IACd,eAAe;IACf,sBAAsB;IACtB,+BAA+B;IAC/B,iCAAiC;IACjC,sBAAsB;IACtB,qBAAqB;IACrB,mBAAmB;IACnB,uBAAuB;IACvB,qBAAqB;IACrB,+BAA+B;IAC/B,+BAA+B;IAC/B,qCAAqC;IACrC,wCAAwC;IACxC,sBAAsB;IACtB,wBAAwB;IACxB,sBAAsB;IACtB,wBAAwB;IACxB,wBAAwB;IACxB,qBAAqB;IACrB,sBAAsB;IACtB,4BAA4B;IAC5B,sBAAsB;CACvB,EACD,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CACnC,CAAC;AAEF,SAAS,oBAAoB,CAAC,wBAAgC;IAC5D,OAAO,IAAI;SACR,MAAM,CAAC,wBAAwB,CAAC;SAChC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC;SACvB,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,GAAG,IAAI,OAAO,EAAY,CAAC;AAC3C,MAAM,UAAU,GAAG,IAAI,OAAO,EAAY,CAAC;AAO3C,SAAS,YAAY,CAAmB,GAAM,EAAE,EAAE,IAAI,EAAE,aAAa,GAAG,KAAK,EAAqB;IAChG,IAAI,KAAK,GAAM,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAEnC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,KAAK,GAAG,IAAI,KAAK,CAAC,GAAG,EAAE;YACrB,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ;gBACxB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;gBACzC,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;gBAElD,IAAI,aAAa,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,KAAK,KAAK,SAAS,CAAC,EAAE,CAAC;oBAChE,mFAAmF;oBACnF,2EAA2E;oBAC3E,qFAAqF;oBACrF,OAAO,YAAY,CAAC;qBAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzE,CAAC;gBAED,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,UAAU,CAAC,EAAE,CAAC;oBACjF,OAAO,YAAY,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC,CAAC;gBAChE,CAAC;gBAED,OAAO,KAAK,CAAC;YACf,CAAC;YAED,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI;gBACzB,IAAI,aAAa,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;oBAClD,MAAM,wBAAwB,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAChD,MAAM,WAAW,GAAG,oBAAoB,CAAC,wBAAwB,CAAC,CAAC;oBACnE,MAAM,IAAI,KAAK,CACb,GAAG,wBAAwB,qCAAqC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACzF,CAAC;gBACJ,CAAC;gBAED,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,OAAO,EAAE,IAAI,CAAC,CAAC;YACzE,CAAC;YAED,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS;gBAC/B,IAAI,aAAa,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;oBAClD,MAAM,wBAAwB,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAChD,MAAM,WAAW,GAAG,oBAAoB,CAAC,wBAAwB,CAAC,CAAC;oBACnE,MAAM,IAAI,KAAK,CACb,GAAG,wBAAwB,wCAAwC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC5F,CAAC;gBACJ,CAAC;gBAED,OAAO,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YACpD,CAAC;SACF,CAAC,CAAC;QAEH,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC3B,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,UAAU,CAAC,IAAY,EAAE,KAAc;IAC9C,IAAI,CAAC,CAAC,KAAK,YAAY,KAAK,CAAC;QAAE,OAAO;IACtC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC;IAC/E,IAAI,CAAC;QACH,yEAAyE;QACzE,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1E,iCAAiC;QACjC,MAAM,IAAI,GACR,UAAU;YACV,IAAI;iBACD,KAAK,CAAC,IAAI,CAAC;iBACX,EAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;gBACjC,EAAE,IAAI,EAAE,CAAC;QACb,OAAO,IAAI,CAAC,CAAC,CAAC,GAAG,OAAO,eAAe,UAAU,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;IAC7E,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,OAAO,CAAC;IACjB,CAAC;AACH,CAAC;AAED,MAAM,KAAK,GAAG,KAAK,EAAE,GAAY,EAAqB,EAAE;IACtD,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA0C,CAAC;IAEnF,MAAM,iBAAiB,GAAG,IAAI,CAAC,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACnE,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,MAAM,OAAO,GACX,IAAI,CAAC,CAAC;YACJ,iDAAiD;YACnD,CAAC,CAAC,kFAAkF,CAAC;QACvF,OAAO,QAAQ,CAAC,IAAI,CAClB;YACE,QAAQ,EAAE,IAAI;YACd,MAAM,EAAE,GAAG,OAAO,0GAA0G;YAC5H,SAAS,EAAE,EAAE;YACb,SAAS,EAAE,EAAE;SACS,EACxB,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,sBAAsB,EAAE,CACpD,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAC3C,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,OAAO,QAAQ,CAAC,IAAI,CAClB;YACE,QAAQ,EAAE,IAAI;YACd,MAAM,EAAE,8CAA8C,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;YAC9E,SAAS,EAAE,EAAE;YACb,SAAS,EAAE,EAAE;SACS,EACxB,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,sBAAsB,EAAE,CACpD,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,2BAAa,CAAC;QAC/B,GAAG,IAAI;KACR,CAAC,CAAC;IAEH,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,MAAM,OAAO,GAAG;QACd,GAAG,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE;YAC1B,SAAS,CAAC,IAAI,CAAC,mBAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QACvC,CAAC;QACD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE;YAC5B,SAAS,CAAC,IAAI,CAAC,mBAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QACvC,CAAC;KACF,CAAC;IACF,IAAI,CAAC;QACH,IAAI,IAAI,GAAG,KAAK,EAAE,MAAW,EAAE,EAAE,GAAE,CAAC,CAAC;QACrC,IAAI,CAAC,GAAG,IAAI,eAAe,CAAC,CAAC;QAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QACtE,OAAO,QAAQ,CAAC,IAAI,CAAC;YACnB,QAAQ,EAAE,KAAK;YACf,MAAM;YACN,SAAS;YACT,SAAS;SACa,CAAC,CAAC;IAC5B,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,OAAO,QAAQ,CAAC,IAAI,CAClB;YACE,QAAQ,EAAE,IAAI;YACd,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;YAC3B,SAAS;YACT,SAAS;SACa,EACxB,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,sBAAsB,EAAE,CACpD,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAEF,kBAAe,EAAE,KAAK,EAAE,CAAC"}

package/code-tool-worker.mjs

@@ -1,38 +1,240 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+import path from 'node:path';
 import util from 'node:util';
+import Fuse from 'fuse.js';
+import ts from 'typescript';
 import { BeeperDesktop } from '@beeper/desktop-api';
+function getRunFunctionSource(code) {
+    const sourceFile = ts.createSourceFile('code.ts', code, ts.ScriptTarget.Latest, true);
+    const printer = ts.createPrinter();
+    for (const statement of sourceFile.statements) {
+        // Check for top-level function declarations
+        if (ts.isFunctionDeclaration(statement)) {
+            if (statement.name?.text === 'run') {
+                return {
+                    type: 'declaration',
+                    client: statement.parameters[0]?.name.getText(),
+                    code: printer.printNode(ts.EmitHint.Unspecified, statement.body, sourceFile),
+                };
+            }
+        }
+        // Check for variable declarations: const run = () => {} or const run = function() {}
+        if (ts.isVariableStatement(statement)) {
+            for (const declaration of statement.declarationList.declarations) {
+                if (ts.isIdentifier(declaration.name) &&
+                    declaration.name.text === 'run' &&
+                    // Check if it's initialized with a function
+                    declaration.initializer &&
+                    (ts.isFunctionExpression(declaration.initializer) || ts.isArrowFunction(declaration.initializer))) {
+                    return {
+                        type: 'expression',
+                        client: declaration.initializer.parameters[0]?.name.getText(),
+                        code: printer.printNode(ts.EmitHint.Unspecified, declaration.initializer, sourceFile),
+                    };
+                }
+            }
+        }
+    }
+    return null;
+}
+function getTSDiagnostics(code) {
+    const functionSource = getRunFunctionSource(code);
+    const codeWithImport = [
+        'import { BeeperDesktop } from "@beeper/desktop-api";',
+        functionSource.type === 'declaration' ?
+            `async function run(${functionSource.client}: BeeperDesktop)`
+            : `const run: (${functionSource.client}: BeeperDesktop) => Promise<unknown> =`,
+        functionSource.code,
+    ].join('\n');
+    const sourcePath = path.resolve('code.ts');
+    const ast = ts.createSourceFile(sourcePath, codeWithImport, ts.ScriptTarget.Latest, true);
+    const options = ts.getDefaultCompilerOptions();
+    options.target = ts.ScriptTarget.Latest;
+    options.module = ts.ModuleKind.NodeNext;
+    options.moduleResolution = ts.ModuleResolutionKind.NodeNext;
+    const host = ts.createCompilerHost(options, true);
+    const newHost = {
+        ...host,
+        getSourceFile: (...args) => {
+            if (path.resolve(args[0]) === sourcePath) {
+                return ast;
+            }
+            return host.getSourceFile(...args);
+        },
+        readFile: (...args) => {
+            if (path.resolve(args[0]) === sourcePath) {
+                return codeWithImport;
+            }
+            return host.readFile(...args);
+        },
+        fileExists: (...args) => {
+            if (path.resolve(args[0]) === sourcePath) {
+                return true;
+            }
+            return host.fileExists(...args);
+        },
+    };
+    const program = ts.createProgram({
+        options,
+        rootNames: [sourcePath],
+        host: newHost,
+    });
+    const diagnostics = ts.getPreEmitDiagnostics(program, ast);
+    return diagnostics.map((d) => {
+        const message = ts.flattenDiagnosticMessageText(d.messageText, '\n');
+        if (!d.file || !d.start)
+            return `- ${message}`;
+        const { line: lineNumber } = ts.getLineAndCharacterOfPosition(d.file, d.start);
+        const line = codeWithImport.split('\n').at(lineNumber)?.trim();
+        return line ? `- ${message}\n    ${line}` : `- ${message}`;
+    });
+}
+const fuse = new Fuse([
+    'client.focus',
+    'client.search',
+    'client.accounts.list',
+    'client.accounts.contacts.list',
+    'client.accounts.contacts.search',
+    'client.chats.archive',
+    'client.chats.create',
+    'client.chats.list',
+    'client.chats.retrieve',
+    'client.chats.search',
+    'client.chats.reminders.create',
+    'client.chats.reminders.delete',
+    'client.chats.messages.reactions.add',
+    'client.chats.messages.reactions.delete',
+    'client.messages.list',
+    'client.messages.search',
+    'client.messages.send',
+    'client.messages.update',
+    'client.assets.download',
+    'client.assets.serve',
+    'client.assets.upload',
+    'client.assets.uploadBase64',
+    'client.info.retrieve',
+], { threshold: 1, shouldSort: true });
+function getMethodSuggestions(fullyQualifiedMethodName) {
+    return fuse
+        .search(fullyQualifiedMethodName)
+        .map(({ item }) => item)
+        .slice(0, 5);
+}
+const proxyToObj = new WeakMap();
+const objToProxy = new WeakMap();
+function makeSdkProxy(obj, { path, isBelievedBad = false }) {
+    let proxy = objToProxy.get(obj);
+    if (!proxy) {
+        proxy = new Proxy(obj, {
+            get(target, prop, receiver) {
+                const propPath = [...path, String(prop)];
+                const value = Reflect.get(target, prop, receiver);
+                if (isBelievedBad || (!(prop in target) && value === undefined)) {
+                    // If we're accessing a path that doesn't exist, it will probably eventually error.
+                    // Let's proxy it and mark it bad so that we can control the error message.
+                    // We proxy an empty class so that an invocation or construction attempt is possible.
+                    return makeSdkProxy(class {
+                    }, { path: propPath, isBelievedBad: true });
+                }
+                if (value !== null && (typeof value === 'object' || typeof value === 'function')) {
+                    return makeSdkProxy(value, { path: propPath, isBelievedBad });
+                }
+                return value;
+            },
+            apply(target, thisArg, args) {
+                if (isBelievedBad || typeof target !== 'function') {
+                    const fullyQualifiedMethodName = path.join('.');
+                    const suggestions = getMethodSuggestions(fullyQualifiedMethodName);
+                    throw new Error(`${fullyQualifiedMethodName} is not a function. Did you mean: ${suggestions.join(', ')}`);
+                }
+                return Reflect.apply(target, proxyToObj.get(thisArg) ?? thisArg, args);
+            },
+            construct(target, args, newTarget) {
+                if (isBelievedBad || typeof target !== 'function') {
+                    const fullyQualifiedMethodName = path.join('.');
+                    const suggestions = getMethodSuggestions(fullyQualifiedMethodName);
+                    throw new Error(`${fullyQualifiedMethodName} is not a constructor. Did you mean: ${suggestions.join(', ')}`);
+                }
+                return Reflect.construct(target, args, newTarget);
+            },
+        });
+        objToProxy.set(obj, proxy);
+        proxyToObj.set(proxy, obj);
+    }
+    return proxy;
+}
+function parseError(code, error) {
+    if (!(error instanceof Error))
+        return;
+    const message = error.name ? `${error.name}: ${error.message}` : error.message;
+    try {
+        // Deno uses V8; the first "<anonymous>:LINE:COLUMN" is the top of stack.
+        const lineNumber = error.stack?.match(/<anonymous>:([0-9]+):[0-9]+/)?.[1];
+        // -1 for the zero-based indexing
+        const line = lineNumber &&
+            code
+                .split('\n')
+                .at(parseInt(lineNumber, 10) - 1)
+                ?.trim();
+        return line ? `${message}\n  at line ${lineNumber}\n    ${line}` : message;
+    }
+    catch {
+        return message;
+    }
+}
 const fetch = async (req) => {
     const { opts, code } = (await req.json());
+    const runFunctionSource = code ? getRunFunctionSource(code) : null;
+    if (!runFunctionSource) {
+        const message = code ?
+            'The code is missing a top-level `run` function.'
+            : 'The code argument is missing. Provide one containing a top-level `run` function.';
+        return Response.json({
+            is_error: true,
+            result: `${message} Write code within this template:\n\n\`\`\`\nasync function run(client) {\n  // Fill this out\n}\n\`\`\``,
+            log_lines: [],
+            err_lines: [],
+        }, { status: 400, statusText: 'Code execution error' });
+    }
+    const diagnostics = getTSDiagnostics(code);
+    if (diagnostics.length > 0) {
+        return Response.json({
+            is_error: true,
+            result: `The code contains TypeScript diagnostics:\n${diagnostics.join('\n')}`,
+            log_lines: [],
+            err_lines: [],
+        }, { status: 400, statusText: 'Code execution error' });
+    }
     const client = new BeeperDesktop({
         ...opts,
     });
-    const logLines = [];
-    const errLines = [];
+    const log_lines = [];
+    const err_lines = [];
     const console = {
         log: (...args) => {
-            logLines.push(util.format(...args));
+            log_lines.push(util.format(...args));
         },
         error: (...args) => {
-            errLines.push(util.format(...args));
+            err_lines.push(util.format(...args));
         },
     };
     try {
         let run_ = async (client) => { };
-        eval(`
-      ${code}
-      run_ = run;
-    `);
-        const result = await run_(client);
+        eval(`${code}\nrun_ = run;`);
+        const result = await run_(makeSdkProxy(client, { path: ['client'] }));
         return Response.json({
+            is_error: false,
             result,
-            logLines,
-            errLines,
+            log_lines,
+            err_lines,
         });
     }
     catch (e) {
-        const message = e instanceof Error ? e.message : undefined;
         return Response.json({
-            message,
+            is_error: true,
+            result: parseError(code, e),
+            log_lines,
+            err_lines,
         }, { status: 400, statusText: 'Code execution error' });
     }
 };

package/code-tool-worker.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"code-tool-worker.mjs","sourceRoot":"","sources":["src/code-tool-worker.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,IAAI,MAAM,WAAW;OAErB,EAAE,aAAa,EAAE,MAAM,qBAAqB;AAEnD,MAAM,KAAK,GAAG,KAAK,EAAE,GAAY,EAAqB,EAAE;IACtD,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAgB,CAAC;IACzD,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC;QAC/B,GAAG,IAAI;KACR,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,OAAO,GAAG;QACd,GAAG,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE;YAC1B,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;QACD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE;YAC5B,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;KACF,CAAC;IACF,IAAI,CAAC;QACH,IAAI,IAAI,GAAG,KAAK,EAAE,MAAW,EAAE,EAAE,GAAE,CAAC,CAAC;QACrC,IAAI,CAAC;QACD,IAAI;;KAEP,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,CAAC;QAClC,OAAO,QAAQ,CAAC,IAAI,CAAC;YACnB,MAAM;YACN,QAAQ;YACR,QAAQ;SACe,CAAC,CAAC;IAC7B,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3D,OAAO,QAAQ,CAAC,IAAI,CAClB;YACE,OAAO;SACc,EACvB,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,sBAAsB,EAAE,CACpD,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAEF,eAAe,EAAE,KAAK,EAAE,CAAC"}
+{"version":3,"file":"code-tool-worker.mjs","sourceRoot":"","sources":["src/code-tool-worker.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,IAAI,MAAM,WAAW;OACrB,IAAI,MAAM,WAAW;OACrB,IAAI,MAAM,SAAS;OACnB,EAAE,MAAM,YAAY;OAEpB,EAAE,aAAa,EAAiB,MAAM,qBAAqB;AAElE,SAAS,oBAAoB,CAAC,IAAY;IAKxC,MAAM,UAAU,GAAG,EAAE,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IACtF,MAAM,OAAO,GAAG,EAAE,CAAC,aAAa,EAAE,CAAC;IAEnC,KAAK,MAAM,SAAS,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;QAC9C,4CAA4C;QAC5C,IAAI,EAAE,CAAC,qBAAqB,CAAC,SAAS,CAAC,EAAE,CAAC;YACxC,IAAI,SAAS,CAAC,IAAI,EAAE,IAAI,KAAK,KAAK,EAAE,CAAC;gBACnC,OAAO;oBACL,IAAI,EAAE,aAAa;oBACnB,MAAM,EAAE,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE;oBAC/C,IAAI,EAAE,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,SAAS,CAAC,IAAK,EAAE,UAAU,CAAC;iBAC9E,CAAC;YACJ,CAAC;QACH,CAAC;QAED,qFAAqF;QACrF,IAAI,EAAE,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,CAAC;YACtC,KAAK,MAAM,WAAW,IAAI,SAAS,CAAC,eAAe,CAAC,YAAY,EAAE,CAAC;gBACjE,IACE,EAAE,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC;oBACjC,WAAW,CAAC,IAAI,CAAC,IAAI,KAAK,KAAK;oBAC/B,4CAA4C;oBAC5C,WAAW,CAAC,WAAW;oBACvB,CAAC,EAAE,CAAC,oBAAoB,CAAC,WAAW,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,EACjG,CAAC;oBACD,OAAO;wBACL,IAAI,EAAE,YAAY;wBAClB,MAAM,EAAE,WAAW,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE;wBAC7D,IAAI,EAAE,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC,WAAW,EAAE,UAAU,CAAC;qBACtF,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAY;IACpC,MAAM,cAAc,GAAG,oBAAoB,CAAC,IAAI,CAAE,CAAC;IACnD,MAAM,cAAc,GAAG;QACrB,sDAAsD;QACtD,cAAc,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;YACrC,sBAAsB,cAAc,CAAC,MAAM,kBAAkB;YAC/D,CAAC,CAAC,eAAe,cAAc,CAAC,MAAM,wCAAwC;QAC9E,cAAc,CAAC,IAAI;KACpB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACb,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3C,MAAM,GAAG,GAAG,EAAE,CAAC,gBAAgB,CAAC,UAAU,EAAE,cAAc,EAAE,EAAE,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC1F,MAAM,OAAO,GAAG,EAAE,CAAC,yBAAyB,EAAE,CAAC;IAC/C,OAAO,CAAC,MAAM,GAAG,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC;IACxC,OAAO,CAAC,MAAM,GAAG,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;IACxC,OAAO,CAAC,gBAAgB,GAAG,EAAE,CAAC,oBAAoB,CAAC,QAAQ,CAAC;IAC5D,MAAM,IAAI,GAAG,EAAE,CAAC,kBAAkB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAClD,MAAM,OAAO,GAAgB;QAC3B,GAAG,IAAI;QACP,aAAa,EAAE,CAAC,GAAG,IAAI,EAAE,EAAE;YACzB,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,EAAE,CAAC;gBACzC,OAAO,GAAG,CAAC;YACb,CAAC;YACD,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,CAAC;QACrC,CAAC;QACD,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,EAAE;YACpB,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,EAAE,CAAC;gBACzC,OAAO,cAAc,CAAC;YACxB,CAAC;YACD,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;QAChC,CAAC;QACD,UAAU,EAAE,CAAC,GAAG,IAAI,EAAE,EAAE;YACtB,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,EAAE,CAAC;gBACzC,OAAO,IAAI,CAAC;YACd,CAAC;YACD,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,CAAC;QAClC,CAAC;KACF,CAAC;IACF,MAAM,OAAO,GAAG,EAAE,CAAC,aAAa,CAAC;QAC/B,OAAO;QACP,SAAS,EAAE,CAAC,UAAU,CAAC;QACvB,IAAI,EAAE,OAAO;KACd,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,EAAE,CAAC,qBAAqB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IAC3D,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC3B,MAAM,OAAO,GAAG,EAAE,CAAC,4BAA4B,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QACrE,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK;YAAE,OAAO,KAAK,OAAO,EAAE,CAAC;QAC/C,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;QAC/E,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC;QAC/D,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,OAAO,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,OAAO,EAAE,CAAC;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,IAAI,GAAG,IAAI,IAAI,CACnB;IACE,cAAc;IACd,eAAe;IACf,sBAAsB;IACtB,+BAA+B;IAC/B,iCAAiC;IACjC,sBAAsB;IACtB,qBAAqB;IACrB,mBAAmB;IACnB,uBAAuB;IACvB,qBAAqB;IACrB,+BAA+B;IAC/B,+BAA+B;IAC/B,qCAAqC;IACrC,wCAAwC;IACxC,sBAAsB;IACtB,wBAAwB;IACxB,sBAAsB;IACtB,wBAAwB;IACxB,wBAAwB;IACxB,qBAAqB;IACrB,sBAAsB;IACtB,4BAA4B;IAC5B,sBAAsB;CACvB,EACD,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CACnC,CAAC;AAEF,SAAS,oBAAoB,CAAC,wBAAgC;IAC5D,OAAO,IAAI;SACR,MAAM,CAAC,wBAAwB,CAAC;SAChC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC;SACvB,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,GAAG,IAAI,OAAO,EAAY,CAAC;AAC3C,MAAM,UAAU,GAAG,IAAI,OAAO,EAAY,CAAC;AAO3C,SAAS,YAAY,CAAmB,GAAM,EAAE,EAAE,IAAI,EAAE,aAAa,GAAG,KAAK,EAAqB;IAChG,IAAI,KAAK,GAAM,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAEnC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,KAAK,GAAG,IAAI,KAAK,CAAC,GAAG,EAAE;YACrB,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ;gBACxB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;gBACzC,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;gBAElD,IAAI,aAAa,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,KAAK,KAAK,SAAS,CAAC,EAAE,CAAC;oBAChE,mFAAmF;oBACnF,2EAA2E;oBAC3E,qFAAqF;oBACrF,OAAO,YAAY,CAAC;qBAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzE,CAAC;gBAED,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,UAAU,CAAC,EAAE,CAAC;oBACjF,OAAO,YAAY,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC,CAAC;gBAChE,CAAC;gBAED,OAAO,KAAK,CAAC;YACf,CAAC;YAED,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI;gBACzB,IAAI,aAAa,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;oBAClD,MAAM,wBAAwB,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAChD,MAAM,WAAW,GAAG,oBAAoB,CAAC,wBAAwB,CAAC,CAAC;oBACnE,MAAM,IAAI,KAAK,CACb,GAAG,wBAAwB,qCAAqC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACzF,CAAC;gBACJ,CAAC;gBAED,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,OAAO,EAAE,IAAI,CAAC,CAAC;YACzE,CAAC;YAED,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS;gBAC/B,IAAI,aAAa,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;oBAClD,MAAM,wBAAwB,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAChD,MAAM,WAAW,GAAG,oBAAoB,CAAC,wBAAwB,CAAC,CAAC;oBACnE,MAAM,IAAI,KAAK,CACb,GAAG,wBAAwB,wCAAwC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC5F,CAAC;gBACJ,CAAC;gBAED,OAAO,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YACpD,CAAC;SACF,CAAC,CAAC;QAEH,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC3B,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,UAAU,CAAC,IAAY,EAAE,KAAc;IAC9C,IAAI,CAAC,CAAC,KAAK,YAAY,KAAK,CAAC;QAAE,OAAO;IACtC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC;IAC/E,IAAI,CAAC;QACH,yEAAyE;QACzE,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1E,iCAAiC;QACjC,MAAM,IAAI,GACR,UAAU;YACV,IAAI;iBACD,KAAK,CAAC,IAAI,CAAC;iBACX,EAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;gBACjC,EAAE,IAAI,EAAE,CAAC;QACb,OAAO,IAAI,CAAC,CAAC,CAAC,GAAG,OAAO,eAAe,UAAU,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;IAC7E,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,OAAO,CAAC;IACjB,CAAC;AACH,CAAC;AAED,MAAM,KAAK,GAAG,KAAK,EAAE,GAAY,EAAqB,EAAE;IACtD,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA0C,CAAC;IAEnF,MAAM,iBAAiB,GAAG,IAAI,CAAC,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACnE,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,MAAM,OAAO,GACX,IAAI,CAAC,CAAC;YACJ,iDAAiD;YACnD,CAAC,CAAC,kFAAkF,CAAC;QACvF,OAAO,QAAQ,CAAC,IAAI,CAClB;YACE,QAAQ,EAAE,IAAI;YACd,MAAM,EAAE,GAAG,OAAO,0GAA0G;YAC5H,SAAS,EAAE,EAAE;YACb,SAAS,EAAE,EAAE;SACS,EACxB,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,sBAAsB,EAAE,CACpD,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAC3C,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,OAAO,QAAQ,CAAC,IAAI,CAClB;YACE,QAAQ,EAAE,IAAI;YACd,MAAM,EAAE,8CAA8C,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;YAC9E,SAAS,EAAE,EAAE;YACb,SAAS,EAAE,EAAE;SACS,EACxB,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,sBAAsB,EAAE,CACpD,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC;QAC/B,GAAG,IAAI;KACR,CAAC,CAAC;IAEH,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,MAAM,OAAO,GAAG;QACd,GAAG,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE;YAC1B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QACvC,CAAC;QACD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE;YAC5B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QACvC,CAAC;KACF,CAAC;IACF,IAAI,CAAC;QACH,IAAI,IAAI,GAAG,KAAK,EAAE,MAAW,EAAE,EAAE,GAAE,CAAC,CAAC;QACrC,IAAI,CAAC,GAAG,IAAI,eAAe,CAAC,CAAC;QAC7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QACtE,OAAO,QAAQ,CAAC,IAAI,CAAC;YACnB,QAAQ,EAAE,KAAK;YACf,MAAM;YACN,SAAS;YACT,SAAS;SACa,CAAC,CAAC;IAC5B,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,OAAO,QAAQ,CAAC,IAAI,CAClB;YACE,QAAQ,EAAE,IAAI;YACd,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;YAC3B,SAAS;YACT,SAAS;SACa,EACxB,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,sBAAsB,EAAE,CACpD,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAEF,eAAe,EAAE,KAAK,EAAE,CAAC"}

package/code-tool.d.mts

@@ -1,12 +1,21 @@
-import { Endpoint } from "./tools/types.mjs";
+import { McpTool } from "./types.mjs";
+import { SdkMethod } from "./methods.mjs";
+import { McpCodeExecutionMode } from "./options.mjs";
 /**
  * A tool that runs code against a copy of the SDK.
  *
- * Instead of exposing every endpoint as it's own tool, which uses up too many tokens for LLMs to use at once,
+ * Instead of exposing every endpoint as its own tool, which uses up too many tokens for LLMs to use at once,
  * we expose a single tool that can be used to search for endpoints by name, resource, operation, or tag, and then
  * a generic endpoint that can be used to invoke any endpoint with the provided arguments.
  *
- * @param endpoints - The endpoints to include in the list.
+ * @param blockedMethods - The methods to block for code execution. Blocking is done by simple string
+ * matching, so it is not secure against obfuscation. For stronger security, block in the downstream API
+ * with limited API keys.
+ * @param codeExecutionMode - Whether to execute code in a local Deno environment or in a remote
+ * sandbox environment hosted by Stainless.
  */
-export declare function codeTool(): Promise<Endpoint>;
+export declare function codeTool({ blockedMethods, codeExecutionMode, }: {
+    blockedMethods: SdkMethod[] | undefined;
+    codeExecutionMode: McpCodeExecutionMode;
+}): McpTool;
 //# sourceMappingURL=code-tool.d.mts.map

package/code-tool.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"code-tool.d.mts","sourceRoot":"","sources":["src/code-tool.ts"],"names":[],"mappings":"OAKO,EAAE,QAAQ,EAA0B;AAM3C;;;;;;;;GAQG;AACH,wBAAsB,QAAQ,IAAI,OAAO,CAAC,QAAQ,CAAC,CA6HlD"}
+{"version":3,"file":"code-tool.d.mts","sourceRoot":"","sources":["src/code-tool.ts"],"names":[],"mappings":"OAOO,EAGL,OAAO,EAKR;OAKM,EAAE,SAAS,EAAE;OACb,EAAE,oBAAoB,EAAE;AA8B/B;;;;;;;;;;;;GAYG;AACH,wBAAgB,QAAQ,CAAC,EACvB,cAAc,EACd,iBAAiB,GAClB,EAAE;IACD,cAAc,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC;IACxC,iBAAiB,EAAE,oBAAoB,CAAC;CACzC,GAAG,OAAO,CAuEV"}

package/code-tool.d.ts

@@ -1,12 +1,21 @@
-import { Endpoint } from "./tools/types.js";
+import { McpTool } from "./types.js";
+import { SdkMethod } from "./methods.js";
+import { McpCodeExecutionMode } from "./options.js";
 /**
  * A tool that runs code against a copy of the SDK.
  *
- * Instead of exposing every endpoint as it's own tool, which uses up too many tokens for LLMs to use at once,
+ * Instead of exposing every endpoint as its own tool, which uses up too many tokens for LLMs to use at once,
  * we expose a single tool that can be used to search for endpoints by name, resource, operation, or tag, and then
  * a generic endpoint that can be used to invoke any endpoint with the provided arguments.
  *
- * @param endpoints - The endpoints to include in the list.
+ * @param blockedMethods - The methods to block for code execution. Blocking is done by simple string
+ * matching, so it is not secure against obfuscation. For stronger security, block in the downstream API
+ * with limited API keys.
+ * @param codeExecutionMode - Whether to execute code in a local Deno environment or in a remote
+ * sandbox environment hosted by Stainless.
  */
-export declare function codeTool(): Promise<Endpoint>;
+export declare function codeTool({ blockedMethods, codeExecutionMode, }: {
+    blockedMethods: SdkMethod[] | undefined;
+    codeExecutionMode: McpCodeExecutionMode;
+}): McpTool;
 //# sourceMappingURL=code-tool.d.ts.map

package/code-tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"code-tool.d.ts","sourceRoot":"","sources":["src/code-tool.ts"],"names":[],"mappings":"OAKO,EAAE,QAAQ,EAA0B;AAM3C;;;;;;;;GAQG;AACH,wBAAsB,QAAQ,IAAI,OAAO,CAAC,QAAQ,CAAC,CA6HlD"}
+{"version":3,"file":"code-tool.d.ts","sourceRoot":"","sources":["src/code-tool.ts"],"names":[],"mappings":"OAOO,EAGL,OAAO,EAKR;OAKM,EAAE,SAAS,EAAE;OACb,EAAE,oBAAoB,EAAE;AA8B/B;;;;;;;;;;;;GAYG;AACH,wBAAgB,QAAQ,CAAC,EACvB,cAAc,EACd,iBAAiB,GAClB,EAAE;IACD,cAAc,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC;IACxC,iBAAiB,EAAE,oBAAoB,CAAC;CACzC,GAAG,OAAO,CAuEV"}