Lzh on GitHub

Stability AI 图像生成

Spring AI 支持 Stability AI 的 文本生成图像模型。

Spring AI 支持 Stability AI 的 文本生成图像模型

前置条件

你需要创建一个 Stability AI 的 API 密钥才能访问他们的 AI 模型。请参考 Stability AI 的 入门文档 获取 API 密钥。

Spring AI 项目定义了一个配置属性 spring.ai.stabilityai.api-key,你应将其设置为从 Stability AI 获取的 API 密钥值。

可以在 application.properties 文件中设置此配置属性:

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

为了在处理敏感信息(如 API 密钥)时增强安全性,你可以使用 Spring 表达式语言(SpEL)引用自定义环境变量:

# 在 application.yml 中
spring:
  ai:
    stabilityai:
      api-key: ${STABILITYAI_API_KEY}

# 在你的环境或 .env 文件中
export STABILITYAI_API_KEY=<your-stabilityai-api-key>

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

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

自动配置

Spring AI 自动配置和启动器模块的 artifact 名称发生了重大变化,请参考 升级说明 获取更多信息。

Spring AI 为 Stability AI 图像生成客户端提供了 Spring Boot 自动配置。要启用该功能,请将以下依赖添加到项目的 Maven pom.xml 文件中:

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

或者添加到 Gradle build.gradle 文件中:

dependencies {
    implementation 'org.springframework.ai:spring-ai-starter-model-stability-ai'
}
请参考 依赖管理 部分,将 Spring AI BOM 添加到构建文件中。

图像生成属性

spring.ai.stabilityai 是用于连接 Stability AI 的配置属性前缀。

属性描述默认值
spring.ai.stabilityai.base-url连接的 URLapi.stability.ai/v1
spring.ai.stabilityai.api-keyAPI 密钥-
图像自动配置的启用与禁用现在通过顶级属性 spring.ai.model.image 进行配置:
  • 启用:spring.ai.model.image=stabilityai(默认启用)
  • 禁用:spring.ai.model.image=none(或任何不匹配 stabilityai 的值)
此更改允许配置多个模型。

spring.ai.stabilityai.image 是用于配置 Stability AI ImageModel 实现的属性前缀。

属性描述默认值
spring.ai.stabilityai.image.enabled启用 Stability AI 图像模型(已移除,不再有效)true
spring.ai.model.image启用 Stability AI 图像模型stabilityai
spring.ai.stabilityai.image.base-url可选,覆盖 spring.ai.openai.base-url 提供特定 URLhttps://api.stability.ai/v1
spring.ai.stabilityai.image.api-key可选,覆盖 spring.ai.openai.api-key 提供特定 API Key-
spring.ai.stabilityai.image.option.n要生成的图像数量,必须在 1~10 之间1
spring.ai.stabilityai.image.option.model使用的 Stability AI 引擎/模型,模型通过 URL 作为路径参数传入stable-diffusion-v1-6
spring.ai.stabilityai.image.option.width生成图像的宽度(像素),需为 64 的倍数512
spring.ai.stabilityai.image.option.height生成图像的高度(像素),需为 64 的倍数512
spring.ai.stabilityai.image.option.responseFormat返回生成图像的格式,可为 "application/json""image/png"-
spring.ai.stabilityai.image.option.cfg_scale扩散过程严格遵循提示文本的程度,范围 0~357
spring.ai.stabilityai.image.option.clip_guidance_preset用于引导图像模型生成特定风格的样式预设NONE
spring.ai.stabilityai.image.option.sampler扩散过程使用的采样器,若未设置则自动选择-
spring.ai.stabilityai.image.option.seed随机噪声种子(可省略或使用 0 随机生成),有效范围 0~42949672950
spring.ai.stabilityai.image.option.steps扩散步骤数,范围 10~5030
spring.ai.stabilityai.image.option.style_preset用于引导图像模型生成特定风格的样式预设-

运行时选项

StabilityAiImageOptions.java 提供了模型配置选项,例如使用的模型、风格、尺寸等。

在应用启动时,可以通过 StabilityAiImageModel(StabilityAiApi stabilityAiApi, StabilityAiImageOptions options) 构造函数配置默认选项。或者,也可以使用之前介绍的 spring.ai.openai.image.options.* 属性进行配置。

在运行时,你可以通过在 ImagePrompt 调用中添加特定请求的选项来覆盖默认配置。例如,要覆盖 Stability AI 特定的选项,如图像质量和生成图像的数量,可以使用如下代码示例:

ImageResponse response = stabilityaiImageModel.call(
    new ImagePrompt("A light cream colored mini golden doodle",
        StabilityAiImageOptions.builder()
            .stylePreset("cinematic")
            .N(4)
            .height(1024)
            .width(1024)
            .build())
);
除了模型特定的 StabilityAiImageOptions 外,你还可以使用通用的 ImageOptions 实例,通过 ImageOptionsBuilder#builder() 创建。