智谱

本站(springdoc.cn)中的内容来源于 spring.io ,原始版权归属于 spring.io。由 springdoc.cn 进行翻译,整理。可供个人学习、研究,未经许可,不得进行任何转载、商用或与之相关的行为。 商标声明:Spring 是 Pivotal Software, Inc. 在美国以及其他国家的商标。

Spring AI 支持智谱 AI(ZhiPu AI)的多种语言模型。你可以通过智谱 AI 模型交互,构建基于智谱 AI 的多语言对话助手。

先决条件

你需要创建智谱 AI(ZhiPuAI)的 API Key 以访问其语言模型。

智谱 AI 注册页面 创建账户,并于 API Key 页面 生成访问令牌。

Spring AI 项目定义了名为 spring.ai.zhipuai.api-key 的配置属性,应将其设置为从 API Key 页面获取的密钥值。

你可以在 application.properties 文件中设置此配置属性:

spring.ai.zhipuai.api-key=<your-zhipuai-api-key>

为提升处理 API Key 等敏感信息的安全性,可使用 Spring 表达式语言 (SpEL)引用自定义环境变量:

# In application.yml
spring:
  ai:
    zhipuai:
      api-key: ${ZHIPUAI_API_KEY}
# In your environment or .env file
export ZHIPUAI_API_KEY=<your-zhipuai-api-key>

你也可以在应用程序代码中以编程方式设置此配置:

// Retrieve API key from a secure source or environment variable
String apiKey = System.getenv("ZHIPUAI_API_KEY");

添加仓库和 BOM

Spring AI 构件已发布至 Maven Central 和 Spring Snapshot 仓库。请参阅 构件仓库 章节将这些仓库添加到构建系统中。

为统一依赖版本管理,Spring AI 提供 BOM 确保项目中使用一致的 Spring AI 版本。请参考 依赖管理 章节将 Spring AI BOM 添加到构建系统中。

自动配置

Spring AI 自动配置及 Starter 模块的 Artifact 命名已发生重大变更。具体升级说明请参阅 更新文档

Spring AI 为 智谱 AI 聊天客户端提供了 Spring Boot 自动配置。要启用该功能,请将以下依赖项添加到你项目的 Maven pom.xml 文件中:

<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-starter-model-zhipuai</artifactId>
</dependency>

或添加到你的 Gradle build.gradle 构建文件中:

dependencies {
    implementation 'org.springframework.ai:spring-ai-starter-model-zhipuai'
}
请参阅 依赖管理 部分,将 Spring AI BOM 添加到你的构建文件中。

聊天配置

重试配置

前缀 spring.ai.retry 用作属性前缀,用于配置智谱 AI 聊天模型的重试机制。

属性 说明 默认值

spring.ai.retry.max-attempts

最大重试尝试次数。

10

spring.ai.retry.backoff.initial-interval

指数退避策略的初始休眠时长。

2 sec.

spring.ai.retry.backoff.multiplier

退避间隔乘数。

5

spring.ai.retry.backoff.max-interval

最大退避时长。

3 min.

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

连接属性

前缀 spring.ai.zhipuai 用作连接至智谱 AI 的属性配置前缀。

属性 说明 默认值

spring.ai.zhipuai.base-url

The URL to connect to

open.bigmodel.cn/api/paas

spring.ai.zhipuai.api-key

The API Key

-

配置属性

聊天自动配置的启用和禁用现在通过顶级属性配置,前缀为 spring.ai.model.chat

  • 启用:spring.ai.model.chat=zhipuai(默认启用)

  • 禁用:spring.ai.model.chat=none(或任何不匹配 zhipuai 的值)

此变更是为了支持多模型配置。

前缀 spring.ai.zhipuai.chat 是用于配置智谱 AI 聊天模型实现的属性前缀。

属性 说明 默认值

spring.ai.zhipuai.chat.enabled (已移除,不再有效)

启用智谱AI聊天模型。

true

spring.ai.model.chat

启用智谱 AI 聊天模型。

