xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-03-13 20:53:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 根据是否有附件和是否自动，自动选择模型并且获取服务；

Browse Source
This commit is contained in:
likunlong
2025-08-19 10:32:17 +08:00
parent 8751bb5104
commit 07cb351807
2 changed files with 16 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/request/ChatRequest.java
View File

@@ -67,4 +67,14 @@ public class ChatRequest {
*/
private Long uuid;
/**
* 是否有附件
*/
private Boolean hasAttachment;
/**
* 是否自动切换模型
*/
private Boolean autoSelectModel;
}

15
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/SseServiceImpl.java
View File

@@ -128,23 +128,20 @@ public class SseServiceImpl implements ISseService {
*/
private IChatService autoSelectModelAndGetService(ChatRequest chatRequest) {
try {
// 处理特殊模型类型
if ("gpt-image".equals(chatRequest.getModel())) {
if (Boolean.TRUE.equals(chatRequest.getHasAttachment())) {
chatModelVo = selectModelByCategory("image");
return chatServiceFactory.getChatService(chatModelVo.getCategory());
} else if (Boolean.TRUE.equals(chatRequest.getAutoSelectModel())) {
chatModelVo = selectModelByCategory("chat");
} else {
chatModelVo = chatModelService.selectModelByName(chatRequest.getModel());
}
// 根据模型名称获取模型分类，然后获取该分类下优先级最高的模型
ChatModelVo tempModel = chatModelService.selectModelByName(chatRequest.getModel());
if (tempModel == null) {
if (chatModelVo == null) {
throw new IllegalStateException("未找到模型名称：" + chatRequest.getModel());
}
chatModelVo = selectModelByCategory(tempModel.getCategory());
// 直接返回对应的聊天服务
return chatServiceFactory.getChatService(chatModelVo.getCategory());
} catch (Exception e) {
log.error("模型选择和服务获取失败: {}", e.getMessage(), e);
throw new IllegalStateException("模型选择和服务获取失败: " + e.getMessage());