fix(mcp): correct code tool api output types

Author: stainless-app[bot]

packages/mcp-server/src/code-tool-types.ts

@@ -3,16 +3,13 @@
 import { ClientOptions } from '@imagekit/nodejs';
 
 export type WorkerInput = {
-  opts: ClientOptions;
+  project_name: string;
   code: string;
+  client_opts: ClientOptions;
 };
-export type WorkerSuccess = {
+export type WorkerOutput = {
+  is_error: boolean;
   result: unknown | null;
-  logLines: string[];
-  errLines: string[];
-};
-export type WorkerError = {
-  message: string | undefined;
-  logLines: string[];
-  errLines: string[];
+  log_lines: string[];
+  err_lines: string[];
 };

packages/mcp-server/src/code-tool.ts

@@ -1,9 +1,9 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-import { McpTool, Metadata, ToolCallResult, asTextContentResult } from './types';
+import { McpTool, Metadata, ToolCallResult, asErrorResult, asTextContentResult } from './types';
 import { Tool } from '@modelcontextprotocol/sdk/types.js';
 import { readEnv } from './server';
-import { WorkerSuccess } from './code-tool-types';
+import { WorkerInput, WorkerOutput } from './code-tool-types';
 /**
  * A tool that runs code against a copy of the SDK.
  *
@@ -44,9 +44,9 @@ export function codeTool(): McpTool {
       },
       body: JSON.stringify({
         project_name: 'imagekit',
-        client_opts: {},
         code,
-      }),
+        client_opts: {},
+      } satisfies WorkerInput),
     });
 
     if (!res.ok) {
@@ -57,7 +57,17 @@ export function codeTool(): McpTool {
       );
     }
 
-    return asTextContentResult((await res.json()) as WorkerSuccess);
+    const { is_error, result, log_lines, err_lines } = (await res.json()) as WorkerOutput;
+    const hasLogs = log_lines.length > 0 || err_lines.length > 0;
+    const output = {
+      result,
+      ...(log_lines.length > 0 && { log_lines }),
+      ...(err_lines.length > 0 && { err_lines }),
+    };
+    if (is_error) {
+      return asErrorResult(typeof result === 'string' && !hasLogs ? result : JSON.stringify(output, null, 2));
+    }
+    return asTextContentResult(output);
   };
 
   return { metadata, tool, handler };