xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-01 22:13:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

上传图片支持使用后台image分类模型

Browse Source
This commit is contained in:
酒亦
2025-06-02 08:11:22 +08:00
parent bd94a3eae8
commit d635e30b4a
5 changed files with 100 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/service/IChatModelService.java
View File

@@ -53,9 +53,13 @@ public interface IChatModelService {
* 通过模型名称获取模型信息
*/
ChatModelVo selectModelByName(String modelName);
/**
* 通过模型分类获取模型信息
*/
ChatModelVo selectModelByCategory(String image);
/**
* 获取ppt模型信息
*/
ChatModel getPPT();
}

7
ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/service/impl/ChatModelServiceImpl.java
View File

@@ -129,6 +129,13 @@ public class ChatModelServiceImpl implements IChatModelService {
public ChatModelVo selectModelByName(String modelName) {
return baseMapper.selectVoOne(Wrappers.<ChatModel>lambdaQuery().eq(ChatModel::getModelName, modelName));
}
/**
* 通过模型分类获取模型信息
*/
@Override
public ChatModelVo selectModelByCategory(String category) {
return baseMapper.selectVoOne(Wrappers.<ChatModel>lambdaQuery().eq(ChatModel::getCategory, category));
}
@Override
public ChatModel getPPT() {