Commit ced06bc4 authored by alex yao's avatar alex yao

fix[智能问数]: csv对话

parent 75bca6fb
...@@ -298,6 +298,7 @@ public class AiBiServiceImpl implements AiBiService { ...@@ -298,6 +298,7 @@ public class AiBiServiceImpl implements AiBiService {
if (csvChainResult != null) { if (csvChainResult != null) {
sseUtil.send(JsonUtils.serialize(csvChainResult));
if (StringUtils.isNotBlank(csvChainResult.getResult())) { if (StringUtils.isNotBlank(csvChainResult.getResult())) {
JSONObject echartJSONObject = new JSONObject(); JSONObject echartJSONObject = new JSONObject();
echartJSONObject.put("sql", StringUtils.EMPTY); echartJSONObject.put("sql", StringUtils.EMPTY);
......
package cn.com.poc.thirdparty.resource.demand.ai.entity.largemodel; package cn.com.poc.thirdparty.resource.demand.ai.entity.largemodel;
import cn.com.poc.agent_application.entity.KnowledgeContentResult; import cn.com.poc.agent_application.entity.KnowledgeContentResult;
import cn.com.poc.thirdparty.resource.demand.ai.entity.csvchain.CSVChainResult;
import cn.com.poc.thirdparty.resource.demand.ai.entity.dbchain.DBChainResult; import cn.com.poc.thirdparty.resource.demand.ai.entity.dbchain.DBChainResult;
import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.ToolFunction; import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.ToolFunction;
import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.Usage; import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.Usage;
...@@ -19,6 +20,8 @@ public class LargeModelDemandResult extends AbstractResult implements Serializab ...@@ -19,6 +20,8 @@ public class LargeModelDemandResult extends AbstractResult implements Serializab
private ToolFunction function; private ToolFunction function;
private CSVChainResult csvChainResult;
private List<DBChainResult> dbChainResult; private List<DBChainResult> dbChainResult;
private List<KnowledgeContentResult> knowledgeContentResult; private List<KnowledgeContentResult> knowledgeContentResult;
...@@ -43,6 +46,14 @@ public class LargeModelDemandResult extends AbstractResult implements Serializab ...@@ -43,6 +46,14 @@ public class LargeModelDemandResult extends AbstractResult implements Serializab
this.knowledgeContentResult = knowledgeContentResult; this.knowledgeContentResult = knowledgeContentResult;
} }
public CSVChainResult getCsvChainResult() {
return csvChainResult;
}
public void setCsvChainResult(CSVChainResult csvChainResult) {
this.csvChainResult = csvChainResult;
}
public String getCode() { public String getCode() {
return code; return code;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment