千帆聊天

Spring AI 支援來自千帆的各種 AI 語言模型。您可以使用千帆語言模型互動,並基於千帆模型建立多語言對話助理。

先決條件

您需要使用千帆建立 API 以存取千帆語言模型。

千帆註冊頁面 建立帳戶,並在 API 金鑰頁面 產生權杖。Spring AI 專案定義了名為 spring.ai.qianfan.api-keyspring.ai.qianfan.secret-key 的組態屬性。您應將其設定為從 API 金鑰頁面 取得的 API 金鑰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 成品發佈在 Spring Milestone 和 Snapshot 儲存庫中。請參閱 儲存庫 章節,將這些儲存庫新增至您的建置系統。

為了協助管理依賴項,Spring AI 提供了 BOM(物料清單),以確保在整個專案中使用一致的 Spring AI 版本。請參閱 依賴項管理 章節,將 Spring AI BOM 新增至您的建置系統。

自動組態

Spring AI 為 Azure 千帆嵌入用戶端提供 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.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)。

連線屬性

前綴 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

Secret Key

-

組態屬性

前綴 spring.ai.qianfan.embedding 是屬性前綴,用於組態千帆的 EmbeddingClient 實作。

屬性 描述 預設值

spring.ai.qianfan.embedding.enabled

啟用千帆嵌入用戶端。

true

spring.ai.qianfan.embedding.base-url

可選覆寫 spring.ai.qianfan.base-url 以提供嵌入特定的 URL

-

spring.ai.qianfan.embedding.api-key

可選覆寫 spring.ai.qianfan.api-key 以提供嵌入特定的 API 金鑰

-

spring.ai.qianfan.embedding.secret-key

可選覆寫 spring.ai.qianfan.secret-key 以提供嵌入特定的 Secret Key

-

spring.ai.qianfan.embedding.options.model

要使用的模型

bge_large_zh

您可以為 ChatModelEmbeddingModel 實作覆寫通用的 spring.ai.qianfan.base-urlspring.ai.qianfan.api-keyspring.ai.qianfan.secret-key。如果設定了 spring.ai.qianfan.chat.base-urlspring.ai.qianfan.chat.api-keyspring.ai.qianfan.chat.secret-key 屬性,則優先於通用屬性。同樣地,如果設定了 spring.ai.qianfan.chat.base-urlspring.ai.qianfan.chat.api-keyspring.ai.qianfan.chat.secret-key 屬性,則優先於通用屬性。如果您想為不同的模型和不同的模型端點使用不同的千帆帳戶,這會很有用。
所有帶有前綴 spring.ai.qianfan.embedding.options 的屬性都可以在執行時被覆寫,方法是在 EmbeddingRequest 呼叫中新增請求特定的 執行階段選項

執行階段選項

QianFanEmbeddingOptions.java 提供了千帆組態,例如要使用的模型等等。

預設選項也可以使用 spring.ai.qianfan.embedding.options 屬性進行組態。

在啟動時,使用 QianFanEmbeddingModel 建構子來設定用於所有嵌入請求的預設選項。在執行時,您可以使用 QianFanEmbeddingOptions 實例作為 EmbeddingRequest 的一部分來覆寫預設選項。

例如,為特定請求覆寫預設模型名稱

EmbeddingResponse embeddingResponse = embeddingClient.call(
    new EmbeddingRequest(List.of("Hello World", "World is big and salvation is near"),
        QianFanEmbeddingOptions.builder()
            .withModel("Different-Embedding-Model-Deployment-Name")
        .build()));

範例控制器

這將建立一個 EmbeddingClient 實作,您可以將其注入到您的類別中。以下是一個簡單的 @Controller 類別範例,該類別使用 EmbeddingClient 實作。

spring.ai.qianfan.api-key=YOUR_API_KEY
spring.ai.qianfan.secret-key=YOUR_SECRET_KEY
spring.ai.qianfan.embedding.options.model=tao_8k
@RestController
public class EmbeddingController {

    private final EmbeddingClient embeddingClient;

    @Autowired
    public EmbeddingController(EmbeddingClient embeddingClient) {
        this.embeddingClient = embeddingClient;
    }

    @GetMapping("/ai/embedding")
    public Map embed(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
        EmbeddingResponse embeddingResponse = this.embeddingClient.embedForResponse(List.of(message));
        return Map.of("embedding", embeddingResponse);
    }
}

手動組態

如果您未使用 Spring Boot,您可以手動組態千帆嵌入用戶端。為此,請將 spring-ai-qianfan 依賴項新增至您專案的 Maven pom.xml 檔案

<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-qianfan</artifactId>
</dependency>

或新增至您的 Gradle build.gradle 建置檔案。

dependencies {
    implementation 'org.springframework.ai:spring-ai-qianfan'
}
請參閱 依賴項管理 章節,將 Spring AI BOM 新增至您的建置檔案。
spring-ai-qianfan 依賴項也提供對 QianFanChatModel 的存取。有關 QianFanChatModel 的更多資訊,請參閱 千帆聊天用戶端 章節。

接下來,建立一個 QianFanEmbeddingModel 實例,並使用它來計算兩個輸入文字之間的相似度

var qianFanApi = new QianFanApi(System.getenv("MINIMAX_API_KEY"), System.getenv("QIANFAN_SECRET_KEY"));

var embeddingClient = new QianFanEmbeddingModel(qianFanApi)
    .withDefaultOptions(QianFanChatOptions.build()
        .withModel("bge_large_en")
        .build());

EmbeddingResponse embeddingResponse = this.embeddingClient
	.embedForResponse(List.of("Hello World", "World is big and salvation is near"));

QianFanEmbeddingOptions 提供了嵌入請求的組態資訊。選項類別提供了 builder() 以方便建立選項。