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

'commit'

Browse Source
main
HuangHai 2 months ago
parent d7425679a2
commit 5004788c89
5 changed files with 122 additions and 81 deletions
Show Stats Download Patch File Download Diff File

BIN
dsAi/src/main/java/com/dsideal/Res/Test/61B49F1B-DF15-4ed1-9C8C-3CF1E1499748.txt
View File

Binary file not shown.

118
dsAi/src/main/java/com/dsideal/Res/Test/CallDeepSeek.java
Unescape View File

@ -0,0 +1,118 @@
package com.dsideal.Res.Test;
import cn.hutool.core.io.FileUtil;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSONArray;
import com.jfinal.kit.PathKit;
import java.io.File;
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";
public static void callDeepSeekStream(String prompt, SSEListener listener) {
new Thread(() -> {
StringBuilder fullResponse = new StringBuilder();
try {
// 修改提示词确保只返回HTML
String htmlPrompt = prompt + "请只返回HTML代码，不要包含任何解释或Markdown格式。" +
"确保HTML是完整的，包含<!DOCTYPE html>和<html>标签。";
JSONObject json = new JSONObject();
json.set("model", "deepseek-chat");
// 修改为对象形式构建messages
JSONObject message = new JSONObject();
message.set("role", "user");
message.set("content", htmlPrompt);
JSONArray messages = new JSONArray();
messages.add(message);
json.set("messages", messages);
json.set("stream", true);
System.out.println(json); // 打印最终请求体
HttpRequest request = HttpRequest.post(API_URL)
.header("Content-Type", "application/json")
.header("Authorization", "Bearer " + API_KEY)
.header("Accept", "text/event-stream")
.body(json.toString());
HttpResponse response = request.execute();
if (response.isOk()) {
String[] lines = response.body().split("\\r?\\n");
for (String line : lines) {
if (line.startsWith("data:")) {
String data = line.substring(5).trim();
if (!data.equals("[DONE]")) {
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);
}
}
}
}
}
// 保存HTML到文件
String htmlContent = fullResponse.toString();
if (htmlContent.contains("<!DOCTYPE html>") && htmlContent.contains("</html>")) {
FileUtil.writeString(htmlContent, new File("C:\\1.html"), "UTF-8");
listener.onComplete("HTML已成功保存到C:\\1.html");
} else {
listener.onError("返回内容不是有效的HTML");
}
} else {
listener.onError("API请求失败: " + response.getStatus());
}
} catch (Exception e) {
listener.onError(e.getMessage());
}
}).start();
}
public static void main(String[] args) {
String sql = FileUtil.readUtf8String(new File(PathKit.getRootClassPath()+"/XueYuan.sql"));
String prompt = "你是一个数据库SQL的血缘关系分析专家，我将提供SQL语句给你，帮我整理数据血缘关系，并且绘制HTML页面，以展示最终的可视化展现。";
callDeepSeekStream(prompt + "\n" + sql, new SSEListener() {
@Override
public void onData(String data) {
System.out.print(data);
}
@Override
public void onComplete(String fullResponse) {
System.out.println("\n\n完整回复: " + fullResponse);
}
@Override
public void onError(String error) {
System.err.println("错误: " + error);
}
});
try {
Thread.sleep(30000);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
public interface SSEListener {
void onData(String data);
void onComplete(String fullResponse);
void onError(String error);
}
}

81
dsAi/src/main/java/com/dsideal/Res/Util/CallDeepSeek.java
Unescape View File

@ -1,81 +0,0 @@
package com.dsideal.Res.Util;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
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";
public static void callDeepSeekStream(String prompt, SSEListener listener) {
new Thread(() -> {
try {
JSONObject json = new JSONObject();
json.set("model", "deepseek-chat");
json.set("messages", JSONUtil.parseArray(
"[{\"role\":\"user\",\"content\":\"" + prompt + "\"}]"
));
json.set("stream", true); // 启用流式响应
HttpRequest request = HttpRequest.post(API_URL)
.header("Content-Type", "application/json")
.header("Authorization", "Bearer " + API_KEY)
.header("Accept", "text/event-stream") // SSE支持
.body(json.toString());
HttpResponse response = request.execute();
if (response.isOk()) {
// 处理流式响应
String[] lines = response.body().split("\\r?\\n");
for (String line : lines) {
if (line.startsWith("data:")) {
String data = line.substring(5).trim();
if (!data.equals("[DONE]")) {
listener.onData(data);
}
}
}
listener.onComplete();
} else {
listener.onError("API请求失败: " + response.getStatus());
}
} catch (Exception e) {
listener.onError(e.getMessage());
}
}).start();
}
public static void main(String[] args) {
callDeepSeekStream("你好", new SSEListener() {
@Override
public void onData(String data) {
System.out.println("收到数据: " + data);
}
@Override
public void onComplete() {
System.out.println("流式传输完成");
}
@Override
public void onError(String error) {
System.err.println("错误: " + error);
}
});
// 保持主线程运行
try {
Thread.sleep(30000); // 等待30秒
} catch (InterruptedException e) {
e.printStackTrace();
}
}
public interface SSEListener {
void onData(String data);
void onComplete();
void onError(String error);
}
}

2
dsAi/src/main/resources/XueYuan.sql
Unescape View File

@ -0,0 +1,2 @@
select tsl.person_id,tsl.person_name,tsl.mz,tdm.mz_name,tsl.xb,tdx.xb_name from t_sys_loginperson as tsl inner join t_dm_mz as tdm on tsl.mz=tdm.mz_id
inner join t_dm_xb as tdx on tsl.xb=tdx.xb_id

2
dsAi/target/classes/XueYuan.sql
Unescape View File

@ -0,0 +1,2 @@
select tsl.person_id,tsl.person_name,tsl.mz,tdm.mz_name,tsl.xb,tdx.xb_name from t_sys_loginperson as tsl inner join t_dm_mz as tdm on tsl.mz=tdm.mz_id
inner join t_dm_xb as tdx on tsl.xb=tdx.xb_id
Write Preview
Loading…
Cancel
Save