From 9fc2d4a2e1f195f218c4a2719a9486dcf96ffe44 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Fri, 20 Sep 2024 15:54:57 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=B7=B7=E5=85=83?= =?UTF-8?q?=E5=A4=A7=E6=A8=A1=E5=9E=8B=E4=BD=BF=E7=94=A8system=E8=A7=92?= =?UTF-8?q?=E8=89=B2=E6=8A=A5=E9=94=99?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../impl/tencent_model_provider/model/hunyuan.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/apps/setting/models_provider/impl/tencent_model_provider/model/hunyuan.py b/apps/setting/models_provider/impl/tencent_model_provider/model/hunyuan.py index 38983a72f..7e913f9ad 100644 --- a/apps/setting/models_provider/impl/tencent_model_provider/model/hunyuan.py +++ b/apps/setting/models_provider/impl/tencent_model_provider/model/hunyuan.py @@ -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}")