RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-02-07 15:38:27 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #2648 from xyhshen/patch-1

Browse Source 
Fix two bugs in Minimax's model worker that affect usage
This commit is contained in:
zR 2024-01-13 16:37:05 +08:00 committed by GitHub
commit f28cd59557
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