ZF/weiyu
1
0
Fork 0
mirror of https://gitee.com/270580156/weiyu.git synced 2025-12-30 10:52:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
jack ning
2025-11-19 11:34:45 +08:00
parent e9f23632c8
commit 8a20b8a9d2
2 changed files with 30 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/ai/src/main/java/com/bytedesk/ai/robot/RobotEntity.java
View File

@@ -107,9 +107,6 @@ public class RobotEntity extends BaseEntity {
@Column(name = "is_kb_source_enabled")
private Boolean kbSourceEnabled = true;
// @Builder.Default
// private String defaultReply = I18Consts.I18N_ROBOT_DEFAULT_REPLY;
/**
* 是否启用知识库问答
*/
@@ -140,8 +137,6 @@ public class RobotEntity extends BaseEntity {
// @Builder.Default
// private Boolean published = false;
// 知识库相关字段已迁移至 RobotSettingsEntity
/**
* Whether workflow/flow integration is enabled
*/

41
modules/ai/src/main/java/com/bytedesk/ai/springai/service/BaseSpringAIService.java
View File

@@ -145,18 +145,38 @@ public abstract class BaseSpringAIService implements SpringAIService {
true,
true);
return;
}
List<Message> messages = promptHelper.buildMessagesForSse(query, "", robot, messageProtobufQuery);
Prompt aiPrompt = promptHelper.toPrompt(messages);
try {
processPromptSse(aiPrompt, robot, messageProtobufQuery, messageProtobufReply, new ArrayList<>(),
emitter);
} catch (Exception e) {
log.error("processPromptSse 异常(无知识库)", e);
sseMessageHelper.handleSseError(new Exception(I18Consts.I18N_ROBOT_PROCESSING_ERROR),
} else if (robot.getLlm().getEnabled()) {
// 开启大模型对话，且无知识库
List<Message> messages = promptHelper.buildMessagesForSse(query, "", robot, messageProtobufQuery);
Prompt aiPrompt = promptHelper.toPrompt(messages);
try {
processPromptSse(aiPrompt, robot, messageProtobufQuery, messageProtobufReply, new ArrayList<>(),
emitter);
} catch (Exception e) {
log.error("processPromptSse 异常(无知识库)", e);
sseMessageHelper.handleSseError(new Exception(I18Consts.I18N_ROBOT_PROCESSING_ERROR),
messageProtobufQuery,
messageProtobufReply, emitter);
}
} else {
// 未开启大模型对话，且无知识库：直接返回默认回复并结束 SSE
String answer = robot.getLlm() != null && robot.getLlm().getDefaultReply() != null
? robot.getLlm().getDefaultReply()
: I18Consts.I18N_ROBOT_DEFAULT_REPLY;
String robotStreamContent = promptHelper.createRobotStreamContentAnswer(query, answer,
new ArrayList<>(), robot);
sseMessageHelper.sendStreamMessage(
messageProtobufQuery,
messageProtobufReply, emitter);
messageProtobufReply,
emitter,
robotStreamContent,
null,
null,
true,
true,
true);
}
return;
}
@@ -250,7 +270,6 @@ public abstract class BaseSpringAIService implements SpringAIService {
}
RobotContent streamContent = streamContentBuilder.build();
sseMessageHelper.sendStreamMessage(
messageProtobufQuery,
messageProtobufReply,