FAL Image/Video MCP Server

Instructions

Implementation Reference

• src/index.ts:495-563 (handler)

  Handler function that implements the core logic for the 'imagen4' tool. It processes input arguments, calls the FAL endpoint 'fal-ai/imagen4/preview' via fal.subscribe, handles the image results, downloads/processes images, and returns formatted content.

  private async handleImageGeneration(args: any, model: any) {
    const {
      prompt,
      image_size = 'landscape_4_3',
      num_inference_steps = 25,
      guidance_scale = 3.5,
      num_images = 1,
      negative_prompt,
      safety_tolerance,
      raw,
    } = args;
  
    try {
      // Configure FAL client lazily with query config override
      configureFalClient(this.currentQueryConfig);
      const inputParams: any = { prompt };
      
      // Add common parameters
      if (image_size) inputParams.image_size = image_size;
      if (num_images > 1) inputParams.num_images = num_images;
      
      // Add model-specific parameters based on model capabilities
      if (model.id.includes('flux') || model.id.includes('stable_diffusion')) {
        if (num_inference_steps) inputParams.num_inference_steps = num_inference_steps;
        if (guidance_scale) inputParams.guidance_scale = guidance_scale;
      }
      if ((model.id.includes('stable_diffusion') || model.id === 'ideogram_v3') && negative_prompt) {
        inputParams.negative_prompt = negative_prompt;
      }
      if (model.id.includes('flux_pro') && safety_tolerance) {
        inputParams.safety_tolerance = safety_tolerance;
      }
      if (model.id === 'flux_pro_ultra' && raw !== undefined) {
        inputParams.raw = raw;
      }
  
      const result = await fal.subscribe(model.endpoint, { input: inputParams });
      const imageData = result.data as FalImageResult;
  
      const processedImages = await downloadAndProcessImages(imageData.images, model.id);
  
      return {
        content: [
          {
            type: 'text',
            text: JSON.stringify({
              model: model.name,
              id: model.id,
              endpoint: model.endpoint,
              prompt,
              images: processedImages,
              metadata: inputParams,
              download_path: DOWNLOAD_PATH,
              data_url_settings: {
                enabled: ENABLE_DATA_URLS,
                max_size_mb: Math.round(MAX_DATA_URL_SIZE / 1024 / 1024),
              },
              autoopen_settings: {
                enabled: AUTOOPEN,
                note: AUTOOPEN ? "Files automatically opened with default application" : "Auto-open disabled"
              },
            }, null, 2),
          },
        ],
      };
    } catch (error) {
      throw new Error(`${model.name} generation failed: ${error}`);
    }
  }

• src/index.ts:101-101 (registration)

  Registration entry for the 'imagen4' tool in MODEL_REGISTRY.imageGeneration array, defining its ID, endpoint, name, and description.

  { id: 'imagen4', endpoint: 'fal-ai/imagen4/preview', name: 'Imagen 4', description: 'Google\'s latest text-to-image model' },

• src/index.ts:357-373 (schema)

  Dynamic input schema definition for imageGeneration tools like 'imagen4', including prompt, image_size, num_images, and model-specific parameters.

  if (category === 'imageGeneration') {
    baseSchema.inputSchema.properties = {
      prompt: { type: 'string', description: 'Text prompt for image generation' },
      image_size: { type: 'string', enum: ['square_hd', 'square', 'portrait_4_3', 'portrait_16_9', 'landscape_4_3', 'landscape_16_9'], default: 'landscape_4_3' },
      num_images: { type: 'number', default: 1, minimum: 1, maximum: 4 },
    };
    baseSchema.inputSchema.required = ['prompt'];
    
    // Add model-specific parameters
    if (model.id.includes('flux') || model.id.includes('stable_diffusion')) {
      baseSchema.inputSchema.properties.num_inference_steps = { type: 'number', default: 25, minimum: 1, maximum: 50 };
      baseSchema.inputSchema.properties.guidance_scale = { type: 'number', default: 3.5, minimum: 1, maximum: 20 };
    }
    if (model.id.includes('stable_diffusion') || model.id === 'ideogram_v3') {
      baseSchema.inputSchema.properties.negative_prompt = { type: 'string', description: 'Negative prompt' };
    }
  } else if (category === 'textToVideo') {

• src/index.ts:140-143 (helper)

  Helper function used to look up the model configuration (endpoint, etc.) by tool name 'imagen4' during tool call dispatch.

  function getModelById(id: string) {
    const allModels = getAllModels();
    return allModels.find(model => model.id === id);
  }

• src/index.ts:400-402 (registration)

  Registration of 'imagen4' tool schema in the listTools response by iterating over imageGeneration models.

  for (const model of MODEL_REGISTRY.imageGeneration) {
    tools.push(this.generateToolSchema(model, 'imageGeneration'));
  }