ZhiPuAI 圖像生成

Spring AI 支援來自 ZhiPuAI 的圖像生成模型 CogView。

先決條件

您需要向 ZhiPuAI 建立 API 以存取 ZhiPu AI 語言模型。

ZhiPu AI 註冊頁面 建立帳戶,並在 API 金鑰頁面 上產生權杖。Spring AI 專案定義了一個名為 spring.ai.zhipuai.api-key 的組態屬性,您應該將其設定為從 API 金鑰頁面 取得的 API 金鑰 值。匯出環境變數是設定該組態屬性的一種方法

export SPRING_AI_ZHIPU_AI_API_KEY=<INSERT KEY HERE>

新增儲存庫和 BOM

Spring AI 成品發佈在 Spring Milestone 和 Snapshot 儲存庫中。請參閱 儲存庫 章節,將這些儲存庫新增至您的建置系統。

為了協助管理依賴性,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)。

empty

spring.ai.retry.on-http-codes

應觸發重試的 HTTP 狀態代碼清單(例如,擲回 TransientAiException)。

empty

執行期選項

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

);
除了模型特定的 ZhiPuAiImageOptions 之外,您還可以使用可移植的 ImageOptions 實例,該實例是使用 ImageOptionsBuilder#builder() 建立的。