掘金 人工智能 04月28日 13:42
Spring AI应用系列——基于DeepSeek客户端的Chat应用
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

本文深入探讨了如何使用Spring AI框架中的DeepSeek客户端构建聊天应用。文章详细分析了DeepSeekChatClientController类的各个参数,包括chatModel、MessageChatMemoryAdvisor、SimpleLoggerAdvisor和OpenAiChatOptions,以及它们在实现聊天记忆、日志记录和参数设置中的作用。通过对自定义参数调用、默认提示语生成和流式生成等功能的分析,展示了如何利用DeepSeek模型生成响应。同时,通过单元测试验证了应用的稳定性和可靠性,为开发者提供了一个全面的实践指南。

🛠️ Spring AI框架提供了一套强大的工具集,用于构建基于AI的应用程序,本文重点介绍了如何利用该框架集成DeepSeek客户端。

🧠 DeepSeekChatClientController是应用的核心,它通过注入的DeepSeek模型实例(chatModel)生成聊天响应,并利用MessageChatMemoryAdvisor实现聊天记忆功能,保持多次交互的上下文。

⚙️ 通过设置OpenAiChatOptions中的temperature参数(例如0.7),可以平衡生成结果的创造性和稳定性。自定义参数调用允许开发者灵活地控制生成内容的主题和风格。

📡 应用支持流式生成,通过stream()方法启用流式模式,适合处理长文本生成任务,并能实时获取分块响应,提升用户体验。

一、引言

在当前的AI开发领域中,Spring AI框架提供了一套强大的工具集,用于构建基于AI的应用程序。本文将深入探讨如何使用Spring AI框架中的DeepSeek客户端来构建一个聊天应用。我们将详细分析DeepSeekChatClientController类中的各个参数,并通过单元测试验证其功能,确保应用的稳定性和可靠性。


二、项目结构与核心控制器分析

2.1 项目结构

在提供的代码文件中,我们可以看到以下关键文件:

    pom.xml:Maven项目的配置文件,定义了项目的依赖和构建信息。
<?xml version="1.0" encoding="UTF-8"?><!--   Copyright 2023-2024 the original author or authors.   Licensed under the Apache License, Version 2.0 (the "License");   you may not use this file except in compliance with the License.   You may obtain a copy of the License at        https://www.apache.org/licenses/LICENSE-2.0   Unless required by applicable law or agreed to in writing, software   distributed under the License is distributed on an "AS IS" BASIS,   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   See the License for the specific language governing permissions and   limitations under the License.--><project xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"         xmlns="http://maven.apache.org/POM/4.0.0"         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">    <modelVersion>4.0.0</modelVersion>    <parent>        <groupId>com.alibaba.cloud.ai</groupId>        <artifactId>deepseek-chat</artifactId>        <version>${revision}</version>        <relativePath>../pom.xml</relativePath>    </parent>    <artifactId>deepseek-chat-client</artifactId>    <version>${revision}</version>    <description>Spring AI Alibaba DeepSeek Chat Client Example</description>    <name>Spring AI Alibaba DeepSeek Chat Client Examples</name>    <build>        <plugins>            <plugin>                <groupId>org.springframework.boot</groupId>                <artifactId>spring-boot-maven-plugin</artifactId>                <version>${spring-boot.version}</version>            </plugin>            <plugin>                <groupId>org.apache.maven.plugins</groupId>                <artifactId>maven-deploy-plugin</artifactId>                <version>${maven-deploy-plugin.version}</version>            </plugin>        </plugins>    </build></project>
    DeepseekChatClientApplication.java:Spring Boot应用的入口类,负责启动应用程序。
package com.alibaba.cloud.ai.example.chat.deepseek;import org.springframework.boot.SpringApplication;import org.springframework.boot.autoconfigure.SpringBootApplication;@SpringBootApplicationpublic class DeepseekChatClientApplication {    public static void main (String[] args) {        SpringApplication.run(DeepseekChatClientApplication.class, args);    }}
    DeepSeekChatClientController.java:核心控制器,负责处理与DeepSeek模型的交互逻辑。
/* * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements.  See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You under the Apache License, Version 2.0 * (the "License"); you may not use this file except in compliance with * the License.  You may obtain a copy of the License at * *     http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */package com.alibaba.cloud.ai.example.chat.deepseek.controller;import org.springframework.ai.chat.client.ChatClient;import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;import org.springframework.ai.chat.memory.InMemoryChatMemory;import org.springframework.ai.chat.model.ChatModel;import org.springframework.ai.chat.model.ChatResponse;import org.springframework.ai.openai.OpenAiChatModel;import org.springframework.ai.openai.OpenAiChatOptions;import org.springframework.web.bind.annotation.GetMapping;import org.springframework.web.bind.annotation.RestController;import reactor.core.publisher.Flux;import java.util.Map;/** * @author 北极星 */@RestControllerpublic class DeepSeekChatClientController {    private static final String DEFAULT_PROMPT = "你好,介绍下你自己!";    private final ChatModel chatModel;    private final ChatClient DeepSeekChatClient;    public DeepSeekChatClientController (OpenAiChatModel chatModel) {        this.chatModel = chatModel;        this.DeepSeekChatClient = ChatClient.builder(chatModel).defaultAdvisors(new MessageChatMemoryAdvisor(new InMemoryChatMemory()))                // 实现 Logger 的 Advisor                .defaultAdvisors(new SimpleLoggerAdvisor())                // 设置 ChatClient 中 ChatModel 的 Options 参数                .defaultOptions(OpenAiChatOptions.builder().temperature(0.7d).build()).build();    }    /**     * 使用自定义参数调用DeepSeek模型     *     * @return ChatResponse 包含模型响应结果的封装对象     * @apiNote 当前硬编码指定模型为deepseek-chat,温度参数0.7以平衡生成结果的创造性和稳定性     */    @GetMapping(value = "/ai/customOptions")    public ChatResponse testDeepSeekCustomOptions () {        return this.DeepSeekChatClient.prompt("Generate the names of 5 famous pirates.").call().chatResponse();    }    /**     * 执行默认提示语的AI生成请求     *     * @return Map 包含生成结果的键值对,格式为{ "generation": 响应内容 }     */    @GetMapping("/ai/generate")    public Map<String, Object> testEasyChat () {        return Map.of("generation", this.DeepSeekChatClient.prompt(DEFAULT_PROMPT).call());    }    /**     * 流式生成接口 - 支持实时获取生成过程的分块响应     *     * @return Flux<ChatResponse> 响应式流对象,包含分块的模型响应数据     * @see Flux 基于Project Reactor的响应式流对象     */    @GetMapping("/ai/stream")    public Flux<ChatResponse> testDeepSeekGenerateWithStream () {        return this.DeepSeekChatClient.prompt(DEFAULT_PROMPT).stream().chatResponse();    }}
    application.yml:配置文件,定义了API密钥、基础URL等参数。
server:  port: 10001spring:  application:    name: spring-ai-alibaba-deepseek-chat-client-example  ai:    openai:      api-key: ${AI_DEEPSEEK_API_KEY:sk-8b9werererererwrw1a68995d}      base-url: https://api.deepseek.com      chat:        options:          model: deepseek-chat      embedding:        enabled: false
2.2 核心控制器:DeepSeekChatClientController

DeepSeekChatClientController是整个应用的核心,负责与DeepSeek模型进行交互。以下是其主要功能和参数的详细分析:

    构造函数

    public DeepSeekChatClientController(OpenAiChatModel chatModel) {    this.chatModel = chatModel;    this.DeepSeekChatClient = ChatClient.builder(chatModel)            .defaultAdvisors(new MessageChatMemoryAdvisor(new InMemoryChatMemory()))            .defaultAdvisors(new SimpleLoggerAdvisor())            .defaultOptions(OpenAiChatOptions.builder().temperature(0.7d).build())            .build();}
      chatModel:注入的DeepSeek模型实例,用于生成聊天响应。MessageChatMemoryAdvisor:实现了聊天记忆功能,允许在多次交互中保持上下文。SimpleLoggerAdvisor:记录日志,便于调试和监控。OpenAiChatOptions:设置模型的参数,例如temperature(温度参数),用于控制生成结果的创造性和稳定性。

    API接口

      testDeepSeekCustomOptions:使用自定义参数调用DeepSeek模型,生成特定的响应。testEasyChat:执行默认提示语的AI生成请求,返回生成结果。testDeepSeekGenerateWithStream:支持流式生成,实时获取分块响应。

三、功能实现

3.1 自定义参数调用
@GetMapping(value = "/ai/customOptions")public ChatResponse testDeepSeekCustomOptions() {    return this.DeepSeekChatClient.prompt("Generate the names of 5 famous pirates.").call().chatResponse();}
3.2 默认提示语生成
@GetMapping("/ai/generate")public Map<String, Object> testEasyChat() {    return Map.of("generation", this.DeepSeekChatClient.prompt(DEFAULT_PROMPT).call());}
3.3 流式生成
@GetMapping("/ai/stream")public Flux<ChatResponse> testDeepSeekGenerateWithStream() {    return this.DeepSeekChatClient.prompt(DEFAULT_PROMPT).stream().chatResponse();}

四、单元测试

为了验证DeepSeekChatClientController的功能,我们编写了以下单元测试:

@Testvoid testCustomOptions() {    Mockito.when(mockChatModel.call(any(Prompt.class))).thenReturn(createMockChatResponse("Pirate Names: Jack Sparrow, Blackbeard, Captain Hook"));    ChatResponse result = controller.testDeepSeekCustomOptions();    assertEquals("Pirate Names: Jack Sparrow, Blackbeard, Captain Hook", result.getResult().getOutput().getText());}

五、总结

通过本文的分析,我们深入探讨了基于DeepSeek客户端的聊天应用的实现细节。DeepSeekChatClientController中的参数设计合理,能够满足不同场景下的需求。单元测试结果表明,该应用具有良好的稳定性和可靠性。未来,我们可以通过进一步优化参数配置和扩展功能,提升应用的用户体验。


Fish AI Reader

Fish AI Reader

AI辅助创作,多种专业模板,深度分析,高质量内容生成。从观点提取到深度思考,FishAI为您提供全方位的创作支持。新版本引入自定义参数,让您的创作更加个性化和精准。

FishAI

FishAI

鱼阅,AI 时代的下一个智能信息助手,助你摆脱信息焦虑

联系邮箱 441953276@qq.com

相关标签

Spring AI DeepSeek 聊天应用 AI框架
相关文章