zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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, ChatMessage,
ChatMessageChunk, ChatMessageChunk,
HumanMessage, HumanMessage,
HumanMessageChunk, HumanMessageChunk, SystemMessage,
) )
from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
from langchain_core.pydantic_v1 import Field, SecretStr, root_validator 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} message_dict = {"Role": "user", "Content": message.content}
elif isinstance(message, AIMessage): elif isinstance(message, AIMessage):
message_dict = {"Role": "assistant", "Content": message.content} message_dict = {"Role": "assistant", "Content": message.content}
elif isinstance(message, SystemMessage):
message_dict = {"Role": "system", "Content": message.content}
else: else:
raise TypeError(f"Got unknown type {message}") raise TypeError(f"Got unknown type {message}")