October 11, 2026

How to Build an Image Generation Plugin with the WordPress AI Client

← AI for Your Work

WebCrimp analysis and practical explanation. AI-assisted; reviewed before publication.

Quick Summary

This tutorial guides developers through building a WordPress plugin that integrates the built-in AI Client API to generate images directly within the Media Library. It explains the provider-agnostic architecture, prompt configuration, REST API creation, and frontend integration required to implement AI image generation seamlessly in WordPress.

What Happened?

The article details the development of the 'AI Client ImageGen' plugin for WordPress 7.0. It demonstrates how to use the `wp_ai_client_prompt()` function to build requests, configure image generation parameters (like orientation and output type), and handle the generated images via REST API endpoints. The plugin allows users to input text prompts, preview generated images, and save them as media attachments, all while leveraging the abstraction layer provided by the WordPress AI Client.

Why It Matters

The WordPress AI Client provides a unified, provider-agnostic interface for AI interactions, preventing vendor lock-in and simplifying plugin development. By abstracting the underlying AI providers (such as OpenAI, Google, or Anthropic), developers can write code once that works across multiple services. This reduces maintenance overhead, enhances portability, and gives site owners centralized control over their API credentials and model preferences, fostering a more scalable and flexible AI ecosystem within WordPress.

Who Can Use It?

AI/ML · Software Development · Graphic Design · Web Development · WordPress

How Can I Use This?

If you work in AI/ML, Software Development, Graphic Design, Web Development, WordPress

  • Software Developers and Web Developers can use this to add AI image generation features to custom sites or client projects without writing provider-specific integrations. AI/ML Engineers can leverage the model preference system (`using_model_preference()`) to optimize outputs for specific tasks while relying on fallbacks. Graphic Designers and Content Creators can benefit from faster visual asset generation directly within the WordPress editorial workflow. Product Managers can evaluate this abstraction for cost-effective feature implementation and easier provider migration.

AI Tip

Always perform a support check using `is_supported_for_image_generation()` before rendering AI features in the UI. This ensures the site is properly configured with a compatible provider, preventing runtime errors and avoiding unnecessary API calls or costs for unsupported capabilities.

AI Trick

Use `FileTypeEnum::inline()` to request base64-encoded image data directly in the API response. This allows you to display the generated image instantly in a preview modal without needing to upload it to a temporary server location first, streamlining the user review process before saving to the Media Library.

Example Workflow

  1. 1. Set up prerequisites: Install WordPress 7.0+ and configure an AI provider (e.g., OpenAI) in Settings > Connectors. 2. Create the plugin directory structure (`ai-client-imagegen/` with `includes/`, `src/`, and `build/` folders). 3. Write the prompt builder function in `includes/prompt.php` using `wp_ai_client_prompt()` and chain methods like `with_text()`, `as_output_file_type()`, and `as_output_media_orientation()`. 4. Register REST API routes in `includes/rest-api.php` with appropriate permissions (e.g., `upload_files`). 5. Develop the frontend UI in `src/index.ts` to handle prompt input and API calls. 6. Build the JavaScript assets, register them in `includes/admin.php`, and enqueue them conditionally on the Media Library screen based on support checks. Note: This workflow requires verification and testing in a staging environment before deploying to production.

Copy This Prompt

I am building a WordPress plugin to generate images in the Media Library. Write the PHP function to configure the AI Client prompt for a text prompt: '[prompt_text]', with an optional aspect ratio of '[orientation]' (square, landscape, or portrait). Use `FileTypeEnum::inline()` to get the image as base64 data, and set a model preference order of 'gpt-5.4', 'gemini-3.1-pro-preview', then fallback to any available model. Include a support check to ensure image generation is available before executing.

Recommended Tools

  • WordPress 7.0+
  • PHP 7.4+
  • Node.js 20+
  • Composer
  • AI Provider Account (OpenAI, Google, etc.)

Check official documentation, availability and pricing before using these suggestions.

Difficulty

Intermediate

Estimated Time Saved

Not measured; validate with your own workflow.

Source

WordPress Developer Blog – AI
How to build an image generation plugin with the WordPress AI Client
Original publication date: 2026-05-14T14:54:00+00:00