SpringBoot对接DeepSeek

Spring Boot 集成 DeepSeek API 详细步骤

1. 创建API Key

1.访问 DeepSeek控制台 并登录。

2.点击 Create API Key 生成新密钥。

3.复制并保存密钥（需在Spring Boot配置文件中使用）。

2. 创建Spring Boot工程

使用 Spring Initializr 创建项目，选择以下依赖：

• Spring Web

• Lombok（简化实体类代码）

3. 配置项目依赖

在 pom.xml 中添加必要依赖：

<!-- OkHttp 实现HTTP请求 -->
<dependency><groupId>com.squareup.okhttp3</groupId><artifactId>okhttp</artifactId><version>4.12.0</version>
</dependency><!-- FastJSON 用于JSON解析 -->
<dependency><groupId>com.alibaba</groupId><artifactId>fastjson</artifactId><version>2.0.48</version>
</dependency><!-- SSE 事件流支持 -->
<dependency><groupId>com.squareup.okhttp3</groupId><artifactId>okhttp-sse</artifactId><version>4.12.0</version>
</dependency>

4. 核心代码实现

4.1 定义响应实体类

@Data
public class AiResult {private Integer code;private String message;private String sid;private String id;private Long created;private List<AiResultChoices> choices;private AiResultUsage usage;
}@Data
public class AiResultChoices {private AiResultDelta delta;private Integer index;
}@Data
public class AiResultDelta {private String role;private String content;
}@Data
public class AiResultUsage {private Integer prompt_tokens;private Integer completion_tokens;private Integer total_tokens;
}@Data
public class ContentDto {private String content;
}

4.2 控制器实现（支持SSE流式响应）

