千帆图像生成
Spring AI 支持 QianFan 的图像生成模型 CogView。
先决条件
你需要创建一个 API 来访问 QianFan 语言模型。
在 QianFan 注册页面 创建一个账户,并在 API Keys 页面 生成 token。Spring AI 项目定义了一个名为 spring.ai.qianfan.api-key
和 spring.ai.qianfan.secret-key
的配置属性。你应该将从 API Keys 页面 获取的 API Key
和 Secret Key
的值设置为这些属性。导出环境变量是设置该配置属性的一种方式:
export SPRING_AI_QIANFAN_API_KEY=<INSERT API KEY HERE>
export SPRING_AI_QIANFAN_SECRET_KEY=<INSERT SECRET KEY HERE>
添加仓库和 BOM
Spring AI 的工件发布在 Maven Central 和 Spring Snapshot 仓库中。请参考仓库部分,将这些仓库添加到您的构建系统中。
为了帮助进行依赖管理,Spring AI 提供了一个 BOM(物料清单),以确保在整个项目中使用一致的 Spring AI 版本。请参考 依赖管理 部分,将 Spring AI BOM 添加到您的构建系统中。
自动配置
Spring AI 为 QianFan 聊天客户端提供了 Spring Boot 自动配置。要启用它,请将以下依赖项添加到项目的 Maven pom.xml
文件中:
<dependency>
<groupId>org.springframework.ai</groupId>
<artifactId>spring-ai-qianfan-spring-boot-starter</artifactId>
</dependency>
或者添加到你的 Gradle build.gradle
构建文件中。
dependencies {
implementation 'org.springframework.ai:spring-ai-qianfan-spring-boot-starter'
}
请参考依赖管理部分,将 Spring AI BOM 添加到您的构建文件中。
图像生成属性
前缀 spring.ai.qianfan.image
是一个属性前缀,用于配置 QianFan 的 ImageModel
实现。
属性 | 描述 | 默认值 |
---|---|---|
spring.ai.qianfan.image.enabled | 启用 QianFan 图像模型。 | true |
spring.ai.qianfan.image.base-url | 可选,覆盖 spring.ai.qianfan.base-url 以提供特定于聊天的 URL | - |
spring.ai.qianfan.image.api-key | 可选,覆盖 spring.ai.qianfan.api-key 以提供特定于聊天的 API 密钥 | - |
spring.ai.qianfan.image.secret-key | 可选,覆盖 spring.ai.qianfan.secret-key 以提供特定于聊天的密钥 | - |
spring.ai.qianfan.image.options.model | 用于图像生成的模型。 | sd_xl |
spring.ai.qianfan.image.options.user | 代表最终用户的唯一标识符,可帮助 QianFan 监控和检测滥用行为。 | - |
连接属性
前缀 spring.ai.qianfan
用作属性前缀,允许你连接到 QianFan。
属性 | 描述 | 默认值 |
---|---|---|
spring.ai.qianfan.base-url | 连接的 URL | aip.baidubce.com/rpc/2.0/ai_custom |
spring.ai.qianfan.api-key | API 密钥 | - |
spring.ai.qianfan.secret-key | 密钥 | - |
配置属性
重试属性
前缀 spring.ai.retry
用作属性前缀,允许你配置 QianFan Image 客户端的重试机制。
属性 | 描述 | 默认值 |
---|---|---|
spring.ai.retry.max-attempts | 最大重试次数。 | 10 |
spring.ai.retry.backoff.initial-interval | 指数退避策略的初始休眠时长。 | 2 秒 |
spring.ai.retry.backoff.multiplier | 退避间隔的乘数。 | 5 |
spring.ai.retry.backoff.max-interval | 最大退避时长。 | 3 分钟 |
spring.ai.retry.on-client-errors | 如果为 false,抛出 NonTransientAiException,并且不对 4xx 客户端错误代码进行重试。 | false |
spring.ai.retry.exclude-on-http-codes | 不应触发重试的 HTTP 状态码列表(例如抛出 NonTransientAiException)。 | 空 |
spring.ai.retry.on-http-codes | 应触发重试的 HTTP 状态码列表(例如抛出 TransientAiException)。 | 空 |
运行时选项
QianFanImageOptions.java 提供了模型配置,例如使用的模型、质量、大小等。
在启动时,默认选项可以通过 QianFanImageModel(QianFanImageApi qianFanImageApi)
构造函数和 withDefaultOptions(QianFanImageOptions defaultOptions)
方法进行配置。或者,也可以使用前面描述的 spring.ai.qianfan.image.options.*
属性进行配置。
在运行时,你可以通过向 ImagePrompt
调用添加新的、特定于请求的选项来覆盖默认选项。例如,要覆盖 QianFan 特定的选项,如质量和要生成的图像数量,可以使用以下代码示例:
ImageResponse response = qianFanImageModel.call(
new ImagePrompt("A light cream colored mini golden doodle",
QianFanImageOptions.builder()
.N(4)
.height(1024)
.width(1024).build())
);
除了模型特定的 QianFanImageOptions 外,你还可以使用一个可移植的 ImageOptions 实例,该实例可以通过 ImageOptionsBuilder#builder() 创建。