tsurumure 5 месяцев назад
Родитель
Сommit
e0af34a3a2

+ 2 - 2
db/ai_chat_model.sql

@@ -16,7 +16,7 @@ CREATE TABLE `ai_chat_model` (
 ) ENGINE=INNODB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4 COMMENT='AI助手模型表';
 
 INSERT INTO ai_chat_model(id, parent_id, label, value, disabled, status) VALUES
-    (1, -1, 'DeepSeek R1', 'OLLAMA', -1, 1),
+    (1, -1, 'DeepSeek R1', 'DEEPSEEK', -1, 1),
         (2, 1, 'deepseek-r1:1.5b', 'deepseek-r1:1.5b', -1, 1),
         (3, 1, 'deepseek-r1:70b', 'deepseek-r1:70b', -1, 1),
         (4, 1, 'deepseek-r1:671b', 'deepseek-r1:671b', 1, 1),
@@ -24,6 +24,6 @@ INSERT INTO ai_chat_model(id, parent_id, label, value, disabled, status) VALUES
         (6, 5, 'deepseek-chat', 'deepseek-chat', -1, 1),
         (7, 5, 'deepseek-reasoner', 'deepseek-reasoner', -1, 1),
     (8, -1, '混元大模型', 'HUNYUAN', -1, 1),
-    (9, -1, 'Gemma3', 'OLLAMA', -1, 1),
+    (9, -1, 'Gemma3', 'GEMMA', -1, 1),
         (10, 9, 'gemma3:27b', 'gemma3:27b', -1, 1)
 ;

+ 1 - 1
src/main/java/com/backendsys/modules/ai/chat/service/impl/ChatServiceImpl.java

@@ -131,7 +131,7 @@ public class ChatServiceImpl implements ChatService {
 
                 ChatResult chatResult = null;
                 // -- [Deepseek R1] -------------------------------------------------------
-                if ("OLLAMA".equals(model)) {
+                if ("DEEPSEEK".equals(model) || "GEMMA".equals(model)) {
                     chatResult = ollamaUtil.chatDeepSeek(user_id, model_version, prompt, final_history_code, chatList);
                 }
                 // -- [Deepseek Api] ------------------------------------------------------