From f79b4ec012347ba01daf4e6fa8569a9954e836e1 Mon Sep 17 00:00:00 2001
From: ageerle <32251822+ageerle@users.noreply.github.com>
Date: 星期四, 05 六月 2025 13:53:09 +0800
Subject: [PATCH] Merge pull request #114 from Code-Mr-Jiu/jiuyi-dev
---
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