Function execute

apps/sim/lib/copilot/tools/server/image/generate-image.ts:56–204

(
    params: GenerateImageArgs,
    context?: ServerToolContext
  )

  name: GenerateImage.id,

  async execute(
    params: GenerateImageArgs,
    context?: ServerToolContext
  ): Promise<GenerateImageResult> {
    const withMessageId = (message: string) =>
      context?.messageId ? `${message} [messageId:${context.messageId}]` : message

    if (!context?.userId) {
      throw new Error('Authentication required')
    }
    const workspaceId = context.workspaceId
    if (!workspaceId) {
      return { success: false, message: 'Workspace ID is required' }
    }

    const { prompt } = params
    if (!prompt) {
      return { success: false, message: 'prompt is required' }
    }

    try {
      const apiKey = getRotatingApiKey('gemini')
      const ai = new GoogleGenAI({ apiKey })

      const aspectRatio = params.aspectRatio || '1:1'
      const sizeHint = ASPECT_RATIO_TO_SIZE[aspectRatio]

      const parts: Part[] = []

      const referencePaths = params.inputs?.files?.map((file) => file.path) ?? []

      if (referencePaths.length) {
        for (const filePath of referencePaths) {
          try {
            const fileRecord = await resolveWorkspaceFileReference(workspaceId, filePath)
            if (fileRecord) {
              const buffer = await fetchWorkspaceFileBuffer(fileRecord)
              const base64 = buffer.toString('base64')
              const mime = fileRecord.type || 'image/png'
              parts.push({
                inlineData: { mimeType: mime, data: base64 },
              })
              logger.info('Loaded reference image', {
                filePath,
                name: fileRecord.name,
                size: buffer.length,
                mimeType: mime,
              })
            } else {
              logger.warn('Reference file not found, skipping', { filePath })
            }
          } catch (err) {
            logger.warn('Failed to load reference image, skipping', {
              filePath,
              error: toError(err).message,
            })
          }
        }