Support files in agent block

Author: Sg312

apps/sim/blocks/blocks.test.ts

@@ -163,6 +163,42 @@ describe.concurrent('Blocks Module', () => {
     })
   })
 
+  describe('Agent block', () => {
+    it('should expose canonical file attachments and normalize file params', () => {
+      const block = getBlock('agent')
+
+      expect(block).toBeDefined()
+      const uploadSubBlock = block?.subBlocks.find((subBlock) => subBlock.id === 'attachmentFiles')
+      const advancedSubBlock = block?.subBlocks.find((subBlock) => subBlock.id === 'files')
+
+      expect(uploadSubBlock?.type).toBe('file-upload')
+      expect(uploadSubBlock?.canonicalParamId).toBe('files')
+      expect(uploadSubBlock?.multiple).toBe(true)
+      expect(advancedSubBlock?.canonicalParamId).toBe('files')
+      expect(block?.inputs.files).toEqual({
+        type: 'array',
+        description: 'Files to include with the latest user message',
+      })
+
+      expect(
+        block?.tools.config?.params?.({
+          model: 'gpt-4o',
+          files:
+            '[{"id":"file-1","key":"workspace/ws-1/example.png","name":"example.png","url":"/api/files/serve/workspace%2Fws-1%2Fexample.png?context=workspace","size":123,"type":"image/png"}]',
+        })
+      ).toMatchObject({
+        files: [
+          {
+            id: 'file-1',
+            key: 'workspace/ws-1/example.png',
+            name: 'example.png',
+            type: 'image/png',
+          },
+        ],
+      })
+    })
+  })
+
   describe('getBlocksByCategory', () => {
     it('should return blocks in the "blocks" category', () => {
       const blocks = getBlocksByCategory('blocks')

apps/sim/blocks/blocks/agent.ts

@@ -5,6 +5,7 @@ import { AuthMode, IntegrationType } from '@/blocks/types'
 import {
   getModelOptions,
   getProviderCredentialSubBlocks,
+  normalizeFileInput,
   RESPONSE_FORMAT_WAND_CONFIG,
 } from '@/blocks/utils'
 import {
@@ -133,6 +134,25 @@ Return ONLY the JSON array.`,
       defaultValue: 'claude-sonnet-4-6',
       options: getModelOptions,
     },
+    {
+      id: 'attachmentFiles',
+      title: 'Files',
+      type: 'file-upload',
+      canonicalParamId: 'files',
+      placeholder: 'Upload files for the agent',
+      multiple: true,
+      mode: 'basic',
+      required: false,
+    },
+    {
+      id: 'files',
+      title: 'Files',
+      type: 'short-input',
+      canonicalParamId: 'files',
+      placeholder: 'Reference files from previous blocks',
+      mode: 'advanced',
+      required: false,
+    },
     {
       id: 'reasoningEffort',
       title: 'Reasoning Effort',
@@ -472,6 +492,9 @@ Return ONLY the JSON array.`,
         return tool
       },
       params: (params: Record<string, any>) => {
+        const normalizedFiles = normalizeFileInput(params.files)
+        const baseParams = normalizedFiles ? { ...params, files: normalizedFiles } : params
+
         // If tools array is provided, handle tool usage control
         if (params.tools && Array.isArray(params.tools)) {
           // Transform tools to include usageControl
@@ -506,9 +529,9 @@ Return ONLY the JSON array.`,
             logger.info('Filtered out tools set to none', { tools: filteredOutTools.join(', ') })
           }
 
-          return { ...params, tools: transformedTools }
+          return { ...baseParams, tools: transformedTools }
         }
-        return params
+        return baseParams
       },
     },
   },
@@ -518,6 +541,7 @@ Return ONLY the JSON array.`,
       description:
         'Array of message objects with role and content: [{ role: "system", content: "..." }, { role: "user", content: "..." }]',
     },
+    files: { type: 'array', description: 'Files to include with the latest user message' },
     memoryType: {
       type: 'string',
       description:

apps/sim/executor/execution/engine.ts

@@ -449,19 +449,6 @@ export class ExecutionEngine {
 
     const readyNodes = this.edgeManager.processOutgoingEdges(node, output, false)
 
-    this.execLogger.info('Processing outgoing edges', {
-      nodeId,
-      outgoingEdgesCount: node.outgoingEdges.size,
-      outgoingEdges: Array.from(node.outgoingEdges.entries()).map(([id, e]) => ({
-        id,
-        target: e.target,
-        sourceHandle: e.sourceHandle,
-      })),
-      output,
-      readyNodesCount: readyNodes.length,
-      readyNodes,
-    })
-
     this.addMultipleToQueue(readyNodes)
 
     if (this.context.pendingDynamicNodes && this.context.pendingDynamicNodes.length > 0) {

apps/sim/executor/handlers/agent/agent-handler.test.ts

@@ -264,6 +264,78 @@ describe('AgentBlockHandler', () => {
       expect(result).toEqual(expectedOutput)
     })
 
+    it('should attach files to the last user message only', async () => {
+      const inputs = {
+        model: 'gpt-4o',
+        messages: [
+          { role: 'system' as const, content: 'You are helpful.' },
+          { role: 'user' as const, content: 'Earlier question' },
+          { role: 'assistant' as const, content: 'Earlier answer' },
+          { role: 'user' as const, content: 'Analyze this file' },
+        ],
+        files: [
+          {
+            id: 'file-1',
+            key: 'workspace/ws-1/example.png',
+            name: 'example.png',
+            url: '/api/files/serve/workspace%2Fws-1%2Fexample.png?context=workspace',
+            size: 128,
+            type: 'image/png',
+            base64: 'aW1hZ2U=',
+          },
+        ],
+        apiKey: 'test-api-key',
+      }
+
+      mockGetProviderFromModel.mockReturnValue('openai')
+
+      await handler.execute(mockContext, mockBlock, inputs)
+
+      const requestBody = mockExecuteProviderRequest.mock.calls[0][1]
+      expect(requestBody.messages[1]).toMatchObject({
+        role: 'user',
+        content: 'Earlier question',
+      })
+      expect(requestBody.messages[1].files).toBeUndefined()
+      expect(requestBody.messages[3]).toMatchObject({
+        role: 'user',
+        content: 'Analyze this file',
+        files: [
+          {
+            id: 'file-1',
+            name: 'example.png',
+            type: 'image/png',
+            base64: 'aW1hZ2U=',
+          },
+        ],
+      })
+    })
+
+    it('should reject files for providers without attachment support', async () => {
+      const inputs = {
+        model: 'deepseek-chat',
+        messages: [{ role: 'user' as const, content: 'Analyze this file' }],
+        files: [
+          {
+            id: 'file-1',
+            key: 'workspace/ws-1/example.png',
+            name: 'example.png',
+            url: '/api/files/serve/workspace%2Fws-1%2Fexample.png?context=workspace',
+            size: 128,
+            type: 'image/png',
+            base64: 'aW1hZ2U=',
+          },
+        ],
+        apiKey: 'test-api-key',
+      }
+
+      mockGetProviderFromModel.mockReturnValue('deepseek')
+
+      await expect(handler.execute(mockContext, mockBlock, inputs)).rejects.toThrow(
+        'File attachments are not supported for provider "deepseek"'
+      )
+    })
+
     it('should preserve usageControl for custom tools and filter out "none"', async () => {
       const inputs = {
         model: 'gpt-4o',

apps/sim/executor/handlers/agent/agent-handler.ts

@@ -6,9 +6,12 @@ import { sleep } from '@sim/utils/helpers'
 import { and, eq, inArray, isNull } from 'drizzle-orm'
 import { normalizeStringRecord, normalizeWorkflowVariables } from '@/lib/core/utils/records'
 import { createMcpToolId } from '@/lib/mcp/utils'
+import { processFilesToUserFiles, type RawFileInput } from '@/lib/uploads/utils/file-utils'
+import { hydrateUserFilesWithBase64 } from '@/lib/uploads/utils/user-file-base64.server'
 import { getCustomToolById } from '@/lib/workflows/custom-tools/operations'
 import { getAllBlocks } from '@/blocks'
 import type { BlockOutput } from '@/blocks/types'
+import { normalizeFileInput } from '@/blocks/utils'
 import {
   validateBlockType,
   validateCustomToolsAllowed,
@@ -36,6 +39,7 @@ import { buildAPIUrl, buildAuthHeaders } from '@/executor/utils/http'
 import { stringifyJSON } from '@/executor/utils/json'
 import { resolveVertexCredential } from '@/executor/utils/vertex-credential'
 import { executeProviderRequest } from '@/providers'
+import { getAttachmentProvider, getProviderAttachmentMaxBytes } from '@/providers/attachments'
 import { getProviderFromModel, transformBlockTool } from '@/providers/utils'
 import type { SerializedBlock } from '@/serializer/types'
 import { filterSchemaForLLM, type ToolSchema } from '@/tools/params'
@@ -87,12 +91,18 @@ export class AgentBlockHandler implements BlockHandler {
 
     const streamingConfig = this.getStreamingConfig(ctx, block)
     const messages = await this.buildMessages(ctx, filteredInputs, skillMetadata)
+    const messagesWithFiles = await this.attachFilesToLastUserMessage(
+      ctx,
+      messages,
+      filteredInputs.files,
+      providerId
+    )
 
     const providerRequest = this.buildProviderRequest({
       ctx,
       providerId,
       model,
-      messages,
+      messages: messagesWithFiles,
       inputs: filteredInputs,
       formattedTools,
       responseFormat,
@@ -672,6 +682,62 @@ export class AgentBlockHandler implements BlockHandler {
     return messages.length > 0 ? messages : undefined
   }
 
+  private async attachFilesToLastUserMessage(
+    ctx: ExecutionContext,
+    messages: Message[] | undefined,
+    filesInput: unknown,
+    providerId: string
+  ): Promise<Message[] | undefined> {
+    const normalizedFiles = normalizeFileInput(filesInput)
+    if (!normalizedFiles || normalizedFiles.length === 0) {
+      return messages
+    }
+
+    if (!messages || messages.length === 0) {
+      throw new Error('Files require at least one user message in the agent prompt')
+    }
+
+    if (!getAttachmentProvider(providerId)) {
+      throw new Error(`File attachments are not supported for provider "${providerId}"`)
+    }
+
+    let lastUserMessageIndex = -1
+    for (let index = messages.length - 1; index >= 0; index--) {
+      if (messages[index].role === 'user') {
+        lastUserMessageIndex = index
+        break
+      }
+    }
+    if (lastUserMessageIndex === -1) {
+      throw new Error('Files require at least one user message in the agent prompt')
+    }
+
+    const requestId = ctx.executionId || ctx.workflowId || 'agent-files'
+    const userFiles = processFilesToUserFiles(normalizedFiles as RawFileInput[], requestId, logger)
+    if (userFiles.length === 0) {
+      throw new Error('Files must include at least one valid file object')
+    }
+
+    const hydratedFiles = await hydrateUserFilesWithBase64(userFiles, {
+      requestId,
+      workspaceId: ctx.workspaceId,
+      workflowId: ctx.workflowId,
+      executionId: ctx.executionId,
+      userId: ctx.userId,
+      logger,
+      maxBytes: getProviderAttachmentMaxBytes(providerId),
+    })
+
+    const lastUserMessage = messages[lastUserMessageIndex]
+    const nextMessages = [...messages]
+    nextMessages[lastUserMessageIndex] = {
+      ...lastUserMessage,
+      files: [...(lastUserMessage.files ?? []), ...hydratedFiles],
+    }
+
+    return nextMessages
+  }
+
   private extractValidMessages(messages?: Message[]): Message[] {
     if (!messages || !Array.isArray(messages)) return []
 

apps/sim/executor/handlers/agent/types.ts

@@ -1,3 +1,5 @@
+import type { UserFile } from '@/executor/types'
+
 export interface SkillInput {
   skillId: string
   name?: string
@@ -37,6 +39,7 @@ export interface AgentInputs {
   reasoningEffort?: string
   verbosity?: string
   thinkingLevel?: string
+  files?: unknown
 }
 
 /**
@@ -66,6 +69,7 @@ export interface ToolInput {
 export interface Message {
   role: 'system' | 'user' | 'assistant'
   content: string
+  files?: UserFile[]
   executionId?: string
   function_call?: any
   tool_calls?: any[]

apps/sim/lib/uploads/utils/file-utils.ts

@@ -172,7 +172,7 @@ export function createFileContentFromBase64(base64: string, mimeType: string): M
     return null
   }
 
-  if (contentType === 'image' && !CLAUDE_SUPPORTED_IMAGE_MIME_TYPES.has(mimeType.toLowerCase())) {
+  if (contentType === 'image' && !MODEL_SUPPORTED_IMAGE_MIME_TYPES.has(mimeType.toLowerCase())) {
     return null
   }
 
@@ -186,7 +186,7 @@ export function createFileContentFromBase64(base64: string, mimeType: string): M
   }
 }
 
-const CLAUDE_SUPPORTED_IMAGE_MIME_TYPES = new Set([
+export const MODEL_SUPPORTED_IMAGE_MIME_TYPES = new Set([
   'image/jpeg',
   'image/jpg',
   'image/png',

apps/sim/providers/anthropic/core.ts

@@ -9,6 +9,7 @@ import {
   checkForForcedToolUsage,
   createReadableStreamFromAnthropicStream,
 } from '@/providers/anthropic/utils'
+import { buildAnthropicMessageContent } from '@/providers/attachments'
 import {
   getMaxOutputTokensForModel,
   getThinkingCapability,
@@ -229,9 +230,10 @@ export async function executeAnthropicProviderRequest(
           ],
         })
       } else {
+        const content = buildAnthropicMessageContent(msg.content, msg.files, config.providerId)
         messages.push({
           role: msg.role === 'assistant' ? 'assistant' : 'user',
-          content: msg.content ? [{ type: 'text', text: msg.content }] : [],
+          content: content as unknown as Anthropic.Messages.ContentBlockParam[],
         })
       }
     })