package com.smartor;
|
|
;
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
import okhttp3.*;
|
|
import java.io.IOException;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
public class DeepSeekApi {
|
// 替换为你自己的key
|
private static final String API_KEY = "sk-";
|
private static final String API_URL = "https://api.deepseek.com/v1/chat/completions";
|
|
public static void main(String[] args) {
|
String resp = chatCompletion("用Java写一个冒泡排序");
|
System.out.println("DeepSeek返回结果：\n" + resp);
|
}
|
|
public static String chatCompletion(String userPrompt) {
|
OkHttpClient client = new OkHttpClient();
|
|
// 1. 组装消息体
|
List<Map<String, String>> messages = new ArrayList<>();
|
// 系统角色（可选，用于设定AI身份）
|
Map<String, String> sysMsg = new HashMap<>();
|
sysMsg.put("role", "system");
|
sysMsg.put("content", "你是专业Java开发工程师，回答简洁规范");
|
messages.add(sysMsg);
|
// 用户提问
|
Map<String, String> userMsg = new HashMap<>();
|
userMsg.put("role", "user");
|
userMsg.put("content", userPrompt);
|
messages.add(userMsg);
|
|
// 2. 请求参数
|
Map<String, Object> requestBody = new HashMap<>();
|
requestBody.put("model", "deepseek-chat");
|
requestBody.put("messages", messages);
|
requestBody.put("temperature", 0.7); // 随机性0~1，越低越严谨
|
requestBody.put("max_tokens", 1024);
|
|
// 3. 构建请求
|
String jsonBody = JSON.toJSONString(requestBody);
|
RequestBody body = RequestBody.create(MediaType.parse("application/json"), jsonBody);
|
|
Request request = new Request.Builder()
|
.url(API_URL)
|
.header("Authorization", "Bearer " + API_KEY)
|
.header("Content-Type", "application/json")
|
.post(body)
|
.build();
|
|
// 4. 发送请求
|
try (Response response = client.newCall(request).execute()) {
|
if (!response.isSuccessful()) {
|
return "请求失败，状态码：" + response.code() + "，错误信息：" + response.body().string();
|
}
|
String resJson = response.body().string();
|
JSONObject jsonObj = JSON.parseObject(resJson);
|
// 提取AI回答内容
|
return jsonObj.getJSONArray("choices")
|
.getJSONObject(0)
|
.getJSONObject("message")
|
.getString("content");
|
} catch (IOException e) {
|
e.printStackTrace();
|
return "接口调用异常：" + e.getMessage();
|
}
|
}
|
}
|
