From 33f71e742bafd82fb19be72d5a573238719ba2fe Mon Sep 17 00:00:00 2001 From: Van0313 <60689272+Van0313@users.noreply.github.com> Date: Thu, 8 May 2025 22:24:28 +0800 Subject: [PATCH] =?UTF-8?q?=E9=87=8D=E6=9E=84=E8=AF=84=E8=AE=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../java/cn/van/business/util/JDUtil.java | 16 +++- .../van/business/util/ds/GPTClientUtil.java | 94 +++++++++++++++++++ 2 files changed, 105 insertions(+), 5 deletions(-) create mode 100644 src/main/java/cn/van/business/util/ds/GPTClientUtil.java diff --git a/src/main/java/cn/van/business/util/JDUtil.java b/src/main/java/cn/van/business/util/JDUtil.java index 4426c60..c8d15b8 100644 --- a/src/main/java/cn/van/business/util/JDUtil.java +++ b/src/main/java/cn/van/business/util/JDUtil.java @@ -8,6 +8,7 @@ import cn.van.business.model.pl.Comment; import cn.van.business.repository.CommentRepository; import cn.van.business.repository.OrderRowRepository; import cn.van.business.util.ds.DeepSeekClientUtil; +import cn.van.business.util.ds.GPTClientUtil; import com.alibaba.fastjson2.JSON; import com.alibaba.fastjson2.JSONArray; import com.alibaba.fastjson2.JSONObject; @@ -115,6 +116,7 @@ public class JDUtil { private final OrderUtil orderUtil; private final DeepSeekClientUtil deepSeekClientUtil; + private final GPTClientUtil gptClientUtil; // 添加ObjectMapper来序列化和反序列化UserInteractionState private final ObjectMapper objectMapper = new ObjectMapper(); private final ConcurrentHashMap userInteractionStates = new ConcurrentHashMap<>(); @@ -122,7 +124,7 @@ public class JDUtil { // 构造函数中注入StringRedisTemplate @Autowired - public JDUtil(StringRedisTemplate redisTemplate, OrderRowRepository orderRowRepository, WXUtil wxUtil, OrderUtil orderUtil, DeepSeekClientUtil deepSeekClientUtil, CommentRepository commentRepository) { + public JDUtil(StringRedisTemplate redisTemplate, OrderRowRepository orderRowRepository, WXUtil wxUtil, OrderUtil orderUtil, DeepSeekClientUtil deepSeekClientUtil, CommentRepository commentRepository, GPTClientUtil gptClientUtil) { this.redisTemplate = redisTemplate; this.orderRowRepository = orderRowRepository; @@ -130,6 +132,7 @@ public class JDUtil { this.orderUtil = orderUtil; this.deepSeekClientUtil = deepSeekClientUtil; this.commentRepository = commentRepository; + this.gptClientUtil = gptClientUtil; } private List filterOrdersByDate(List orderRows, int daysBack) { @@ -1767,15 +1770,18 @@ public class JDUtil { } // 调用 DeepSeek 生成新的评论内容 String deepSeekPrompt = COMMENT_TEMPLATES_DS + commentToUse.getCommentText(); - String deepSeekResponse = ""; + //String deepSeekResponse = ""; + String gptResponse = ""; try { - deepSeekResponse = deepSeekClientUtil.getDeepSeekResponse(deepSeekPrompt); + //deepSeekResponse = deepSeekClientUtil.getDeepSeekResponse(deepSeekPrompt); + gptResponse = gptClientUtil.getGPTResponse(deepSeekPrompt); } catch (IOException e) { logger.error("生成评论异常 - 用户: {}", fromWxid, e); - wxUtil.sendTextMessage(fromWxid, "DS评论生成失败", 1, fromWxid, false); + wxUtil.sendTextMessage(fromWxid, "AI 评论生成失败", 1, fromWxid, false); } // 发送生成的评论文本 - wxUtil.sendTextMessage(fromWxid, "DS回复：\n" + deepSeekResponse, 1, fromWxid, true); + //wxUtil.sendTextMessage(fromWxid, "DS回复：\n" + deepSeekResponse, 1, fromWxid, true); + wxUtil.sendTextMessage(fromWxid, "GPT回复：\n" + gptResponse, 1, fromWxid, true); // 更新评论状态为已使用 commentToUse.setIsUse(1); diff --git a/src/main/java/cn/van/business/util/ds/GPTClientUtil.java b/src/main/java/cn/van/business/util/ds/GPTClientUtil.java new file mode 100644 index 0000000..88dcc56 --- /dev/null +++ b/src/main/java/cn/van/business/util/ds/GPTClientUtil.java @@ -0,0 +1,94 @@ +package cn.van.business.util.ds; + +/** + * @author Leo + * @version 1.0 + * @create 2025/5/8 22:19 + * @description： + */ +package cn.van.business.util.ds; + +import cn.hutool.http.HttpRequest; +import cn.hutool.http.HttpResponse; +import cn.hutool.http.Method; +import com.fasterxml.jackson.databind.JsonNode; +import com.fasterxml.jackson.databind.ObjectMapper; +import org.springframework.stereotype.Component; + +import java.io.IOException; +import java.net.InetSocketAddress; +import java.net.Proxy; +import java.util.HashMap; +import java.util.Map; + +@Component +public class GPTClientUtil { + // logger + private static final org.slf4j.Logger logger = org.slf4j.LoggerFactory.getLogger(GPTClientUtil.class); + + private static final String GPT_API_URL = "https://api.openai.com/v1/chat/completions"; // GPT API 地址 + private static final String GPT_API_KEY = "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"; // 替换为你的 GPT API 密钥 + private static final ObjectMapper objectMapper = new ObjectMapper(); // Jackson JSON 解析器 + private static final String PROXY_HOST = "192.168.8.9"; // 本地代理地址 + private static final int PROXY_PORT = 1070; // 本地代理端口 + + /** + * 调用 GPT API 并返回第一次回复的文本内容 + * + * @param inputText 用户输入的文本 + * @return API 返回的第一个回复内容 + * @throws IOException 如果网络请求或 JSON 解析失败 + * @throws IllegalArgumentException 如果输入为空或过长 + */ + public String getGPTResponse(String inputText) throws IOException { + // 1. 输入校验 + if (inputText == null || inputText.trim().isEmpty()) { + throw new IllegalArgumentException("输入文本不能为空"); + } + if (inputText.length() > 10000) { + throw new IllegalArgumentException("输入文本过长"); + } + + // 2. 构建请求体 + Map requestBody = new HashMap<>(); + requestBody.put("model", "gpt-4o"); + requestBody.put("messages", new Map[]{ + Map.of("role", "user", "content", inputText) + }); + requestBody.put("temperature", 0.7); + + String jsonBody = objectMapper.writeValueAsString(requestBody); + + // 3. 使用 Hutool HTTP 发送请求，设置代理 + Proxy proxy = new Proxy(Proxy.Type.HTTP, new InetSocketAddress(PROXY_HOST, PROXY_PORT)); + HttpRequest request = HttpRequest.of(GPT_API_URL) + .method(Method.POST) + .header("Content-Type", "application/json") + .header("Authorization", "Bearer " + GPT_API_KEY) + .header("Accept", "application/json") + .body(jsonBody) + .setProxy(proxy); + + logger.info("请求 GPT API: URL={}, Body={}", GPT_API_URL, jsonBody); + + HttpResponse response = request.execute(); + + // 4. 检查 HTTP 状态码 + if (response.getStatus() != 200) { + logger.error("GPT API 调用失败！状态码={}, 响应={}", response.getStatus(), response.body()); + throw new IOException("API 调用失败，HTTP 状态码: " + response.getStatus()); + } + + // 5. 解析 JSON 响应 + JsonNode rootNode = objectMapper.readTree(response.body()); + JsonNode choices = rootNode.path("choices"); + if (choices.isEmpty()) { + throw new IOException("API 返回数据格式异常，未找到回复内容"); + } + + return choices.get(0) + .path("message") + .path("content") + .asText(); + } +}