Commit fe7e6177 authored by alex yao's avatar alex yao

fix[车险]:修复对话结束无[DONE]输出问题

parent 19a3f26f
...@@ -340,7 +340,7 @@ ...@@ -340,7 +340,7 @@
<dependency> <dependency>
<groupId>com.baidubce</groupId> <groupId>com.baidubce</groupId>
<artifactId>appbuilder</artifactId> <artifactId>appbuilder</artifactId>
<version>0.9.4</version> <version>1.1.0</version>
</dependency> </dependency>
<dependency> <dependency>
......
...@@ -339,6 +339,7 @@ public class AiDialoguesServiceImpl implements AiDialoguesService { ...@@ -339,6 +339,7 @@ public class AiDialoguesServiceImpl implements AiDialoguesService {
StringBuilder reasoningContentBuilder = new StringBuilder(); StringBuilder reasoningContentBuilder = new StringBuilder();
long inputTimeStamp = System.currentTimeMillis(); long inputTimeStamp = System.currentTimeMillis();
try {
if (dialoguesType.equals(AiDialoguesTypeEnum.CAR_INSURANCE.getType())) {//AppBuilder if (dialoguesType.equals(AiDialoguesTypeEnum.CAR_INSURANCE.getType())) {//AppBuilder
BizAiDialogues3rdRelationEntity bizAiDialogues3rdRelationEntity = new BizAiDialogues3rdRelationEntity(); BizAiDialogues3rdRelationEntity bizAiDialogues3rdRelationEntity = new BizAiDialogues3rdRelationEntity();
bizAiDialogues3rdRelationEntity.setDialoguesId(dialoguesId); bizAiDialogues3rdRelationEntity.setDialoguesId(dialoguesId);
...@@ -352,6 +353,7 @@ public class AiDialoguesServiceImpl implements AiDialoguesService { ...@@ -352,6 +353,7 @@ public class AiDialoguesServiceImpl implements AiDialoguesService {
while (iterator.hasNext()) { while (iterator.hasNext()) {
AppBuilderClientResult res = iterator.next(); AppBuilderClientResult res = iterator.next();
LargeModelDemandResult largeModelDemandResult = new LargeModelDemandResult(); LargeModelDemandResult largeModelDemandResult = new LargeModelDemandResult();
largeModelDemandResult.setCode("0");
if (ArrayUtils.isNotEmpty(res.getEvents())) { if (ArrayUtils.isNotEmpty(res.getEvents())) {
Event event = res.getEvents()[0]; Event event = res.getEvents()[0];
String eventType = event.getEventType(); String eventType = event.getEventType();
...@@ -369,11 +371,13 @@ public class AiDialoguesServiceImpl implements AiDialoguesService { ...@@ -369,11 +371,13 @@ public class AiDialoguesServiceImpl implements AiDialoguesService {
} }
sseUtil.send(JsonUtils.serialize(largeModelDemandResult)); sseUtil.send(JsonUtils.serialize(largeModelDemandResult));
} }
}
} catch (Exception e) {
logger.error(e.getMessage());
sseUtil.completeByError(e.getMessage());
} finally {
sseUtil.send("[DONE]"); sseUtil.send("[DONE]");
sseUtil.complete(); sseUtil.complete();
}
//保存对话记录 //保存对话记录
//用户输入记录 //用户输入记录
BizAgentApplicationDialoguesRecordEntity inputRecord = new BizAgentApplicationDialoguesRecordEntity(); BizAgentApplicationDialoguesRecordEntity inputRecord = new BizAgentApplicationDialoguesRecordEntity();
...@@ -397,7 +401,7 @@ public class AiDialoguesServiceImpl implements AiDialoguesService { ...@@ -397,7 +401,7 @@ public class AiDialoguesServiceImpl implements AiDialoguesService {
outputRecord.setContent(contentBuilder.toString()); outputRecord.setContent(contentBuilder.toString());
outputRecord.setReasoningContent(reasoningContentBuilder.toString()); outputRecord.setReasoningContent(reasoningContentBuilder.toString());
bizAgentApplicationDialoguesRecordService.save(outputRecord); bizAgentApplicationDialoguesRecordService.save(outputRecord);
}
return contentBuilder.toString(); return contentBuilder.toString();
} }
} }
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