千帆图像生成

Spring AI 支持 CogView,这是来自千帆的图像生成模型。

先决条件

您需要使用千帆创建一个 API 才能访问千帆语言模型。

千帆注册页面 创建一个帐户,并在 API 密钥页面 生成令牌。Spring AI 项目定义了一个名为 spring.ai.qianfan.api-keyspring.ai.qianfan.secret-key 的配置属性。您应该将其设置为从 API 密钥页面 获取的 API 密钥密钥 的值。导出环境变量是设置该配置属性的一种方法。

export SPRING_AI_QIANFAN_API_KEY=<INSERT API KEY HERE>
export SPRING_AI_QIANFAN_SECRET_KEY=<INSERT SECRET KEY HERE>

添加存储库和 BOM

Spring AI 工件发布在 Spring 里程碑和快照存储库中。请参阅 存储库 部分,将这些存储库添加到您的构建系统中。

为了帮助进行依赖项管理,Spring AI 提供了一个 BOM(物料清单)以确保在整个项目中使用一致版本的 Spring AI。请参阅 依赖项管理 部分,将 Spring AI BOM 添加到您的构建系统中。

自动配置

Spring AI 为千帆聊天客户端提供 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 是允许您配置千帆的 ImageModel 实现的属性前缀。

属性

描述

默认值

spring.ai.qianfan.image.enabled

启用千帆图像模型。

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-key

-

spring.ai.qianfan.image.secret-key

可选地覆盖 spring.ai.qianfan.secret-key 以提供特定于聊天的 secret-key

-

spring.ai.qianfan.image.options.model

用于图像生成的模型。

sd_xl

spring.ai.qianfan.image.options.user

表示您的最终用户的唯一标识符,这可以帮助千帆监控和检测滥用行为。

-

连接属性

前缀 spring.ai.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 用作属性前缀,允许您配置千帆图像客户端的重试机制。

属性 描述 默认值

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()
                .withN(4)
                .withHeight(1024)
                .withWidth(1024).build())

);
除了特定于模型的QianFanImageOptions之外,您还可以使用可移植的ImageOptions实例,该实例使用ImageOptionsBuilder#builder()创建。