feat: adds image generation capability to Workers AI API

[unsync]

Author Description

Implements support for an image generation feature

Summary By MatterAI

🔄 What Changed

This PR adds image generation capability to the Workers AI API by implementing a new endpoint. The implementation includes configuration for handling image generation requests with parameters like prompt, negative_prompt, steps, size, and seed. It also includes response transformation to properly format the image data returned from the Workers AI service.

🔍 Impact of the Change

This change extends the API's capabilities to include image generation, allowing clients to generate images using the Workers AI models. The implementation follows the same pattern as other Workers AI features, reusing common error handling and response transformation patterns.

📁 Total Files Changed

7 files changed with 133 additions and 100 deletions. Key changes include:

• Added new imageGenerate.ts file with configuration and response transformation
• Added common utils.ts file to centralize error handling
• Updated API configuration to support the new endpoint
• Refactored existing files to use the centralized error handling

🧪 Test Added

N/A - No tests were included in this PR.

🔒 Security Vulnerabilities

No security vulnerabilities detected.

Type of Change

• New feature (non-breaking change which adds functionality)
• Bug fix (non-breaking change which fixes an issue)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• Documentation update
• Refactoring (no functional changes)

How Has This Been Tested?

• Unit Tests
• Integration Tests
• Manual Testing

Screenshots (if applicable)

N/A

Checklist

• My code follows the style guidelines of this project
• I have performed a self-review of my own code
• I have made corresponding changes to the documentation
• My changes generate no new warnings
• I have added tests that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes

Related Issues

N/A

Quality Recommendations

1. Add input validation for the image size parameter to ensure it's in the correct format before parsing

2. Consider adding error handling for the case when size parameter is malformed

3. Add unit tests for the new image generation functionality

4. Add documentation for the supported image models and their specific requirements

Sequence Diagram

sequenceDiagram
    participant Client
    participant API
    participant WorkersAiAPI
    participant ImageGenerate
    
    Client->>API: POST /imageGenerate
    Note over Client,API: prompt, negative_prompt, steps, size, seed
    
    API->>WorkersAiAPI: Route to Workers AI API
    Note over API,WorkersAiAPI: Configure endpoint using model
    
    WorkersAiAPI->>ImageGenerate: Process request
    Note over WorkersAiAPI,ImageGenerate: Transform parameters
    
    ImageGenerate->>WorkersAiAPI: Generate image
    Note over ImageGenerate,WorkersAiAPI: Process with model parameters
    
    WorkersAiAPI-->>Client: Return response
    Note over WorkersAiAPI,Client: b64_json image data

Loading

[narengogi]

LGTM, only one comment

[matter-code-review]

Summary By MatterAI

🔄 What Changed

This PR adds image generation capability to the Workers AI API by implementing a new endpoint and the necessary configuration and response transformation logic. The implementation includes parameter handling for prompt, negative_prompt, steps, size, and seed parameters, as well as proper error handling and response formatting.

🔍 Impact of the Change

This feature enables users to generate images through the Workers AI API, expanding the platform's capabilities beyond text-based operations. The implementation follows the existing patterns for other API features while adding specific handling for image generation responses.

📁 Total Files Changed

7 files changed with 133 additions and 100 deletions. Key changes include:

• Added new imageGenerate.ts file with configuration and response transformation
• Added new utils.ts file to centralize error handling
• Updated API configuration to support the new endpoint
• Refactored error handling across multiple files

🧪 Test Added

N/A - No tests were included in this PR

🔒Security Vulnerabilities

N/A - No security vulnerabilities detected

Type of Change

• New feature (non-breaking change which adds functionality)
• Bug fix (non-breaking change which fixes an issue)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• Documentation update
• Refactoring (no functional changes)

How Has This Been Tested?

• Unit Tests
• Integration Tests
• Manual Testing

Checklist

• My code follows the style guidelines of this project
• I have performed a self-review of my own code
• I have added tests that prove my fix is effective or that my feature works
• My changes generate no new warnings

Sequence Diagram

sequenceDiagram
    participant Client
    participant Gateway
    participant WorkersAI_API
    
    Client->>Gateway: POST /imageGenerate
    Note over Client,Gateway: Payload: prompt, negative_prompt, steps, size, seed
    
    Gateway->>WorkersAI_API: Forward request to /{model}
    Note over Gateway,WorkersAI_API: Transform parameters according to WorkersAiImageGenerateConfig
    
    WorkersAI_API-->>Gateway: Return image generation response
    Note over WorkersAI_API,Gateway: Response contains base64 image data
    
    Gateway->>Gateway: Transform response with WorkersAiImageGenerateResponseTransform
    Note over Gateway: Convert to standardized ImageGenerateResponse format
    
    Gateway-->>Client: Return standardized response
    Note over Gateway,Client: Response: {created, data[{b64_json}], provider}

Loading