Spring AI 使用 OpenAI 图像生成

Spring AI 支持 OpenAI 的图像生成模型 DALL-E。

前提条件

你需要创建 OpenAI API 密钥才能访问 ChatGPT 模型。

在 OpenAI 注册页面创建账号，并在 API 密钥页面生成令牌。

Spring AI 项目定义了一个名为 spring.ai.openai.api-key 的配置属性，你需要将其设置为从 openai.com 获取的 API 密钥值。

你可以在 application.properties 文件中设置该配置属性：

spring.ai.openai.api-key=<your-openai-api-key>

为了在处理 API 密钥等敏感信息时提高安全性，你可以使用 Spring 表达式语言（SpEL）引用自定义环境变量：

# 在 application.yml 中
spring:
  ai:
    openai:
      api-key: ${OPENAI_API_KEY}
# 在你的环境变量或 .env 文件中
export OPENAI_API_KEY=<your-openai-api-key>

你也可以在应用程序代码中以编程方式设置此配置：

// 从安全来源或环境变量中获取 API 密钥
String apiKey = System.getenv("OPENAI_API_KEY");

自动配置

Spring AI 自动配置、启动器模块的构件名称发生了重大变化。有关详细信息，请参阅升级说明。

Spring AI 为 OpenAI 图像生成客户端提供了 Spring Boot 自动配置。要启用它，请在项目的 Maven pom.xml 文件中添加以下依赖：

<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-starter-model-openai</artifactId>
</dependency>

或者添加到 Gradle build.gradle 构建文件中：

dependencies {
    implementation 'org.springframework.ai:spring-ai-starter-model-openai'
}

请参阅依赖管理章节，将 Spring AI BOM 添加到构建文件中。

图像生成配置属性

连接属性

前缀 spring.ai.openai 用于配置连接到 OpenAI 的属性。

对于属于多个组织的用户（或通过旧版用户 API 密钥访问项目的用户），可以选择指定用于 API 请求的组织和项目。这些 API 请求的使用量将计入指定的组织和项目。

重试属性

前缀 spring.ai.retry 用于配置 OpenAI 图像客户端的重试机制。

配置属性

图像自动配置的启用和禁用现在通过前缀为 spring.ai.model.image 的顶级属性配置。

启用：spring.ai.model.image=openai（默认启用）

禁用：spring.ai.model.image=none（或任何不等于 openai 的值）

此更改用于支持多模型配置。

前缀 spring.ai.openai.image 用于配置 OpenAI 的 ImageModel 实现。

你可以覆盖通用的 spring.ai.openai.base-url、spring.ai.openai.api-key、spring.ai.openai.organization-id 和 spring.ai.openai.project-id 属性。spring.ai.openai.image 下的对应属性如果设置，优先级高于通用属性。如果你想为不同模型和模型端点使用不同的 OpenAI 账号，这会非常有用。

所有以 spring.ai.openai.image.options 为前缀的属性都可以在运行时覆盖。

运行时选项

OpenAiImageOptions.java 提供模型配置，例如使用的模型、质量、尺寸等。

启动时，可以通过 OpenAiImageModel(OpenAiImageApi openAiImageApi) 构造函数和 withDefaultOptions(OpenAiImageOptions defaultOptions) 方法配置默认选项。或者使用前文所述的 spring.ai.openai.image.options.* 属性。

运行时，你可以通过向 ImagePrompt 调用添加新的请求特定选项来覆盖默认选项。例如，要覆盖 OpenAI 特定选项（如质量和生成图像数量），请使用以下代码示例：

ImageResponse response = openaiImageModel.call(
        new ImagePrompt("A light cream colored mini golden doodle",
        OpenAiImageOptions.builder()
                .quality("hd")
                .N(4)
                .height(1024)
                .width(1024).build())

);

除了模型特定的 OpenAiImageOptions，你还可以使用通过 ImageOptionsBuilder#builder() 创建的通用 ImageOptions 实例。