/*** DeepSeek AI接口请求控制器* 功能：处理SSE（Server-Sent Events）流式请求，与DeepSeek API交互* 主要职责：* 1. 接收前端SSE请求* 2. 构造DeepSeek API请求* 3. 处理流式响应并转发给客户端* 4. 管理连接生命周期和异常处理*/
@RestController
public class SeekController {// 日志记录器private static final Logger log = LoggerFactory.getLogger(SeekController.class);// 流式结束标识符（符合OpenAI标准）private static final String DONE_FLAG = "[DONE]";// HTTP请求超时时间（单位：秒）private static final int TIMEOUT_SECONDS = 60;// DeepSeek API端点private static final String AI_URL = "https://api.deepseek.com/chat/completions";// JSON媒体类型常量private static final MediaType JSON_MEDIA_TYPE = MediaType.parse("application/json");// 从配置文件注入API密钥@Value("${api.key}")private String apiKey;// OkHttp客户端配置（线程安全）private final OkHttpClient httpClient = new OkHttpClient.Builder().connectTimeout(TIMEOUT_SECONDS, TimeUnit.SECONDS)  // 连接超时.readTimeout(TIMEOUT_SECONDS, TimeUnit.SECONDS)     // 读取超时.writeTimeout(TIMEOUT_SECONDS, TimeUnit.SECONDS)    // 写入超时.build();/*** SSE请求处理入口* @param message 用户输入消息* @param response HTTP响应对象* @throws IOException 当响应流出现问题时抛出*/@GetMapping("/stream")public void handleSse(@RequestParam String message, HttpServletResponse response) throws IOException {// 配置SSE响应头configureSSEResponse(response);PrintWriter pw = null;try {// 获取响应输出流pw = response.getWriter();// 创建同步锁（初始计数器为1）CountDownLatch latch = new CountDownLatch(1);// 执行SSE请求executeSSERequest(pw, buildRequest(message), latch);// 等待异步操作完成（最大等待TIMEOUT_SECONDS秒）if (!latch.await(TIMEOUT_SECONDS, TimeUnit.SECONDS)) {log.warn("SSE请求超时");}} catch (InterruptedException e) {// 正确处理线程中断Thread.currentThread().interrupt();log.error("请求被中断", e);response.sendError(503, "服务不可用");} catch (Exception e) {log.error("SSE请求异常", e);response.sendError(500, "服务器内部错误");} finally {// 确保关闭输出流if (pw != null) {try {pw.close();} catch (Exception e) {log.warn("关闭PrintWriter异常", e);}}}}/*** 配置SSE响应头* @param response HTTP响应对象*/private void configureSSEResponse(HttpServletResponse response) {response.setContentType("text/event-stream");  // MIME类型response.setCharacterEncoding("UTF-8");        // 字符编码response.setHeader("Connection", "keep-alive"); // 保持长连接response.setHeader("Cache-Control", "no-cache"); // 禁用缓存}/*** 构建DeepSeek API请求* @param content 用户输入内容* @return 构造好的Request对象*/private Request buildRequest(String content) {// 构造请求参数Map<String, Object> params = new HashMap<>();params.put("model", "deepseek-chat");    // 使用的模型params.put("stream", true);              // 启用流式传输// 构造消息列表（使用双括号初始化语法）params.put("messages", Collections.singletonList(new HashMap<String, String>() {{put("role", "user");         // 消息角色put("content", content);     // 消息内容}}));// 构建OkHttp请求return new Request.Builder().url(AI_URL)  // API地址.post(RequestBody.create(JSON.toJSONString(params), JSON_MEDIA_TYPE)) // JSON请求体.addHeader("Authorization", "Bearer " + apiKey) // 认证头.addHeader("Accept", "text/event-stream")       // 接受SSE流.build();}/*** 执行SSE请求核心方法* @param pw 响应输出流* @param request 构造好的API请求* @param latch 线程同步锁*/private void executeSSERequest(PrintWriter pw, Request request, CountDownLatch latch) {// 创建事件源监听器RealEventSource eventSource = new RealEventSource(request, new EventSourceListener() {/*** 事件处理回调* @param eventSource 事件源* @param id 事件ID（未使用）* @param type 事件类型（未使用）* @param data 接收到的数据*/@Overridepublic void onEvent(EventSource eventSource, String id, String type, String data) {try {// 处理结束标志if (DONE_FLAG.equals(data)) {sendData(pw, new ContentDto(DONE_FLAG));return;}// 解析并发送有效内容String content = parseContent(data);if (!content.isEmpty()) {sendData(pw, new ContentDto(content));}} catch (Exception e) {log.error("事件处理异常", e);}}/*** 失败处理回调*/@Overridepublic void onFailure(EventSource eventSource, Throwable t, Response response) {log.error("API调用失败: {}，响应码：{}", t.getMessage(), (response != null ? response.code() : "N/A"));latch.countDown(); // 释放同步锁}/*** 连接关闭回调*/@Overridepublic void onClosed(EventSource eventSource) {try {sendData(pw, new ContentDto(DONE_FLAG)); // 发送结束标志} finally {latch.countDown(); // 确保释放同步锁}}});// 发起异步连接eventSource.connect(httpClient);}/*** 线程安全的数据发送方法* @param pw 响应输出流* @param dto 要发送的数据传输对象*/private synchronized void sendData(PrintWriter pw, ContentDto dto) {try {// 按照SSE格式发送数据pw.write("data:" + JSON.toJSONString(dto) + "\n\n");pw.flush(); // 立即刷新缓冲区} catch (Exception e) {log.error("数据发送失败", e);}}/*** 解析API响应内容* @param data 原始响应字符串* @return 解析出的内容文本*/private String parseContent(String data) {try {// 使用FastJSON反序列化AiResult result = JSON.parseObject(data, AiResult.class);// 使用Optional处理可能为空的字段return Optional.ofNullable(result.getChoices()).flatMap(list -> list.stream().findFirst())  // 取第一个choice.map(AiResultChoices::getDelta)              // 获取delta对象.map(AiResultDelta::getContent)              // 获取content字段.orElse("");                                 // 默认返回空字符串} catch (Exception e) {log.error("数据解析异常，原始数据：{}", data, e);return "";}}
}

5. 配置文件说明

在 application.yml 中添加配置：

deepseek:api-key: your_api_key_here  # 替换为实际API Key

6. 测试与验证

6.1启动应用后，通过浏览器或工具发送请求：

GET http://localhost:8080/stream?message=你好，介绍一下你自己

6.2预期输出（流式响应）：