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-01 09:39:09 +08:00
parent 09f12fdd8c
commit 5916c0af59
1 changed files with 20 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
modules/ai/src/main/java/com/bytedesk/ai/robot/RobotRestService.java
View File

@@ -30,7 +30,10 @@ import com.bytedesk.ai.provider.LlmProviderEntity;
import com.bytedesk.ai.provider.LlmProviderRestService;
import com.bytedesk.ai.robot.RobotJsonLoader.Robot;
import com.bytedesk.ai.robot.RobotJsonLoader.RobotConfiguration;
import com.bytedesk.ai.robot_settings.RobotLlmRequest;
import com.bytedesk.ai.robot_settings.RobotSettingsEntity;
import com.bytedesk.ai.robot_settings.RobotSettingsRequest;
import com.bytedesk.ai.robot_settings.RobotSettingsResponse;
import com.bytedesk.ai.utils.ConvertAiUtils;
import com.bytedesk.core.base.BaseRestServiceWithExport;
import com.bytedesk.core.category.CategoryTypeEnum;
@@ -435,22 +438,27 @@ public class RobotRestService extends BaseRestServiceWithExport<RobotEntity, Rob
return convertToResponse(findByUid(robotUid).get());
}
// 创建带有默认提示词的 Settings 并绑定
com.bytedesk.ai.robot_settings.RobotSettingsRequest settingsReq = com.bytedesk.ai.robot_settings.RobotSettingsRequest
RobotSettingsRequest settingsReq = RobotSettingsRequest
.builder()
.orgUid(orgUid)
.name("空白智能体配置")
.llm(com.bytedesk.ai.robot_settings.RobotLlmRequest.builder().prompt("请回答用户提出的问题").build())
.llm(RobotLlmRequest.builder().prompt("请回答用户提出的问题").build())
.build();
com.bytedesk.ai.robot_settings.RobotSettingsResponse settingsResp = robotSettingsRestService
RobotSettingsResponse settingsResp = robotSettingsRestService
.create(settingsReq);
// 使用已持久化的 Settings 实体，避免出现未保存的瞬态引用
RobotSettingsEntity persistedSettings = robotSettingsRestService
.findByUid(settingsResp.getUid())
.orElseThrow(() -> new RuntimeException("persisted settings not found: " + settingsResp.getUid()));
RobotEntity robot = RobotEntity.builder()
.uid(robotUid)
.name(RobotConsts.ROBOT_NAME_VOID_AGENT)
.nickname("空白智能体")
.type(RobotTypeEnum.LLM.name())
.orgUid(orgUid)
.settings(modelMapper.map(settingsResp, com.bytedesk.ai.robot_settings.RobotSettingsEntity.class))
.settings(persistedSettings)
.build();
//
RobotEntity updatedRobot = save(robot);
@@ -497,17 +505,21 @@ public class RobotRestService extends BaseRestServiceWithExport<RobotEntity, Rob
: robotJson.getI18n().getEn();
// 创建 Settings 并设置 LLM 提示词
com.bytedesk.ai.robot_settings.RobotSettingsRequest settingsReq = com.bytedesk.ai.robot_settings.RobotSettingsRequest
RobotSettingsRequest settingsReq = RobotSettingsRequest
.builder()
.orgUid(orgUid)
.name(robotJson.getName() + "配置")
.llm(com.bytedesk.ai.robot_settings.RobotLlmRequest.builder().prompt(localeData.getPrompt())
.llm(RobotLlmRequest.builder().prompt(localeData.getPrompt())
.build())
.build();
com.bytedesk.ai.robot_settings.RobotSettingsResponse settingsResp = robotSettingsRestService
RobotSettingsResponse settingsResp = robotSettingsRestService
.create(settingsReq);
// Create RobotEntity with data from both Robot and LocaleData
RobotSettingsEntity persistedSettings = robotSettingsRestService
.findByUid(settingsResp.getUid())
.orElseThrow(() -> new RuntimeException(
"persisted settings not found: " + settingsResp.getUid()));
RobotEntity robot = RobotEntity.builder()
.uid(uidUtils.getUid())
.name(robotJson.getName())
@@ -518,8 +530,7 @@ public class RobotRestService extends BaseRestServiceWithExport<RobotEntity, Rob
.categoryUid(categoryUid)
.level(level)
.orgUid(orgUid)
.settings(
modelMapper.map(settingsResp, com.bytedesk.ai.robot_settings.RobotSettingsEntity.class))
.settings(persistedSettings)
.system(true)
.build();
//