zhipuai

spring.ai.zhipuai.chat.base-url

可选参数,用于覆盖 spring.ai.zhipuai.base-url 以指定聊天专用URL

open.bigmodel.cn/api/paas

spring.ai.zhipuai.chat.api-key

可选参数,用于覆盖 spring.ai.zhipuai.api-key 以指定聊天专用 API Key

-

spring.ai.zhipuai.chat.options.model

这是要使用的智谱 AI 聊天模型。

GLM-3-Turbo(GLM-3-Turbo、GLM-4、GLM-4-Air、GLM-4-AirX、GLM-4-Flash 和 GLM-4V 均指向最新模型版本)

spring.ai.zhipuai.chat.options.maxTokens

聊天补全中生成的最大 Token 数。输入 Token 和生成 Token 的总长度受模型上下文长度限制。

-

spring.ai.zhipuai.chat.options.temperature

使用的采样温度值,范围在 0 到 1 之间。较高的值(如 0.8)会使输出更随机,而较低的值(如 0.2)会使输出更集中和确定。通常建议调整此参数或 top_p 中的一个,而非同时调整两者。

0.7

spring.ai.zhipuai.chat.options.topP

温度采样的替代方案,称为核采样(nucleus sampling),模型仅考虑概率质量之和达到 top_p 的 token 结果。例如 0.1 表示仅考虑概率质量前 10% 的 token。通常建议调整此参数或 temperature 中的一个,而非同时调整两者。

1.0

spring.ai.zhipuai.chat.options.stop

模型将在遇到 stop 指定的字符时停止生成,当前仅支持一个停止词,格式为 ["stop_word1"]

-

spring.ai.zhipuai.chat.options.user

一个代表你的终端用户的唯一标识符,可帮助智谱 AI 监控和检测滥用行为。

-

spring.ai.zhipuai.chat.options.requestId

该参数由客户端传入,必须保证唯一性,用于区分每次请求的唯一标识。若客户端未提供,平台将默认生成。

-

spring.ai.zhipuai.chat.options.doSample

do_sample 设置为 true 时,将启用采样策略。如果 do_samplefalse,则采样策略参数 temperaturetop_p 将不会生效。

true

spring.ai.zhipuai.chat.options.proxy-tool-calls

若为 true,Spring AI 将不会在内部处理函数调用,而是将其代理给客户端。此时由客户端负责处理这些函数调用,将其分发到相应的函数并返回结果。若为 false(默认值),Spring AI 将在内部处理函数调用。该配置仅适用于支持函数调用的聊天模型。

false

你可以为 ChatModel 实现覆盖通用的 spring.ai.zhipuai.base-urlspring.ai.zhipuai.api-key 配置。若设置了 spring.ai.zhipuai.chat.base-urlspring.ai.zhipuai.chat.api-key 属性,它们将优先于通用配置生效。此功能适用于需要为不同模型和不同模型端点使用不同智谱 AI 账户的场景。
所有以 spring.ai.zhipuai.chat.options 为前缀的属性,都可通过在 Prompt 调用中添加请求特定的 运行时选项 来覆盖。

运行时选项

ZhiPuAiChatOptions.java 提供模型配置参数,包括使用的模型、温度值(Temperature)、频率惩罚(Frequency Penalty)等。

启动时,默认选项可通过 ZhiPuAiChatModel(api, options) 构造函数或 spring.ai.zhipuai.chat.options.* 属性进行配置。

运行时,你可以通过在 Prompt 调用中添加新的请求特定选项来覆盖默认配置。例如,针对特定请求覆盖默认模型和温度值:

ChatResponse response = chatModel.call(
    new Prompt(
        "Generate the names of 5 famous pirates.",
        ZhiPuAiChatOptions.builder()
            .model(ZhiPuAiApi.ChatModel.GLM_3_Turbo.getValue())
            .temperature(0.5)
        .build()
    ));
除了模型专用的 ZhiPuAiChatOptions 外,你还可以使用通过 ChatOptionsBuilder#builder() 创建的可移植 ChatOptions 实例。

