scorecard-ai-mcp 2.5.0 → 2.6.0

package/src/code-tool-worker.ts

@@ -1,225 +0,0 @@
-// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-import util from 'node:util';
-
-import Fuse from 'fuse.js';
-import ts from 'typescript';
-
-import { WorkerInput, WorkerSuccess, WorkerError } from './code-tool-types';
-import { Scorecard } from 'scorecard-ai';
-
-function getRunFunctionNode(
-  code: string,
-): ts.FunctionDeclaration | ts.FunctionExpression | ts.ArrowFunction | null {
-  const sourceFile = ts.createSourceFile('code.ts', code, ts.ScriptTarget.Latest, true);
-
-  for (const statement of sourceFile.statements) {
-    // Check for top-level function declarations
-    if (ts.isFunctionDeclaration(statement)) {
-      if (statement.name?.text === 'run') {
-        return statement;
-      }
-    }
-
-    // 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
-          if (
-            declaration.initializer &&
-            (ts.isFunctionExpression(declaration.initializer) || ts.isArrowFunction(declaration.initializer))
-          ) {
-            return declaration.initializer;
-          }
-        }
-      }
-    }
-  }
-
-  return null;
-}
-
-const fuse = new Fuse(
-  [
-    'client.projects.create',
-    'client.projects.list',
-    'client.testsets.create',
-    'client.testsets.delete',
-    'client.testsets.get',
-    'client.testsets.list',
-    'client.testsets.update',
-    'client.testcases.create',
-    'client.testcases.delete',
-    'client.testcases.get',
-    'client.testcases.list',
-    'client.testcases.update',
-    'client.runs.create',
-    'client.runs.get',
-    'client.runs.list',
-    'client.metrics.create',
-    'client.metrics.delete',
-    'client.metrics.get',
-    'client.metrics.list',
-    'client.metrics.update',
-    'client.records.create',
-    'client.records.delete',
-    'client.records.list',
-    'client.scores.upsert',
-    'client.systems.delete',
-    'client.systems.get',
-    'client.systems.list',
-    'client.systems.update',
-    'client.systems.upsert',
-    'client.systems.versions.get',
-    'client.systems.versions.upsert',
-  ],
-  { threshold: 1, shouldSort: true },
-);
-
-function getMethodSuggestions(fullyQualifiedMethodName: string): string[] {
-  return fuse
-    .search(fullyQualifiedMethodName)
-    .map(({ item }) => item)
-    .slice(0, 5);
-}
-
-const proxyToObj = new WeakMap<any, any>();
-const objToProxy = new WeakMap<any, any>();
-
-type ClientProxyConfig = {
-  path: string[];
-  isBelievedBad?: boolean;
-};
-
-function makeSdkProxy<T extends object>(obj: T, { path, isBelievedBad = false }: ClientProxyConfig): T {
-  let proxy: T = 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: string, error: unknown): string | undefined {
-  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: Request): Promise<Response> => {
-  const { opts, code } = (await req.json()) as WorkerInput;
-  if (code == null) {
-    return Response.json(
-      {
-        message:
-          'The code param is missing. Provide one containing a top-level `run` function. Write code within this template:\n\n```\nasync function run(client) {\n  // Fill this out\n}\n```',
-      } satisfies WorkerError,
-      { status: 400, statusText: 'Code execution error' },
-    );
-  }
-
-  const runFunctionNode = getRunFunctionNode(code);
-  if (!runFunctionNode) {
-    return Response.json(
-      {
-        message:
-          'The code is missing a top-level `run` function. Write code within this template:\n\n```\nasync function run(client) {\n  // Fill this out\n}\n```',
-      } satisfies WorkerError,
-      { status: 400, statusText: 'Code execution error' },
-    );
-  }
-
-  const client = new Scorecard({
-    ...opts,
-  });
-
-  const logLines: string[] = [];
-  const errLines: string[] = [];
-  const console = {
-    log: (...args: unknown[]) => {
-      logLines.push(util.format(...args));
-    },
-    error: (...args: unknown[]) => {
-      errLines.push(util.format(...args));
-    },
-  };
-  try {
-    let run_ = async (client: any) => {};
-    eval(`${code}\nrun_ = run;`);
-    const result = await run_(makeSdkProxy(client, { path: ['client'] }));
-    return Response.json({
-      result,
-      logLines,
-      errLines,
-    } satisfies WorkerSuccess);
-  } catch (e) {
-    return Response.json(
-      {
-        message: parseError(code, e),
-      } satisfies WorkerError,
-      { status: 400, statusText: 'Code execution error' },
-    );
-  }
-};
-
-export default { fetch };