增加了套餐包类目区分
This commit is contained in:
1215525055@qq.com
parent
f1ffb08941
commit
470f1b9610
@ -0,0 +1,241 @@
|
||||
package cn.com.tenlion.operator.util;
|
||||
|
||||
import com.alibaba.fastjson2.JSONArray;
|
||||
import com.alibaba.fastjson2.JSON;
|
||||
import com.alibaba.fastjson2.JSONObject;
|
||||
import okhttp3.*;
|
||||
import javax.crypto.Mac;
|
||||
import javax.crypto.spec.SecretKeySpec;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.util.Base64;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
public class XinghuoAssistantClient {
|
||||
private final OkHttpClient client;
|
||||
private final String appId;
|
||||
private final String apiKey;
|
||||
private final String apiSecret; // 新增：星火鉴权必须的API Secret
|
||||
private final String assistantId;
|
||||
private final String wsUrl; // WebSocket接口地址
|
||||
|
||||
// 构造方法：新增apiSecret参数（你的Secret是ZGI1M2M4MDBlNjRhNDhhMTg1YTM4OWE1）
|
||||
public XinghuoAssistantClient(String appId, String apiKey, String apiSecret, String assistantId) {
|
||||
this.appId = appId;
|
||||
this.apiKey = apiKey;
|
||||
this.apiSecret = apiSecret;
|
||||
this.assistantId = assistantId;
|
||||
this.wsUrl = "wss://spark-openapi.cn-huabei-1.xf-yun.com/v1/assistants/" + assistantId;
|
||||
|
||||
// 初始化OkHttp客户端（保持超时配置）
|
||||
this.client = new OkHttpClient.Builder()
|
||||
.connectTimeout(30, TimeUnit.SECONDS)
|
||||
.readTimeout(60, TimeUnit.SECONDS)
|
||||
.writeTimeout(30, TimeUnit.SECONDS)
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 向星火助手发送消息并获取AI回复（支持流式分段响应）
|
||||
* @param userQuery 用户输入的问题文本
|
||||
* @return 完整的AI回复
|
||||
* @throws Exception 网络异常、鉴权失败等
|
||||
*/
|
||||
public String askQuestion(String userQuery) throws Exception {
|
||||
final StringBuilder aiResponseBuilder = new StringBuilder(); // 拼接流式响应
|
||||
final boolean[] isResponseComplete = {false}; // 标记响应是否完成
|
||||
final boolean[] isWebSocketClosed = {false}; // 标记连接是否关闭
|
||||
final Exception[] requestException = {null}; // 捕获回调中的异常
|
||||
|
||||
// 1. 生成鉴权头（核心：替换Bearer Token）
|
||||
String authHeader = generateAuthHeader();
|
||||
|
||||
// 2. 构建请求体JSON（已修正domain为generalv3）
|
||||
String requestBodyJson = buildRequestBodyJson(userQuery);
|
||||
|
||||
// 3. 创建WebSocket请求（无RequestBody，仅携带鉴权头）
|
||||
Request request = new Request.Builder()
|
||||
.url(wsUrl)
|
||||
.addHeader("Authorization", authHeader) // 使用签名鉴权
|
||||
.addHeader("Content-Type", "application/json")
|
||||
.build();
|
||||
|
||||
WebSocket webSocket = null;
|
||||
try {
|
||||
// 4. 建立WebSocket连接，在onOpen中发送请求体
|
||||
webSocket = client.newWebSocket(request, new WebSocketListener() {
|
||||
@Override
|
||||
public void onOpen(WebSocket webSocket, Response response) {
|
||||
System.out.println("✅ WebSocket connected successfully!");
|
||||
// 连接建立后，发送请求体（关键修正：这里才是正确的发送时机）
|
||||
webSocket.send(requestBodyJson);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onMessage(WebSocket webSocket, String text) {
|
||||
try {
|
||||
// 解析星火的响应（支持流式分段返回，status=2表示响应完成）
|
||||
JSONObject jsonObj = JSON.parseObject(text);
|
||||
int headerCode = jsonObj.getJSONObject("header").getIntValue("code");
|
||||
|
||||
// 先判断服务端是否返回错误（如鉴权失败、参数错误）
|
||||
if (headerCode != 0) {
|
||||
String errorMsg = jsonObj.getJSONObject("header").getString("message");
|
||||
requestException[0] = new RuntimeException("Server error: " + headerCode + " - " + errorMsg);
|
||||
webSocket.close(1001, "Server error");
|
||||
return;
|
||||
}
|
||||
|
||||
// 解析AI回复（流式返回可能分多段，需拼接）
|
||||
JSONObject payload = jsonObj.getJSONObject("payload");
|
||||
if (payload != null && payload.containsKey("choices")) {
|
||||
JSONObject choices = payload.getJSONObject("choices");
|
||||
int choicesStatus = choices.getIntValue("status"); // status=1：分段，status=2：完成
|
||||
JSONArray textArray = choices.getJSONArray("text");
|
||||
|
||||
for (int i = 0; i < textArray.size(); i++) {
|
||||
JSONObject item = textArray.getJSONObject(i);
|
||||
if ("assistant".equals(item.getString("role"))) {
|
||||
String segment = item.getString("content");
|
||||
aiResponseBuilder.append(segment); // 拼接分段内容
|
||||
System.out.println("💬 AI Segment: " + segment);
|
||||
}
|
||||
}
|
||||
|
||||
// 响应完成（status=2），标记结束
|
||||
if (choicesStatus == 2) {
|
||||
isResponseComplete[0] = true;
|
||||
System.out.println("✅ AI response completed!");
|
||||
}
|
||||
}
|
||||
} catch (Exception e) {
|
||||
requestException[0] = e; // 捕获解析异常
|
||||
webSocket.close(1002, "Parse error");
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onClosing(WebSocket webSocket, int code, String reason) {
|
||||
System.out.println("⚠️ Closing connection: code=" + code + ", reason=" + reason);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onClosed(WebSocket webSocket, int code, String reason) {
|
||||
System.out.println("🔒 Connection closed: code=" + code + ", reason=" + reason);
|
||||
isWebSocketClosed[0] = true;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onFailure(WebSocket webSocket, Throwable t, Response response) {
|
||||
requestException[0] = new RuntimeException("WebSocket failure", t); // 捕获连接失败异常
|
||||
isWebSocketClosed[0] = true;
|
||||
}
|
||||
});
|
||||
|
||||
// 5. 等待响应完成或超时（延长超时时间到30秒，适配流式返回）
|
||||
long startTime = System.currentTimeMillis();
|
||||
while (!isResponseComplete[0] && !isWebSocketClosed[0] && requestException[0] == null) {
|
||||
Thread.sleep(100); // 每100ms检查一次状态
|
||||
long elapsed = System.currentTimeMillis() - startTime;
|
||||
// 超时时间设为30秒（星火流式返回可能需要较长时间）
|
||||
if (elapsed > 30 * 1000) {
|
||||
throw new RuntimeException("Timeout waiting for AI response (elapsed: " + elapsed + "ms)");
|
||||
}
|
||||
}
|
||||
|
||||
// 6. 检查是否有异常（如鉴权失败、解析错误）
|
||||
if (requestException[0] != null) {
|
||||
throw requestException[0];
|
||||
}
|
||||
|
||||
// 返回完整的AI回复
|
||||
return aiResponseBuilder.toString();
|
||||
} finally {
|
||||
// 关闭WebSocket连接（避免资源泄漏）
|
||||
if (webSocket != null && !isWebSocketClosed[0] ) {
|
||||
webSocket.close(1000, "Normal closure");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成星火API要求的鉴权头（核心：HMAC-SHA256签名）
|
||||
* 鉴权逻辑参考：https://www.xfyun.cn/doc/spark/Web.html#_2-1-%E9%89%B4%E6%9D%83%E8%AF%B4%E6%98%8E
|
||||
*/
|
||||
private String generateAuthHeader() throws Exception {
|
||||
// 1. 获取当前时间戳（秒级）
|
||||
String timestamp = String.valueOf(System.currentTimeMillis() / 1000);
|
||||
// 2. 拼接签名原文（格式：apiKey + timestamp）
|
||||
String signatureOrigin = apiKey + timestamp;
|
||||
// 3. HMAC-SHA256加密（密钥是apiSecret，编码UTF-8）
|
||||
Mac mac = Mac.getInstance("HmacSHA256");
|
||||
mac.init(new SecretKeySpec(apiSecret.getBytes(StandardCharsets.UTF_8), "HmacSHA256"));
|
||||
byte[] signatureBytes = mac.doFinal(signatureOrigin.getBytes(StandardCharsets.UTF_8));
|
||||
// 4. Base64编码签名结果
|
||||
String signature = Base64.getEncoder().encodeToString(signatureBytes);
|
||||
// 5. 拼接鉴权头（格式：api_key="xxx",timestamp="xxx",signature="xxx"）
|
||||
return String.format("api_key=\"%s\",timestamp=\"%s\",signature=\"%s\"",
|
||||
apiKey, timestamp, signature);
|
||||
}
|
||||
|
||||
/**
|
||||
* 构建符合星火API规范的请求体JSON（已修正domain为generalv3）
|
||||
*/
|
||||
private String buildRequestBodyJson(String userQuery) {
|
||||
// 1. header部分（必传appId，uid可选）
|
||||
JSONObject header = new JSONObject();
|
||||
header.put("app_id", appId);
|
||||
header.put("uid", "user_123456"); // 建议传唯一用户ID（便于问题排查）
|
||||
|
||||
// 2. parameter部分（domain修正为generalv3，其他参数保持默认）
|
||||
JSONObject chatParams = new JSONObject();
|
||||
chatParams.put("domain", "generalv3"); // 关键修正：通用模型必须传generalv3
|
||||
chatParams.put("temperature", 0.5); // 采样阈值（0-1，值越大越随机）
|
||||
chatParams.put("top_k", 4); // TopK选择（1-6，默认4）
|
||||
chatParams.put("max_tokens", 2048); // 最大回复长度（1-4096，默认2048）
|
||||
|
||||
JSONObject parameter = new JSONObject();
|
||||
parameter.put("chat", chatParams);
|
||||
|
||||
// 3. payload部分（用户问题，role固定为user）
|
||||
JSONArray textArray = new JSONArray();
|
||||
JSONObject userMsg = new JSONObject();
|
||||
userMsg.put("role", "user");
|
||||
userMsg.put("content", userQuery);
|
||||
textArray.add(userMsg);
|
||||
|
||||
JSONObject message = new JSONObject();
|
||||
message.put("text", textArray);
|
||||
|
||||
JSONObject payload = new JSONObject();
|
||||
payload.put("message", message);
|
||||
|
||||
// 4. 组装根JSON
|
||||
JSONObject rootJson = new JSONObject();
|
||||
rootJson.put("header", header);
|
||||
rootJson.put("parameter", parameter);
|
||||
rootJson.put("payload", payload);
|
||||
|
||||
return rootJson.toJSONString();
|
||||
}
|
||||
|
||||
// 主方法：测试调用（替换为你的实际参数）
|
||||
public static void main(String[] args) {
|
||||
try {
|
||||
// 你的实际参数（从星火平台获取）
|
||||
String appId = "a9514070";
|
||||
String apiKey = "20a8e5c33dc8694925d4a5777cb19d53";
|
||||
String apiSecret = "ZGI1M2M4MDBlNjRhNDhhMTg1YTM4OWE1"; // 你的API Secret
|
||||
String assistantId = "hz96ujdxn91w_v1";
|
||||
|
||||
// 创建客户端并调用
|
||||
XinghuoAssistantClient client = new XinghuoAssistantClient(appId, apiKey, apiSecret, assistantId);
|
||||
String answer = client.askQuestion("今天的天气怎么样？");
|
||||
|
||||
// 打印最终结果
|
||||
System.out.println("\n📌 Final AI Answer:\n" + answer);
|
||||
} catch (Exception e) {
|
||||
System.err.println("❌ Call failed: ");
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user