Spring AI学习笔记三十八:聊天模型之ZhiPu AI(二)

spring.ai.zhipuai.api-key=YOUR_API_KEYspring.ai.zhipuai.chat.options.model=glm-4-airspring.ai.zhipuai.chat.options.temperature=0.7

@RestControllerpublic 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(value = "/ai/generateStream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)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);    }}

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

dependencies {    implementation 'org.springframework.ai:spring-ai-zhipuai'}

var zhiPuAiApi = new ZhiPuAiApi(System.getenv("ZHIPU_AI_API_KEY"));var chatModel = new ZhiPuAiChatModel(this.zhiPuAiApi, ZhiPuAiChatOptions.builder()                .model(ZhiPuAiApi.ChatModel.GLM_4_Air.getValue())                .temperature(0.4)                .maxTokens(200)                .build());ChatResponse response = this.chatModel.call(    new Prompt("Generate the names of 5 famous pirates."));// Or with streaming responsesFlux<ChatResponse> streamResponse = this.chatModel.stream(    new Prompt("Generate the names of 5 famous pirates."));

ZhiPuAiApi zhiPuAiApi =    new ZhiPuAiApi(System.getenv("ZHIPU_AI_API_KEY"));ChatCompletionMessage chatCompletionMessage =    new ChatCompletionMessage("Hello world", Role.USER);// Sync requestResponseEntity<ChatCompletion> response = this.zhiPuAiApi.chatCompletionEntity(    new ChatCompletionRequest(List.of(this.chatCompletionMessage), ZhiPuAiApi.ChatModel.GLM_4_Air.getValue(), 0.7, false));// Streaming requestFlux<ChatCompletionChunk> streamResponse = this.zhiPuAiApi.chatCompletionStream(        new ChatCompletionRequest(List.of(this.chatCompletionMessage), ZhiPuAiApi.ChatModel.GLM_4_Air.getValue(), 0.7, true));

import java.util.List;import java.util.Objects;import org.junit.jupiter.api.Test;import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;import reactor.core.publisher.Flux;import org.springframework.ai.zhipuai.api.ZhiPuAiApi.ChatCompletion;import org.springframework.ai.zhipuai.api.ZhiPuAiApi.ChatCompletionChunk;import org.springframework.ai.zhipuai.api.ZhiPuAiApi.ChatCompletionMessage;import org.springframework.ai.zhipuai.api.ZhiPuAiApi.ChatCompletionMessage.Role;import org.springframework.ai.zhipuai.api.ZhiPuAiApi.ChatCompletionRequest;import org.springframework.ai.zhipuai.api.ZhiPuAiApi.Embedding;import org.springframework.ai.zhipuai.api.ZhiPuAiApi.EmbeddingList;import org.springframework.http.ResponseEntity;import static org.assertj.core.api.Assertions.assertThat;/** * @author Geng Rong * @author YunKui Lu */@EnabledIfEnvironmentVariable(named = "ZHIPU_AI_API_KEY", matches = ".+")public class ZhiPuAiApiIT {ZhiPuAiApi zhiPuAiApi = ZhiPuAiApi.builder().apiKey(System.getenv("ZHIPU_AI_API_KEY")).build();@Testvoid chatCompletionEntity() {ChatCompletionMessage chatCompletionMessage = new ChatCompletionMessage("Hello world", Role.USER);		ResponseEntity<ChatCompletion> response = this.zhiPuAiApi			.chatCompletionEntity(new ChatCompletionRequest(List.of(chatCompletionMessage), "glm-3-turbo", 0.7, false));		assertThat(response).isNotNull();		assertThat(response.getBody()).isNotNull();	}@Testvoid chatCompletionEntityWithMoreParams() {ChatCompletionMessage chatCompletionMessage = new ChatCompletionMessage("Hello world", Role.USER);		ResponseEntity<ChatCompletion> response = this.zhiPuAiApi			.chatCompletionEntity(new ChatCompletionRequest(List.of(chatCompletionMessage), "glm-4-flash", 1024, null,false, 0.95, 0.7, null, null, null, "test_request_id", false, null, null));		assertThat(response).isNotNull();		assertThat(response.getBody()).isNotNull();		assertThat(response.getBody().usage()).isNotNull();	}@Testvoid chatCompletionStream() {ChatCompletionMessage chatCompletionMessage = new ChatCompletionMessage("Hello world", Role.USER);		Flux<ChatCompletionChunk> response = this.zhiPuAiApi			.chatCompletionStream(new ChatCompletionRequest(List.of(chatCompletionMessage), "glm-4-flash", 0.7, true));		assertThat(response).isNotNull();		List<ChatCompletionChunk> chunks = response.collectList().block();		assertThat(chunks).isNotNull();		assertThat(chunks.get(chunks.size() - 1).usage()).isNotNull();	}@Testvoid embeddings() {		ResponseEntity<EmbeddingList<Embedding>> response = this.zhiPuAiApi			.embeddings(new ZhiPuAiApi.EmbeddingRequest<>("Hello world"));		assertThat(response).isNotNull();		assertThat(Objects.requireNonNull(response.getBody()).data()).hasSize(1);		assertThat(response.getBody().data().get(0).embedding()).hasSize(1024);	}@Testvoid embeddingsWithDimensions() {		ResponseEntity<EmbeddingList<Embedding>> response = this.zhiPuAiApi			.embeddings(new ZhiPuAiApi.EmbeddingRequest<>("Hello world",					ZhiPuAiApi.EmbeddingModel.Embedding_3.getValue(), 1536));		assertThat(response).isNotNull();		assertThat(Objects.requireNonNull(response.getBody()).data()).hasSize(1);		assertThat(response.getBody().data().get(0).embedding()).hasSize(1536);	}}