From d635e30b4aa0ec38844f1094038193308dd8fdd8 Mon Sep 17 00:00:00 2001
From: 酒亦 <mobaitongxue@163.com>
Date: 星期一, 02 六月 2025 08:11:22 +0800
Subject: [PATCH] 上传图片支持使用后台image分类模型
---
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/SseServiceImpl.java | 11 ++++++++++-
1 files changed, 10 insertions(+), 1 deletions(-)
diff --git a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/SseServiceImpl.java b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/SseServiceImpl.java
index 472fd7b..f7dee9a 100644
--- a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/SseServiceImpl.java
+++ b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/SseServiceImpl.java
@@ -125,7 +125,16 @@
*/
private void buildChatMessageList(ChatRequest chatRequest){
String sysPrompt;
- chatModelVo = chatModelService.selectModelByName(chatRequest.getModel());
+ // 鐭妯″瀷鍚嶇О 濡傛灉鏄痝pt-image 鍒欐煡璇mage绫诲瀷妯″瀷 鑾峰彇妯″瀷鍚嶇О
+ if(chatRequest.getModel().equals("gpt-image")) {
+ chatModelVo = chatModelService.selectModelByCategory("image");
+ if (chatModelVo == null) {
+ log.error("鏈壘鍒癷mage绫诲瀷鐨勬ā鍨嬮厤缃�");
+ throw new IllegalStateException("鏈壘鍒癷mage绫诲瀷鐨勬ā鍨嬮厤缃�");
+ }// chatRequest.setModel(chatModelVo.getModelName());
+ }else{
+ chatModelVo = chatModelService.selectModelByName(chatRequest.getModel());
+ }
// 鑾峰彇瀵硅瘽娑堟伅鍒楄〃
List<Message> messages = chatRequest.getMessages();
// 鏌ヨ鍚戦噺搴撶浉鍏充俊鎭姞鍏ュ埌涓婁笅鏂�
--
Gitblit v1.9.3