当前位置：首页 > news >正文

springboot-ai接入DeepSeek

news 来源：原创 2025/6/22 22:47:51

1、引入pom依赖

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


 <dependencyManagement>
     <dependencies>
         <dependency>
             <groupId>org.springframework.ai</groupId>
             <artifactId>spring-ai-bom</artifactId>
             <version>1.0.0-SNAPSHOT</version>
             <type>pom</type>
             <scope>import</scope>
         </dependency>
     </dependencies>
 </dependencyManagement>

2 、配置LLM

2.1 到deepseek 申请api-key

https://platform.deepseek.com/api_keys 首次注册会送10元限时额度

2.2 添加springboot-ai配置

spring.ai.openai.base-url=https://api.deepseek.com
spring.ai.openai.api-key=xxxx

3、编写MemoryService 记录历史对话

import org.springframework.ai.chat.messages.AssistantMessage;
import org.springframework.ai.chat.messages.Message;
import org.springframework.ai.chat.messages.UserMessage;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

@Service
public class MemoryService {
    HashMap<String, List<Message>> memory = new HashMap<>();
    public List<Message> getMemory(String userId) {
        List<Message> messages = memory.get(userId);
        if (messages == null) {
            return new ArrayList<>();
        }
        return messages;
    }
    public void addMemory(String userId, Message message) {
        List<Message> messages = memory.get(userId);
        if (messages == null) {
            messages = new ArrayList<>();
        }
        messages.add(message);
        memory.put(userId, messages);
    }
    public String buildHisChat(String userId){
        StringBuilder builder = new StringBuilder();
        for (Message message : memory.get(userId)) {
            if (message instanceof UserMessage) {
                builder.append("[user]").append(message.getText()).append("\n");
            } else if (message instanceof AssistantMessage) {
                builder.append("[assistant]").append(message.getText()).append("\n");
                builder.append("===============================================\n");
            }
        }
        return builder.toString();
    }

    public void cleanMemory(String userId) {
        memory.remove(userId);
    }
}

4、编写ChatService 调用LLM

import org.springframework.ai.chat.messages.Message;
import org.springframework.ai.chat.messages.UserMessage;
import org.springframework.ai.chat.model.ChatResponse;
import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.ai.model.Media;
import org.springframework.ai.openai.OpenAiChatModel;
import org.springframework.ai.openai.OpenAiChatOptions;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;
import org.springframework.util.MimeTypeUtils;
import org.springframework.web.multipart.MultipartFile;

import java.util.ArrayList;
import java.util.List;

@Service
public class ChatService {
    private static final String MODEL = "deepseek-chat";
    private final OpenAiChatModel chatModel;
    public ChatService(OpenAiChatModel chatModel) {
        this.chatModel = chatModel;
    }

    public Message buildMessageWithImage(String question, List<MultipartFile> files) {
        if (!CollectionUtils.isEmpty(files)) {
            ArrayList<Media> mediaList = new ArrayList<>();
            for (MultipartFile file : files) {
                mediaList.add(new Media(MimeTypeUtils.IMAGE_PNG, file.getResource()));
            }
            return new UserMessage(question, mediaList);
        } else {
            return new UserMessage(question);
        }
    }
    public String chat(String userId, List<Message> messages) {
        Prompt prompt = new Prompt(messages,
                OpenAiChatOptions.builder()
                        .model(MODEL)
                        .build());
        ChatResponse response = chatModel.call(prompt);
        return response.getResults().get(0).getOutput().getText();
    }
}

5、UI层代码

@RestController
@RequestMapping("/ai/chat")
@Slf4j
public class ChatController {


    private final MemoryService memory;
    private final ChatService chatService;

    @Autowired
    public ChatController(ChatService chatService, MemoryService service) {
        this.chatService = chatService;
        this.memory = service;
    }


    @GetMapping()
    public String chat(String userId, String question) {
        memory.addMemory(userId, new UserMessage(question));
        List<Message> messages = memory.getMemory(userId);
        String text = chatService.chat(userId, messages);
        memory.addMemory(userId, new AssistantMessage(text));
        return memory.buildHisChat(userId);
    }


    @GetMapping("image")
    public String chatWithImage(String userId, String question, String imageType, List<MultipartFile> files) {
        memory.addMemory(userId, chatService.buildMessageWithImage(question,files));
        List<Message> messages = memory.getMemory(userId);
        String text = chatService.chat(userId, messages);
        memory.addMemory(userId, new AssistantMessage(text));
        return memory.buildHisChat(userId);
    }
    @DeleteMapping("memory")
    public String cleanMemory(String userId) {
        memory.cleanMemory(userId);
        return "ok";
    }