From ebda1056e54c5a5fdf9f8b2aacc063573741d4de Mon Sep 17 00:00:00 2001
From: xieyh <540140031@qq.com>
Date: Sat, 13 Jan 2024 15:44:26 +0800
Subject: [PATCH] Fix two bugs in Minimax's model worker that affect usage,
 namely do_ Embeddings should return result for multiple documents, and the
 other option is validate_ Messages, 'role': 'user' should be the uppercase
 letter 'USER'

---
 server/model_workers/minimax.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/server/model_workers/minimax.py b/server/model_workers/minimax.py
index deae07ea..ba610d52 100644
--- a/server/model_workers/minimax.py
+++ b/server/model_workers/minimax.py
@@ -28,7 +28,7 @@ class MiniMaxWorker(ApiModelWorker):
 
     def validate_messages(self, messages: List[Dict]) -> List[Dict]:
         role_maps = {
-            "user": self.user_role,
+            "USER": self.user_role,
             "assistant": self.ai_role,
             "system": "system",
         }
@@ -140,7 +140,7 @@ class MiniMaxWorker(ApiModelWorker):
                     self.logger.error(f"请求 MiniMax API 时发生错误：{data}")
                     return data
                 i += batch_size
-            return {"code": 200, "data": embeddings}
+            return {"code": 200, "data": result}
 
     def get_embeddings(self, params):
         # TODO: 支持embeddings