调用dify，解决时间问题，源语料为空的问题
This commit is contained in:
spllzh
308
src/main/java/com/rj/service/DifyWorkflowService.java
Normal file
308
src/main/java/com/rj/service/DifyWorkflowService.java
Normal file
@@ -0,0 +1,308 @@
|
||||
package com.rj.service;
|
||||
|
||||
import com.alibaba.fastjson.JSON;
|
||||
import com.alibaba.fastjson.JSONObject;
|
||||
import com.rj.entity.bz.DifyWorkflowAnalysis;
|
||||
import com.rj.service.biz.IDifyWorkflowAnalysisService;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.http.*;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.web.client.RestTemplate;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* Dify工作流服务类
|
||||
* 用于调用Dify平台的工作流API
|
||||
*
|
||||
* @author 李中华
|
||||
* @date 2025/1/3
|
||||
*/
|
||||
@Slf4j
|
||||
@Service
|
||||
public class DifyWorkflowService {
|
||||
|
||||
@Value("${dify.api.base-url}")
|
||||
private String difyBaseUrl;
|
||||
|
||||
@Value("${dify.api.workflow-endpoint-consultingScenar}")
|
||||
private String workflowEndpoint;
|
||||
|
||||
@Value("${dify.api.summary-qiwei-token}")// 企微（一句话+分类别） app-WPuiaYg0iVLc2ws0iOfsAUC6
|
||||
private String summaryQiweiToken;
|
||||
@Value("${dify.api.summary-ddc-token}")// DDC（一句话+分类别） app-rgaQbIir7vrVb1473Z3Puz6w
|
||||
private String summaryddcToken;
|
||||
@Value("${dify.api.summary-nameplate-token}")// 铭牌（一句话+分类别） //app-cv5glaYrY4zgjq0XIidSoJea
|
||||
private String summaryNameplateToken;
|
||||
|
||||
|
||||
|
||||
private final RestTemplate restTemplate;
|
||||
|
||||
@Autowired
|
||||
private IDifyWorkflowAnalysisService difyWorkflowAnalysisService;
|
||||
|
||||
public DifyWorkflowService() {
|
||||
this.restTemplate = new RestTemplate();
|
||||
}
|
||||
|
||||
/**
|
||||
* 调用企微对话分析工作流
|
||||
*
|
||||
* @param request 工作流请求参数
|
||||
* @return 工作流响应结果
|
||||
*/
|
||||
public DifyWorkflowResponse callConsultingScenarioWorkflow(DifyWorkflowRequest request) {
|
||||
try {
|
||||
String url = difyBaseUrl + workflowEndpoint;
|
||||
|
||||
// 构建请求头
|
||||
HttpHeaders headers = new HttpHeaders();
|
||||
headers.setContentType(MediaType.APPLICATION_JSON);
|
||||
headers.set("Authorization", "Bearer " + summaryQiweiToken);
|
||||
|
||||
// 构建请求体
|
||||
Map<String, Object> requestBody = new HashMap<>();
|
||||
requestBody.put("inputs", request.getInputs());
|
||||
requestBody.put("response_mode", "blocking");
|
||||
requestBody.put("user", request.getUserId());
|
||||
|
||||
HttpEntity<Map<String, Object>> entity = new HttpEntity<>(requestBody, headers);
|
||||
|
||||
log.info("调用Dify工作流API: {}", url);
|
||||
log.info("请求参数: {}", JSON.toJSONString(requestBody));
|
||||
|
||||
ResponseEntity<String> response = restTemplate.exchange(
|
||||
url,
|
||||
HttpMethod.POST,
|
||||
entity,
|
||||
String.class
|
||||
);
|
||||
|
||||
log.info("Dify工作流响应状态: {}", response.getStatusCode());
|
||||
log.info("Dify工作流响应内容: {}", response.getBody());
|
||||
|
||||
if (response.getStatusCode() == HttpStatus.OK) {
|
||||
JSONObject responseJson = JSON.parseObject(response.getBody());
|
||||
return parseWorkflowResponse(responseJson, request);
|
||||
} else {
|
||||
throw new RuntimeException("Dify工作流调用失败，状态码: " + response.getStatusCode());
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error("调用Dify工作流异常", e);
|
||||
throw new RuntimeException("调用Dify工作流异常: " + e.getMessage(), e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 解析工作流响应并保存数据
|
||||
*/
|
||||
private DifyWorkflowResponse parseWorkflowResponse(JSONObject responseJson, DifyWorkflowRequest request) {
|
||||
DifyWorkflowResponse response = new DifyWorkflowResponse();
|
||||
|
||||
if (responseJson.containsKey("data")) {
|
||||
JSONObject data = responseJson.getJSONObject("data");
|
||||
response.setWorkflowRunId(data.getString("workflow_run_id"));
|
||||
response.setTaskId(data.getString("task_id"));
|
||||
response.setData(data);
|
||||
|
||||
// 解析并保存分析结果到数据库
|
||||
if (data.containsKey("outputs")) {
|
||||
JSONObject outputs = data.getJSONObject("outputs");
|
||||
log.info("Dify工作流响应数据 outputs : {}", outputs);
|
||||
|
||||
// 保存分析结果到数据库
|
||||
saveAnalysisResultToDatabase(outputs, request);
|
||||
}
|
||||
}
|
||||
|
||||
if (responseJson.containsKey("metadata")) {
|
||||
response.setMetadata(responseJson.getJSONObject("metadata"));
|
||||
}
|
||||
|
||||
return response;
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存分析结果到数据库
|
||||
*/
|
||||
private void saveAnalysisResultToDatabase(JSONObject outputs, DifyWorkflowRequest request) {
|
||||
try {
|
||||
DifyWorkflowAnalysis analysis = new DifyWorkflowAnalysis();
|
||||
|
||||
// 从outputs中提取数据
|
||||
if (outputs.containsKey("data")) {
|
||||
JSONObject analysisData = outputs.getJSONObject("data");
|
||||
|
||||
// 提取分析结果摘要
|
||||
if (analysisData.containsKey("analysisResult")) {
|
||||
analysis.setAnalysisResult(analysisData.getString("analysisResult"));
|
||||
}
|
||||
|
||||
// 解析详细分析结果
|
||||
if (analysisData.containsKey("analysisDetail")) {
|
||||
JSONObject analysisDetail = analysisData.getJSONObject("analysisDetail");
|
||||
|
||||
// 解析客户需求
|
||||
if (analysisDetail.containsKey("customerNeeds")) {
|
||||
JSONObject customerNeeds = analysisDetail.getJSONObject("customerNeeds");
|
||||
analysis.setCustomerSource(customerNeeds.getString("customerSource"));
|
||||
analysis.setCustomerOccupation(customerNeeds.getString("customerOccupation"));
|
||||
analysis.setCustomerHobbies(customerNeeds.getString("customerHobbies"));
|
||||
analysis.setHomeAddress(customerNeeds.getString("homeAddress"));
|
||||
analysis.setCarPurchaseNeed(customerNeeds.getString("carPurchase"));
|
||||
analysis.setPurchaseType(customerNeeds.getString("purchaseType"));
|
||||
analysis.setPurchaseBuyer(customerNeeds.getString("purchaseBuyer"));
|
||||
analysis.setCarUser(customerNeeds.getString("carUser"));
|
||||
analysis.setIntendedCarModel(customerNeeds.getString("intendedCarModel"));
|
||||
analysis.setCarQualifications(customerNeeds.getString("carQualifications"));
|
||||
analysis.setCarBudget(customerNeeds.getString("carBudget"));
|
||||
analysis.setFinancialInstallment(customerNeeds.getString("financialInstallment"));
|
||||
analysis.setPurchaseCycle(customerNeeds.getString("purchaseCycle"));
|
||||
analysis.setFocusPoints(customerNeeds.getString("focus"));
|
||||
analysis.setConcerns(customerNeeds.getString("concerns"));
|
||||
analysis.setCarContrast(customerNeeds.getString("carContrast"));
|
||||
}
|
||||
|
||||
// 解析顾问服务
|
||||
if (analysisDetail.containsKey("customerService")) {
|
||||
JSONObject customerService = analysisDetail.getJSONObject("customerService");
|
||||
analysis.setProductDescription(customerService.getString("productDesc"));
|
||||
analysis.setSolution(customerService.getString("solution"));
|
||||
analysis.setQuotation(customerService.getString("quotation"));
|
||||
}
|
||||
|
||||
// 解析后续行动和未解决问题
|
||||
analysis.setAgreedFollowUpActions(analysisDetail.getString("agreedFollowUpActions"));
|
||||
analysis.setUnresolvedIssues(analysisDetail.getString("unresolvedIssues"));
|
||||
}
|
||||
}
|
||||
|
||||
// 从outputs中提取其他字段
|
||||
if (outputs.containsKey("unionId")) {
|
||||
analysis.setUnionId(outputs.getString("unionId"));
|
||||
}
|
||||
if (outputs.containsKey("consultantId")) {
|
||||
analysis.setConsultantId(outputs.getString("consultantId"));
|
||||
}
|
||||
if (outputs.containsKey("communicateDate")) {
|
||||
try {
|
||||
DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
|
||||
analysis.setCommunicateDate(LocalDateTime.parse(outputs.getString("communicateDate"), formatter));
|
||||
} catch (Exception e) {
|
||||
log.warn("解析沟通时间失败: {}", outputs.getString("communicateDate"), e);
|
||||
analysis.setCommunicateDate(LocalDateTime.now());
|
||||
}
|
||||
}
|
||||
if (outputs.containsKey("analysisScene")) {
|
||||
analysis.setAnalysisScene(outputs.getString("analysisScene"));
|
||||
}
|
||||
if (outputs.containsKey("analysisRecordId")) {
|
||||
analysis.setAnalysisRecordId(outputs.getString("analysisRecordId"));
|
||||
}
|
||||
if (outputs.containsKey("version")) {
|
||||
analysis.setVersion(outputs.getInteger("version"));
|
||||
}
|
||||
|
||||
// 从request的inputs中提取原始对话内容
|
||||
if (request.getInputs() != null && request.getInputs().containsKey("chat")) {
|
||||
analysis.setOriginalCorpus((String) request.getInputs().get("chat"));
|
||||
}
|
||||
|
||||
// 设置创建时间和更新时间
|
||||
LocalDateTime now = LocalDateTime.now();
|
||||
analysis.setCreatedAt(now);
|
||||
analysis.setUpdatedAt(now);
|
||||
|
||||
// 保存到数据库
|
||||
boolean saveResult = difyWorkflowAnalysisService.save(analysis);
|
||||
if (saveResult) {
|
||||
log.info("分析结果保存成功，ID: {}", analysis.getId());
|
||||
} else {
|
||||
log.error("分析结果保存失败");
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error("保存分析结果到数据库失败", e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Dify工作流请求参数
|
||||
*/
|
||||
public static class DifyWorkflowRequest {
|
||||
private Map<String, Object> inputs;
|
||||
private String userId;
|
||||
|
||||
public DifyWorkflowRequest() {}
|
||||
|
||||
public DifyWorkflowRequest(Map<String, Object> inputs, String userId) {
|
||||
this.inputs = inputs;
|
||||
this.userId = userId;
|
||||
}
|
||||
|
||||
public Map<String, Object> getInputs() {
|
||||
return inputs;
|
||||
}
|
||||
|
||||
public void setInputs(Map<String, Object> inputs) {
|
||||
this.inputs = inputs;
|
||||
}
|
||||
|
||||
public String getUserId() {
|
||||
return userId;
|
||||
}
|
||||
|
||||
public void setUserId(String userId) {
|
||||
this.userId = userId;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Dify工作流响应结果
|
||||
*/
|
||||
public static class DifyWorkflowResponse {
|
||||
private String workflowRunId;
|
||||
private String taskId;
|
||||
private JSONObject data;
|
||||
private JSONObject metadata;
|
||||
|
||||
public String getWorkflowRunId() {
|
||||
return workflowRunId;
|
||||
}
|
||||
|
||||
public void setWorkflowRunId(String workflowRunId) {
|
||||
this.workflowRunId = workflowRunId;
|
||||
}
|
||||
|
||||
public String getTaskId() {
|
||||
return taskId;
|
||||
}
|
||||
|
||||
public void setTaskId(String taskId) {
|
||||
this.taskId = taskId;
|
||||
}
|
||||
|
||||
public JSONObject getData() {
|
||||
return data;
|
||||
}
|
||||
|
||||
public void setData(JSONObject data) {
|
||||
this.data = data;
|
||||
}
|
||||
|
||||
public JSONObject getMetadata() {
|
||||
return metadata;
|
||||
}
|
||||
|
||||
public void setMetadata(JSONObject metadata) {
|
||||
this.metadata = metadata;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user