OpenAI Image Generation
Spring AI supports DALL-E, the Image generation model from OpenAI.
Prerequisites
You will need to create an API key with OpenAI to access ChatGPT models.
Create an account at OpenAI signup page and generate the token on the API Keys page.
The Spring AI project defines a configuration property named spring.ai.openai.api-key
that you should set to the value of the API Key
obtained from openai.com.
Exporting an environment variable is one way to set that configuration property:
export SPRING_AI_OPENAI_API_KEY=<INSERT KEY HERE>
Auto-configuration
Spring AI provides Spring Boot auto-configuration for the OpenAI Image Generation Client.
To enable it add the following dependency to your project’s Maven pom.xml
file:
<dependency>
<groupId>org.springframework.ai</groupId>
<artifactId>spring-ai-openai-spring-boot-starter</artifactId>
</dependency>
or to your Gradle build.gradle
build file.
dependencies {
implementation 'org.springframework.ai:spring-ai-openai-spring-boot-starter'
}
Refer to the Dependency Management section to add the Spring AI BOM to your build file. |
Image Generation Properties
Connection Properties
The prefix spring.ai.openai
is used as the property prefix that lets you connect to OpenAI.
Property |
Description |
Default |
spring.ai.openai.base-url |
The URL to connect to |
|
spring.ai.openai.api-key |
The API Key |
- |
spring.ai.openai.organization-id |
Optionally you can specify which organization used for an API request. |
- |
spring.ai.openai.project-id |
Optionally, you can specify which project is used for an API request. |
- |
For users that belong to multiple organizations (or are accessing their projects through their legacy user API key), optionally, you can specify which organization and project is used for an API request. Usage from these API requests will count as usage for the specified organization and project. |
Retry Properties
The prefix spring.ai.retry
is used as the property prefix that lets you configure the retry mechanism for the OpenAI Image client.
Property | Description | Default |
---|---|---|
spring.ai.retry.max-attempts |
Maximum number of retry attempts. |
10 |
spring.ai.retry.backoff.initial-interval |
Initial sleep duration for the exponential backoff policy. |
2 sec. |
spring.ai.retry.backoff.multiplier |
Backoff interval multiplier. |
5 |
spring.ai.retry.backoff.max-interval |
Maximum backoff duration. |
3 min. |
spring.ai.retry.on-client-errors |
If false, throw a NonTransientAiException, and do not attempt retry for |
false |
spring.ai.retry.exclude-on-http-codes |
List of HTTP status codes that should not trigger a retry (e.g. to throw NonTransientAiException). |
empty |
spring.ai.retry.on-http-codes |
List of HTTP status codes that should trigger a retry (e.g. to throw TransientAiException). |
empty |
Configuration Properties
The prefix spring.ai.openai.image
is the property prefix that lets you configure the ImageModel
implementation for OpenAI.
Property |
Description |
Default |
spring.ai.openai.image.enabled |
Enable OpenAI image model. |
true |
spring.ai.openai.image.base-url |
Optional overrides the spring.ai.openai.base-url to provide chat specific url |
- |
spring.ai.openai.image.api-key |
Optional overrides the spring.ai.openai.api-key to provide chat specific api-key |
- |
spring.ai.openai.image.organization-id |
Optionally you can specify which organization used for an API request. |
- |
spring.ai.openai.image.project-id |
Optionally, you can specify which project is used for an API request. |
- |
spring.ai.openai.image.options.n |
The number of images to generate. Must be between 1 and 10. For dall-e-3, only n=1 is supported. |
- |
spring.ai.openai.image.options.model |
The model to use for image generation. |
OpenAiImageApi.DEFAULT_IMAGE_MODEL |
spring.ai.openai.image.options.quality |
The quality of the image that will be generated. HD creates images with finer details and greater consistency across the image. This parameter is only supported for dall-e-3. |
- |
spring.ai.openai.image.options.response_format |
The format in which the generated images are returned. Must be one of URL or b64_json. |
- |
|
The size of the generated images. Must be one of 256x256, 512x512, or 1024x1024 for dall-e-2. Must be one of 1024x1024, 1792x1024, or 1024x1792 for dall-e-3 models. |
- |
|
The width of the generated images. Must be one of 256, 512, or 1024 for dall-e-2. |
- |
|
The height of the generated images. Must be one of 256, 512, or 1024 for dall-e-2. |
- |
|
The style of the generated images. Must be one of vivid or natural. Vivid causes the model to lean towards generating hyper-real and dramatic images. Natural causes the model to produce more natural, less hyper-real looking images. This parameter is only supported for dall-e-3. |
- |
|
A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse. |
- |
You can override the common spring.ai.openai.base-url , spring.ai.openai.api-key , spring.ai.openai.organization-id and spring.ai.openai.project-id properties.
The spring.ai.openai.image.base-url , spring.ai.openai.image.api-key , spring.ai.openai.image.organization-id and spring.ai.openai.image.project-id properties if set take precedence over the common properties.
This is useful if you want to use different OpenAI accounts for different models and different model endpoints.
|
All properties prefixed with spring.ai.openai.image.options can be overridden at runtime.
|
Runtime Options
The OpenAiImageOptions.java provides model configurations, such as the model to use, the quality, the size, etc.
On start-up, the default options can be configured with the OpenAiImageModel(OpenAiImageApi openAiImageApi)
constructor and the withDefaultOptions(OpenAiImageOptions defaultOptions)
method. Alternatively, use the spring.ai.openai.image.options.*
properties described previously.
At runtime you can override the default options by adding new, request specific, options to the ImagePrompt
call.
For example to override the OpenAI specific options such as quality and the number of images to create, use the following code example:
ImageResponse response = openaiImageModel.call(
new ImagePrompt("A light cream colored mini golden doodle",
OpenAiImageOptions.builder()
.withQuality("hd")
.withN(4)
.withHeight(1024)
.withWidth(1024).build())
);
In addition to the model specific OpenAiImageOptions you can use a portable ImageOptions instance, created with the ImageOptionsBuilder#builder(). |