diff --git a/src/main/java/com/dsideal/base/Tools/AiGenerate.java b/src/main/java/com/dsideal/base/Tools/AiGenerate.java new file mode 100644 index 00000000..de9d4297 --- /dev/null +++ b/src/main/java/com/dsideal/base/Tools/AiGenerate.java @@ -0,0 +1,37 @@ +package com.dsideal.base.Tools; + +import com.dsideal.base.BaseApplication; +import com.dsideal.base.DataEase.Model.DataEaseModel; +import com.dsideal.base.Plugin.YamlProp; +import com.dsideal.base.Tools.Util.LocalMysqlConnectUtil; +import com.jfinal.plugin.activerecord.Db; +import com.jfinal.plugin.activerecord.Record; + +import java.util.List; + +public class AiGenerate { + public static DataEaseModel dm = new DataEaseModel(); + + + public static void main(String[] args) { + //加载配置文件 + String configFile = "application.yaml"; + BaseApplication.PropKit = new YamlProp(configFile); + LocalMysqlConnectUtil.Init(); + + String sql = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'dataease' AND TABLE_NAME LIKE 'excel_报告-教育资源配置发展预测%';"; + List tableList = Db.find(sql); + String[] biJiao = new String[]{"文山州", "楚雄州"}; + for (Record record : tableList) { + String tableName = record.getStr("TABLE_NAME"); + + for (int j = 0; j < biJiao.length; j++) { + sql = "select * from `" + tableName + "` where `行政区划`=?"; + List listConetent = Db.use(DataEaseModel.DB_NAME).find(sql, biJiao[j]); + if (!listConetent.isEmpty()) { + System.out.println(listConetent); + } + } + } + } +} diff --git a/src/main/java/com/dsideal/base/Tools/Util/CallDeepSeek.java b/src/main/java/com/dsideal/base/Tools/Util/CallDeepSeek.java new file mode 100644 index 00000000..f89f4708 --- /dev/null +++ b/src/main/java/com/dsideal/base/Tools/Util/CallDeepSeek.java @@ -0,0 +1,179 @@ +package com.dsideal.base.Tools.Util; + +import cn.hutool.core.io.FileUtil; +import cn.hutool.json.JSONObject; +import cn.hutool.json.JSONUtil; +import com.alibaba.fastjson.JSONArray; + +import java.io.File; +import java.net.URI; +import java.net.http.HttpClient; +import java.net.http.HttpResponse; +import java.nio.charset.StandardCharsets; +import java.util.concurrent.CompletableFuture; +import java.util.stream.Stream; + +public class CallDeepSeek { + private static final String API_KEY = "sk-44ae895eeb614aa1a9c6460579e322f1"; // 请替换为您的API KEY + private static final String API_URL = "https://api.deepseek.com/v1/chat/completions"; + + /** + * 通用的DeepSeek API调用方法（流式响应） + * + * @param prompt 用户提示词 + * @param listener 响应监听器 + */ + public static void callDeepSeekStream(String prompt, SSEListener listener) { + callDeepSeekStream(prompt, listener, null, false); + } + + /** + * 通用的DeepSeek API调用方法（流式响应） + * + * @param prompt 用户提示词 + * @param listener 响应监听器 + * @param outputPath 输出文件路径（可选） + * @param saveToFile 是否保存到文件 + */ + public static void callDeepSeekStream(String prompt, SSEListener listener, String outputPath, boolean saveToFile) { + new Thread(() -> { + StringBuilder fullResponse = new StringBuilder(); + try { + JSONObject jsonPayload = createRequestPayload(prompt); + HttpClient client = createHttpClient(); + java.net.http.HttpRequest request = createHttpRequest(jsonPayload); + + CompletableFuture future = client.sendAsync(request, HttpResponse.BodyHandlers.ofLines()) + .thenAccept(response -> { + handleStreamResponse(response, fullResponse, listener, outputPath, saveToFile); + }).exceptionally(e -> { + listener.onError("请求或处理异常: " + e.getMessage()); + e.printStackTrace(); + return null; + }); + future.join(); + + } catch (Exception e) { + listener.onError("发生意外错误: " + e.getMessage()); + e.printStackTrace(); + } + }).start(); + } + + + /** + * 创建HTTP请求载荷 + */ + private static JSONObject createRequestPayload(String prompt) { + JSONObject jsonPayload = new JSONObject(); + jsonPayload.set("model", "deepseek-chat"); + JSONObject message = new JSONObject(); + message.set("role", "user"); + message.set("content", prompt); + JSONArray messages = new JSONArray(); + messages.add(message); + jsonPayload.set("messages", messages); + jsonPayload.set("stream", true); + return jsonPayload; + } + + /** + * 创建HTTP客户端 + */ + private static HttpClient createHttpClient() { + return HttpClient.newBuilder() + .version(HttpClient.Version.HTTP_1_1) + .build(); + } + + /** + * 创建HTTP请求 + */ + private static java.net.http.HttpRequest createHttpRequest(JSONObject jsonPayload) { + return java.net.http.HttpRequest.newBuilder() + .uri(URI.create(API_URL)) + .header("Content-Type", "application/json") + .header("Authorization", "Bearer " + API_KEY) + .header("Accept", "text/event-stream") + .POST(java.net.http.HttpRequest.BodyPublishers.ofString(jsonPayload.toString(), StandardCharsets.UTF_8)) + .build(); + } + + /** + * 处理流式响应 + */ + private static void handleStreamResponse(HttpResponse> response, + StringBuilder fullResponse, + SSEListener listener, + String outputPath, + boolean saveToFile) { + if (response.statusCode() == 200) { + response.body().forEach(line -> { + if (line.startsWith("data:")) { + String data = line.substring(5).trim(); + if (!data.equals("[DONE]")) { + try { + JSONObject jsonData = JSONUtil.parseObj(data); + if (jsonData.containsKey("choices")) { + String content = jsonData.getJSONArray("choices") + .getJSONObject(0) + .getJSONObject("delta") + .getStr("content", ""); + if (content != null && !content.isEmpty()) { + fullResponse.append(content); + listener.onData(content); + } + } + } catch (Exception e) { + System.err.println("解析SSE JSON数据错误: " + data + " \nError: " + e.getMessage()); + } + } + } + }); + + // 流结束后的处理 + String responseContent = fullResponse.toString(); + if (saveToFile && outputPath != null) { + FileUtil.writeString(responseContent, new File(outputPath), "UTF-8"); + listener.onComplete("内容已成功保存到" + outputPath); + } else { + listener.onComplete(responseContent); + } + } else { + listener.onError("API请求失败: " + response.statusCode() + " Body: " + response.body().toString()); + } + } + + + public static void main(String[] args) { + String prompt = "你好，你是谁?"; + + // 使用通用方法 + callDeepSeekStream(prompt, new SSEListener() { + @Override + public void onData(String data) { + System.out.print(data); + System.out.flush(); + } + + @Override + public void onComplete(String fullResponse) { + System.out.println("\n\n完整回复: " + fullResponse); + } + + @Override + public void onError(String error) { + System.err.println("错误: " + error); + } + }); + } + + public interface SSEListener { + void onData(String data); + + void onComplete(String fullResponse); + + void onError(String error); + } +} +