From 607de74aceef61840589cd5727979e870fbf00f3 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Tue, 16 Jul 2024 10:55:33 +0800
Subject: [PATCH] fix minimax bug (#1528)

### What problem does this PR solve?

#1353

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 api/db/init_data.py | 19 +++++++------------
 rag/llm/__init__.py |  1 +
 2 files changed, 8 insertions(+), 12 deletions(-)

diff --git a/api/db/init_data.py b/api/db/init_data.py
index ec295d57a..e1e79057d 100644
--- a/api/db/init_data.py
+++ b/api/db/init_data.py
@@ -573,42 +573,35 @@ def init_llm_factory():
         # ------------------------ Minimax -----------------------
         {
             "fid": factory_infos[13]["name"],
-            "llm_name": "abab6.5-chat",
+            "llm_name": "abab6.5",
             "tags": "LLM,CHAT,8k",
             "max_tokens": 8192,
             "model_type": LLMType.CHAT.value
         },
         {
             "fid": factory_infos[13]["name"],
-            "llm_name": "abab6.5s-chat",
+            "llm_name": "abab6.5s",
             "tags": "LLM,CHAT,245k",
             "max_tokens": 245760,
             "model_type": LLMType.CHAT.value
         },
         {
             "fid": factory_infos[13]["name"],
-            "llm_name": "abab6.5t-chat",
+            "llm_name": "abab6.5t",
             "tags": "LLM,CHAT,8k",
             "max_tokens": 8192,
             "model_type": LLMType.CHAT.value
         },
         {
             "fid": factory_infos[13]["name"],
-            "llm_name": "abab6.5g-chat",
+            "llm_name": "abab6.5g",
             "tags": "LLM,CHAT,8k",
             "max_tokens": 8192,
             "model_type": LLMType.CHAT.value
         },
         {
             "fid": factory_infos[13]["name"],
-            "llm_name": "abab5.5-chat",
-            "tags": "LLM,CHAT,16k",
-            "max_tokens": 16384,
-            "model_type": LLMType.CHAT.value
-        },
-        {
-            "fid": factory_infos[13]["name"],
-            "llm_name": "abab5.5s-chat",
+            "llm_name": "abab5.5s",
             "tags": "LLM,CHAT,8k",
             "max_tokens": 8192,
             "model_type": LLMType.CHAT.value
@@ -987,6 +980,8 @@ def init_llm_factory():
             LLMFactoriesService.save(**info)
         except Exception as e:
             pass
+
+    LLMService.filter_delete([(LLM.fid == "MiniMax" or LLM.fid == "Minimax")])
     for info in llm_infos:
         try:
             LLMService.save(**info)
diff --git a/rag/llm/__init__.py b/rag/llm/__init__.py
index 50b541f41..4d780b4e8 100644
--- a/rag/llm/__init__.py
+++ b/rag/llm/__init__.py
@@ -61,6 +61,7 @@ ChatModel = {
     "VolcEngine": VolcEngineChat,
     "BaiChuan": BaiChuanChat,
     "MiniMax": MiniMaxChat,
+    "Minimax": MiniMaxChat,
     "Mistral": MistralChat,
     'Gemini' : GeminiChat,
     "Bedrock": BedrockChat,