智谱 AI 图像生成
Spring AI 支持智谱 AI 的 CogView 图像生成模型。
先决条件
你需要创建一个 API 来访问 ZhiPuAI 的语言模型。
在 ZhiPu AI 注册页面 创建一个账户,并在 API Keys 页面 生成 token。Spring AI 项目定义了一个名为 spring.ai.zhipuai.api-key
的配置属性,您应该将其设置为从 API Keys 页面 获取的 API Key
的值。导出环境变量是设置该配置属性的一种方式:
export SPRING_AI_ZHIPU_AI_API_KEY=<INSERT KEY HERE>
添加仓库和 BOM
Spring AI 的构件已发布在 Maven Central 和 Spring Snapshot 仓库中。请参考 Repositories 部分,将这些仓库添加到您的构建系统中。
为了帮助进行依赖管理,Spring AI 提供了一个 BOM(物料清单)来确保在整个项目中使用一致的 Spring AI 版本。请参考 依赖管理 部分,将 Spring AI BOM 添加到你的构建系统中。
自动配置
Spring AI 为 ZhiPuAI 聊天客户端提供了 Spring Boot 自动配置。要启用它,请将以下依赖项添加到项目的 Maven pom.xml
文件中:
<dependency>
<groupId>org.springframework.ai</groupId>
<artifactId>spring-ai-zhipuai-spring-boot-starter</artifactId>
</dependency>
或添加到你的 Gradle build.gradle
构建文件中。
dependencies {
implementation 'org.springframework.ai:spring-ai-zhipuai-spring-boot-starter'
}
请参考 依赖管理 部分,将 Spring AI BOM 添加到您的构建文件中。
图像生成属性
前缀 spring.ai.zhipuai.image
是用于配置 ZhiPuAI 的 ImageModel
实现的属性前缀。
属性 | 描述 | 默认值 |
---|---|---|
spring.ai.zhipuai.image.enabled | 启用 ZhiPuAI 图像模型。 | true |
spring.ai.zhipuai.image.base-url | 可选,覆盖 spring.ai.zhipuai.base-url 以提供特定于聊天的 URL | - |
spring.ai.zhipuai.image.api-key | 可选,覆盖 spring.ai.zhipuai.api-key 以提供特定于聊天的 API 密钥 | - |
spring.ai.zhipuai.image.options.model | 用于图像生成的模型。 | cogview-3 |
spring.ai.zhipuai.image.options.user | 代表最终用户的唯一标识符,可以帮助 ZhiPuAI 监控和检测滥用行为。 | - |
连接属性
前缀 spring.ai.zhipuai
用作属性前缀,允许你连接到 ZhiPuAI。
属性 | 描述 | 默认值 |
---|---|---|
spring.ai.zhipuai.base-url | 连接的 URL | open.bigmodel.cn/api/paas |
spring.ai.zhipuai.api-key | API 密钥 | - |
配置属性
重试属性
前缀 spring.ai.retry
用作属性前缀,允许你为 ZhiPuAI 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)。 | 空 |
运行时选项
ZhiPuAiImageOptions.java 提供了模型配置,例如使用的模型、质量、大小等。
在启动时,默认选项可以通过 ZhiPuAiImageModel(ZhiPuAiImageApi zhiPuAiImageApi)
构造函数和 withDefaultOptions(ZhiPuAiImageOptions defaultOptions)
方法进行配置。或者,也可以使用之前描述的 spring.ai.zhipuai.image.options.*
属性进行配置。
在运行时,你可以通过在 ImagePrompt
调用中添加新的、特定于请求的选项来覆盖默认选项。例如,要覆盖 ZhiPuAI 的特定选项,如质量和生成的图像数量,可以使用以下代码示例:
ImageResponse response = zhiPuAiImageModel.call(
new ImagePrompt("A light cream colored mini golden doodle",
ZhiPuAiImageOptions.builder()
.quality("hd")
.N(4)
.height(1024)
.width(1024).build())
);
除了特定于模型的 ZhiPuAiImageOptions,你还可以使用一个可移植的 ImageOptions 实例,该实例通过 ImageOptionsBuilder#builder() 创建。