Image generation

In Agents-Flex, the ability to generate images through AI is built-in.

Large model support

Agents-Flex image generation models support the following:

Large Language Model Name	Support Status	Description
Openai	✅ Supported	-
Stability	✅ Supported	-
GiteeAI - stable-diffusion-3-medium	✅ Supported	-
GiteeAI - FLUX.1-schnell	✅ Supported	-
GiteeAI - stable-diffusion-xl-base-1.0	✅ Supported	-
GiteeAI - Kolors	✅ Supported	-
SiliconFlow - Flux.1-schnell	✅ Supported	-
SiliconFlow - Stable Diffusion 3	✅ Supported	-
SiliconFlow - Stable Diffusion XL	✅ Supported	-
SiliconFlow - Stable Diffusion 2.1	✅ Supported	-
SiliconFlow - Stable Diffusion Turbo	✅ Supported	-
SiliconFlow - Stable Diffusion XL Turbo	✅ Supported	-
SiliconFlow - Stable Diffusion XL Lighting	✅ Supported	-
More	Planning...	Looking forward to PR

Example Code

 @Test
public void testGenImage(){
    GiteeImageModelConfig config = new GiteeImageModelConfig();
    config.setApiKey("****");

    //Step 1: Create an ImageModel
    ImageModel imageModel = new GiteeImageModel(config);

    //Step 2: Create pictures to generate prompt words and parameters
    GenerateImageRequest request = new GenerateImageRequest();
    request.setPrompt("A cute little tiger standing in the high-speed train");
    request.setSize(1024, 1024);

    //Step 3: Generate images through large models
    ImageResponse generate = imageModel.generate(request);
    System.out.println(generate);

    int index = 0;
    for (Image image : generate.getImages()) {
        //Step 4: Save the image locally
        image.writeToFile(new File("/image-path/"+(index++)+".jpg"));
    }
}

或者使用 OpenAI ImageModel

 @Test
public void testGenImage(){

    //Or use OpenAI ImageModel
    OpenAIImageModelConfig config = new OpenAIImageModelConfig();
    config.setApiKey("sk-5gqOclb****");
    ImageModel imageModel = new OpenAIImageModel(config);


    GenerateImageRequest request = new GenerateImageRequest();
    request.setPrompt("A cute little tiger standing in the high-speed train");
    request.setSize(1024, 1024);

    ImageResponse generate = imageModel.generate(request);
    System.out.println(generate);

    int index = 0;
    for (Image image : generate.getImages()) {
        image.writeToFile(new File("/image-path/"+(index++)+".jpg"));
    }
}

Or use SiliconFlowImageModel

 @Test
public void testGenImage(){

    //Or use  SiliconFlowImageModel
    SiliconflowImageModelConfig config = new SiliconflowImageModelConfig();
    config.setModel(SiliconflowImageModels.Stable_Diffusion_XL);
    config.setApiKey("sk-****");
    ImageModel imageModel = new OpenAIImageModel(config);


    GenerateImageRequest request = new GenerateImageRequest();
    request.setPrompt("A cute little tiger standing in the high-speed train");
    request.setSize(1024, 1024);

    ImageResponse generate = imageModel.generate(request);
    System.out.println(generate);

    int index = 0;
    for (Image image : generate.getImages()) {
        image.writeToFile(new File("/image-path/"+(index++)+".jpg"));
    }
}