ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/service/IChatModelService.java
@@ -3,6 +3,7 @@ import org.ruoyi.core.page.TableDataInfo; import org.ruoyi.core.page.PageQuery; import org.ruoyi.domain.ChatModel; import org.ruoyi.domain.bo.ChatModelBo; import org.ruoyi.domain.vo.ChatModelVo; @@ -53,4 +54,8 @@ */ ChatModelVo selectModelByName(String modelName); /** * 获取ppt模型信息 */ ChatModel getPPT(); } ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/service/impl/ChatModelServiceImpl.java
@@ -120,8 +120,11 @@ */ @Override public ChatModelVo selectModelByName(String modelName) { return baseMapper.selectVoOne(Wrappers.<ChatModel>lambdaQuery().eq(ChatModel::getModelName, modelName)); return baseMapper.selectVoOne(Wrappers.<ChatModel>lambdaQuery().eq(ChatModel::getModelName, modelName)); } @Override public ChatModel getPPT() { return baseMapper.selectOne(Wrappers.<ChatModel>lambdaQuery().eq(ChatModel::getModelName, "ppt")); } } ruoyi-modules-api/ruoyi-knowledge-api/pom.xml
@@ -33,6 +33,17 @@ <dependencies> <dependency> <groupId>commons-io</groupId> <artifactId>commons-io</artifactId> <version>2.17.0</version> </dependency> <dependency> <groupId>org.ruoyi</groupId> <artifactId>ruoyi-system-api</artifactId> </dependency> <!-- pdf解析器 --> <dependency> <groupId>org.apache.pdfbox</groupId> @@ -52,16 +63,9 @@ <artifactId>langchain4j</artifactId> </dependency> <dependency> <groupId>dev.langchain4j</groupId> <artifactId>langchain4j-weaviate</artifactId> </dependency> <dependency> <groupId>dev.langchain4j</groupId> <artifactId>langchain4j-embeddings-all-minilm-l6-v2</artifactId> </dependency> <dependency> @@ -82,42 +86,8 @@ <dependency> <groupId>dev.langchain4j</groupId> <artifactId>langchain4j-milvus</artifactId> </dependency> <dependency> <groupId>org.testcontainers</groupId> <artifactId>milvus</artifactId> <version>1.19.6</version> </dependency> <dependency> <groupId>dev.langchain4j</groupId> <artifactId>langchain4j-qdrant</artifactId> </dependency> <dependency> <groupId>org.testcontainers</groupId> <artifactId>qdrant</artifactId> <version>1.19.6</version> </dependency> <dependency> <groupId>dev.langchain4j</groupId> <artifactId>langchain4j-document-parser-apache-tika</artifactId> </dependency> <!-- ruoyi-knowledge-api/pom.xml --> <dependency> <groupId>commons-io</groupId> <artifactId>commons-io</artifactId> <version>2.17.0</version> </dependency> <dependency> <groupId>org.ruoyi</groupId> <artifactId>ruoyi-system-api</artifactId> </dependency> </dependencies> ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/config/OkHttpConfig.java
@@ -25,11 +25,13 @@ public void init() { initializeOkHttpUtil("suno"); initializeOkHttpUtil("luma"); initializeOkHttpUtil("ppt"); } private void initializeOkHttpUtil(String modelName) { ChatModelVo chatModelVo = chatModelService.selectModelByName(modelName); if(chatModelVo==null){ return; } OkHttpUtil okHttpUtil = new OkHttpUtil(); okHttpUtil.setApiHost(chatModelVo.getApiHost()); okHttpUtil.setApiKey(chatModelVo.getApiKey()); ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/controller/chat/ChatModelController.java
@@ -10,6 +10,7 @@ import org.ruoyi.common.excel.utils.ExcelUtil; import org.ruoyi.common.idempotent.annotation.RepeatSubmit; import org.ruoyi.core.page.TableDataInfo; import org.ruoyi.domain.ChatModel; import org.ruoyi.domain.bo.ChatModelBo; import org.ruoyi.domain.vo.ChatModelVo; import org.ruoyi.service.IChatModelService; @@ -55,6 +56,15 @@ return R.ok(chatModelService.queryList(bo)); } /** * 查询ppt模型信息 */ @GetMapping("/getPPT") public R<ChatModel> getPPT() { return R.ok(chatModelService.getPPT()); } /** * 导出聊天模型列表 */