15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: 修复混元大模型使用system角色报错

This commit is contained in:
shaohuzhang1 2024-09-20 15:54:57 +08:00 committed by shaohuzhang1
parent 3cce902a7a
commit 9fc2d4a2e1
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/setting/models_provider/impl/tencent_model_provider/model/hunyuan.py
View File

@ -15,7 +15,7 @@ from langchain_core.messages import (
ChatMessage,
ChatMessageChunk,
HumanMessage,
HumanMessageChunk,
HumanMessageChunk, SystemMessage,
)
from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
from langchain_core.pydantic_v1 import Field, SecretStr, root_validator
@ -37,6 +37,8 @@ def _convert_message_to_dict(message: BaseMessage) -> dict:
message_dict = {"Role": "user", "Content": message.content}
elif isinstance(message, AIMessage):
message_dict = {"Role": "assistant", "Content": message.content}
elif isinstance(message, SystemMessage):
message_dict = {"Role": "system", "Content": message.content}
else:
raise TypeError(f"Got unknown type {message}")