AI大模型进阶系列(02)基于Spring AI实现AI chatbot助理|一句话让deepseek实现

package com.example.config;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.client.RestClient;
@Configuration
public class DeepSeekConfig {
    @Value("${deepseek.api.key}")
    private String apiKey;
    @Value("${deepseek.api.url}")
    private String apiUrl;
    @Bean
    public RestClient deepSeekRestClient() {
        return RestClient.builder()
                .baseUrl(apiUrl)
                .defaultHeader("Authorization", "Bearer " + apiKey)
                .defaultHeader("Content-Type", "application/json")
                .build();
    }
}

package com.example.model;
import java.util.List;
public class DeepSeekResponse {
    private String id;
    private String object;
    private long created;
    private String model;
    private List<Choice> choices;
    private Usage usage;
    // Getters and setters
    public String getId() {
        return id;
    }
    public void setId(String id) {
        this.id = id;
    }
    public String getObject() {
        return object;
    }
    public void setObject(String object) {
        this.object = object;
    }
    public long getCreated() {
        return created;
    }
    public void setCreated(long created) {
        this.created = created;
    }
    public String getModel() {
        return model;
    }
    public void setModel(String model) {
        this.model = model;
    }
    public List<Choice> getChoices() {
        return choices;
    }
    public void setChoices(List<Choice> choices) {
        this.choices = choices;
    }
    public Usage getUsage() {
        return usage;
    }
    public void setUsage(Usage usage) {
        this.usage = usage;
    }
    public static class Choice {
        private int index;
        private Message message;
        private String finish_reason;
        // Getters and setters
        public int getIndex() {
            return index;
        }
        public void setIndex(int index) {
            this.index = index;
        }
        public Message getMessage() {
            return message;
        }
        public void setMessage(Message message) {
            this.message = message;
        }
        public String getFinish_reason() {
            return finish_reason;
        }
        public void setFinish_reason(String finish_reason) {
            this.finish_reason = finish_reason;
        }
    }
    public static class Message {
        private String role;
        private String content;
        // Getters and setters
        public String getRole() {
            return role;
        }
        public void setRole(String role) {
            this.role = role;
        }
        public String getContent() {
            return content;
        }
        public void setContent(String content) {
            this.content = content;
        }
    }
    public static class Usage {
        private int prompt_tokens;
        private int completion_tokens;
        private int total_tokens;
        // Getters and setters
        public int getPrompt_tokens() {
            return prompt_tokens;
        }
        public void setPrompt_tokens(int prompt_tokens) {
            this.prompt_tokens = prompt_tokens;
        }
        public int getCompletion_tokens() {
            return completion_tokens;
        }
        public void setCompletion_tokens(int completion_tokens) {
            this.completion_tokens = completion_tokens;
        }
        public int getTotal_tokens() {
            return total_tokens;
        }
        public void setTotal_tokens(int total_tokens) {
            this.total_tokens = total_tokens;
        }
    }
}

package com.example.model;
import java.util.ArrayList;
import java.util.List;
public class DeepSeekRequest {
    private String model = "deepseek-chat";
    private List<Message> messages;
    private double temperature = 0.7;
    private int max_tokens = 2000;
    public DeepSeekRequest(String prompt) {
        this.messages = new ArrayList<>();
        this.messages.add(new Message("user", prompt));
    }
    // Getters and setters
    public String getModel() {
        return model;
    }
    public void setModel(String model) {
        this.model = model;
    }
    public List<Message> getMessages() {
        return messages;
    }
    public void setMessages(List<Message> messages) {
        this.messages = messages;
    }
    public double getTemperature() {
        return temperature;
    }
    public void setTemperature(double temperature) {
        this.temperature = temperature;
    }
    public int getMax_tokens() {
        return max_tokens;
    }
    public void setMax_tokens(int max_tokens) {
        this.max_tokens = max_tokens;
    }
    public static class Message {
        private String role;
        private String content;
        public Message(String role, String content) {
            this.role = role;
            this.content = content;
        }
        // Getters and setters
        public String getRole() {
            return role;
        }
        public void setRole(String role) {
            this.role = role;
        }
        public String getContent() {
            return content;
        }
        public void setContent(String content) {
            this.content = content;
        }
    }
}

package com.example.model;
public class ChatMessage {
    private String role; // "user" or "ai"
    private String content;
    public ChatMessage() {
    }
    public ChatMessage(String role, String content) {
        this.role = role;
        this.content = content;
    }
    // Getters and setters
    public String getRole() {
        return role;
    }
    public void setRole(String role) {
        this.role = role;
    }
    public String getContent() {
        return content;
    }
    public void setContent(String content) {
        this.content = content;
    }
}

package com.example.service;
import com.example.model.DeepSeekRequest;
import com.example.model.DeepSeekResponse;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.client.RestClient;
@Service
public class DeepSeekService {
    private final RestClient deepSeekRestClient;
    @Autowired
    public DeepSeekService(RestClient deepSeekRestClient) {
        this.deepSeekRestClient = deepSeekRestClient;
    }
    public String generateResponse(String prompt) {
        DeepSeekRequest request = new DeepSeekRequest(prompt);
        DeepSeekResponse response = deepSeekRestClient.post()
                .body(request)
                .retrieve()
                .body(DeepSeekResponse.class);
        return response != null ? response.getChoices().get(0).getMessage().getContent() : "No response from DeepSeek";
    }
}

package com.example.controller;
import com.example.model.ChatMessage;
import com.example.service.DeepSeekService;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.PostMapping;
import java.util.ArrayList;
import java.util.List;
@Controller
public class ChatController {
    private final DeepSeekService deepSeekService;
    private List<ChatMessage> chatHistory = new ArrayList<>();
    public ChatController(DeepSeekService deepSeekService) {
        this.deepSeekService = deepSeekService;
    }
    @GetMapping("/")
    public String chat(Model model) {
        model.addAttribute("chatMessage", new ChatMessage());
        model.addAttribute("chatHistory", chatHistory);
        return "chat";
    }
    @PostMapping("/send")
    public String sendMessage(@ModelAttribute ChatMessage chatMessage, Model model) {
        // Add user message to chat history
        chatHistory.add(new ChatMessage("user", chatMessage.getContent()));
        // Get response from DeepSeek
        String response = deepSeekService.generateResponse(chatMessage.getContent());
        // Add AI response to chat history
        chatHistory.add(new ChatMessage("ai", response));
        model.addAttribute("chatMessage", new ChatMessage());
        model.addAttribute("chatHistory", chatHistory);
        return "chat";
    }
}