RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-02-05 22:33:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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'

Browse Source
This commit is contained in:
xieyh 2024-01-13 15:44:26 +08:00 committed by GitHub
parent 0a37fe93b8
commit ebda1056e5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/model_workers/minimax.py
View File

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