diff --git a/apps/sim/tools/function/execute.ts b/apps/sim/tools/function/execute.ts
index eb121ecbc6..af015aae68 100644
--- a/apps/sim/tools/function/execute.ts
+++ b/apps/sim/tools/function/execute.ts
@@ -28,6 +28,18 @@ export const functionExecuteTool: ToolConfig<CodeExecutionInput, CodeExecutionOu
       description: 'Environment variables to make available during execution',
       default: {},
     },
+    blockData: {
+      type: 'object',
+      required: false,
+      description: 'Block output data for variable resolution',
+      default: {},
+    },
+    blockNameMapping: {
+      type: 'object',
+      required: false,
+      description: 'Mapping of block names to block IDs',
+      default: {},
+    },
   },
 
   request: {
diff --git a/apps/sim/tools/function/types.ts b/apps/sim/tools/function/types.ts
index dbb9555a34..06e673d5ee 100644
--- a/apps/sim/tools/function/types.ts
+++ b/apps/sim/tools/function/types.ts
@@ -5,6 +5,8 @@ export interface CodeExecutionInput {
   timeout?: number
   memoryLimit?: number
   envVars?: Record<string, string>
+  blockData?: Record<string, any>
+  blockNameMapping?: Record<string, string>
   _context?: {
     workflowId?: string
   }