huanghai
/
YunNanDsBase
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

'commit'

Browse Source
main
HuangHai 1 month ago
parent 88e3c6ebe2
commit 42a75c317f
2 changed files with 216 additions and 0 deletions
Show Stats Download Patch File Download Diff File

37
src/main/java/com/dsideal/base/Tools/AiGenerate.java
Unescape View File

@ -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<Record> 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<Record> listConetent = Db.use(DataEaseModel.DB_NAME).find(sql, biJiao[j]);
if (!listConetent.isEmpty()) {
System.out.println(listConetent);
}
}
}
}
}

179
src/main/java/com/dsideal/base/Tools/Util/CallDeepSeek.java
Unescape View File

@ -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<Void> 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<Stream<String>> 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);
}
}
Write Preview
Loading…
Cancel
Save