示例 Controller

创建 一个新的 Spring Boot 项目,并将 spring-ai-starter-model-zhipuai 添加到你的 pom(或gradle)依赖中。

src/main/resources 目录下添加 application.properties 文件,用于启用和配置智谱 AI 聊天模型:

spring.ai.zhipuai.api-key=YOUR_API_KEY
spring.ai.zhipuai.chat.options.model=glm-4-air
spring.ai.zhipuai.chat.options.temperature=0.7
api-key 替换为你的智谱 AI 凭证。

这将创建一个可注入到你类中的 ZhiPuAiChatModel 实现。以下是一个简单的 @Controller 类示例,该类使用聊天模型进行文本生成。

@RestController
public class ChatController {

    private final ZhiPuAiChatModel chatModel;

    @Autowired
    public ChatController(ZhiPuAiChatModel chatModel) {
        this.chatModel = chatModel;
    }

    @GetMapping("/ai/generate")
    public Map generate(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
        return Map.of("generation", this.chatModel.call(message));
    }

    @GetMapping("/ai/generateStream")
	public Flux<ChatResponse> generateStream(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
        var prompt = new Prompt(new UserMessage(message));
        return this.chatModel.stream(prompt);
    }
}

手动配置

ZhiPuAiChatModel 实现了 ChatModelStreamingChatModel 接口,并使用 底层智谱 Api 客户端 连接至智谱 AI 服务。

spring-ai-zhipuai 依赖添加到你项目的 Maven pom.xml 文件中:

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

或添加到你的 Gradle build.gradle 构建文件中。

dependencies {
    implementation 'org.springframework.ai:spring-ai-zhipuai'
}
请参考 依赖管理 部分,将 Spring AI BOM 添加到你的构建文件中。

接下来,创建一个 ZhiPuAiChatModel 并使用它进行文本生成:

var zhiPuAiApi = new ZhiPuAiApi(System.getenv("ZHIPU_AI_API_KEY"));

var chatModel = new ZhiPuAiChatModel(this.zhiPuAiApi, ZhiPuAiChatOptions.builder()
                .model(ZhiPuAiApi.ChatModel.GLM_3_Turbo.getValue())
                .temperature(0.4)
                .maxTokens(200)
                .build());

ChatResponse response = this.chatModel.call(
    new Prompt("Generate the names of 5 famous pirates."));

// Or with streaming responses
Flux<ChatResponse> streamResponse = this.chatModel.stream(
    new Prompt("Generate the names of 5 famous pirates."));

ZhiPuAiChatOptions 为聊天请求提供配置信息,其内置的 ZhiPuAiChatOptions.Builder 为 Fluent 式选项构建器。

底层智谱 Api 客户端

ZhiPuAiApi 是面向 ZhiPu AI API 的轻量级 Java 客户端。

以下是通过代码使用该 API 的简单示例:

ZhiPuAiApi zhiPuAiApi =
    new ZhiPuAiApi(System.getenv("ZHIPU_AI_API_KEY"));

ChatCompletionMessage chatCompletionMessage =
    new ChatCompletionMessage("Hello world", Role.USER);

// Sync request
ResponseEntity<ChatCompletion> response = this.zhiPuAiApi.chatCompletionEntity(
    new ChatCompletionRequest(List.of(this.chatCompletionMessage), ZhiPuAiApi.ChatModel.GLM_3_Turbo.getValue(), 0.7, false));

// Streaming request
Flux<ChatCompletionChunk> streamResponse = this.zhiPuAiApi.chatCompletionStream(
        new ChatCompletionRequest(List.of(this.chatCompletionMessage), ZhiPuAiApi.ChatModel.GLM_3_Turbo.getValue(), 0.7, true));

更多详细信息请参阅 ZhiPuAiApi.java 的 JavaDoc 文档。

ZhiPuAiApi 示例

ZhiPuAiApiIT.java 测试类提供了使用该轻量级库的通用示例。